import { Command, Option } from 'clipanion'
import * as typanion from 'typanion'
import chalk from 'chalk'

import { Solution } from '../../services/Solution.js'
import { GitAffected } from '../../services/GitAffected.js'
import { template } from '../../services/Template.js'
import { Test } from '../../services/Test.js'
import { SolutionTestsResult } from '../../services/SolutionTestsResult.js'
import { TemporaryFolder } from '../../services/TemporaryFolder.js'

export class RunTestCommand extends Command {
  public static override paths = [['run', 'test']]

  public static override usage = {
    description:
      'Test if the solution is correct and display where it succeeds and fails.'
  }

  public programmingLanguage = Option.String('--language', {
    description: 'The programming language used to solve the challenge.',
    validator: typanion.isString()
  })

  public challenge = Option.String('--challenge', {
    description: 'The challenge name where you want to test your solution.',
    validator: typanion.isString()
  })

  public solutionName = Option.String('--solution', {
    description: 'The solution name to run.',
    validator: typanion.isString()
  })

  public affected = Option.Boolean('--affected', false, {
    description: 'Only run the tests for the affected files in `git`.'
  })

  public all = Option.Boolean('--all', false, {
    description: 'Run the tests for all the solutions.'
  })

  public base = Option.String('--base', {
    description: 'Base of the current branch (usually master).'
  })

  public async execute(): Promise<number> {
    console.log()
    try {
      await TemporaryFolder.cleanAll()
      if (this.programmingLanguage != null) {
        await template.verifySupportedProgrammingLanguage(
          this.programmingLanguage
        )
      }
      if (this.all) {
        const solutions = await Solution.getManyByProgrammingLanguages(
          this.programmingLanguage != null ? [this.programmingLanguage] : undefined
        )
        return await Test.runManyWithSolutions(solutions)
      }
      if (this.affected) {
        const gitAffected = new GitAffected({
          base: this.base
        })
        const solutions = await gitAffected.getAffectedSolutionsFromGit()
        return await Test.runManyWithSolutions(solutions)
      }
      if (
        this.solutionName == null ||
        this.challenge == null ||
        this.programmingLanguage == null
      ) {
        throw new Error(
          'You must specify all the options (`--challenge`, `--solution`,  `--language`).'
        )
      }
      const solution = await Solution.get({
        name: this.solutionName,
        challengeName: this.challenge,
        programmingLanguageName: this.programmingLanguage
      })
      const result = await solution.test()
      result.print({
        shouldPrintBenchmark: true,
        shouldPrintTableResult: true
      })
      await TemporaryFolder.cleanAll()
      if (result.isSuccess)  {
        console.log(SolutionTestsResult.SUCCESS_MESSAGE)
        return 0
      }
      return 1
    } catch (error) {
      if (error instanceof Error) {
        console.error(`${chalk.bold.red('Error:')} ${error.message}`)
      }
      await TemporaryFolder.cleanAll()
      return 1
    }
  }
}