fix: auto create bucket when uploading file recursively

supabase · Oct 24, 2023 · 691eff6 · 691eff6
1 parent 3f0e2d0
commit 691eff6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/cmd/storage.go b/cmd/storage.go
@@ -35,7 +35,7 @@ var (
 
 	cpCmd = &cobra.Command{
 		Use: "cp <src> <dst>",
-		Example: `cp readme.md ss:///bucket
+		Example: `cp readme.md ss:///bucket/readme.md
 cp -r docs ss:///bucket/docs
 cp -r ss:///bucket/docs .
 `,

diff --git a/internal/storage/cp/cp.go b/internal/storage/cp/cp.go
@@ -108,7 +108,8 @@ func UploadStorageObjectAll(ctx context.Context, projectRef, remotePath, localPa
 		dstPath := remotePath
 		// Copying single file
 		if relPath == "." {
-			if IsDir(dstPath) || (dirExists && !fileExists) {
+			_, prefix := storage.SplitBucketPrefix(dstPath)
+			if IsDir(prefix) || (dirExists && !fileExists) {
 				dstPath = path.Join(dstPath, info.Name())
 			}
 		} else {
@@ -132,6 +133,6 @@ func UploadStorageObjectAll(ctx context.Context, projectRef, remotePath, localPa
 	})
 }
 
-func IsDir(objectPath string) bool {
-	return len(objectPath) == 0 || strings.HasSuffix(objectPath, "/")
+func IsDir(objectPrefix string) bool {
+	return len(objectPrefix) == 0 || strings.HasSuffix(objectPrefix, "/")
 }