bwt 0.2.4

use std::collections::HashMap;
use std::sync::{mpsc, Arc};
use std::{fmt, thread, time};

use serde::Serialize;

use bitcoin::{Address, BlockHash, OutPoint, Txid};
use bitcoincore_rpc::json::GetTransactionResultDetailCategory as TxCategory;
use bitcoincore_rpc::{Client as RpcClient, RpcApi};

use crate::error::Result;
use crate::store::{FundingInfo, MemoryStore, SpendingInfo, TxEntry};
use crate::types::{BlockId, InPoint, RescanSince, ScriptHash, TxStatus};
use crate::util::bitcoincore_ext::{ListTransactionResult, RpcApiExt};
use crate::util::progress::Progress;
use crate::wallet::{KeyOrigin, WalletWatcher};

pub struct Indexer {
    rpc: Arc<RpcClient>,
    watcher: WalletWatcher,
    store: MemoryStore,
    tip: Option<BlockId>,
    fixed_listsinceblock: bool,
}

impl Indexer {
    pub fn new(rpc: Arc<RpcClient>, watcher: WalletWatcher) -> Result<Self> {
        // listsinceblock was racey in Bitcoin Core < 0.21: https://github.com/bitcoin/bitcoin/issues/19338
        let fixed_listsinceblock = rpc.version()? >= 210000;

        Ok(Indexer {
            rpc,
            watcher,
            store: MemoryStore::new(),
            tip: None,
            fixed_listsinceblock,
        })
    }

    pub fn store(&self) -> &MemoryStore {
        &self.store
    }

    pub fn watcher(&self) -> &WalletWatcher {
        &self.watcher
    }

    pub fn tip(&self) -> Option<BlockId> {
        self.tip
    }

    // continue to sync transactions and import addresses (with rescan) until no more new addresses
    // need to be imported. the initial sync does not collect the Changelog and does not emit updates.
    pub fn initial_sync(&mut self, progress_tx: Option<mpsc::Sender<Progress>>) -> Result<()> {
        let timer = time::Instant::now();

        debug!("starting initial sync");
        self.watcher.check_imports(&self.rpc)?;

        let mut changelog = Changelog::new(false);
        let mut synced_tip;

        let shutdown_progress_thread = spawn_send_progress_thread(self.rpc.clone(), progress_tx);

        while {
            synced_tip = self.sync_transactions(true, &mut changelog)?;
            self.watcher.do_imports(&self.rpc, /*rescan=*/ true)?
        } { /* do while */ }

        shutdown_progress_thread.send(()).ok();

        self.sync_mempool(/*force_refresh=*/ true);

        let stats = self.store.stats();
        info!(
            "completed initial sync in {:?} up to height {} (total {} transactions and {} addresses)",
            timer.elapsed(),
            synced_tip.0,
            stats.transaction_count,
            stats.scripthash_count,
        );
        self.tip = Some(synced_tip);
        Ok(())
    }

    // initiate a regular sync to catch up with updates and import new addresses (no rescan)
    pub fn sync(&mut self) -> Result<Vec<IndexChange>> {
        let mut changelog = Changelog::new(self.tip.is_some());

        // detect reorgs and sync the whole history from scratch when they happen
        // XXX the reorg test is racey
        if let Some(BlockId(tip_height, ref tip_hash)) = self.tip {
            let best_chain_hash = self.rpc.get_block_hash(tip_height as u64)?;
            if best_chain_hash != *tip_hash {
                warn!(
                    "reorg detected, block height {} was {} and now is {}. fetching history from scratch...",
                    tip_height, tip_hash, best_chain_hash
                );

                // notify clients about the reorg, but don't collect additional events (apart from
                // ChainTip, added below)
                changelog.push(|| IndexChange::Reorg(tip_height, *tip_hash, best_chain_hash));
                changelog.track = false;

                // XXX start syncing from N blocks backs instead of from the beginning?
                self.tip = None;
            }
        }

        let synced_tip = self.sync_transactions(false, &mut changelog)?;
        let tip_updated = self.tip != Some(synced_tip);
        self.sync_mempool(/*force_refresh=*/ tip_updated);
        self.watcher.do_imports(&self.rpc, /*rescan=*/ false)?;

        let mut changelog = changelog.into_vec();

        if tip_updated {
            info!(
                "synced up to height {}{}",
                synced_tip.0,
                iif!(changelog.is_empty(), "", " (found wallet activity)")
            );

            changelog.push(IndexChange::ChainTip(synced_tip));
            self.tip = Some(synced_tip);
        }

        if !changelog.is_empty() && log_enabled!(log::Level::Debug) {
            for update in &changelog {
                debug!("  - {:?}", update);
            }
        }

        Ok(changelog)
    }

