1 files changed, 10 insertions, 58 deletions

diff --git a/internal/storage/s3.go b/internal/storage/s3.go
index 1b191f675..1ead7efe9 100644
--- a/internal/storage/s3.go
+++ b/internal/storage/s3.go
@@ -19,80 +19,32 @@
 package storage
 
 import (
-	"bytes"
 	"context"
-	"fmt"
-	"io"
 	"mime"
 	"net/url"
 	"path"
 	"time"
 
-	"github.com/minio/minio-go/v7"
+	"codeberg.org/gruf/go-store/v2/kv"
+	"codeberg.org/gruf/go-store/v2/storage"
 )
 
 type S3 struct {
-	mc     *minio.Client
-	bucket string
-	proxy  bool
-}
-
-func NewS3(mc *minio.Client, bucket string, proxy bool) *S3 {
-	return &S3{
-		mc:     mc,
-		bucket: bucket,
-		proxy:  proxy,
-	}
-}
-
-func (s *S3) Get(ctx context.Context, key string) ([]byte, error) {
-	r, err := s.GetStream(ctx, key)
-	if err != nil {
-		return nil, err
-	}
-	defer r.Close()
-	b, err := io.ReadAll(r)
-	if err != nil {
-		return nil, fmt.Errorf("reading data from s3: %w", err)
-	}
-	return b, nil
-}
-
-func (s *S3) GetStream(ctx context.Context, key string) (io.ReadCloser, error) {
-	o, err := s.mc.GetObject(ctx, s.bucket, key, minio.GetObjectOptions{})
-	if err != nil {
-		err = fmt.Errorf("retrieving object from s3: %w", err)
-	}
-	return o, err
-}
-
-func (s *S3) PutStream(ctx context.Context, key string, r io.Reader) error {
-	if _, err := s.mc.PutObject(ctx, s.bucket, key, r, -1, minio.PutObjectOptions{}); err != nil {
-		return fmt.Errorf("uploading data stream: %w", err)
-	}
-	return nil
-}
-
-func (s *S3) Put(ctx context.Context, key string, value []byte) error {
-	if _, err := s.mc.PutObject(ctx, s.bucket, key, bytes.NewBuffer(value), -1, minio.PutObjectOptions{}); err != nil {
-		return fmt.Errorf("uploading data slice: %w", err)
-	}
-	return nil
-}
-
-func (s *S3) Delete(ctx context.Context, key string) error {
-	return s.mc.RemoveObject(ctx, s.bucket, key, minio.RemoveObjectOptions{})
+	Proxy   bool
+	Bucket  string
+	Storage *storage.S3Storage
+	*kv.KVStore
 }
 
 func (s *S3) URL(ctx context.Context, key string) *url.URL {
-	if s.proxy {
+	if s.Proxy {
 		return nil
 	}
 
-	// it's safe to ignore the error here, as we just fall back to fetching the
-	// file if the url request fails
-	url, _ := s.mc.PresignedGetObject(ctx, s.bucket, key, time.Hour, url.Values{
+	// it's safe to ignore the error here, as we just fall back to fetching the file if URL request fails
+	url, _ := s.Storage.Client().PresignedGetObject(ctx, s.Bucket, key, time.Hour, url.Values{
 		"response-content-type": []string{mime.TypeByExtension(path.Ext(key))},
 	})
+
 	return url
 }