programming-challenges/cli/services/Solution.ts

import { fileURLToPath } from 'node:url'
import path from 'node:path'
import fs from 'node:fs'
import { performance } from 'node:perf_hooks'

import chalk from 'chalk'
import ora from 'ora'

import { isExistingPath } from '../utils/isExistingPath.js'
import { Challenge } from './Challenge.js'
import { copyDirectory } from '../utils/copyDirectory.js'
import { template } from './Template.js'
import { docker } from './Docker.js'
import { Test } from './Test.js'
import { SolutionTestsResult } from './SolutionTestsResult.js'
import { TemporaryFolder } from './TemporaryFolder.js'

export interface GetSolutionOptions {
  programmingLanguageName: string
  challengeName: string
  name: string
}

export interface GenerateSolutionOptions extends GetSolutionOptions {
  githubUser: string
}

export interface SolutionOptions {
  programmingLanguageName: string
  challenge: Challenge
  name: string
}

export class Solution implements SolutionOptions {
  public programmingLanguageName: string
  public challenge: Challenge
  public name: string
  public path: string
  public temporaryFolder: TemporaryFolder

  constructor(options: SolutionOptions) {
    const { programmingLanguageName, challenge, name } = options
    this.programmingLanguageName = programmingLanguageName
    this.challenge = challenge
    this.name = name
    this.path = path.join(
      challenge.path,
      'solutions',
      programmingLanguageName,
      name
    )
    this.temporaryFolder = new TemporaryFolder()
  }

  private async setup(): Promise<void> {
    await this.temporaryFolder.create()
    await copyDirectory(this.path, this.temporaryFolder.path)
    await template.docker({
      programmingLanguage: this.programmingLanguageName,
      destination: this.temporaryFolder.path
    })
    process.chdir(this.temporaryFolder.path)
    try {
      await docker.build(this.temporaryFolder.id)
    } catch (error: any) {
      throw new Error(`solution: ${this.path}\n${error.message as string}\n`)
    }
  }

  public async test(): Promise<SolutionTestsResult> {
    await this.setup()
    return await Test.runAll(this)
  }

  public async run(input: string, output: boolean = false): Promise<void> {
    await this.setup()
    const loader = ora('Running...').start()
    try {
      const start = performance.now()
      const { stdout } = await docker.run(input, this.temporaryFolder.id)
      const end = performance.now()
      const elapsedTimeMilliseconds = end - start
      loader.succeed(chalk.bold.green('Success!'))
      SolutionTestsResult.printBenchmark(elapsedTimeMilliseconds)
      if (output) {
        console.log(`${chalk.bold('Output:')}`)
        console.log(stdout)
      }
    } catch (error: any) {
      loader.fail()
      throw new Error(`solution: ${this.path}\n${error.message as string}\n`)
    }
  }

  static async generate(options: GenerateSolutionOptions): Promise<Solution> {
    const { name, challengeName, programmingLanguageName, githubUser } = options
    const challenge = new Challenge({ name: challengeName })
    if (!(await isExistingPath(challenge.path))) {
      throw new Error(`The challenge doesn't exist yet: ${challenge.name}.`)
    }
    const solution = new Solution({
      name,
      challenge,
      programmingLanguageName
    })
    if (await isExistingPath(solution.path)) {
      throw new Error(`The solution already exists: ${name}.`)
    }
    await template.solution({
      challengeName: challenge.name,
      destination: solution.path,
      githubUser,
      programmingLanguageName: solution.programmingLanguageName,
      name: solution.name
    })
    return solution
  }

  static async get(options: GetSolutionOptions): Promise<Solution> {
    const { name, challengeName, programmingLanguageName } = options
    const challenge = new Challenge({
      name: challengeName
    })
    const solution = new Solution({
      name,
      challenge,
      programmingLanguageName
    })
    if (!(await isExistingPath(solution.path))) {
      throw new Error('The solution was not found.')
    }
    return solution
  }

  static async getManyByChallenge(challenge: Challenge): Promise<Solution[]> {
    const solutionsPath = path.join(challenge.path, 'solutions')
    const languagesSolution = (await fs.promises.readdir(solutionsPath)).filter(
      (name) => {
        return name !== '.gitkeep'
      }
    )
    const paths: string[] = []
    for (const language of languagesSolution) {
      const solutionPath = (
        await fs.promises.readdir(path.join(solutionsPath, language))
      ).map((solutionName) => {
        return `challenges/${challenge.name}/solutions/${language}/${solutionName}`
      })
      paths.push(...solutionPath)
    }
    return await Solution.getManyByPaths(paths)
  }

  static async getManyByProgrammingLanguages(
    programmingLanguages?: string[]
  ): Promise<Solution[]> {
    const languages =
      programmingLanguages ?? (await template.getProgrammingLanguages())
    const challengesPath = fileURLToPath(
      new URL('../../challenges', import.meta.url)
    )
    const challenges = await fs.promises.readdir(challengesPath)
    const paths: string[] = []
    for (const challenge of challenges) {
      const solutionsPath = path.join(challengesPath, challenge, 'solutions')
      const languagesSolution = (
        await fs.promises.readdir(solutionsPath)
      ).filter((name) => {
        return name !== '.gitkeep' && languages.includes(name)
      })
      for (const language of languagesSolution) {
        const solutionPath = (
          await fs.promises.readdir(path.join(solutionsPath, language))
        ).map((solutionName) => {
          return `challenges/${challenge}/solutions/${language}/${solutionName}`
        })
        paths.push(...solutionPath)
      }
    }
    return await Solution.getManyByPaths(paths)
  }

  /**
   * Get Solutions by relative paths.
   * @param paths relative to `challenges` (e.g: `challenges/hello-world/solutions/c/function`)
   * @returns
   */
  static async getManyByPaths(paths: string[]): Promise<Solution[]> {
    const solutions: string[] = []
    for (const path of paths) {
      if (await isExistingPath(path)) {
        solutions.push(path)
      }
    }
    return solutions.map((solution) => {
      const [, challengeName, , programmingLanguageName, solutionName] =
        solution.replaceAll('\\', '/').split('/')

      if (
        challengeName == null ||
        programmingLanguageName == null ||
        solutionName == null
      ) {
        throw new Error(`Invalid solution path: ${solution}`)
      }

      return new Solution({
        challenge: new Challenge({
          name: challengeName
        }),
        name: solutionName,
        programmingLanguageName
      })
    })
  }
}