seaweedfs/weed/shell/commands.go

123 lines
2.6 KiB
Go
Raw Normal View History

package shell
import (
"fmt"
"io"
"net/url"
"path/filepath"
"strconv"
"strings"
2019-12-13 08:22:37 +00:00
"google.golang.org/grpc"
2020-03-24 05:54:02 +00:00
"github.com/chrislusf/seaweedfs/weed/pb"
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb"
2020-03-23 07:01:34 +00:00
"github.com/chrislusf/seaweedfs/weed/util"
"github.com/chrislusf/seaweedfs/weed/wdclient"
)
type ShellOptions struct {
2019-10-25 14:45:12 +00:00
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{
2020-03-02 06:39:08 +00:00
env: make(map[string]string),
MasterClient: wdclient.NewMasterClient(options.GrpcDialOption, "shell", 0, strings.Split(*options.Masters, ",")),
2020-03-02 06:39:08 +00:00
option: options,
}
}
2020-03-24 04:26:15 +00:00
func (ce *CommandEnv) parseUrl(input string) (path string, err error) {
if strings.HasPrefix(input, "http") {
err = fmt.Errorf("http://<filer>:<port> prefix is not supported any more")
return
}
if !strings.HasPrefix(input, "/") {
input = filepath.ToSlash(filepath.Join(ce.option.Directory, input))
}
2020-03-24 04:26:15 +00:00
return input, err
}
2020-03-24 04:26:15 +00:00
func (ce *CommandEnv) isDirectory(path string) bool {
2020-03-24 04:26:15 +00:00
return ce.checkDirectory(path) == nil
}
2020-03-24 04:26:15 +00:00
func (ce *CommandEnv) checkDirectory(path string) error {
2020-03-23 07:01:34 +00:00
dir, name := util.FullPath(path).DirAndName()
2020-03-24 04:36:39 +00:00
exists, err := filer_pb.Exists(ce, dir, name, true)
if !exists {
return fmt.Errorf("%s is not a directory", path)
}
2020-03-24 04:26:15 +00:00
return err
}
2020-03-24 05:54:02 +00:00
func (ce *CommandEnv) WithFilerClient(fn func(filer_pb.SeaweedFilerClient) error) error {
filerGrpcAddress := fmt.Sprintf("%s:%d", ce.option.FilerHost, ce.option.FilerPort+10000)
return pb.WithGrpcFilerClient(filerGrpcAddress, ce.option.GrpcDialOption, fn)
}
func (ce *CommandEnv) AdjustedUrl(hostAndPort string) string {
return hostAndPort
}
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 /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
}