1 files changed, 68 insertions, 0 deletions
diff --git a/s3/path.go b/s3/path.go
new file mode 100644
index 0000000..b1a981b
--- /dev/null
+++ b/s3/path.go
@@ -0,0 +1,68 @@
+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,
+	}
+}