s3browser-backend/internal/mutations.go
2021-08-16 22:16:42 +02:00

135 lines
2.6 KiB
Go

package s3browser
import (
"context"
"fmt"
"strings"
"github.com/minio/minio-go/v7"
)
func deleteMutation(ctx context.Context, id string) error {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
if !ok {
return fmt.Errorf("Failed to get s3Client from context")
}
// TODO: it is posible to remove multiple objects with a single call.
// Is it better to batch this?
err := s3Client.RemoveObject(ctx, bucketName, id, minio.RemoveObjectOptions{})
if err != nil {
return err
}
// Invalidate cache
return invalidateCache(ctx, id)
}
func copyMutation(ctx context.Context, src, dest string) (*File, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
}
// Check if dest is a file or a dir
if strings.HasSuffix(dest, "/") {
// create new dest id
// TODO: What if a file with this id already exists?
dest += getFilenameFromID(src)
}
info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
Bucket: bucketName,
Object: dest,
}, minio.CopySrcOptions{
Bucket: bucketName,
Object: src,
})
if err != nil {
return nil, err
}
// Invalidate cache
// TODO: check error
invalidateCache(ctx, info.Key)
return &File{
ID: info.Key,
}, nil
}
func moveMutation(ctx context.Context, src, dest string) (*File, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
}
// Check if dest is a file or a dir
if strings.HasSuffix(dest, "/") {
// create new dest id
// TODO: What if a file with this id already exists?
dest += getFilenameFromID(src)
}
// There is no (spoon) move. Only copy and delete
info, err := s3Client.CopyObject(ctx, minio.CopyDestOptions{
Bucket: bucketName,
Object: dest,
}, minio.CopySrcOptions{
Bucket: bucketName,
Object: src,
})
if err != nil {
return nil, err
}
err = deleteMutation(ctx, src)
if err != nil {
return nil, err
}
invalidateCache(ctx, info.Key)
return &File{
ID: info.Key,
}, nil
}
func createDirectory(ctx context.Context, path string) (*Directory, error) {
s3Client, ok := ctx.Value("s3Client").(*minio.Client)
if !ok {
return nil, fmt.Errorf("Failed to get s3Client from context")
}
if !strings.HasSuffix(path, "/") {
path += "/"
}
info, err := s3Client.PutObject(ctx, bucketName, path, strings.NewReader(""), 0, minio.PutObjectOptions{
ContentType: "application/x-directory",
})
if err != nil {
return nil, err
}
// Invalidate cache
// TODO: check error
invalidateCacheForDir(ctx, info.Key)
return &Directory{
ID: info.Key,
}, nil
}