forked from Deuxfleurs/bagage
69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
|
package s3
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/minio/minio-go/v7"
|
||
|
|
)
|
||
|
|
|
||
|
|
type S3Class int
|
||
|
|
|
||
|
|
const (
|
||
|
|
ROOT S3Class = 1 << iota
|
||
|
|
BUCKET
|
||
|
|
COMMON_PREFIX
|
||
|
|
OBJECT
|
||
|
|
OPAQUE_KEY
|
||
|
|
|
||
|
|
KEY = COMMON_PREFIX | OBJECT | OPAQUE_KEY
|
||
|
|
)
|
||
|
|
|
||
|
|
type S3Path struct {
|
||
|
|
Path string
|
||
|
|
Class S3Class
|
||
|
|
Bucket string
|
||
|
|
Key string
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewS3Path(p string) S3Path {
|
||
|
|
// Remove first dot, eq. relative directory == "/"
|
||
|
|
if len(p) > 0 && p[0] == '.' {
|
||
|
|
p = p[1:]
|
||
|
|
}
|
||
|
|
|
||
|
|
// Add the first slash if missing
|
||
|
|
p = "/" + p
|
||
|
|
|
||
|
|
// Clean path using golang tools
|
||
|
|
p = path.Clean(p)
|
||
|
|
|
||
|
|
exploded_path := strings.SplitN(p, "/", 3)
|
||
|
|
|
||
|
|
// If there is no bucket name (eg. "/")
|
||
|
|
if len(exploded_path) < 2 || exploded_path[1] == "" {
|
||
|
|
return S3Path{p, ROOT, "", ""}
|
||
|
|
}
|
||
|
|
|
||
|
|
// If there is no key
|
||
|
|
if len(exploded_path) < 3 || exploded_path[2] == "" {
|
||
|
|
return S3Path{p, BUCKET, exploded_path[1], ""}
|
||
|
|
}
|
||
|
|
|
||
|
|
return S3Path{p, OPAQUE_KEY, exploded_path[1], exploded_path[2]}
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewTrustedS3Path(bucket string, obj minio.ObjectInfo) S3Path {
|
||
|
|
cl := OBJECT
|
||
|
|
if obj.Key[len(obj.Key)-1:] == "/" {
|
||
|
|
cl = COMMON_PREFIX
|
||
|
|
}
|
||
|
|
|
||
|
|
return S3Path{
|
||
|
|
Path: path.Join("/", bucket, obj.Key),
|
||
|
|
Bucket: bucket,
|
||
|
|
Key: obj.Key,
|
||
|
|
Class: cl,
|
||
|
|
}
|
||
|
|
}
|