Organize code (ordinals#10)

src/arguments.rs

@@ -0,0 +1,16 @@
+use super::*;
+
+#[derive(StructOpt)]
+pub enum Arguments {
+ Catalog,
+ Supply,
+}
+
+impl Arguments {
+ pub fn run(self) -> Result<()> {
+ match self {
+ Self::Catalog => catalog::run(),
+ Self::Supply => supply::run(),
+ }
+ }
+}

src/catalog.rs

@@ -0,0 +1,72 @@
+use super::*;
+
+pub fn run() -> Result<()> {
+ let client = client::initialize()?;
+
+ let tip_height = client
+ .get_block_header_info(&client.get_best_block_hash()?)?
+ .height as u64;
+
+ eprintln!("Scanning for atoms up to height {}…", tip_height);
+
+ let mut atoms: BTreeMap<OutPoint, u64> = BTreeMap::new();
+
+ for height in 0..tip_height {
+ eprintln!("Scanning for atoms in block {}…", height);
+ let hash = client.get_block_hash(height)?;
+
+ let block = client.get_block(&hash)?;
+
+ for (i, transaction) in block.txdata.iter().enumerate() {
+ let txid = transaction.txid();
+ if i == 0 {
+ atoms.insert(OutPoint { txid, vout: 0 }, height);
+ } else {
+ let mut transferred = transaction
+ .input
+ .iter()
+ .map(|txin| atoms.remove_entry(&txin.previous_output))
+ .flatten()
+ .collect::<Vec<(OutPoint, u64)>>();
+
+ if transferred.is_empty() {
+ continue;
+ }
+
+ eprintln!(
+ "Transferring {} atoms: {:?}",
+ transferred.len(),
+ transferred,
+ );
+
+ let total = transaction
+ .output
+ .iter()
+ .map(|txout| txout.value)
+ .sum::<u64>();
+
+ let mut pending = 0;
+ for (vout, output) in transaction.output.iter().enumerate() {
+ let vout = vout as u32;
+
+ pending += output.value * transferred.len() as u64;
+
+ while pending >= total {
+ let (old_outpoint, atom) = transferred.remove(0);
+ let new_outpoint = OutPoint { vout, txid };
+ eprintln!(
+ "Transferring atom {} from {} to {}",
+ atom, old_outpoint, new_outpoint
+ );
+ atoms.insert(new_outpoint, atom);
+ pending -= total;
+ }
+ }
+
+ assert!(transferred.is_empty());
+ }
+ }
+ }
+
+ Ok(())
+}

src/client.rs

@@ -0,0 +1,22 @@
+use super::*;
+
+pub fn initialize() -> Result<Client> {
+ let home = dirs::home_dir().ok_or("Failed to retrieve home dir.")?;
+
+ let cookiefile = home.join("Library/Application Support/Bitcoin/.cookie");
+
+ if !cookiefile.is_file() {
+ return Err(
+ format!(
+ "Bitcoin RPC cookiefile does not exist. Path: {}",
+ cookiefile.display()
+ )
+ .into(),
+ );
+ }
+
+ Ok(Client::new(
+ "https://localhost:8332",
+ Auth::CookieFile(cookiefile),
+ )?)
+}

src/main.rs

@@ -1,179 +1,20 @@
 use {
+ arguments::Arguments,
  bitcoin::blockdata::transaction::OutPoint,
  bitcoincore_rpc::{Auth, Client, RpcApi},
  serde::Deserialize,
  std::collections::BTreeMap,
  structopt::StructOpt,
 };
 
-fn client() -> Result<Client> {
- let home = dirs::home_dir().ok_or("Failed to retrieve home dir.")?;
-
- let cookiefile = home.join("Library/Application Support/Bitcoin/.cookie");
-
- if !cookiefile.is_file() {
- return Err(
- format!(
- "Bitcoin RPC cookiefile does not exist. Path: {}",
- cookiefile.display()
- )
- .into(),
- );
- }
-
- Ok(Client::new(
- "https://localhost:8332",
- Auth::CookieFile(cookiefile),
- )?)
-}
-
-fn price() -> Result<f64> {
- #[derive(Deserialize)]
- struct Response {
- bpi: Bpi,
- }
-
- #[derive(Deserialize)]
- struct Bpi {
- #[serde(rename = "USD")]
- usd: Usd,
- }
-
- #[derive(Deserialize)]
- struct Usd {
- rate_float: f64,
- }
-
- Ok(
- reqwest::blocking::get("https://api.coindesk.com/v1/bpi/currentprice/usd.json")?
- .json::<Response>()?
- .bpi
- .usd
- .rate_float,
- )
-}
+mod arguments;
+mod catalog;
+mod client;
+mod price;
+mod supply;
 
 type Result<T, E = Box<dyn std::error::Error>> = std::result::Result<T, E>;
 
-#[derive(StructOpt)]
-enum Arguments {
- Catalog,
- Supply,
-}
-
-impl Arguments {
- fn run(self) -> Result<()> {
- match self {
- Self::Catalog => Self::catalog(),
- Self::Supply => Self::supply(),
- }
- }
-
- fn catalog() -> Result<()> {
- let client = client()?;
-
- let tip_height = client
- .get_block_header_info(&client.get_best_block_hash()?)?
- .height as u64;
-
- eprintln!("Scanning for atoms up to height {}…", tip_height);
-
- let mut atoms: BTreeMap<OutPoint, u64> = BTreeMap::new();
-
- for height in 0..tip_height {
- eprintln!("Scanning for atoms in block {}…", height);
- let hash = client.get_block_hash(height)?;
-
- let block = client.get_block(&hash)?;
-
- for (i, transaction) in block.txdata.iter().enumerate() {
- let txid = transaction.txid();
- if i == 0 {
- atoms.insert(OutPoint { txid, vout: 0 }, height);
- } else {
- let mut transferred = transaction
- .input
- .iter()
- .map(|txin| atoms.remove_entry(&txin.previous_output))
- .flatten()
- .collect::<Vec<(OutPoint, u64)>>();
-
- if transferred.is_empty() {
- continue;
- }
-
- eprintln!(
- "Transferring {} atoms: {:?}",
- transferred.len(),
- transferred,
- );
-
- let total = transaction
- .output
- .iter()
- .map(|txout| txout.value)
- .sum::<u64>();
-
- let mut pending = 0;
- for (vout, output) in transaction.output.iter().enumerate() {
- let vout = vout as u32;
-
- pending += output.value * transferred.len() as u64;
-
- while pending >= total {
- let (old_outpoint, atom) = transferred.remove(0);
- let new_outpoint = OutPoint { vout, txid };
- eprintln!(
- "Transferring atom {} from {} to {}",
- atom, old_outpoint, new_outpoint
- );
- atoms.insert(new_outpoint, atom);
- pending -= total;
- }
- }
-
- assert!(transferred.is_empty());
- }
- }
- }
-
- Ok(())
- }
-
- fn supply() -> Result<()> {
- let client = client()?;
-
- let height = client
- .get_block_header_info(&client.get_best_block_hash()?)?
- .height as u64;
-
- let mut supply = 0u64;
-
- let mut reward = 50 * 100_000_000;
-
- for i in 0..height {
- supply += reward;
-
- if i > 0 && i % 210_000 == 0 {
- reward /= 2;
- }
- }
-
- let atoms = height + 1;
- let supply = supply / 100_000_000;
- let supply_per_atom = supply as f64 / atoms as f64;
-
- let price = price()?;
-
- eprintln!("Bitcoin mined: {}", supply);
- eprintln!("Atoms discovered: {}", atoms);
- eprintln!("Bitcoin per atom: {:.02}", supply_per_atom);
- eprintln!("USD per atom: {:.02}", supply_per_atom * price);
-
- Ok(())
- }
-}
-
 fn main() -> Result<()> {
  Arguments::from_args().run()
 }

src/price.rs

@@ -0,0 +1,27 @@
+use super::*;
+
+pub fn get() -> Result<f64> {
+ #[derive(Deserialize)]
+ struct Response {
+ bpi: Bpi,
+ }
+
+ #[derive(Deserialize)]
+ struct Bpi {
+ #[serde(rename = "USD")]
+ usd: Usd,
+ }
+
+ #[derive(Deserialize)]
+ struct Usd {
+ rate_float: f64,
+ }
+
+ Ok(
+ reqwest::blocking::get("https://api.coindesk.com/v1/bpi/currentprice/usd.json")?
+ .json::<Response>()?
+ .bpi
+ .usd
+ .rate_float,
+ )
+}

src/supply.rs

@@ -0,0 +1,34 @@
+use super::*;
+
+pub fn run() -> Result<()> {
+ let client = client::initialize()?;
+
+ let height = client
+ .get_block_header_info(&client.get_best_block_hash()?)?
+ .height as u64;
+
+ let mut supply = 0u64;
+
+ let mut reward = 50 * 100_000_000;
+
+ for i in 0..height {
+ supply += reward;
+
+ if i > 0 && i % 210_000 == 0 {
+ reward /= 2;
+ }
+ }
+
+ let atoms = height + 1;
+ let supply = supply / 100_000_000;
+ let supply_per_atom = supply as f64 / atoms as f64;
+
+ let price = price::get()?;
+
+ eprintln!("Bitcoin mined: {}", supply);
+ eprintln!("Atoms discovered: {}", atoms);
+ eprintln!("Bitcoin per atom: {:.02}", supply_per_atom);
+ eprintln!("USD per atom: {:.02}", supply_per_atom * price);
+
+ Ok(())
+}