import "dotenv/config";
import os from "os";
import path from "path";
import { Connection, PublicKey } from "@_koii/web3.js";
import { borsh_bpf_js_deserialize } from "./wasm/bincode_js.cjs";
import { TASK_ID, WEBPACKED_FILE_PATH, TEST_KEYWORDS } from "./config";

class Debugger {
  /*
  Create .env file with following variables or directly input values to be used in live-debugging mode.
  */
  static taskID = TASK_ID;
  static webpackedFilePath = WEBPACKED_FILE_PATH;
  static keywords = TEST_KEYWORDS;
  static nodeDir: string;

  static async getConfig() {
    Debugger.nodeDir = await this.getNodeDirectory();

    let destinationPath = "executables/" + (await this.getAuditProgram()) + ".js";
    let logPath = "namespace/" + TASK_ID + "/task.log";

    console.log("Debugger.nodeDir", Debugger.nodeDir);

    return {
      webpackedFilePath: Debugger.webpackedFilePath,
      destinationPath: destinationPath,
      keywords: Debugger.keywords,
      logPath: logPath,
      nodeDir: Debugger.nodeDir,
      taskID: Debugger.taskID,
    };
  }

  static async getNodeDirectory() {
    if (Debugger.nodeDir) {
      return Debugger.nodeDir;
    }
    const homeDirectory = os.homedir();
    let nodeDirectory: string;

    switch (os.platform()) {
      case "linux":
        nodeDirectory = path.join(homeDirectory, ".config", "KOII-Desktop-Node");
        break;
      case "darwin":
        nodeDirectory = path.join(homeDirectory, "Library", "Application Support", "KOII-Desktop-Node");
        break;
      default:
        // Windows is the default
        nodeDirectory = path.join(homeDirectory, "AppData", "Roaming", "KOII-Desktop-Node");
    }

    return nodeDirectory;
  }

  static async getAuditProgram() {
    const connection = new Connection("https://mainnet.koii.network");
    const taskId = Debugger.taskID;
    const accountInfo = await connection.getAccountInfo(new PublicKey(taskId));
    if (!accountInfo?.data) {
      console.log(`${taskId} doesn't contain any distribution list data`);
      return null;
    }
    let data;
    const owner = accountInfo.owner.toBase58();
    if (owner === "Koiitask22222222222222222222222222222222222") {
      data = JSON.parse(accountInfo.data.toString());
    } else if (owner === "KPLTRVs6jA7QTthuJH2cEmyCEskFbSV2xpZw46cganN") {
      const buffer = accountInfo.data;
      data = borsh_bpf_js_deserialize(buffer);
      data = parseTaskState(data);
    } else {
      console.error(`Not a valid Task ID ${taskId}`);
      return null;
    }

    console.log("data.task_audit_program", data.task_audit_program);
    return data.task_audit_program;
  }
}

function parseTaskState(taskState: any) {
  taskState.stake_list = objectify(taskState.stake_list, true);
  taskState.ip_address_list = objectify(taskState.ip_address_list, true);
  taskState.distributions_audit_record = objectify(taskState.distributions_audit_record, true);
  taskState.distributions_audit_trigger = objectify(taskState.distributions_audit_trigger, true);
  taskState.submissions = objectify(taskState.submissions, true);
  taskState.submissions_audit_trigger = objectify(taskState.submissions_audit_trigger, true);
  taskState.distribution_rewards_submission = objectify(taskState.distribution_rewards_submission, true);
  taskState.available_balances = objectify(taskState.available_balances, true);
  return taskState;
}

function objectify(data: any, recursive = false) {
  if (data instanceof Map) {
    const obj = Object.fromEntries(data);
    if (recursive) {
      for (const key in obj) {
        if (obj[key] instanceof Map) {
          obj[key] = objectify(obj[key], true);
        } else if (typeof obj[key] === "object" && obj[key] !== null) {
          obj[key] = objectify(obj[key], true);
        }
      }
    }
    return obj;
  }
  return data;
}

export default Debugger;