    fn sync_transactions(
        &mut self,
        refresh_outgoing: bool,
        changelog: &mut Changelog,
    ) -> Result<BlockId> {
        // To deal with listsinceblock's non-atomicity in bitcoin < 0.21, we need to record the
        // tip before and make sure it didn't move during the listsinceblock rpc call.
        let tip_sanity_check = if !self.fixed_listsinceblock {
            let height = self.rpc.get_block_count()?;
            Some(BlockId(height as u32, self.rpc.get_block_hash(height)?))
        } else {
            None
        };

        let prev_indexed_tip = self.tip.as_ref().map(|tip| tip.hash());
        let result = self.rpc.list_since_block_(prev_indexed_tip)?;

        // Ensure consistent results for `listsinceblock` in < 0.21.
        if tip_sanity_check.map_or(false, |tip| result.lastblock != *tip.hash()) {
            warn!("chain tip moved while reading listsinceblock, retrying...");
            return self.sync_transactions(refresh_outgoing, changelog);
        }

        // When `listsinceblock` is atomic in >=0.21, use its returned 'lastblock' as the tip.
        let tip = tip_sanity_check.map_or_else(
            || -> Result<_> {
                if prev_indexed_tip != Some(&result.lastblock) {
                    let height = self.rpc.get_block_header_info(&result.lastblock)?.height;
                    Ok(BlockId(height as u32, result.lastblock))
                } else {
                    Ok(self.tip.unwrap()) // the tip didn't move
                }
            },
            Ok,
        )?;

        trace!(
            "fetched {} transactions between blocks {} and {}",
            result.transactions.len(),
            self.tip.map_or(0, |t| t.height()),
            tip.height()
        );

        for ltx in result.removed {
            // transactions that were re-added in the active chain will appear in `removed`
            // but with a positive confirmation count, ignore these.
            if ltx.info.confirmations < 0 {
                self.purge_tx(&ltx.info.txid, changelog);
            }
        }

        let mut buffered_outgoing: HashMap<Txid, i32> = HashMap::new();
        let mut cached_conflicted = HashMap::new();

        for ltx in result.transactions {
            if self.is_conflicted(&ltx, &mut cached_conflicted)? {
                self.purge_tx(&ltx.info.txid, changelog);
                continue;
            }

            // "listtransactions"/"listsinceblock" in fact lists transaction outputs and not transactions.
            // for "receive" txs, it returns one entry per wallet-owned output in the tx.
            // for "send" txs, it returns one entry for every output in the tx, owned or not.
            match ltx.detail.category {
                TxCategory::Receive => {
                    // incoming txouts are easy: bitcoind tells us the associated
                    // address and label, giving us all the information we need in
                    // order to save the txo to the index.
                    self.process_incoming_txo(ltx, tip.height(), changelog);
                }
                TxCategory::Send => {
                    // indexing outgoing txs require fetching the list of spent prevouts and
                    // comparing them against the wallet's known funded outputs. we can't do that
                    // straightaway because the prevouts being spent might not be indexed yet, so
                    // the outgoing txs are buffered and processed at the end, after the txs funding
                    // the prevouts are guarranted to be indexed.
                    buffered_outgoing.insert(ltx.info.txid, ltx.info.confirmations);
                }
                // ignore mining-related transactions
                TxCategory::Generate | TxCategory::Immature | TxCategory::Orphan => (),
            };
        }

        for (txid, confirmations) in buffered_outgoing {
            let status = TxStatus::from_confirmations(confirmations, tip.height());
            self.process_outgoing_tx(txid, status, refresh_outgoing, changelog)
                .map_err(|err| warn!("failed processing outgoing payment: {:?}", err))
                .ok();
        }

        Ok(tip)
    }

