Invoke CodeBuild from webhook handler Lambda function

Author: toricls

sam.yml

@@ -156,8 +156,9 @@ Resources:
                   - "logs:PutLogEvents"
                 Resource: "arn:aws:logs:*:*:*"
               - Effect: Allow
-                Action: "states:StartExecution"
-                Resource: !Ref BuildStateMachine
+                Action:
+                  - "codebuild:StartBuild"
+                Resource: !Sub "arn:aws:codebuild:${AWS::Region}:${AWS::AccountId}:project/${CodeBuildProjectName}"
   GitHubWebhookHandler:
     Type: "AWS::Serverless::Function"
     Properties:
@@ -176,6 +177,7 @@ Resources:
         Variables:
           DO_NOT_RUN: false
           CODEBUILD_PROJECT_REGION: !Ref CodeBuildRegion
+          CODEBUILD_PROJECT_NAME: !Ref CodeBuildProjectName
           GITHUB_TOKEN: !Ref GitHubPersonalAccessToken
           GITHUB_REPOSITORY_URL: !Ref GitHubRepositoryUrl
           GITHUB_TARGET_RESOURCE: !Ref GitHubTargetResource

src/functions/github-webhook-handler/index.js

@@ -1,7 +1,7 @@
 'use strict'
 
 const AWS = require('aws-sdk'),
-    stepfunctions = new AWS.StepFunctions()
+    codebuild = new AWS.CodeBuild()
 const GitHubApi = require('github'),
     github = new GitHubApi({version: '3.0.0'})
 const ghUrl = require('parse-github-url'),
@@ -26,39 +26,45 @@ exports.handler = (event, context, callback) => {
         callback()
     }
 
-    // Create GitHub pending status
+    // Execute CodeBuild Project
     const sha = ghEvent.pull_request ? ghEvent.pull_request.head.sha : ghEvent.head_commit.id
-    const p = new Promise((resolve) => {
-        github.repos.createStatus({
-            owner: repo.owner,
-            repo: repo.name,
-            sha: sha,
-            state: 'pending',
-            target_url: `https://${region}.console.aws.amazon.com/codebuild/home?region=${region}#/builds`,
-            context: 'codebuild',
-            description: 'AWS CodeBuild is preparing your tests...'
-        }).then((data) => {
-            console.log(data)
-            resolve('status created')
-        }).catch((err) => {
-            console.log(err)
-            resolve('status creation failed')
-        })
-    })
-
-    // Execute Step Functions
-    p.then(() => {
-        const params = {
-            stateMachineArn: process.env.STEP_FUNCTIONS_ARN,
-            input: JSON.stringify(ghEvent)
-        }
-        stepfunctions.startExecution(params, function(err, data) {
-            if (err) {
-                console.log(err, err.stack)
+    const params = {
+        projectName: process.env.CODEBUILD_PROJECT_NAME,
+        sourceVersion: sha
+    }
+    codebuild.startBuild(params, (err, data) => {
+        // Create GitHub status
+        if (err) {
+            // If CodeBuild couldn't start a build, save GitHub status as 'error'
+            console.log(err, err.stack)
+            github.repos.createStatus({
+                owner: repo.owner,
+                repo: repo.name,
+                sha: sha,
+                state: 'error',
+                target_url: `https://${region}.console.aws.amazon.com/codebuild/home?region=${region}#/builds`,
+                context: 'codebuild',
+                description: 'AWS CodeBuild has failed to start your tests'
+            }).then(() => {
                 callback(err)
-            } else {
+            }).catch((err) => {
+                callback(err)
+            })
+        } else {
+            github.repos.createStatus({
+                owner: repo.owner,
+                repo: repo.name,
+                sha: sha,
+                state: 'pending',
+                target_url: `https://${region}.console.aws.amazon.com/codebuild/home?region=${region}#/builds/${data.build.id}/view/new`,
+                context: 'codebuild',
+                description: 'AWS CodeBuild is running your tests'
+            }).then((data) => {
                 callback(null, data)
-            }
-        })
+            }).catch((err) => {
+                console.log(err, err.stack)
+                callback(err)
+            })
+        }
     })
 }