Skip to content

Commit

Permalink
Merge branch 'rebuy-de:main' into main
Browse files Browse the repository at this point in the history
  • Loading branch information
ekristen authored Dec 21, 2021
2 parents b812517 + 93eb15f commit 2675c2a
Show file tree
Hide file tree
Showing 2 changed files with 157 additions and 0 deletions.
65 changes: 65 additions & 0 deletions resources/kendra-indexes.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
package resources

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

type KendraIndex struct {
svc *kendra.Kendra
name string
id string
}

func init() {
register("KendraIndex", ListKendraIndexes)
}

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

params := &kendra.ListIndicesInput{
MaxResults: aws.Int64(100),
}

for {
resp, err := svc.ListIndices(params)
if err != nil {
return nil, err
}
for _, index := range resp.IndexConfigurationSummaryItems {
resources = append(resources, &KendraIndex{
svc: svc,
id: *index.Id,
name: *index.Name,
})
}

if resp.NextToken == nil {
break
}
params.NextToken = resp.NextToken
}
return resources, nil
}

func (i *KendraIndex) Remove() error {
_, err := i.svc.DeleteIndex(&kendra.DeleteIndexInput{
Id: &i.id,
})
return err
}

func (i *KendraIndex) String() string {
return i.id
}

func (i *KendraIndex) Properties() types.Properties {
properties := types.NewProperties()
properties.Set("Name", i.name)

return properties
}
92 changes: 92 additions & 0 deletions resources/lex-model-building-service-bot-alias.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
package resources

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

type LexModelBuildingServiceBotAlias struct {
svc *lexmodelbuildingservice.LexModelBuildingService
name *string
checksum *string
botName *string
}

func init() {
register("LexModelBuildingServiceBotAlias", ListLexModelBuildingServiceBotAliases)
}

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

botParams := &lexmodelbuildingservice.GetBotsInput{
MaxResults: aws.Int64(10),
}

for {
botResp, err := svc.GetBots(botParams)
if err != nil {
return nil, err
}

for _, bot := range botResp.Bots {
for {
aliasParams := &lexmodelbuildingservice.GetBotAliasesInput{
BotName: bot.Name,
MaxResults: aws.Int64(10),
}
aliasResp, err := svc.GetBotAliases(aliasParams)
if err != nil {
continue
}

for _, alias := range aliasResp.BotAliases {
resources = append(resources, &LexModelBuildingServiceBotAlias{
svc: svc,
name: alias.Name,
checksum: alias.Checksum,
botName: bot.Name,
})
}

if aliasResp.NextToken == nil {
break
}
aliasParams.NextToken = aliasResp.NextToken
}
}

if botResp.NextToken == nil {
break
}

botParams.NextToken = botResp.NextToken
}

return resources, nil
}

func (f *LexModelBuildingServiceBotAlias) Remove() error {
params := &lexmodelbuildingservice.DeleteBotAliasInput{
BotName: f.botName,
Name: f.name,
}
_, err := f.svc.DeleteBotAlias(params)
return err
}

func (f *LexModelBuildingServiceBotAlias) String() string {
return *f.name
}

func (f *LexModelBuildingServiceBotAlias) Properties() types.Properties {
properties := types.NewProperties()

properties.Set("Name", f.name)
properties.Set("BotName", f.botName)
properties.Set("Checksum", f.checksum)
return properties
}

0 comments on commit 2675c2a

Please sign in to comment.