    /// Check if the given wallet transaction is conflicted
    fn is_conflicted(
        &self,
        ltx: &ListTransactionResult,
        cached: &mut HashMap<Txid, bool>,
    ) -> Result<bool> {
        // Unconfirmed transactions with wallet conflicts need to be looked up in the mempool to
        // determine if the node considers them to be the active leading one among the conflicts
        // See https://github.com/bitcoin/bitcoin/issues/21018
        if ltx.info.confirmations == 0 && !ltx.info.wallet_conflicts.is_empty() {
            if let Some(is_conflicted) = cached.get(&ltx.info.txid) {
                Ok(*is_conflicted)
            } else {
                let is_active = self.rpc.get_mempool_entry_opt(&ltx.info.txid)?.is_some();
                cached.insert(ltx.info.txid, !is_active);
                if is_active {
                    // if this transaction is the active one, the ones that conflict with it aren't
                    cached.extend(ltx.info.wallet_conflicts.iter().map(|txid| (*txid, true)));
                }
                Ok(!is_active)
            }
        } else {
            Ok(ltx.info.confirmations < 0)
        }
    }

    // upsert the transaction while collecting the changelog
    fn upsert_tx(&mut self, txid: &Txid, status: TxStatus, changelog: &mut Changelog) {
        let tx_updated = self.store.upsert_tx(txid, status);
        if tx_updated {
            changelog.with(|changelog| {
                let tx_entry = self.store.get_tx_entry(txid).unwrap();
                changelog.extend(IndexChange::from_tx(txid, tx_entry));
            });
        }
    }

    fn purge_tx(&mut self, txid: &Txid, changelog: &mut Changelog) {
        let tx_deleted = self.store.purge_tx(&txid);
        if tx_deleted {
            changelog.push(|| IndexChange::TransactionReplaced(*txid));
        }
    }

    fn process_incoming_txo(
        &mut self,
        ltx: ListTransactionResult,
        tip_height: u32,
        changelog: &mut Changelog,
    ) {
        let label = ltx.detail.label.as_ref();
        let origin = some_or_ret!(label.and_then(|l| KeyOrigin::from_label(l)));
        let address = some_or_ret!(ltx.detail.address);

        // XXX we assume that any address with a "bwt/..." label is ours, this may not necessarily be true.

        let txid = ltx.info.txid;
        let vout = ltx.detail.vout;
        let scripthash = ScriptHash::from(&address);
        let status = TxStatus::from_confirmations(ltx.info.confirmations, tip_height);
        let amount = ltx.detail.amount.to_unsigned().unwrap().as_sat(); // safe to unwrap, incoming payments cannot have negative amounts

        trace!(
            "processing incoming txout {}:{} scripthash={} address={} origin={:?} status={:?} amount={}",
            txid, vout, scripthash, address, origin, status, amount
        );

        self.upsert_tx(&txid, status, changelog);

        self.store.index_scripthash(&scripthash, &origin, &address);

        let txo_added =
            self.store
                .index_tx_output_funding(&txid, vout, FundingInfo(scripthash, amount));

        if txo_added {
            changelog.push(|| {
                IndexChange::TxoFunded(OutPoint::new(txid, vout), scripthash, amount, status)
            });
            self.watcher.mark_funded(&origin);
        }
    }

