import DB from '../database';
import logger from '../logger';
import { Ancestor, CpfpInfo } from '../mempool.interfaces';
import cpfpRepository from './CpfpRepository';

class TransactionRepository {
  public async $setCluster(txid: string, clusterRoot: string): Promise<void> {
    try {
      await DB.query(
        `
          INSERT INTO compact_transactions
          (
            txid,
            cluster
          )
          VALUE (UNHEX(?), UNHEX(?))
          ON DUPLICATE KEY UPDATE
            cluster = UNHEX(?)
        ;`,
        [txid, clusterRoot, clusterRoot]
      );
    } catch (e: any) {
      logger.err(`Cannot save transaction cpfp cluster into db. Reason: ` + (e instanceof Error ? e.message : e));
      throw e;
    }
  }

  public async $batchSetCluster(txs): Promise<void> {
    try {
      let query = `
          INSERT IGNORE INTO compact_transactions
          (
            txid,
            cluster
          )
          VALUES
      `;
      query += txs.map(tx => {
        return (' (UNHEX(?), UNHEX(?))');
      }) + ';';
      const values = txs.map(tx => [tx.txid, tx.cluster]).flat();
      await DB.query(
        query,
        values
      );
    } catch (e: any) {
      logger.err(`Cannot save cpfp transactions into db. Reason: ` + (e instanceof Error ? e.message : e));
      throw e;
    }
  }

  public async $getCpfpInfo(txid: string): Promise<CpfpInfo | void> {
    try {
      const [txRows]: any = await DB.query(
        `
          SELECT HEX(txid) as id, HEX(cluster) as root
          FROM compact_transactions
          WHERE txid = UNHEX(?)
        `,
        [txid]
      );
      if (txRows.length && txRows[0].root != null) {
        const txid = txRows[0].id.toLowerCase();
        const clusterId = txRows[0].root.toLowerCase();
        const cluster = await cpfpRepository.$getCluster(clusterId);
        if (cluster) {
          return this.convertCpfp(txid, cluster);
        }
      }
    } catch (e) {
      logger.err('Cannot get transaction cpfp info from db. Reason: ' + (e instanceof Error ? e.message : e));
      throw e;
    }
  }

  public async $removeTransaction(txid: string): Promise<void> {
    try {
      await DB.query(
        `
          DELETE FROM compact_transactions
          WHERE txid = UNHEX(?)
        `,
        [txid]
      );
    } catch (e) {
      logger.warn('Cannot delete transaction cpfp info from db. Reason: ' + (e instanceof Error ? e.message : e));
      throw e;
    }
  }

  private convertCpfp(txid, cluster): CpfpInfo {
    const descendants: Ancestor[] = [];
    const ancestors: Ancestor[] = [];
    let matched = false;

    for (const tx of (cluster?.txs || [])) {
      if (tx.txid === txid) {
        matched = true;
      } else if (!matched) {
        descendants.push(tx);
      } else {
        ancestors.push(tx);
      }
    }
    return {
      descendants,
      ancestors,
    };
  }
}

export default new TransactionRepository();