programming-challenges/cli/services/GitAffected.ts

import { execaCommand } from 'execa'

import { Challenge } from './Challenge.js'
import { Solution } from './Solution.js'
import { parseCommandOutput } from '../utils/parseCommandOutput.js'

const solutionsRegex =
  /challenges\/[\S\s]*\/solutions\/(c|cpp|cs|dart|java|javascript|python|rust|typescript)\/[\S\s]*\/(.*).(c|cpp|cs|dart|java|js|py|rs|ts)/

const dockerRegex =
  /templates\/docker\/(c|cpp|cs|dart|java|javascript|python|rust|typescript)\/(.*)/

const inputOutputRegex =
  /challenges\/[\S\s]*\/test\/(.*)\/(input.txt|output.txt)/

export interface GitAffectedOptions {
  base?: string
}

export class GitAffected implements GitAffectedOptions {
  public base?: string

  constructor(options: GitAffectedOptions = {}) {
    this.base = options.base
  }

  public async getFilesUsingBaseAndHead(
    base: string,
    head: string
  ): Promise<string[]> {
    try {
      const { stdout } = await execaCommand(
        `git diff --name-only --relative ${base} ${head}`
      )
      return parseCommandOutput(stdout)
    } catch {
      return []
    }
  }

  public async getUncommittedFiles(): Promise<string[]> {
    return await this.getFilesUsingBaseAndHead('HEAD', '.')
  }

  public async getLatestPushedCommit(): Promise<string> {
    const latestCommit = this.base != null ? '~1' : ''
    const { stdout } = await execaCommand(
      `git rev-parse origin/master${latestCommit}`
    )
    return stdout
  }

  public async getUnpushedFiles(): Promise<string[]> {
    return await this.getFilesUsingBaseAndHead(
      await this.getLatestPushedCommit(),
      '.'
    )
  }

  public async getAffectedSolutionsFromFiles(
    files: string[]
  ): Promise<Solution[]> {
    const affectedSolutionsPaths = files.filter((filePath) => {
      return solutionsRegex.test(filePath)
    })
    const affectedDockerPaths = files.filter((filePath) => {
      return dockerRegex.test(filePath)
    })
    const affectedLanguages = affectedDockerPaths.map((filePath) => {
      const [, , programmingLanguageName] = filePath
        .replaceAll('\\', '/')
        .split('/')
      if (programmingLanguageName == null) {
        throw new Error('programmingLanguageName is null')
      }
      return programmingLanguageName
    })
    const affectedInputOutput = files.filter((filePath) => {
      return inputOutputRegex.test(filePath)
    })
    const affectedChallengesFromInputOutput = affectedInputOutput.map(
      (filePath) => {
        const [, challengeName] = filePath.replaceAll('\\', '/').split('/')
        if (challengeName == null) {
          throw new Error('challengeName is null')
        }
        return new Challenge({ name: challengeName })
      }
    )
    const solutionsChallenges = await Solution.getManyByPaths(
      affectedSolutionsPaths
    )
    const solutionsDocker = await Solution.getManyByProgrammingLanguages(
      affectedLanguages
    )
    const solutions: Solution[] = [...solutionsDocker, ...solutionsChallenges]
    for (const challenge of affectedChallengesFromInputOutput) {
      const solutionsByChallenge = await Solution.getManyByChallenge(challenge)
      solutions.push(...solutionsByChallenge)
    }
    const solutionsUnique: Solution[] = []
    for (const solution of solutions) {
      const isAlreadyIncluded = solutionsUnique.some((solutionUnique) => {
        return solutionUnique.path === solution.path
      })
      if (!isAlreadyIncluded) {
        solutionsUnique.push(solution)
      }
    }
    return solutionsUnique
  }

  public async getAffectedSolutionsFromGit(): Promise<Solution[]> {
    let files = [
      ...(await this.getUnpushedFiles()),
      ...(await this.getUncommittedFiles())
    ]
    if (this.base != null) {
      files.push(...(await this.getFilesUsingBaseAndHead(this.base, '.')))
    }
    files = Array.from(new Set(files))
    return await this.getAffectedSolutionsFromFiles(files)
  }
}