    fn process_outgoing_tx(
        &mut self,
        txid: Txid,
        status: TxStatus,
        refresh: bool,
        changelog: &mut Changelog,
    ) -> Result<()> {
        trace!("processing outgoing tx txid={} status={:?}", txid, status);

        let has_spends = |tx_entry: &TxEntry| !tx_entry.spending.is_empty();

        if !refresh && self.store.get_tx_entry(&txid).map_or(false, has_spends) {
            // skip indexing spent inputs, but keep the status which might be more recent
            self.upsert_tx(&txid, status, changelog);
            trace!("skipping outgoing tx {}, already indexed", txid);
            return Ok(());
        }

        // TODO use batch rpc to fetch all buffered outgoing txs
        let tx = self.rpc.get_transaction(&txid, Some(true))?.transaction()?;

        let spending: HashMap<u32, SpendingInfo> = tx
            .input
            .iter()
            .enumerate()
            .filter_map(|(vin, input)| {
                let inpoint = InPoint::new(txid, vin as u32);
                let prevout = input.previous_output;
                let FundingInfo(scripthash, amount) = self.store.lookup_txo_fund(&prevout)?;

                #[cfg(feature = "track-spends")]
                self.store.index_txo_spend(prevout, inpoint);

                changelog.push(|| IndexChange::TxoSpent(inpoint, scripthash, prevout, status));

                // we could keep just the previous_output and lookup the scripthash and amount
                // from the corresponding FundingInfo, but we keep it here anyway for quick access
                let spending_info = SpendingInfo(scripthash, prevout, amount);
                Some((vin as u32, spending_info))
            })
            .collect();

        if !spending.is_empty() {
            self.upsert_tx(&txid, status, changelog);
            self.store
                .index_tx_inputs_spending(&txid, spending, refresh);
        }

        Ok(())
    }

    /// Update missing/outdated mempool entries for unconfirmed mempool transactions (or all mempool
    /// entries when force_refresh is set, during the initial sync or following a chain tip update)
    fn sync_mempool(&mut self, force_refresh: bool) {
        let mempool = self.store.mempool_mut();

        for (txid, opt_entry) in mempool.iter_mut() {
            if force_refresh || opt_entry.is_none() {
                match self.rpc.get_mempool_entry(txid) {
                    Ok(rpc_entry) => *opt_entry = Some(rpc_entry.into()),
                    Err(e) => warn!("failed fetching mempool entry for {}: {}", txid, e),
                }
            }
        }

        // TODO use batch rpc
    }

    /// Get historical events that happened at or after `min_block_height`, including unconfirmed,
    /// ordered with oldest first.
    ///
    /// Includes the `Transaction`, `TxoFunded` and `TxoSpent` events, and a *single* `ChainTip`
    /// event with the currently synced tip as the last entry (when bwt is synced).
    pub fn get_changelog_since(&self, min_block_height: u32) -> Vec<IndexChange> {
        self.store
            .get_history_since(min_block_height)
            .into_iter()
            .map(|txhist| {
                let tx_entry = self.store.get_tx_entry(&txhist.txid).unwrap();
                IndexChange::from_tx(&txhist.txid, tx_entry)
            })
            .flatten()
            .chain(self.tip.clone().map(IndexChange::ChainTip).into_iter())
            .collect()
    }

    pub fn track_address(&mut self, address: Address, rescan_since: RescanSince) -> Result<()> {
        self.watcher.track_address(address, rescan_since)
    }
}

#[derive(Clone, Serialize, Debug)]
#[serde(tag = "category", content = "params")]
pub enum IndexChange {
    ChainTip(BlockId),
    Reorg(u32, BlockHash, BlockHash),

    Transaction(Txid, TxStatus),
    TransactionReplaced(Txid),

    TxoFunded(OutPoint, ScriptHash, u64, TxStatus),
    TxoSpent(InPoint, ScriptHash, OutPoint, TxStatus),
}

