added alias properties

oreillymedia · Sep 16, 2024 · 2e67160 · 2e67160
1 parent 1e5142c
commit 2e67160
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/resources/bedrock-agentalias.go b/resources/bedrock-agentalias.go
@@ -4,6 +4,7 @@ import (
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/bedrockagent"
+	"github.com/rebuy-de/aws-nuke/v2/pkg/types"
 )
 
 type BedrockAgentAlias struct {
@@ -90,6 +91,15 @@ func (f *BedrockAgentAlias) Remove() error {
 	return err
 }
 
+func (f *BedrockAgentAlias) Properties() types.Properties {
+	properties := types.NewProperties().
+		Set("AgentId", f.AgentId).
+		Set("AgentAliasId", f.AgentAliasId).
+		Set("AgentAliasName", f.AgentAliasName)
+
+	return properties
+}
+
 func (f *BedrockAgentAlias) String() string {
 	return *f.AgentAliasName
 }
diff --git a/resources/bedrock-flowalias.go b/resources/bedrock-flowalias.go
@@ -7,6 +7,7 @@ import (
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/session"
 	"github.com/aws/aws-sdk-go/service/bedrockagent"
+	"github.com/rebuy-de/aws-nuke/v2/pkg/types"
 )
 
 type BedrockFlowAlias struct {
@@ -100,6 +101,15 @@ func (f *BedrockFlowAlias) Remove() error {
 	return err
 }
 
+func (f *BedrockFlowAlias) Properties() types.Properties {
+	properties := types.NewProperties().
+		Set("FlowId", f.FlowId).
+		Set("FlowAliasId", f.FlowAliasId).
+		Set("FlowAliasName", f.FlowAliasName)
+
+	return properties
+}
+
 func (f *BedrockFlowAlias) String() string {
 	return *f.FlowAliasName
 }