cli + query: improvements

crates/brk_cli/src/run.rs

@@ -0,0 +1,58 @@
+use std::{path::Path, thread::sleep, time::Duration};
+
+use brk_computer::Computer;
+use brk_exit::Exit;
+use brk_indexer::Indexer;
+use brk_parser::rpc::{self, RpcApi};
+use brk_server::tokio;
+use clap::Parser;
+use log::info;
+
+#[derive(Parser, Debug)]
+pub struct RunArgs {
+    name: Option<String>,
+}
+
+pub fn run(mut indexer: Indexer, mut computer: Computer) -> color_eyre::Result<()> {
+    let data_dir = Path::new("../../../bitcoin");
+
+    let rpc = Box::leak(Box::new(rpc::Client::new(
+        "http://localhost:8332",
+        rpc::Auth::CookieFile(Path::new(data_dir).join(".cookie")),
+    )?));
+
+    let exit = Exit::new();
+
+    let parser = brk_parser::Parser::new(data_dir, rpc);
+
+    tokio::runtime::Builder::new_multi_thread()
+        .enable_all()
+        .build()?
+        .block_on(async {
+            let served_indexer = indexer.clone();
+            let served_computer = computer.clone();
+
+            tokio::spawn(async move {
+                brk_server::main(served_indexer, served_computer).await.unwrap();
+            });
+
+            loop {
+                let block_count = rpc.get_block_count()?;
+
+                info!("{block_count} blocks found.");
+
+                let starting_indexes = indexer.index(&parser, rpc, &exit)?;
+
+                computer.compute(&mut indexer, starting_indexes, &exit)?;
+
+                info!("Waiting for new blocks...");
+
+                while block_count == rpc.get_block_count()? {
+                    sleep(Duration::from_secs(1))
+                }
+            }
+
+            #[allow(unreachable_code)]
+            Ok(())
+        })
+}