Create labels when repo is added to installation

index.js

@@ -4,6 +4,7 @@ const issueLabeled = require('./src/issue/labeled')
 const pullLabeled = require('./src/pull/labeled')
 const closed = require('./src/issue/closed')
 const commentDeleted = require('./src/comment/deleted')
+const installationAdded = require('./src/installation/added')
 
 const { CLOSE, MERGE } = require('./src/constants')
 
@@ -72,7 +73,9 @@ module.exports = async (robot, queue = setup()) => {
   // Kill job when issue/pull is closed
   robot.on(['issues.closed', 'pull_request.closed'], closed(queue))
 
-  robot.on(['issue_comment.deleted'], commentDeleted(queue))
+  robot.on('issue_comment.deleted', commentDeleted(queue))
+
+  robot.on(['installation_repositories.added', 'installation.created'], installationAdded(robot))
 
   // For more information on building apps:
   // https://probot.github.io/docs/

src/api.js

@@ -22,3 +22,15 @@ exports.getPullRequest = function getPullRequest(github, data) {
     })
     .then(_ => _.data)
 }
+
+exports.createLabel = function createLabel(github, data) {
+  const { owner, repo, name, color, description } = data
+
+  return github.issues.createLabel({
+    owner,
+    repo,
+    name,
+    color,
+    description
+  })
+}

src/installation/added.js

@@ -0,0 +1,25 @@
+const { createLabel } = require('../api')
+
+module.exports = robot => async context => {
+  try {
+    const github = await robot.auth(context.payload.installation.id)
+
+    const repos = context.payload.repositories_added || context.payload.repositories
+
+    const promises = repos.map(({ name: repo }) => {
+      const data = {
+        owner: context.payload.installation.account.login,
+        repo,
+        name: 'merge when passing',
+        color: 'FF851B',
+        description: 'Merge the PR once all status checks have passed'
+      }
+
+      return createLabel(github, data)
+    })
+
+    return Promise.all(promises)
+  } catch (e) {
+    robot.log.error(e)
+  }
+}

test/fixtures/added.js

@@ -0,0 +1,19 @@
+module.exports = ({
+  repositories_added = [
+    {
+      name: 'ranger'
+    }
+  ]
+}) => ({
+  name: 'installation_repositories',
+  payload: {
+    action: 'added',
+    installation: {
+      id: 533899,
+      account: {
+        login: 'ranger'
+      }
+    },
+    repositories_added
+  }
+})

test/fixtures/created.js

@@ -0,0 +1,19 @@
+module.exports = ({
+  repositories = [
+    {
+      name: 'ranger'
+    }
+  ]
+}) => ({
+  name: 'installation',
+  payload: {
+    action: 'created',
+    installation: {
+      id: 533899,
+      account: {
+        login: 'ranger'
+      }
+    },
+    repositories
+  }
+})

test/index.test.js

@@ -4,6 +4,9 @@ const app = require('..')
 
 const payload = require('./fixtures/labeled')
 const commentPayload = require('./fixtures/comment')
+const addedPayload = require('./fixtures/added')
+const createdPayload = require('./fixtures/created')
+
 const wait = (delay = 0) => new Promise(resolve => setTimeout(resolve, delay))
 
 class MockJob {
@@ -115,6 +118,7 @@ describe('Bot', () => {
     github = {
       issues: {
         createComment: jest.fn(),
+        createLabel: jest.fn(),
         update: jest.fn((_, data) => Promise.resolve({ data }))
       },
       pullRequests: {
@@ -195,8 +199,8 @@ describe('Bot', () => {
         action: 'close'
       }
       expect(queue.createJob).toHaveBeenCalledWith(data)
-      expect(queue.jobs[Object.keys(queue.jobs).slice(-1)[0]].id).toBe('mfix22:test-issue-bot:7:close')
-      expect(queue.jobs[Object.keys(queue.jobs).slice(-1)[0]].data).toEqual(data)
+      expect(queue.jobs[Object.keys(queue.jobs)[0]].id).toBe('mfix22:test-issue-bot:7:close')
+      expect(queue.jobs[Object.keys(queue.jobs)[0]].data).toEqual(data)
     })
 
     test('Will remove the job if an issue is closed', async () => {
@@ -263,8 +267,8 @@ describe('Bot', () => {
       }
 
       expect(queue.createJob).toHaveBeenCalledWith(data)
-      expect(queue.jobs[Object.keys(queue.jobs).slice(-1)[0]].id).toBe('mfix22:test-issue-bot:7:merge')
-      expect(queue.jobs[Object.keys(queue.jobs).slice(-1)[0]].data).toEqual(data)
+      expect(queue.jobs[Object.keys(queue.jobs)[0]].id).toBe('mfix22:test-issue-bot:7:merge')
+      expect(queue.jobs[Object.keys(queue.jobs)[0]].data).toEqual(data)
 
       await wait(2)
 
@@ -358,6 +362,44 @@ describe('Bot', () => {
     })
   })
 
+  describe('installation', () => {
+    test('Will take action when repos are added', async () => {
+      const repositories_added = [{ name: 'ranger-0' }, { name: 'ranger-1' }]
+
+      await robot.receive(addedPayload({ repositories_added }))
+
+      const data = repositories_added.map(({ name: repo }) => [
+        {
+          owner: 'ranger',
+          repo,
+          name: 'merge when passing',
+          color: 'FF851B',
+          description: 'Merge the PR once all status checks have passed'
+        }
+      ])
+
+      expect(github.issues.createLabel.mock.calls).toEqual(data)
+    })
+
+    test('Will take action when an installation is created', async () => {
+      const repositories = [{ name: 'ranger-0' }, { name: 'ranger-1' }]
+
+      await robot.receive(createdPayload({ repositories }))
+
+      const data = repositories.map(({ name: repo }) => [
+        {
+          owner: 'ranger',
+          repo,
+          name: 'merge when passing',
+          color: 'FF851B',
+          description: 'Merge the PR once all status checks have passed'
+        }
+      ])
+
+      expect(github.issues.createLabel.mock.calls).toEqual(data)
+    })
+  })
+
   describe('billing', () => {
     beforeEach(() => {
       robot.log.error = jest.fn()