From d57d4c5f8f32eef65282f95bbb5ef1c9c0f9c2a0 Mon Sep 17 00:00:00 2001
From: Chris Lu <chris.lu@gmail.com>
Date: Sat, 4 Sep 2021 21:37:25 -0700
Subject: [PATCH] shell: add remote.mount.buckets

---
 weed/shell/command_remote_mount.go         |  10 +-
 weed/shell/command_remote_mount_buckets.go | 110 +++++++++++++++++++++
 2 files changed, 112 insertions(+), 8 deletions(-)
 create mode 100644 weed/shell/command_remote_mount_buckets.go

diff --git a/weed/shell/command_remote_mount.go b/weed/shell/command_remote_mount.go
index 75212dfa0..8e12ae3d5 100644
--- a/weed/shell/command_remote_mount.go
+++ b/weed/shell/command_remote_mount.go
@@ -74,7 +74,7 @@ func (c *commandRemoteMount) Do(args []string, commandEnv *CommandEnv, writer io
 	}
 
 	// sync metadata from remote
-	if err = c.syncMetadata(commandEnv, writer, *dir, *nonEmpty, remoteConf, remoteStorageLocation); err != nil {
+	if err = syncMetadata(commandEnv, writer, *dir, *nonEmpty, remoteConf, remoteStorageLocation); err != nil {
 		return fmt.Errorf("pull metadata: %v", err)
 	}
 
@@ -114,13 +114,7 @@ func jsonPrintln(writer io.Writer, message proto.Message) error {
 	return err
 }
 
-func (c *commandRemoteMount) findRemoteStorageConfiguration(commandEnv *CommandEnv, writer io.Writer, remote *remote_pb.RemoteStorageLocation) (conf *remote_pb.RemoteConf, err error) {
-
-	return filer.ReadRemoteStorageConf(commandEnv.option.GrpcDialOption, commandEnv.option.FilerAddress, remote.Name)
-
-}
-
-func (c *commandRemoteMount) syncMetadata(commandEnv *CommandEnv, writer io.Writer, dir string, nonEmpty bool, remoteConf *remote_pb.RemoteConf, remote *remote_pb.RemoteStorageLocation) error {
+func syncMetadata(commandEnv *CommandEnv, writer io.Writer, dir string, nonEmpty bool, remoteConf *remote_pb.RemoteConf, remote *remote_pb.RemoteStorageLocation) error {
 
 	// find existing directory, and ensure the directory is empty
 	err := commandEnv.WithFilerClient(func(client filer_pb.SeaweedFilerClient) error {
diff --git a/weed/shell/command_remote_mount_buckets.go b/weed/shell/command_remote_mount_buckets.go
new file mode 100644
index 000000000..f95143dfd
--- /dev/null
+++ b/weed/shell/command_remote_mount_buckets.go
@@ -0,0 +1,110 @@
+package shell
+
+import (
+	"flag"
+	"fmt"
+	"github.com/chrislusf/seaweedfs/weed/filer"
+	"github.com/chrislusf/seaweedfs/weed/pb/remote_pb"
+	"github.com/chrislusf/seaweedfs/weed/remote_storage"
+	"github.com/chrislusf/seaweedfs/weed/util"
+	"io"
+	"path/filepath"
+)
+
+func init() {
+	Commands = append(Commands, &commandRemoteMountBuckets{})
+}
+
+type commandRemoteMountBuckets struct {
+}
+
+func (c *commandRemoteMountBuckets) Name() string {
+	return "remote.mount.buckets"
+}
+
+func (c *commandRemoteMountBuckets) Help() string {
+	return `mount all buckets in remote storage and pull its metadata
+
+	# assume a remote storage is configured to name "cloud1"
+	remote.configure -name=cloud1 -type=s3 -access_key=xxx -secret_key=yyy
+
+	# mount all buckets
+	remote.mount.buckets -remote=cloud1
+
+	# after mount, start a separate process to write updates to remote storage
+	weed filer.remote.sync -filer=<filerHost>:<filerPort> -createBucketAt=cloud1
+
+`
+}
+
+func (c *commandRemoteMountBuckets) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
+
+	remoteMountBucketsCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
+
+	remote := remoteMountBucketsCommand.String("remote", "", "a already configured storage name")
+	bucketPattern := remoteMountBucketsCommand.String("bucketPattern", "", "match existing bucket name with wildcard characters '*' and '?'")
+	apply := remoteMountBucketsCommand.Bool("apply", false, "apply the mount for listed buckets")
+
+	if err = remoteMountBucketsCommand.Parse(args); err != nil {
+		return nil
+	}
+
+	if *remote == "" {
+		_, err = listExistingRemoteStorageMounts(commandEnv, writer)
+		return err
+	}
+
+	// find configuration for remote storage
+	remoteConf, err := filer.ReadRemoteStorageConf(commandEnv.option.GrpcDialOption, commandEnv.option.FilerAddress, *remote)
+	if err != nil {
+		return fmt.Errorf("find configuration for %s: %v", *remote, err)
+	}
+
+	// get storage client
+	remoteStorageClient, err := remote_storage.GetRemoteStorage(remoteConf)
+	if err != nil {
+		return fmt.Errorf("get storage client for %s: %v", *remote, err)
+	}
+
+	buckets, err := remoteStorageClient.ListBuckets()
+	if err != nil {
+		return fmt.Errorf("list buckets on %s: %v", *remote, err)
+	}
+
+	fillerBucketsPath, err := readFilerBucketsPath(commandEnv)
+	if err != nil {
+		return fmt.Errorf("read filer buckets path: %v", err)
+	}
+
+	for _, bucket := range buckets {
+		if *bucketPattern != "" {
+			if matched, _ := filepath.Match(*bucketPattern, bucket.Name); !matched {
+				continue
+			}
+		}
+
+		fmt.Fprintf(writer, "bucket %s\n", bucket.Name)
+		if *apply {
+
+			dir := util.FullPath(fillerBucketsPath).Child(bucket.Name)
+			remoteStorageLocation := &remote_pb.RemoteStorageLocation{
+				Name:   *remote,
+				Bucket: bucket.Name,
+				Path:   "/",
+			}
+
+			// sync metadata from remote
+			if err = syncMetadata(commandEnv, writer, string(dir), true, remoteConf, remoteStorageLocation); err != nil {
+				return fmt.Errorf("pull metadata on %+v: %v", remoteStorageLocation, err)
+			}
+
+			// store a mount configuration in filer
+			if err = filer.InsertMountMapping(commandEnv, string(dir), remoteStorageLocation); err != nil {
+				return fmt.Errorf("save mount mapping %s to %+v: %v", dir, remoteStorageLocation, err)
+			}
+
+		}
+	}
+
+	return nil
+}