|
|
package shell
import ( "context" "fmt" "github.com/seaweedfs/seaweedfs/weed/cluster" "github.com/seaweedfs/seaweedfs/weed/pb" "github.com/seaweedfs/seaweedfs/weed/pb/filer_pb" "github.com/seaweedfs/seaweedfs/weed/pb/master_pb" "github.com/seaweedfs/seaweedfs/weed/util/grace" "golang.org/x/exp/slices" "io" "math/rand" "os" "path" "regexp" "strings"
"github.com/peterh/liner" )
var ( line *liner.State historyPath = path.Join(os.TempDir(), "weed-shell") )
func RunShell(options ShellOptions) { slices.SortFunc(Commands, func(a, b command) bool { return strings.Compare(a.Name(), b.Name()) < 0 }) line = liner.NewLiner() defer line.Close() grace.OnInterrupt(func() { line.Close() })
line.SetCtrlCAborts(true) line.SetTabCompletionStyle(liner.TabPrints)
setCompletionHandler() loadHistory()
defer saveHistory()
reg, _ := regexp.Compile(`'.*?'|".*?"|\S+`)
commandEnv := NewCommandEnv(&options)
go commandEnv.MasterClient.KeepConnectedToMaster() commandEnv.MasterClient.WaitUntilConnected()
if commandEnv.option.FilerAddress == "" { var filers []pb.ServerAddress commandEnv.MasterClient.WithClient(false, func(client master_pb.SeaweedClient) error { resp, err := client.ListClusterNodes(context.Background(), &master_pb.ListClusterNodesRequest{ ClientType: cluster.FilerType, FilerGroup: *options.FilerGroup, }) if err != nil { return err }
for _, clusterNode := range resp.ClusterNodes { filers = append(filers, pb.ServerAddress(clusterNode.Address)) } return nil }) fmt.Printf("master: %s ", *options.Masters) if len(filers) > 0 { fmt.Printf("filers: %v", filers) commandEnv.option.FilerAddress = filers[rand.Intn(len(filers))] } fmt.Println() }
if commandEnv.option.FilerAddress != "" { commandEnv.WithFilerClient(false, func(filerClient filer_pb.SeaweedFilerClient) error { resp, err := filerClient.GetFilerConfiguration(context.Background(), &filer_pb.GetFilerConfigurationRequest{}) if err != nil { return err } if resp.ClusterId != "" { fmt.Printf(` --- Free Monitoring Data URL: https://cloud.seaweedfs.com/ui/%s
--- `, resp.ClusterId) } return nil }) }
for { cmd, err := line.Prompt("> ") if err != nil { if err != io.EOF { fmt.Printf("%v\n", err) } return }
for _, c := range strings.Split(cmd, ";") { if processEachCmd(reg, c, commandEnv) { return } } } }
func processEachCmd(reg *regexp.Regexp, cmd string, commandEnv *CommandEnv) bool { cmds := reg.FindAllString(cmd, -1)
line.AppendHistory(cmd)
if len(cmds) == 0 { return false } else {
args := make([]string, len(cmds[1:]))
for i := range args { args[i] = strings.Trim(string(cmds[1+i]), "\"'") }
cmd := cmds[0] if cmd == "help" || cmd == "?" { printHelp(cmds) } else if cmd == "exit" || cmd == "quit" { return true } else { foundCommand := false for _, c := range Commands { if c.Name() == cmd || c.Name() == "fs."+cmd { if err := c.Do(args, commandEnv, os.Stdout); err != nil { fmt.Fprintf(os.Stderr, "error: %v\n", err) } foundCommand = true } } if !foundCommand { fmt.Fprintf(os.Stderr, "unknown command: %v\n", cmd) } }
} return false }
func printGenericHelp() { msg := `Type: "help <command>" for help on <command>. Most commands support "<command> -h" also for options. ` fmt.Print(msg)
for _, c := range Commands { helpTexts := strings.SplitN(c.Help(), "\n", 2) fmt.Printf(" %-30s\t# %s \n", c.Name(), helpTexts[0]) } }
func printHelp(cmds []string) { args := cmds[1:] if len(args) == 0 { printGenericHelp() } else if len(args) > 1 { fmt.Println() } else { cmd := strings.ToLower(args[0])
for _, c := range Commands { if c.Name() == cmd { fmt.Printf(" %s\t# %s\n", c.Name(), c.Help()) } } } }
func setCompletionHandler() { line.SetCompleter(func(line string) (c []string) { for _, i := range Commands { if strings.HasPrefix(i.Name(), strings.ToLower(line)) { c = append(c, i.Name()) } } return }) }
func loadHistory() { if f, err := os.Open(historyPath); err == nil { line.ReadHistory(f) f.Close() } }
func saveHistory() { if f, err := os.Create(historyPath); err != nil { fmt.Printf("Error creating history file: %v\n", err) } else { if _, err = line.WriteHistory(f); err != nil { fmt.Printf("Error writing history file: %v\n", err) } f.Close() } }
|