Skip to content

Commit

Permalink
Add CodeBuild Project Cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
tomvachon committed Mar 5, 2018
1 parent 15fefb1 commit b881849
Showing 1 changed file with 57 additions and 0 deletions.
57 changes: 57 additions & 0 deletions resources/codebuild-projects.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,57 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/codebuild"
)

type CodeBuildProject struct {
svc *codebuild.CodeBuild
projectName *string
}

func init() {
register("CodeBuildProject", ListCodeBuildProjects)
}

func ListCodeBuildProjects(sess *session.Session) ([]Resource, error) {
svc := codebuild.New(sess)
resources := []Resource{}

params := &codebuild.ListProjectsInput{}

for {
resp, err := svc.ListProjects(params)
if err != nil {
return nil, err
}

for _, project := range resp.Projects {
resources = append(resources, &CodeBuildProject{
svc: svc,
projectName: project,
})
}

if resp.NextToken == nil {
break
}

params.NextToken = resp.NextToken
}

return resources, nil
}

func (f *CodeBuildProject) Remove() error {

_, err := f.svc.DeleteProject(&codebuild.DeleteProjectInput{
Name: f.projectName,
})

return err
}

func (f *CodeBuildProject) String() string {
return *f.projectName
}

0 comments on commit b881849

Please sign in to comment.