From 792e5a4ba7f051f88d07caaa8c2bf64abde7e5d0 Mon Sep 17 00:00:00 2001
From: Anurag Hazra <hazru.anurag@gmail.com>
Date: Mon, 10 Aug 2020 20:01:19 +0530
Subject: [PATCH] infra: getPrNumber fix (#357)

---
 scripts/preview-theme.js | 22 +++++++++++++---------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/scripts/preview-theme.js b/scripts/preview-theme.js
index 4935b778b2604..81d722535f60f 100644
--- a/scripts/preview-theme.js
+++ b/scripts/preview-theme.js
@@ -3,21 +3,25 @@ const github = require("@actions/github");
 const parse = require("parse-diff");
 require("dotenv").config();
 
-const parsePullRequestId = (githubRef) => {
-  const result = /refs\/pull\/(\d+)\/merge/g.exec(githubRef);
-  if (!result) {
-    console.log("Reference not found.");
-    return 297;
+function getPrNumber() {
+  const pullRequest = github.context.payload.pull_request;
+  if (!pullRequest) {
+    return undefined;
   }
-  const [, pullRequestId] = result;
-  return pullRequestId;
-};
+
+  return pullRequest.number;
+}
 
 async function run() {
   try {
     const token = core.getInput("token");
     const octokit = github.getOctokit(token || process.env.PERSONAL_TOKEN);
-    const pullRequestId = parsePullRequestId(process.env.GITHUB_REF);
+    const pullRequestId = getPrNumber();
+
+    if (!pullRequestId) {
+      console.log("PR not found");
+      return;
+    }
 
     let res = await octokit.pulls.get({
       owner: "anuraghazra",