github.js

const { context, getOctokit } = require('@actions/github')

function getPrNumber () {
  const pullRequest = context.payload.pull_request
  if (!pullRequest) {
    return undefined
  }

  return pullRequest.number
}

function getIssueNumber () {
  const issue = context.payload.issue
  if (!issue) {
    return undefined
  }

  return issue.number
}

function getRepo () {
  return context.repo
}

async function getIssue (token) {
  const octokit = getOctokit(token)
  let issueNumber
  if (getIssueNumber() !== undefined) {
    issueNumber = getIssueNumber()
  } else if (getPrNumber() !== undefined) {
    issueNumber = getPrNumber()
  } else {
    throw new Error('No Issue Provided')
  }

  const { data } = await octokit.rest.issues.get({
    ...getRepo(),
    issue_number: issueNumber
  })

  return data
}

async function createAndInviteToRepo (token, owner, username, repo) {
  const octokit = getOctokit(token)
  try {
    await octokit.rest.repos.createInOrg({
      org: owner,
      name: repo,
      has_issues: false,
      has_projects: false,
      has_wiki: false
    })
  } catch (err) {
    let e = err
    if (e.name === 'HttpError') e = e.response.data
    if (e && e.errors && e.errors.length && e.errors[0].field === 'name') {
      return false
    }
    console.log('嘤嘤嘤', JSON.stringify(err))
    throw err
  }
  for (let i = 0; i < 3; i++) {
    try {
      await octokit.rest.repos.addCollaborator({
        owner,
        repo,
        username,
        permission: 'admin'
      })
      break
    } catch (err) {
      let e = err
      if (e.status === 404) {
        await new Promise(resolve => setTimeout(resolve, 2000))
        continue
      }
      if (e.name === 'HttpError') e = e.response.data
      if (e && e.errors && e.errors.length && e.errors[0].field === 'name') {
        return false
      }
      console.log('呜呜呜', JSON.stringify(err))
      throw err
    }
  }
  return true
}

async function addLabel (token, owner, repo, issueNumber, label) {
  const octokit = getOctokit(token)
  await octokit.rest.issues.addLabels({
    owner,
    repo,
    issue_number: issueNumber,
    labels: [label]
  })
}

async function setLabel (token, owner, repo, issueNumber, label) {
  const octokit = getOctokit(token)
  await octokit.rest.issues.setLabels({
    owner,
    repo,
    issue_number: issueNumber,
    labels: [label]
  })
}

async function leaveComment (token, owner, repo, issueNumber, comment) {
  const octokit = getOctokit(token)
  await octokit.rest.issues.createComment({
    owner,
    repo,
    issue_number: issueNumber,
    body: comment
  })
}

async function closeIssue (token, owner, repo, issueNumber, isCompleted = false) {
  const octokit = getOctokit(token)
  await octokit.rest.issues.update({
    owner,
    repo,
    issue_number: issueNumber,
    state: 'closed',
    state_reason: isCompleted ? 'completed' : 'not_planned'
  })
}

async function lockSpamIssue (token, owner, repo, issueNumber) {
  const octokit = getOctokit(token)
  await octokit.rest.issues.lock({
    owner,
    repo,
    issue_number: issueNumber,
    lock_reason: 'spam'
  })
}

async function orgBlockUser (token, owner, username) {
  const octokit = getOctokit(token)
  await octokit.rest.orgs.blockUser({
    org: owner,
    username
  })
}

async function getUser (token, username) {
  const octokit = getOctokit(token)
  return await octokit.rest.users.getByUsername({
    username
  })
}

module.exports = {
  getRepo,
  getIssue,
  createAndInviteToRepo,
  addLabel,
  setLabel,
  leaveComment,
  closeIssue,
  lockSpamIssue,
  orgBlockUser,
  getUser
}