Merge pull request #194 from tony24681379/ssm

feat: add ssm put and delete

README.md

@@ -648,7 +648,7 @@ aws_secretsmanager:
       path: /prod/billing-svc/vars/mg
 ```
 
-## AWS Paramstore
+## AWS Parameter store
 
 ### Authentication
 
@@ -657,8 +657,8 @@ Your standard `AWS_DEFAULT_REGION`, `AWS_ACCESS_KEY_ID`, `AWS_SECRET_ACCESS_KEY`
 ### Features
 
 - Sync - `no`
-- Mapping - `yes`
-- Modes - `read`, [write: accepting PR](https://github.com/spectralops/teller)
+- Mapping - `no`
+- Modes - `read+write+delete`
 - Key format
   - `env` - path based
   - `decrypt` - available in this provider, will use KMS automatically

pkg/providers/aws_ssm.go

@@ -8,12 +8,15 @@ import (
 	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/config"
 	"github.com/aws/aws-sdk-go-v2/service/ssm"
+	"github.com/aws/aws-sdk-go-v2/service/ssm/types"
 	"github.com/spectralops/teller/pkg/core"
 	"github.com/spectralops/teller/pkg/logging"
 )
 
 type AWSSSMClient interface {
 	GetParameter(ctx context.Context, params *ssm.GetParameterInput, optFns ...func(*ssm.Options)) (*ssm.GetParameterOutput, error)
+	PutParameter(ctx context.Context, params *ssm.PutParameterInput, optFns ...func(*ssm.Options)) (*ssm.PutParameterOutput, error)
+	DeleteParameter(ctx context.Context, params *ssm.DeleteParameterInput, optFns ...func(*ssm.Options)) (*ssm.DeleteParameterOutput, error)
 }
 type AWSSSM struct {
 	client AWSSSMClient
@@ -22,7 +25,7 @@ type AWSSSM struct {
 
 const awsssmName = "aws_ssm"
 
-//nolint
+// nolint
 func init() {
 	metaInfo := core.MetaInfo{
 		Description:    "AWS SSM (aka paramstore)",
@@ -36,7 +39,7 @@ func init() {
         path: /prod/foobar
         decrypt: true
 		`,
-		Ops: core.OpMatrix{Get: true},
+		Ops: core.OpMatrix{Get: true, Put: true, Delete: true},
 	}
 	RegisterProvider(metaInfo, NewAWSSSM)
 }
@@ -66,23 +69,43 @@ func NewAWSSSM(logger logging.Logger) (core.Provider, error) {
 	return &AWSSSM{client: client, logger: logger}, nil
 }
 
-func (a *AWSSSM) Put(p core.KeyPath, val string) error {
-	return fmt.Errorf("provider %q does not implement write yet", awsssmName)
+func (a *AWSSSM) Put(kp core.KeyPath, val string) error {
+	_, err := a.client.PutParameter(context.TODO(), &ssm.PutParameterInput{
+		Name:      &kp.Path,
+		Value:     &val,
+		Overwrite: true,
+		Type:      types.ParameterTypeString,
+	})
+	if err != nil {
+		return err
+	}
+
+	return nil
 }
-func (a *AWSSSM) PutMapping(p core.KeyPath, m map[string]string) error {
-	return fmt.Errorf("provider %q does not implement write yet", awsssmName)
+
+func (a *AWSSSM) PutMapping(kp core.KeyPath, m map[string]string) error {
+	for k, v := range m {
+		ap := kp.SwitchPath(k)
+		err := a.Put(ap, v)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
 }
 
 func (a *AWSSSM) GetMapping(kp core.KeyPath) ([]core.EnvEntry, error) {
 	return nil, fmt.Errorf("does not support full env sync (path: %s)", kp.Path)
 }
 
 func (a *AWSSSM) Delete(kp core.KeyPath) error {
-	return fmt.Errorf("%s does not implement delete yet", awsssmName)
+	_, err := a.client.DeleteParameter(context.TODO(), &ssm.DeleteParameterInput{Name: &kp.Path})
+	return err
 }
 
 func (a *AWSSSM) DeleteMapping(kp core.KeyPath) error {
-	return fmt.Errorf("%s does not implement delete yet", awsssmName)
+	return fmt.Errorf("does not support full env sync (path: %s)", kp.Path)
 }
 
 func (a *AWSSSM) Get(p core.KeyPath) (*core.EnvEntry, error) {