fix: add labeling function

.github/workflows/auto-pull-request.yml

@@ -22,3 +22,4 @@ jobs:
           target-branch: "release/v1"
           title: "Main to Release v1"
           body: "main to Release v1"
+          lables: "release, test"

src/github-utils.ts

@@ -3,9 +3,9 @@ import { getOctokit } from '@actions/github'
 
 type OctoKit = ReturnType<typeof getOctokit>
 
-export type PullRequestCreateParam = Parameters<
-  OctoKit['rest']['pulls']['create']
->[number]
+type PullRequestCreateParam = Parameters<OctoKit['rest']['pulls']['create']>[0]
+
+type issueAddLabelsParam = Parameters<OctoKit['rest']['issues']['addLabels']>[0]
 
 /** Pull Request response in list API */
 export type PullRequestItem = Awaited<
@@ -48,3 +48,17 @@ export async function createPullRequest(
     setFailed(`Error creating pull request: ${error.message}`)
   }
 }
+
+/** Add labels to pull request */
+export async function addLabelsToPullRequest(
+  octokit: OctoKit,
+  param: issueAddLabelsParam
+) {
+  debug(`Adding labels to pull request: ${JSON.stringify(param)}`)
+  try {
+    const response = await octokit.rest.issues.addLabels(param)
+    return response.data
+  } catch (error: any) {
+    setFailed(`Error adding labels to pull request: ${error.message}`)
+  }
+}

src/run.ts

@@ -4,6 +4,7 @@ import { fetchRemoteBranches, hasCommitsBetween } from './git-util'
 import {
   PullRequestDetailItem,
   PullRequestItem,
+  addLabelsToPullRequest,
   createPullRequest,
   fetchExistingPullRequest
 } from './github-utils'
@@ -84,5 +85,16 @@ export async function run(options: OptionParams) {
 
   if (!pullRequest) return
 
-  info(`Pull request created: ${pullRequest.html_url}`)
+  if (options.labels.length > 0) {
+    /** Adds labels to the pull request */
+    const labels = await addLabelsToPullRequest(octokit, {
+      owner,
+      repo,
+      issue_number: pullRequest.number,
+      labels: options.labels
+    })
+    if (labels) {
+      info(`Added labels to pull request: ${labels.map((label) => label.name)}`)
+    }
+  }
 }