seaweedfs/weed/command/shell.go

62 lines
1.1 KiB
Go
Raw Normal View History

package command
2012-08-24 03:56:09 +00:00
import (
2013-01-17 08:56:56 +00:00
"bufio"
"fmt"
2013-01-17 08:56:56 +00:00
"os"
"github.com/chrislusf/seaweedfs/weed/glog"
2012-08-24 03:56:09 +00:00
)
func init() {
2013-01-17 08:56:56 +00:00
cmdShell.Run = runShell // break init cycle
2012-08-24 03:56:09 +00:00
}
var cmdShell = &Command{
2013-01-17 08:56:56 +00:00
UsageLine: "shell",
Short: "run interactive commands, now just echo",
Long: `run interactive commands.
2012-08-24 03:56:09 +00:00
`,
}
2013-01-17 08:56:56 +00:00
var ()
2012-08-24 03:56:09 +00:00
func runShell(command *Command, args []string) bool {
2013-01-17 08:56:56 +00:00
r := bufio.NewReader(os.Stdin)
o := bufio.NewWriter(os.Stdout)
e := bufio.NewWriter(os.Stderr)
prompt := func() {
2013-02-27 06:54:22 +00:00
var err error
if _, err = o.WriteString("> "); err != nil {
glog.V(0).Infoln("error writing to stdout:", err)
2013-02-27 06:54:22 +00:00
}
if err = o.Flush(); err != nil {
glog.V(0).Infoln("error flushing stdout:", err)
2013-02-27 06:54:22 +00:00
}
2013-01-17 08:56:56 +00:00
}
readLine := func() string {
ret, err := r.ReadString('\n')
if err != nil {
fmt.Fprint(e, err)
os.Exit(1)
}
return ret
}
execCmd := func(cmd string) int {
if cmd != "" {
2013-02-27 06:54:22 +00:00
if _, err := o.WriteString(cmd); err != nil {
glog.V(0).Infoln("error writing to stdout:", err)
2013-02-27 06:54:22 +00:00
}
2013-01-17 08:56:56 +00:00
}
return 0
}
2012-08-24 03:56:09 +00:00
2013-01-17 08:56:56 +00:00
cmd := ""
for {
prompt()
cmd = readLine()
execCmd(cmd)
}
2012-08-24 03:56:09 +00:00
}