Merge pull request #2208 from combineads/add_rm_options

weed/shell/command_fs_rm.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"strings"
 
 	"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
 	"github.com/chrislusf/seaweedfs/weed/util"
@@ -21,47 +22,79 @@ func (c *commandFsRm) Name() string {
 }
 
 func (c *commandFsRm) Help() string {
-	return `remove a file or a folder, recursively delete all files and folders
+	return `remove file and directory entries
 
-	fs.rm <entry1>
+	fs.rm [-rf] <entry1> <entry2> ...
 
-	fs.rm /dir/file_name
+	fs.rm /dir/file_name1 dir/file_name2
 	fs.rm /dir
+
+	The option "-r" can be recursive.
+	The option "-f" can be ignored by recursive error.
 `
 }
 
 func (c *commandFsRm) Do(args []string, commandEnv *CommandEnv, writer io.Writer) (err error) {
-
-	if len(args) != 1 {
+	isRecursive := false
+	ignoreRecursiveError := false
+	var entiries []string
+	for _, arg := range args {
+		if !strings.HasPrefix(arg, "-") {
+			entiries = append(entiries, arg)
+			continue
+		}
+		for _, t := range arg {
+			switch t {
+			case 'r':
+				isRecursive = true
+			case 'f':
+				ignoreRecursiveError = true
+			}
+		}
+	}
+	if len(entiries) < 1 {
 		return fmt.Errorf("need to have arguments")
 	}
 
-	targetPath, err := commandEnv.parseUrl(args[0])
+	commandEnv.WithFilerClient(func(client filer_pb.SeaweedFilerClient) error {
+		for _, entry := range entiries {
+			targetPath, err := commandEnv.parseUrl(entry)
 			if err != nil {
-		return err
+				fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, err)
+				continue
 			}
 
 			targetDir, targetName := util.FullPath(targetPath).DirAndName()
 
-	return commandEnv.WithFilerClient(func(client filer_pb.SeaweedFilerClient) error {
+			lookupRequest := &filer_pb.LookupDirectoryEntryRequest{
+				Directory: targetDir,
+				Name:      targetName,
+			}
+			_, err = filer_pb.LookupEntry(client, lookupRequest)
+			if err != nil {
+				fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, err)
+				continue
+			}
 
 			request := &filer_pb.DeleteEntryRequest{
 				Directory:            targetDir,
 				Name:                 targetName,
-			IgnoreRecursiveError: true,
+				IgnoreRecursiveError: ignoreRecursiveError,
 				IsDeleteData:         true,
-			IsRecursive:          true,
+				IsRecursive:          isRecursive,
 				IsFromOtherCluster:   false,
 				Signatures:           nil,
 			}
-		_, err = client.DeleteEntry(context.Background(), request)
-
-		if err == nil {
-			fmt.Fprintf(writer, "remove: %s\n", targetPath)
+			if resp, err := client.DeleteEntry(context.Background(), request); err != nil {
+				fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, err)
+			} else {
+				if resp.Error != "" {
+					fmt.Fprintf(writer, "rm: %s: %v\n", targetPath, resp.Error)
 				}
-
-		return err
-
+			}
+		}
+		return nil
 	})
 
+	return
 }