import { Command, Option } from "commander";
import Mocha from "mocha";
import "ts-mocha";
import path from "path";
import dirTree from "directory-tree";

import { loadConfig, LOCKLIFT_NETWORK_NAME } from "../../config";
import { Locklift } from "../../../index";
import * as utils from "../builder/utils";

import { buildStep } from "../steps/build";

const program = new Command();

program
  .name("test")
  .description("Run mocha tests")
  .option("--disable-build", "Disable automatic contracts build", false)
  .option("-t, --test <test>", "Path to Mocha test folder", "test")
  .option("-c, --contracts <contracts>", "Path to the contracts folder", "contracts")
  .option("-b, --build <build>", "Path to the build folder", "build")
  .option("-f, --force", "Force build contracts", false)
  .option("--disable-include-path", "Disables including node_modules. Use this with old compiler versions", false)
  .option("-n, --network <network>", "Network to use, choose from configuration", LOCKLIFT_NETWORK_NAME)
  .addOption(
    new Option("--config <config>", "Path to the config file")
      .default(() => loadConfig("locklift.config.ts"))
      .argParser(config => () => loadConfig(config)),
  )
  .option("--tests [tests...]", "Set of tests to run, separated by comma")
  .allowUnknownOption()
  .action(async options => {
    const config = options.config();
    if (config.networks[options.network] === undefined) {
      console.error(`Can't find configuration for ${options.network} network!`);

      process.exit(1);
    }
    if (!options.disableBuild) {
      await buildStep(config, options, options.force);
    }
    // Initialize Locklift and pass it into tests context
    const locklift = await Locklift.setup(config, options.network);

    //@ts-ignore
    global.locklift = locklift;

    // Run mocha tests
    if (config.mocha?.tsconfig) {
      process.env.TS_NODE_PROJECT = config.mocha?.tsconfig;
      process.env.TS_CONFIG_PATHS = "true";
    }
    const mocha = new Mocha({ ...config.mocha });

    // Run all .js files in tests or only specified tests
    let testFiles: string[];

    if (Array.isArray(options.tests)) {
      testFiles = options.tests;
    } else {
      const testNestedTree = dirTree(path.resolve(process.cwd(), options.test), { extensions: /\.(js|ts)/ });

      testFiles = utils.flatDirTree(testNestedTree)?.map(t => t.path) || [];
    }
    testFiles.forEach((file: string) => mocha.addFile(file));
    mocha.run(fail => process.exit(fail ? 1 : 0));
  });

export default program;