From 4bfc792376299e0a1fd80a72ba4a54886b1f8007 Mon Sep 17 00:00:00 2001
From: KeisukeYamashita <19yamashita15@gmail.com>
Date: Mon, 1 Aug 2022 14:57:20 +0900
Subject: [PATCH] feat: add module aws registry (#8)

Signed-off-by: KeisukeYamashita <keisuke.yamashita@linecorp.com>
---
 internal/driver/s3/module.go | 56 ++++++++++++++++++++++++++++++++----
 1 file changed, 51 insertions(+), 5 deletions(-)

diff --git a/internal/driver/s3/module.go b/internal/driver/s3/module.go
index 493bc30..a6f3fe4 100644
--- a/internal/driver/s3/module.go
+++ b/internal/driver/s3/module.go
@@ -2,9 +2,11 @@ package s3
 
 import (
 	"context"
+	"fmt"
 	"io"
 	"os"
 
+	"github.com/aws/aws-sdk-go-v2/aws"
 	"github.com/aws/aws-sdk-go-v2/service/s3"
 	"github.com/kerraform/kegistry/internal/driver"
 	"go.uber.org/zap"
@@ -23,21 +25,65 @@ func (d *module) CreateModule(ctx context.Context, namespace, provider, name str
 }
 
 func (d *module) CreateVersion(ctx context.Context, namespace, provider, name, version string) (*driver.CreateModuleVersionResult, error) {
-	return nil, nil
+	filepath := fmt.Sprintf("%s/%s/%s/%s/versions/%s/terraform-%s-%s-%s.tar.gz", driver.ModuleRootPath, namespace, provider, name, version, provider, name, version)
+	psc := s3.NewPresignClient(d.s3)
+	uploadURL, err := psc.PresignPutObject(ctx, &s3.PutObjectInput{
+		Bucket: aws.String(d.bucket),
+		Key:    aws.String(filepath),
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &driver.CreateModuleVersionResult{
+		Upload: uploadURL.URL,
+	}, nil
 }
 
 func (d *module) GetDownloadURL(ctx context.Context, namespace, provider, name, version string) (string, error) {
-	return "", nil
+	filepath := fmt.Sprintf("%s/%s/%s/%s/versions/%s/terraform-%s-%s-%s.tar.gz", driver.ModuleRootPath, namespace, provider, name, version, provider, name, version)
+
+	psc := s3.NewPresignClient(d.s3)
+	downloadURL, err := psc.PresignGetObject(ctx, &s3.GetObjectInput{
+		Bucket: aws.String(d.bucket),
+		Key:    aws.String(filepath),
+	})
+
+	if err != nil {
+		return "", err
+	}
+
+	return downloadURL.URL, nil
 }
 
 func (d *module) GetModule(ctx context.Context, namespace, provider, name, version string) (*os.File, error) {
-	return nil, nil
+	return nil, ErrS3NotAllowed
 }
 
 func (d *module) ListAvailableVersions(ctx context.Context, namespace, provider, name string) ([]string, error) {
-	return nil, nil
+	prefix := fmt.Sprintf("%s/%s/%s/%s/versions", driver.ModuleRootPath, namespace, provider, name)
+	input := &s3.ListObjectsV2Input{
+		Bucket: aws.String(d.bucket),
+		Prefix: aws.String(prefix),
+	}
+
+	resp, err := d.s3.ListObjectsV2(ctx, input)
+	if err != nil {
+		return nil, err
+	}
+
+	d.logger.Debug("found versions",
+		zap.Int("count", len(resp.Contents)),
+	)
+
+	vs := []string{}
+	for _, obj := range resp.Contents {
+		vs = append(vs, *obj.Key)
+	}
+
+	return vs, nil
 }
 
 func (d *module) SavePackage(ctx context.Context, namespace, provider, name, version string, body io.Reader) error {
-	return nil
+	return ErrS3NotAllowed
 }