struct Changelog {
    track: bool,
    changes: Vec<IndexChange>,
}

impl Changelog {
    fn new(track: bool) -> Self {
        Changelog {
            track,
            changes: vec![],
        }
    }
    fn push(&mut self, make_update: impl Fn() -> IndexChange) {
        if self.track {
            self.changes.push(make_update());
        }
    }
    fn with(&mut self, closure: impl Fn(&mut Vec<IndexChange>)) {
        if self.track {
            closure(&mut self.changes)
        }
    }
    fn into_vec(self) -> Vec<IndexChange> {
        self.changes
    }
}
impl IndexChange {
    // the scripthash affected by the update, if any
    pub fn scripthash(&self) -> Option<&ScriptHash> {
        match self {
            IndexChange::TxoFunded(_, ref scripthash, ..) => Some(scripthash),
            IndexChange::TxoSpent(_, ref scripthash, ..) => Some(scripthash),
            _ => None,
        }
    }

    // the outpoint created or spent, if any
    pub fn outpoint(&self) -> Option<&OutPoint> {
        match self {
            IndexChange::TxoFunded(ref outpoint, ..) => Some(outpoint),
            IndexChange::TxoSpent(_, _, ref outpoint, _) => Some(outpoint),
            _ => None,
        }
    }

    pub fn category_str(&self) -> &str {
        match self {
            Self::ChainTip(..) => "ChainTip",
            Self::Reorg(..) => "Reorg",

            Self::Transaction(..) => "Transaction",
            Self::TransactionReplaced(..) => "TransactionReplaced",

            Self::TxoFunded(..) => "TxoFunded",
            Self::TxoSpent(..) => "TxoSpent",
        }
    }

    // create all the changelog events inflicted by the transaction
    fn from_tx(txid: &Txid, tx_entry: &TxEntry) -> Vec<Self> {
        let mut changes = vec![IndexChange::Transaction(*txid, tx_entry.status)];

        changes.extend(tx_entry.funding.iter().map(|(vout, funding_info)| {
            let outpoint = OutPoint::new(*txid, *vout);
            let FundingInfo(scripthash, amount) = funding_info;
            IndexChange::TxoFunded(outpoint, *scripthash, *amount, tx_entry.status)
        }));

        changes.extend(tx_entry.spending.iter().map(|(vin, spending_info)| {
            let inpoint = InPoint::new(*txid, *vin);
            let SpendingInfo(scripthash, prevout, _) = spending_info;
            IndexChange::TxoSpent(inpoint, *scripthash, *prevout, tx_entry.status)
        }));

        changes
    }
}

impl fmt::Display for IndexChange {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{:?}", self)
    }
}

// Spawn a thread to poll getwalletinfo, log progress and send progress updates via mpsc
fn spawn_send_progress_thread(
    rpc: Arc<RpcClient>,
    progress_tx: Option<mpsc::Sender<Progress>>,
) -> mpsc::SyncSender<()> {
    use crate::util::progress::wait_wallet_scan;

    const DELAY: time::Duration = time::Duration::from_millis(250);
    const INTERVAL_SLOW: time::Duration = time::Duration::from_secs(6);
    const INTERVAL_FAST: time::Duration = time::Duration::from_millis(1500);
    // use the fast interval if we're reporting progress to a channel, or the slow one if its only for CLI
    let interval = iif!(progress_tx.is_some(), INTERVAL_FAST, INTERVAL_SLOW);

    let (shutdown_tx, shutdown_rx) = mpsc::sync_channel(1);

    thread::spawn(move || {
        // allow some time for the indexer to start the first set of imports
        thread::sleep(DELAY);

        if shutdown_rx.try_recv() != Err(mpsc::TryRecvError::Empty) {
            return;
        }
        if let Err(e) = wait_wallet_scan(&rpc, progress_tx, Some(shutdown_rx), interval) {
            trace!("progress thread aborted: {:?}", e);
        }
    });

    shutdown_tx
}