fabasoad/nsfw-detection-action

import { context, getOctokit } from '@actions/github'
import LoggerFactory from './LoggerFactory'
import { Logger } from 'winston'
import { WebhookPayload } from '@actions/github/lib/interfaces'
import {
  GetResponseTypeFromEndpointMethod,
  GetResponseDataTypeFromEndpointMethod
} from '@octokit/types'

export class GitHubClient {
  private readonly logger: Logger = LoggerFactory.create(GitHubClient.name)

  public async getChangedFiles(
    gitHubToken: string,
    types: string[],
    extensions: string[]): Promise<Set<string>> {
    const octokit = getOctokit(gitHubToken)

    type CompareCommitsResponseType = GetResponseTypeFromEndpointMethod<
      typeof octokit.rest.repos.compareCommits
    >
    type CompareCommitsResponseDataType = GetResponseDataTypeFromEndpointMethod<
      typeof octokit.rest.repos.compareCommits
    >
    const payload: WebhookPayload = context.payload
    const repo = context.repo.repo
    const owner = context.repo.owner

    const resp: CompareCommitsResponseType =
      await octokit.rest.repos.compareCommits(
        { owner, repo, base: payload.before, head: payload.after }
      )
    const data: CompareCommitsResponseDataType = resp.data
    if (!data.files) {
      throw new Error('Cannot retrieve files list')
    }
    const count = data.files.length;
    this.logger.info(`There ${count > 1 ? 'are' : 'is'} ${count} ` +
      `file${count > 1 ? 's' : ''} found between ${payload.before} and ` +
      `${payload.after} commits`)
    const result = new Set<string>()
    for (const file of data.files) {
      this.logger.debug(`File: ${file.filename}. Status: ${file.status}`)
      if (types.includes(file.status)) {
        const temp: string[] = file.filename.split('.')
        if (extensions.map((e: string) => e.toLowerCase())
          .includes(temp[temp.length - 1].toLowerCase())) {
          result.add(file.filename)
        }
      }
    }
    this.logger.info(`There ${result.size === 1 ? 'is' : 'are'}` +
      ` ${result.size} file${result.size === 1 ? '' : 's'} will be checked`)
    return result
  }
}