Add support for editing policies on buckets

Makefile

@@ -4,7 +4,7 @@ build:
 
 .PHONY: run
 run:
-	go run
+	go run main.go
 
 .PHONY: lint
 lint:

internal/app/s3manager/bucket_policy.go

@@ -0,0 +1,39 @@
+package s3manager
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+
+	"github.com/gorilla/mux"
+)
+
+func HandleGetBucketPolicy(s3 S3) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		bucketName := mux.Vars(r)["bucketName"]
+		policy, err := s3.GetBucketPolicy(r.Context(), bucketName)
+		if err != nil {
+			handleHTTPError(w, fmt.Errorf("error getting bucket policy: %w", err))
+			return
+		}
+		w.Header().Set("Content-Type", "application/json")
+		w.Write([]byte(policy))
+	}
+}
+
+func HandlePutBucketPolicy(s3 S3) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		bucketName := mux.Vars(r)["bucketName"]
+		policy, err := io.ReadAll(r.Body)
+		if err != nil {
+			handleHTTPError(w, fmt.Errorf("error reading request body: %w", err))
+			return
+		}
+		err = s3.SetBucketPolicy(r.Context(), bucketName, string(policy))
+		if err != nil {
+			handleHTTPError(w, fmt.Errorf("error setting bucket policy: %w", err))
+			return
+		}
+		w.WriteHeader(http.StatusNoContent)
+	}
+}

internal/app/s3manager/errors.go

@@ -32,7 +32,7 @@ func handleHTTPError(w http.ResponseWriter, err error) {
 		code = http.StatusNotFound
 	}
 
-	http.Error(w, http.StatusText(code), code)
+	http.Error(w, err.Error(), code)
 
 	// Log if server error
 	if code >= http.StatusInternalServerError {

internal/app/s3manager/s3.go

@@ -21,6 +21,8 @@ type S3 interface {
 	PutObject(ctx context.Context, bucketName, objectName string, reader io.Reader, objectSize int64, opts minio.PutObjectOptions) (minio.UploadInfo, error)
 	RemoveBucket(ctx context.Context, bucketName string) error
 	RemoveObject(ctx context.Context, bucketName, objectName string, opts minio.RemoveObjectOptions) error
+	GetBucketPolicy(ctx context.Context, bucketName string) (string, error)
+	SetBucketPolicy(ctx context.Context, bucketName string, policy string) error
 }
 
 type SSEType struct {

main.go

@@ -187,6 +187,8 @@ func main() {
 	if configuration.AllowDelete {
 		r.Handle("/api/buckets/{bucketName}/objects/{objectName:.*}", s3manager.HandleDeleteObject(s3)).Methods(http.MethodDelete)
 	}
+	r.Handle("/api/buckets/{bucketName}/policy", s3manager.HandleGetBucketPolicy(s3)).Methods(http.MethodGet)
+	r.Handle("/api/buckets/{bucketName}/policy", s3manager.HandlePutBucketPolicy(s3)).Methods(http.MethodPut)
 
 	lr := logging.Handler(os.Stdout)(r)
 	srv := &http.Server{