package shell import ( "context" "fmt" "io" "net/url" "path/filepath" "strconv" "strings" "github.com/chrislusf/seaweedfs/weed/filer2" "github.com/chrislusf/seaweedfs/weed/pb/filer_pb" "github.com/chrislusf/seaweedfs/weed/wdclient" "google.golang.org/grpc" ) type ShellOptions struct { Masters *string GrpcDialOption grpc.DialOption // shell transient context FilerHost string FilerPort int64 Directory string } type CommandEnv struct { env map[string]string MasterClient *wdclient.MasterClient option ShellOptions } type command interface { Name() string Help() string Do([]string, *CommandEnv, io.Writer) error } var ( Commands = []command{} ) func NewCommandEnv(options ShellOptions) *CommandEnv { return &CommandEnv{ env: make(map[string]string), MasterClient: wdclient.NewMasterClient(context.Background(), options.GrpcDialOption, "shell", strings.Split(*options.Masters, ",")), option: options, } } func (ce *CommandEnv) parseUrl(input string) (filerServer string, filerPort int64, path string, err error) { if strings.HasPrefix(input, "http") { return parseFilerUrl(input) } if !strings.HasPrefix(input, "/") { input = filepath.ToSlash(filepath.Join(ce.option.Directory, input)) } return ce.option.FilerHost, ce.option.FilerPort, input, err } func (ce *CommandEnv) isDirectory(ctx context.Context, filerServer string, filerPort int64, path string) bool { return ce.checkDirectory(ctx, filerServer, filerPort, path) == nil } func (ce *CommandEnv) checkDirectory(ctx context.Context, filerServer string, filerPort int64, path string) error { dir, name := filer2.FullPath(path).DirAndName() return ce.withFilerClient(ctx, filerServer, filerPort, func(client filer_pb.SeaweedFilerClient) error { resp, listErr := client.ListEntries(ctx, &filer_pb.ListEntriesRequest{ Directory: dir, Prefix: name, StartFromFileName: name, InclusiveStartFrom: true, Limit: 1, }) if listErr != nil { return listErr } if len(resp.Entries) == 0 { return fmt.Errorf("entry not found") } if resp.Entries[0].Name != name { return fmt.Errorf("not a valid directory, found %s", resp.Entries[0].Name) } if !resp.Entries[0].IsDirectory { return fmt.Errorf("not a directory") } return nil }) } func parseFilerUrl(entryPath string) (filerServer string, filerPort int64, path string, err error) { if strings.HasPrefix(entryPath, "http") { var u *url.URL u, err = url.Parse(entryPath) if err != nil { return } filerServer = u.Hostname() portString := u.Port() if portString != "" { filerPort, err = strconv.ParseInt(portString, 10, 32) } path = u.Path } else { err = fmt.Errorf("path should have full url http://:/path/to/dirOrFile : %s", entryPath) } return } func findInputDirectory(args []string) (input string) { input = "." if len(args) > 0 { input = args[len(args)-1] if strings.HasPrefix(input, "-") { input = "." } } return input }