shell: add commands for bucket

3 files changed, 246 insertions, 0 deletions

diff --git a/weed/shell/command_bucket_create.go b/weed/shell/command_bucket_create.go
new file mode 100644
index 000000000..603e9c564
--- /dev/null
+++ b/weed/shell/command_bucket_create.go
@@ -0,0 +1,90 @@
+package shell
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"io"
+	"os"
+	"time"
+
+	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+)
+
+func init() {
+	Commands = append(Commands, &commandBucketCreate{})
+}
+
+type commandBucketCreate struct {
+}
+
+func (c *commandBucketCreate) Name() string {
+	return "bucket.create"
+}
+
+func (c *commandBucketCreate) Help() string {
+	return `create a bucket with a given name
+
+	Example:
+		bucket.create -name <bucket_name> -replication 001
+`
+}
+
+func (c *commandBucketCreate) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
+
+	bucketCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
+	bucketName := bucketCommand.String("name", "", "bucket name")
+	replication := bucketCommand.String("replication", "", "replication setting for the bucket")
+	if err = bucketCommand.Parse(args); err != nil {
+		return nil
+	}
+
+	if *bucketName == "" {
+		return fmt.Errorf("empty bucket name")
+	}
+
+	filerServer, filerPort, _, parseErr := commandEnv.parseUrl(findInputDirectory(bucketCommand.Args()))
+	if parseErr != nil {
+		return parseErr
+	}
+
+	ctx := context.Background()
+
+	err = commandEnv.withFilerClient(ctx, filerServer, filerPort, func(ctx context.Context, client filer_pb.SeaweedFilerClient) error {
+
+		resp, err := client.GetFilerConfiguration(ctx, &filer_pb.GetFilerConfigurationRequest{})
+		if err != nil {
+			return fmt.Errorf("get filer %s:%d configuration: %v", filerServer, filerPort, err)
+		}
+		filerBucketsPath := resp.DirBuckets
+
+		println("create bucket under", filerBucketsPath)
+
+		entry := &filer_pb.Entry{
+			Name:        *bucketName,
+			IsDirectory: true,
+			Attributes: &filer_pb.FuseAttributes{
+				Mtime:       time.Now().Unix(),
+				Crtime:      time.Now().Unix(),
+				FileMode:    uint32(0777 | os.ModeDir),
+				Collection:  *bucketName,
+				Replication: *replication,
+			},
+		}
+
+		if err := filer_pb.CreateEntry(ctx, client, &filer_pb.CreateEntryRequest{
+			Directory: filerBucketsPath,
+			Entry:     entry,
+		}); err != nil {
+			return err
+		}
+
+		println("created bucket", *bucketName)
+
+		return nil
+
+	})
+
+	return err
+
+}
diff --git a/weed/shell/command_bucket_delete.go b/weed/shell/command_bucket_delete.go
new file mode 100644
index 000000000..9e814ccf9
--- /dev/null
+++ b/weed/shell/command_bucket_delete.go
@@ -0,0 +1,73 @@
+package shell
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"io"
+
+	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+)
+
+func init() {
+	Commands = append(Commands, &commandBucketDelete{})
+}
+
+type commandBucketDelete struct {
+}
+
+func (c *commandBucketDelete) Name() string {
+	return "bucket.delete"
+}
+
+func (c *commandBucketDelete) Help() string {
+	return `delete a bucket by a given name
+
+	bucket.delete -name <bucket_name>
+`
+}
+
+func (c *commandBucketDelete) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
+
+	bucketCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
+	bucketName := bucketCommand.String("name", "", "bucket name")
+	if err = bucketCommand.Parse(args); err != nil {
+		return nil
+	}
+
+	if *bucketName == "" {
+		return fmt.Errorf("empty bucket name")
+	}
+
+	filerServer, filerPort, _, parseErr := commandEnv.parseUrl(findInputDirectory(bucketCommand.Args()))
+	if parseErr != nil {
+		return parseErr
+	}
+
+	ctx := context.Background()
+
+	err = commandEnv.withFilerClient(ctx, filerServer, filerPort, func(ctx context.Context, client filer_pb.SeaweedFilerClient) error {
+
+		resp, err := client.GetFilerConfiguration(ctx, &filer_pb.GetFilerConfigurationRequest{})
+		if err != nil {
+			return fmt.Errorf("get filer %s:%d configuration: %v", filerServer, filerPort, err)
+		}
+		filerBucketsPath := resp.DirBuckets
+
+		if _, err := client.DeleteEntry(ctx, &filer_pb.DeleteEntryRequest{
+			Directory:            filerBucketsPath,
+			Name:                 *bucketName,
+			IsDeleteData:         false,
+			IsRecursive:          true,
+			IgnoreRecursiveError: true,
+		}); err != nil {
+			return err
+		}
+
+		return nil
+
+	})
+
+	return err
+
+}
diff --git a/weed/shell/command_bucket_list.go b/weed/shell/command_bucket_list.go
new file mode 100644
index 000000000..051eeda2d
--- /dev/null
+++ b/weed/shell/command_bucket_list.go
@@ -0,0 +1,83 @@
+package shell
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"io"
+	"math"
+
+	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
+)
+
+func init() {
+	Commands = append(Commands, &commandBucketList{})
+}
+
+type commandBucketList struct {
+}
+
+func (c *commandBucketList) Name() string {
+	return "bucket.list"
+}
+
+func (c *commandBucketList) Help() string {
+	return `list all buckets
+
+`
+}
+
+func (c *commandBucketList) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
+
+	bucketCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
+	if err = bucketCommand.Parse(args); err != nil {
+		return nil
+	}
+
+	filerServer, filerPort, _, parseErr := commandEnv.parseUrl(findInputDirectory(bucketCommand.Args()))
+	if parseErr != nil {
+		return parseErr
+	}
+
+	ctx := context.Background()
+
+	err = commandEnv.withFilerClient(ctx, filerServer, filerPort, func(ctx context.Context, client filer_pb.SeaweedFilerClient) error {
+
+		resp, err := client.GetFilerConfiguration(ctx, &filer_pb.GetFilerConfigurationRequest{})
+		if err != nil {
+			return fmt.Errorf("get filer %s:%d configuration: %v", filerServer, filerPort, err)
+		}
+		filerBucketsPath := resp.DirBuckets
+
+		stream, err := client.ListEntries(ctx, &filer_pb.ListEntriesRequest{
+			Directory: filerBucketsPath,
+			Limit:     math.MaxUint32,
+		})
+		if err != nil {
+			return fmt.Errorf("list buckets under %v: %v", filerBucketsPath, err)
+		}
+
+		for {
+			resp, recvErr := stream.Recv()
+			if recvErr != nil {
+				if recvErr == io.EOF {
+					break
+				} else {
+					return recvErr
+				}
+			}
+
+			if resp.Entry.Attributes.Replication == "" {
+				fmt.Fprintf(writer, "  %s\n", resp.Entry.Name)
+			} else {
+				fmt.Fprintf(writer, "  %s\t\t\treplication: %s\n", resp.Entry.Name, resp.Entry.Attributes.Replication)
+			}
+		}
+
+		return nil
+
+	})
+
+	return err
+
+}