main.ts

import * as core from '@actions/core'
import {getTeamLabel} from './teams'
import {
  getPrNumber,
  getPrAuthor,
  getLabelsConfiguration,
  addLabels,
  createClient
} from './github'

async function run() {
  try {
    const token = core.getInput('repo-token', {required: true})
    const configPath = core.getInput('configuration-path', {required: true})
    const teamsRepo = core.getInput('teams-repo', {required: false})
    const teamsBranch = core.getInput('teams-branch', {required: false})

    const prNumber = getPrNumber()
    if (!prNumber) {
      core.info('Could not get pull request number from context, exiting')
      return
    }

    const author = getPrAuthor()
    if (!author) {
      core.info('Could not get pull request user from context, exiting')
      return
    }

    const client = createClient(token)
    const labelsConfiguration: Map<string, string[]> =
      await getLabelsConfiguration(
        client,
        configPath,
        teamsRepo !== '' ? {repo: teamsRepo, ref: teamsBranch} : undefined
      )

    const labels: string[] = getTeamLabel(labelsConfiguration, `@${author}`)

    if (labels.length > 0) await addLabels(client, prNumber, labels)
  } catch (error) {
    if (error instanceof Error) {
      core.error(error)
      core.setFailed(error.message)
    }
  }
}

run()