Merge pull request #93 from ghinks/feature/close-draft-pr

Author: dominykas

.wiby.json

@@ -1,13 +1,16 @@
 {
   "dependents": [
     {
-      "repository": "https://github.com/wiby-test/partial"
+      "repository": "https://github.com/wiby-test/partial",
+      "pullRequest": false
     },
     {
-      "repository": "git://github.com/wiby-test/fail"
+      "repository": "git://github.com/wiby-test/fail",
+      "pullRequest": false
     },
     {
-      "repository": "git+https://github.com/wiby-test/pass"
+      "repository": "git+https://github.com/wiby-test/pass",
+      "pullRequest": true
     }
   ]
 }

bin/commands/close-pr.js

@@ -0,0 +1,29 @@
+'use strict'
+
+const wiby = require('../..')
+
+exports.desc = 'Use this command to close the PRs raised against your dependents. wiby will go off to the dependent’s repo and close the PRs raised that trigger jobs  `package.json` pointing to your latest version (with the new changes) triggering the dependent’s CI to run.'
+
+exports.builder = (yargs) => yargs
+  .option('dependent', {
+    desc: 'URL of a dependent',
+    type: 'string',
+    conflicts: 'config'
+  })
+  .option('config', {
+    desc: 'Path to the configuration file. By default it will try to load the configuration from the first file it finds in the current working directory: `.wiby.json`, `.wiby.js`',
+    type: 'string'
+  })
+
+exports.handler = async (params) => {
+  const config = params.dependent
+    ? {
+        dependents: [{ repository: params.dependent, pullRequest: true }]
+      }
+    : wiby.validate({ config: params.config })
+
+  const result = await wiby.closePR(config)
+  // TODO, something more like the result process output
+  const output = `${result.length} PRs closed`
+  console.log(output)
+}

bin/commands/result.js

@@ -17,7 +17,9 @@ exports.builder = (yargs) => yargs
 
 exports.handler = async (params) => {
   const config = params.dependent
-    ? { dependents: [{ repository: params.dependent }] }
+    ? {
+        dependents: [{ repository: params.dependent }]
+      }
     : wiby.validate({ config: params.config })
 
   const result = await wiby.result(config)

bin/commands/test.js

@@ -10,14 +10,22 @@ exports.builder = (yargs) => yargs
     type: 'string',
     conflicts: 'config'
   })
+  .option('pull-request', {
+    desc: 'Raise a draft PR in addition to creating a branch',
+    alias: 'pr',
+    type: 'boolean',
+    conflicts: 'config'
+  })
   .option('config', {
     desc: 'Path to the configuration file. By default it will try to load the configuration from the first file it finds in the current working directory: `.wiby.json`, `.wiby.js`',
     type: 'string'
   })
 
 exports.handler = (params) => {
   const config = params.dependent
-    ? { dependents: [{ repository: params.dependent }] }
+    ? {
+        dependents: [{ repository: params.dependent, pullRequest: !!params['pull-request'] }]
+      }
     : wiby.validate({ config: params.config })
 
   return wiby.test(config)

lib/closePR.js

@@ -0,0 +1,49 @@
+'use strict'
+
+const github = require('../lib/github')
+const logger = require('./logger')
+const context = require('./context')
+const gitURLParse = require('git-url-parse')
+
+const debug = logger('wiby:closepr')
+
+module.exports = async ({ dependents }) => {
+  const closedPrs = []
+  for (const { repository: url, pullRequest } of dependents) {
+    if (pullRequest) {
+      const dependentPkgInfo = gitURLParse(url)
+      const parentBranchName = await context.getParentBranchName()
+      const branch = await context.getTestingBranchName(parentBranchName)
+      const resp = await github.getChecks(dependentPkgInfo.owner, dependentPkgInfo.name, branch)
+      const closedPR = await closeDependencyPR(dependentPkgInfo.owner, dependentPkgInfo.name, branch, resp.data.check_runs)
+      if (closedPR) {
+        closedPrs.push(closedPR)
+      }
+    }
+  }
+  return closedPrs
+}
+
+const closeDependencyPR = module.exports.closeDependencyPR = async function closeDependencyPR (owner, repo, branch, checkRuns) {
+  if (!checkRuns) {
+    return
+  }
+  // TODO, in reality multiple checks could exist and they may not all have passed
+  const prsToClose = checkRuns.reduce((acc, check) => {
+    if (check.status === 'completed' &&
+      check.conclusion === 'success' &&
+      check.pull_requests &&
+      check.pull_requests.length !== 0) {
+      check.pull_requests.forEach((pr) => {
+        if (pr.head.ref === branch) {
+          acc.push({
+            number: pr.number
+          })
+        }
+      })
+    }
+    return acc
+  }, [])
+  debug(`Dependent module: ${JSON.stringify(prsToClose, null, 2)}`)
+  return await Promise.all(prsToClose.map((pr) => github.closePR(owner, repo, pr.number)))
+}

lib/config.js

@@ -13,7 +13,8 @@ const dependentSchema = joi.object({
       'git',
       'git+https'
     ]
-  })
+  }),
+  pullRequest: joi.boolean().optional()
 }).unknown(false)
 
 exports.schema = joi.object({

lib/github.js

@@ -42,6 +42,15 @@ module.exports.getWibyBranches = async function (owner, repo) {
   return edges.map(({ node: { branchName } }) => branchName)
 }
 
+module.exports.getDefaultBranch = async function (owner, repo) {
+  const resp = await graphqlWithAuth({
+    query: queries.getDefaultBranch,
+    owner: owner,
+    repo: repo
+  })
+  return resp.repository.defaultBranchRef.name
+}
+
 module.exports.getShas = async function getShas (owner, repo) {
   const resp = await octokit.repos.listCommits({
     owner,
@@ -136,3 +145,24 @@ module.exports.getCommitStatusesForRef = async function getCommitStatusesForRef
     ref: branch
   })
 }
+
+module.exports.createPR = async function createPR (owner, repo, title, head, base, draft, body) {
+  return octokit.pulls.create({
+    owner,
+    repo,
+    title,
+    head,
+    base,
+    draft,
+    body
+  })
+}
+
+module.exports.closePR = async function closePR (owner, repo, pullNumber) {
+  return octokit.rest.pulls.update({
+    owner,
+    repo,
+    pull_number: pullNumber,
+    state: 'closed'
+  })
+}

lib/graphql/getDefaultBranch.graphql

@@ -0,0 +1,7 @@
+query getExistingRepoBranches($owner: String!, $repo: String!) {
+    repository(name: $repo, owner: $owner) {
+        defaultBranchRef {
+            name
+        }
+    }
+}

lib/graphql/index.js

@@ -6,3 +6,5 @@ const path = require('path')
 exports.getPackageJson = fs.readFileSync(path.join(__dirname, 'getPackageJson.graphql')).toString()
 
 exports.getWibyBranches = fs.readFileSync(path.join(__dirname, 'getWibyBranches.graphql')).toString()
+
+exports.getDefaultBranch = fs.readFileSync(path.join(__dirname, 'getDefaultBranch.graphql')).toString()

lib/index.js

@@ -7,3 +7,5 @@ exports.test = require('./test')
 exports.result = require('./result')
 
 exports.validate = require('./config').validate
+
+exports.closePR = require('./closePR')