Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support for additional codebuild resources #17

Merged
merged 5 commits into from
Oct 4, 2023
Merged
Show file tree
Hide file tree
Changes from 4 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
60 changes: 60 additions & 0 deletions resources/codebuild-build-batches.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/codebuild"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type CodeBuildBuildBatch struct {
svc *codebuild.CodeBuild
Id *string
}

func init() {
register("CodeBuildBuildBatch", ListCodeBuildBuildBatch)
}

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

params := &codebuild.ListBuildBatchesInput{}

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

for _, batch := range resp.Ids {
resources = append(resources, &CodeBuildBuildBatch{
svc: svc,
Id: batch,
})
}

if resp.NextToken == nil {
break
}

params.NextToken = resp.NextToken
}

return resources, nil
}

func (f *CodeBuildBuildBatch) Remove() error {
_, err := f.svc.DeleteBuildBatch(&codebuild.DeleteBuildBatchInput{
Id: f.Id,
})

return err
}

func (f *CodeBuildBuildBatch) Properties() types.Properties {
properties := types.NewProperties()
properties.
Set("Id", f.Id)
return properties
}
60 changes: 60 additions & 0 deletions resources/codebuild-builds.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/codebuild"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type CodeBuildBuild struct {
svc *codebuild.CodeBuild
Id *string
}

func init() {
register("CodeBuildBuild", ListCodeBuildBuild)
}

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

params := &codebuild.ListBuildsInput{}

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

for _, build := range resp.Ids {
resources = append(resources, &CodeBuildBuild{
svc: svc,
Id: build,
})
}

if resp.NextToken == nil {
break
}

params.NextToken = resp.NextToken
}

return resources, nil
}

func (f *CodeBuildBuild) Remove() error {
_, err := f.svc.BatchDeleteBuilds(&codebuild.BatchDeleteBuildsInput{
Ids: []*string{f.Id},
})

return err
}

func (f *CodeBuildBuild) Properties() types.Properties {
properties := types.NewProperties()
properties.
Set("Id", f.Id)
return properties
}
60 changes: 60 additions & 0 deletions resources/codebuild-report-group.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
package resources

import (
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/codebuild"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type CodeBuildReportGroup struct {
svc *codebuild.CodeBuild
Arn *string
}

func init() {
register("CodeBuildReportGroup", ListCodeBuildReportGroup)
}

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

params := &codebuild.ListReportGroupsInput{}

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

for _, reportGroup := range resp.ReportGroups {
resources = append(resources, &CodeBuildReportGroup{
svc: svc,
Arn: reportGroup,
})
}

if resp.NextToken == nil {
break
}

params.NextToken = resp.NextToken
}

return resources, nil
}

func (f *CodeBuildReportGroup) Remove() error {
_, err := f.svc.DeleteReportGroup(&codebuild.DeleteReportGroupInput{
Arn: f.Arn,
})

return err
}

func (f *CodeBuildReportGroup) Properties() types.Properties {
properties := types.NewProperties()
properties.
Set("Arn", f.Arn)
return properties
}
57 changes: 57 additions & 0 deletions resources/codebuild-source-credentials.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"
"github.com/rebuy-de/aws-nuke/v2/pkg/types"
)

type CodeBuildSourceCredential struct {
svc *codebuild.CodeBuild
Arn *string
AuthType *string
ServerType *string
}

func init() {
register("CodeBuildSourceCredential", ListCodeBuildSourceCredential)
}

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

params := &codebuild.ListSourceCredentialsInput{}

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

if resp == nil {
return nil, nil
}
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why does this follow a different pattern than the modules above? This is the only one where I see if resp == nil. Reading the SDK docs doesn't clarify to me when the response would ever be nil.

I also notice a lack of pagination, but that appears correct as this endpoint does not seem to be paginated if I'm reading the docs correctly.

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

In a previous version of this module I had an error saying resp could be null. I double checked, and it looks to me that it is actually not needed. Removed it in 6c8489e Thanks for catching this @sstoops

Regarding the pagination, you are right, this endpoint is not paginated, SourceCredentialsInfo doesn't have a NextToken field.


for _, credential := range resp.SourceCredentialsInfos {
resources = append(resources, &CodeBuildSourceCredential{
svc: svc,
Arn: credential.Arn,
})
}

return resources, nil
}

func (f *CodeBuildSourceCredential) Remove() error {
_, err := f.svc.DeleteSourceCredentials(&codebuild.DeleteSourceCredentialsInput{Arn: f.Arn})
return err
}

func (f *CodeBuildSourceCredential) Properties() types.Properties {
properties := types.NewProperties()
properties.Set("Arn", f.Arn)
properties.Set("AuthType", f.AuthType)
properties.Set("ServerType", f.ServerType)

return properties
}
Loading