✨ Add label to PR [skip-ci]

BetaHuhn · Jan 29, 2021 · 789ec9f · 789ec9f
1 parent 8d02dfe
commit 789ec9f
Show file tree

Hide file tree

Showing 4 changed files with 60 additions and 6 deletions.
diff --git a/dist/index.js b/dist/index.js
@@ -25448,6 +25448,9 @@ const getVar = ({ key, default: dft, required = false, type = 'string' }) => {
 		envVar = process.env[key]
 	}
 
+	if (key === 'PR_LABELS' && (coreVar === false || envVar === 'false'))
+		return undefined
+
 	if (coreVar !== undefined && coreVar.length >= 1) {
 		if (type === 'array') return coreVar.split('\n')
 
@@ -25500,6 +25503,11 @@ const context = {
 		type: 'boolean',
 		default: true
 	}),
+	PR_LABELS: getVar({
+		key: 'PR_LABELS',
+		default: [ 'deployed' ],
+		type: 'array'
+	}),
 	VERCEL_SCOPE: getVar({
 		key: 'VERCEL_SCOPE'
 	}),
@@ -25564,7 +25572,8 @@ const {
 	PRODUCTION,
 	PR_NUMBER,
 	REF,
-	LOG_URL
+	LOG_URL,
+	PR_LABELS
 } = __nccwpck_require__(4570)
 
 const init = () => {
@@ -25629,11 +25638,23 @@ const init = () => {
 		return comment.data
 	}
 
+	const addLabel = async () => {
+		const label = await client.issues.addLabels({
+			owner: USER,
+			repo: REPOSITORY,
+			issue_number: PR_NUMBER,
+			labels: PR_LABELS
+		})
+
+		return label.data
+	}
+
 	return {
 		client,
 		createDeployment,
 		updateDeployment,
-		createComment
+		createComment,
+		addLabel
 	}
 }
 
@@ -25703,7 +25724,8 @@ const vercel = __nccwpck_require__(847)
 
 const {
 	GITHUB_DEPLOYMENT,
-	IS_PR
+	IS_PR,
+	PR_LABELS
 } = __nccwpck_require__(4570)
 
 const run = async () => {
@@ -25737,6 +25759,11 @@ const run = async () => {
 			log.info(`Created comment on PR: ${ comment.html_url }`)
 		}
 
+		if (IS_PR && PR_LABELS) {
+			const labels = await github.addLabel()
+			log.info(`Added label(s) ${ labels.map((label) => label.name).join(', ') } to PR`)
+		}
+
 		// Set Action output
 		core.setOutput('PREVIEW_URL', previewUrl)
 		core.setOutput('DEPLOYMENT_CREATED', GITHUB_DEPLOYMENT)

diff --git a/src/config.js b/src/config.js
@@ -25,6 +25,9 @@ const getVar = ({ key, default: dft, required = false, type = 'string' }) => {
 		envVar = process.env[key]
 	}
 
+	if (key === 'PR_LABELS' && (coreVar === false || envVar === 'false'))
+		return undefined
+
 	if (coreVar !== undefined && coreVar.length >= 1) {
 		if (type === 'array') return coreVar.split('\n')
 
@@ -77,6 +80,11 @@ const context = {
 		type: 'boolean',
 		default: true
 	}),
+	PR_LABELS: getVar({
+		key: 'PR_LABELS',
+		default: [ 'deployed' ],
+		type: 'array'
+	}),
 	VERCEL_SCOPE: getVar({
 		key: 'VERCEL_SCOPE'
 	}),

diff --git a/src/github.js b/src/github.js
@@ -9,7 +9,8 @@ const {
 	PRODUCTION,
 	PR_NUMBER,
 	REF,
-	LOG_URL
+	LOG_URL,
+	PR_LABELS
 } = require('./config')
 
 const init = () => {
@@ -74,11 +75,23 @@ const init = () => {
 		return comment.data
 	}
 
+	const addLabel = async () => {
+		const label = await client.issues.addLabels({
+			owner: USER,
+			repo: REPOSITORY,
+			issue_number: PR_NUMBER,
+			labels: PR_LABELS
+		})
+
+		return label.data
+	}
+
 	return {
 		client,
 		createDeployment,
 		updateDeployment,
-		createComment
+		createComment,
+		addLabel
 	}
 }
 

diff --git a/src/index.js b/src/index.js
@@ -12,7 +12,8 @@ const vercel = require('./vercel')
 
 const {
 	GITHUB_DEPLOYMENT,
-	IS_PR
+	IS_PR,
+	PR_LABELS
 } = require('./config')
 
 const run = async () => {
@@ -46,6 +47,11 @@ const run = async () => {
 			log.info(`Created comment on PR: ${ comment.html_url }`)
 		}
 
+		if (IS_PR && PR_LABELS) {
+			const labels = await github.addLabel()
+			log.info(`Added label(s) ${ labels.map((label) => label.name).join(', ') } to PR`)
+		}
+
 		// Set Action output
 		core.setOutput('PREVIEW_URL', previewUrl)
 		core.setOutput('DEPLOYMENT_CREATED', GITHUB_DEPLOYMENT)