brk/crates/brk_cli/src/lib.rs

#![doc = include_str!("../README.md")]

use std::{
    fs,
    io::Cursor,
    path::Path,
    thread::{self, sleep},
    time::Duration,
};

use bitcoincore_rpc::{self, RpcApi};
use brk_bundler::bundle;
use brk_computer::Computer;
use brk_indexer::Indexer;
use brk_interface::Interface;
use brk_server::{Server, VERSION};
use brk_vecs::Exit;
use log::info;

mod bridge;
mod config;
mod paths;
mod website;

use crate::{bridge::Bridge, config::Config, paths::*};

pub fn main() -> color_eyre::Result<()> {
    color_eyre::install()?;

    fs::create_dir_all(dot_brk_path())?;

    brk_logger::init(Some(&dot_brk_log_path()));

    thread::Builder::new()
        .stack_size(256 * 1024 * 1024)
        .spawn(run)?
        .join()
        .unwrap()
}

pub fn run() -> color_eyre::Result<()> {
    let config = Config::import()?;

    let rpc = config.rpc()?;

    let exit = Exit::new();
    exit.set_ctrlc_handler();

    let parser = brk_parser::Parser::new(config.blocksdir(), config.brkdir(), rpc);

    let mut indexer = Indexer::forced_import(&config.brkdir())?;

    let wait_for_synced_node = |rpc_client: &bitcoincore_rpc::Client| -> color_eyre::Result<()> {
        let is_synced = || -> color_eyre::Result<bool> {
            let info = rpc_client.get_blockchain_info()?;
            Ok(info.headers == info.blocks)
        };

        if !is_synced()? {
            info!("Waiting for node to be synced...");
            while !is_synced()? {
                sleep(Duration::from_secs(1))
            }
        }

        Ok(())
    };

    let mut computer = Computer::forced_import(&config.brkdir(), &indexer, config.fetcher())?;

    tokio::runtime::Builder::new_multi_thread()
        .enable_all()
        .build()?
        .block_on(async {
            let interface = Interface::build(&indexer, &computer);

            let website = config.website();

            let downloads_path = config.downloads_dir();

            let bundle_path = if website.is_some() {
                let websites_dev_path = Path::new("../../websites");

                let websites_path = if fs::exists(websites_dev_path)? {
                    websites_dev_path.to_path_buf()
                } else {
                    let downloaded_websites_path =
                        downloads_path.join(format!("brk-{VERSION}")).join("websites");

                    if !fs::exists(&downloaded_websites_path)? {
                        info!("Downloading websites from Github...");

                        let url = format!(
                            "https://github.com/bitcoinresearchkit/brk/archive/refs/tags/v{VERSION}.zip",
                        );

                        let response = minreq::get(url).send()?;
                        let bytes = response.as_bytes();
                        let cursor = Cursor::new(bytes);

                        let mut zip = zip::ZipArchive::new(cursor).unwrap();

                        zip.extract(downloads_path).unwrap();
                    }

                    downloaded_websites_path
                };

                interface.generate_bridge_file(website, websites_path.as_path())?;

                let watch = config.watch();

                Some(bundle(&websites_path, website.to_folder_name(), watch).await?)
            } else {
                None
            };

            let server = Server::new(
                interface,
                bundle_path,
            );

            let mcp = config.mcp();

            tokio::spawn(async move {
                server.serve(mcp).await.unwrap();
            });

            sleep(Duration::from_secs(1));

            loop {
                wait_for_synced_node(rpc)?;

                let block_count = rpc.get_block_count()?;

                info!("{} blocks found.", block_count + 1);

                let starting_indexes =
                    indexer.index(&parser, rpc, &exit, config.check_collisions())?;

                computer.compute(&indexer, starting_indexes, &exit)?;

                if let Some(delay) = config.delay() {
                    sleep(Duration::from_secs(delay))
                }

                info!("Waiting for new blocks...");

                while block_count == rpc.get_block_count()? {
                    sleep(Duration::from_secs(1))
                }
            }
        })
}