Add PR label

.vscode/launch.json

@@ -7,7 +7,7 @@
     {
       "type": "node",
       "request": "launch",
-      "name": "Launch Program",
+      "name": "Run file",
       "program": "${file}"
     },
     {

src/cli.js

@@ -1,5 +1,9 @@
 const github = require('./github');
-const { ensureConfigAndFoldersExists, validateConfig } = require('./configs');
+const {
+  ensureConfigAndFoldersExists,
+  validateConfig,
+  getRepoConfig
+} = require('./configs');
 const {
   promptRepoInfo,
   promptCommit,
@@ -11,7 +15,7 @@ const {
 } = require('./cliService');
 
 function init(config, options) {
-  let commit, versions, reference, owner, repoName;
+  let commit, versions, reference, owner, repoName, repoConfig;
 
   return ensureConfigAndFoldersExists()
     .then(() => validateConfig(config))
@@ -20,14 +24,13 @@ function init(config, options) {
     .then(({ owner: _owner, repoName: _repoName }) => {
       owner = _owner;
       repoName = _repoName;
+      repoConfig = getRepoConfig(owner, repoName, config.repositories);
     })
     .then(() =>
       promptCommit(owner, repoName, options.own ? config.username : null)
     )
     .then(c => (commit = c))
-    .then(() =>
-      promptVersions(owner, repoName, config.repositories, options.multiple)
-    )
+    .then(() => promptVersions(repoConfig.versions, options.multiple))
     .then(v => (versions = v))
     .then(() => getReference(owner, repoName, commit.sha))
     .then(ref => (reference = ref))
@@ -39,7 +42,8 @@ function init(config, options) {
         commit,
         reference,
         versions,
-        username: config.username
+        username: config.username,
+        labels: repoConfig.labels
       })
     )
     .catch(handleErrors);

src/cliService.js

@@ -15,21 +15,15 @@ const {
   setupRepo
 } = require('./git');
 
-function sequentially(items, handler) {
-  return items.reduce(
-    (p, item) => p.then(() => handler(item)),
-    Promise.resolve()
-  );
-}
-
 const service = {};
 service.doBackportVersions = ({
   owner,
   repoName,
   commit,
   reference,
   versions,
-  username
+  username,
+  labels
 }) => {
   return sequentially(versions, version => {
     return service
@@ -39,7 +33,8 @@ service.doBackportVersions = ({
         commit,
         reference,
         version,
-        username
+        username,
+        labels
       })
       .then(res => console.log(`View pull request: ${res.data.html_url}\n`))
       .catch(service.handleErrors);
@@ -52,13 +47,12 @@ service.doBackportVersion = ({
   commit,
   reference,
   version,
-  username
+  username,
+  labels = []
 }) => {
   const backportBranchName = getBackportBranchName(version, reference);
 
-  console.log(
-    `Backporting ${service.getReferenceValue(reference)} to ${version}`
-  );
+  console.log(`Backporting ${getReferenceLong(reference)} to ${version}`);
 
   return withSpinner(
     resetAndPullMaster(owner, repoName).then(() =>
@@ -81,7 +75,14 @@ service.doBackportVersion = ({
         username
       );
       return withSpinner(
-        github.createPullRequest(owner, repoName, payload),
+        github.createPullRequest(owner, repoName, payload).then(res => {
+          if (labels.length > 0) {
+            return github
+              .addLabels(owner, repoName, res.data.number, labels)
+              .then(() => res);
+          }
+          return res;
+        }),
         'Creating pull request'
       );
     });
@@ -108,9 +109,7 @@ service.promptRepoInfo = (repositories, cwd) => {
         console.log(`Repository: ${currentFullRepoName}`);
         return currentFullRepoName;
       }
-      return prompts
-        .listFullRepoName(fullRepoNames)
-        .then(({ fullRepoName }) => fullRepoName);
+      return prompts.listFullRepoName(fullRepoNames);
     })
     .then(fullRepoName => {
       const [owner, repoName] = fullRepoName.split('/');
@@ -140,22 +139,13 @@ service.promptCommit = (owner, repoName, username) => {
     .then(commits => {
       spinner.stop();
       return prompts.listCommits(commits);
-    })
-    .then(({ commit }) => commit);
+    });
 };
 
-service.promptVersions = (
-  owner,
-  repoName,
-  repositories,
-  multipleChoice = false
-) => {
-  const versions = getVersions(owner, repoName, repositories);
-  if (multipleChoice) {
-    return prompts.checkboxVersions(versions).then(({ versions }) => versions);
-  }
-
-  return prompts.listVersions(versions).then(({ version }) => [version]);
+service.promptVersions = (versions, multipleChoice = false) => {
+  return multipleChoice
+    ? prompts.checkboxVersions(versions)
+    : prompts.listVersions(versions);
 };
 
 service.handleErrors = e => {
@@ -207,15 +197,27 @@ service.handleErrors = e => {
   }
 };
 
-service.getReferenceValue = reference => {
-  return reference.type === 'pullRequest'
-    ? `pull request #${reference.value}`
-    : `commit ${reference.value}`;
-};
+function sequentially(items, handler) {
+  return items.reduce(
+    (p, item) => p.then(() => handler(item)),
+    Promise.resolve()
+  );
+}
 
-function getVersions(owner, repoName, repositories) {
-  return repositories.find(repo => repo.name === `${owner}/${repoName}`)
-    .versions;
+function getReferenceValue({ type, value }, { short }) {
+  if (type === 'pullRequest') {
+    return short ? `pr-${value}` : `pull request #${value}`;
+  }
+
+  return short ? `commit-${value}` : `commit ${value}`;
+}
+
+function getReferenceLong(reference) {
+  return getReferenceValue(reference, { short: false });
+}
+
+function getReferenceShort(reference) {
+  return getReferenceValue(reference, { short: true });
 }
 
 function isCherrypickConflict(e) {
@@ -235,7 +237,7 @@ function cherrypickAndPrompt(owner, repoName, sha) {
       throw e;
     }
 
-    return prompts.confirmConflictResolved().then(({ isConflictResolved }) => {
+    return prompts.confirmConflictResolved().then(isConflictResolved => {
       if (!isConflictResolved) {
         const error = new Error(constants.CHERRYPICK_CONFLICT_NOT_HANDLED);
         error.details = e.message;
@@ -246,24 +248,18 @@ function cherrypickAndPrompt(owner, repoName, sha) {
 }
 
 function getBackportBranchName(version, reference) {
-  const refValue = getReferenceValueShort(reference);
+  const refValue = getReferenceShort(reference);
   return `backport/${version}/${refValue}`;
 }
 
-function getReferenceValueShort(reference) {
-  return reference.type === 'pullRequest'
-    ? `pr-${reference.value}`
-    : `commit-${reference.value}`;
-}
-
 function getCurrentFullRepoName(fullRepoNames, cwd) {
   const currentDir = path.basename(cwd);
   return fullRepoNames.find(name => name.endsWith(`/${currentDir}`));
 }
 
 function getPullRequestPayload(commitMessage, version, reference, username) {
   const backportBranchName = getBackportBranchName(version, reference);
-  const refValue = service.getReferenceValue(reference);
+  const refValue = getReferenceLong(reference);
 
   return {
     title: `[${version}] ${commitMessage}`,

src/configs.js

@@ -27,6 +27,10 @@ function ensureConfigAndFoldersExists() {
     });
 }
 
+function getRepoConfig(owner, repoName, repositories) {
+  return repositories.find(repo => repo.name === `${owner}/${repoName}`);
+}
+
 function getConfigTemplate() {
   return utils.readFile(path.join(__dirname, 'configTemplate.json'), 'utf8');
 }
@@ -63,5 +67,6 @@ function getConfig() {
 module.exports = {
   ensureConfigAndFoldersExists,
   getConfig,
+  getRepoConfig,
   validateConfig
 };

src/git.js

@@ -79,16 +79,11 @@ function resetAndPullMaster(owner, repoName) {
   );
 }
 
-function getCommit(repo, sha) {
-  return repo.getCommit(sha);
-}
-
 module.exports = {
   resetAndPullMaster,
   cherrypick,
   cloneRepo,
   createAndCheckoutBranch,
-  getCommit,
   repoExists,
   setupRepo,
   push

src/github.js

@@ -55,6 +55,15 @@ function createPullRequest(owner, repoName, payload) {
     .catch(throwGithubError);
 }
 
+function addLabels(owner, repoName, pullNumber, labels) {
+  return axios
+    .post(
+      `https://api.github.com/repos/${owner}/${repoName}/issues/${pullNumber}/labels?access_token=${accessToken}`,
+      labels
+    )
+    .catch(throwGithubError);
+}
+
 function getPullRequestByCommit(owner, repoName, commitSha) {
   return axios(
     `https://api.github.com/search/issues?q=repo:${owner}/${repoName}+${commitSha}&access_token=${accessToken}`
@@ -69,6 +78,7 @@ function setAccessToken(_accessToken) {
 
 module.exports = {
   setAccessToken,
+  addLabels,
   createPullRequest,
   getCommits,
   getPullRequestByCommit