seaweedfs/go/weed/weed.go

219 lines
4.1 KiB
Go
Raw Normal View History

package main
import (
2012-08-24 03:56:09 +00:00
"encoding/json"
"flag"
"fmt"
"io"
2013-02-27 06:54:22 +00:00
"log"
2012-09-28 16:12:36 +00:00
"math/rand"
2012-08-24 03:56:09 +00:00
"net/http"
"os"
"strings"
"sync"
"text/template"
2012-09-28 16:12:36 +00:00
"time"
"unicode"
"unicode/utf8"
)
var IsDebug *bool
var server *string
var commands = []*Command{
cmdFix,
cmdMaster,
2012-08-24 03:56:09 +00:00
cmdUpload,
cmdShell,
cmdVersion,
cmdVolume,
cmdExport,
}
var exitStatus = 0
var exitMu sync.Mutex
func setExitStatus(n int) {
exitMu.Lock()
if exitStatus < n {
exitStatus = n
}
exitMu.Unlock()
}
func main() {
2012-09-28 16:12:36 +00:00
rand.Seed(time.Now().UnixNano())
flag.Usage = usage
flag.Parse()
args := flag.Args()
if len(args) < 1 {
usage()
}
if args[0] == "help" {
help(args[1:])
2012-08-24 03:56:09 +00:00
for _, cmd := range commands {
2012-09-28 16:12:36 +00:00
if len(args) >= 2 && cmd.Name() == args[1] && cmd.Run != nil {
2012-08-24 03:56:09 +00:00
fmt.Fprintf(os.Stderr, "Default Parameters:\n")
cmd.Flag.PrintDefaults()
}
}
return
}
for _, cmd := range commands {
if cmd.Name() == args[0] && cmd.Run != nil {
cmd.Flag.Usage = func() { cmd.Usage() }
cmd.Flag.Parse(args[1:])
args = cmd.Flag.Args()
2013-01-20 03:58:31 +00:00
IsDebug = cmd.IsDebug
if !cmd.Run(cmd, args) {
2012-08-24 03:56:09 +00:00
fmt.Fprintf(os.Stderr, "\n")
cmd.Flag.Usage()
fmt.Fprintf(os.Stderr, "Default Parameters:\n")
cmd.Flag.PrintDefaults()
}
exit()
return
}
}
fmt.Fprintf(os.Stderr, "weed: unknown subcommand %q\nRun 'weed help' for usage.\n", args[0])
setExitStatus(2)
exit()
}
var usageTemplate = `WeedFS is a software to store billions of files and serve them fast!
Usage:
weed command [arguments]
The commands are:
{{range .}}{{if .Runnable}}
{{.Name | printf "%-11s"}} {{.Short}}{{end}}{{end}}
Use "weed help [command]" for more information about a command.
`
var helpTemplate = `{{if .Runnable}}Usage: weed {{.UsageLine}}
{{end}}
{{.Long}}
`
// tmpl executes the given template text on data, writing the result to w.
func tmpl(w io.Writer, text string, data interface{}) {
t := template.New("top")
t.Funcs(template.FuncMap{"trim": strings.TrimSpace, "capitalize": capitalize})
template.Must(t.Parse(text))
if err := t.Execute(w, data); err != nil {
panic(err)
}
}
func capitalize(s string) string {
if s == "" {
return s
}
r, n := utf8.DecodeRuneInString(s)
return string(unicode.ToTitle(r)) + s[n:]
}
func printUsage(w io.Writer) {
tmpl(w, usageTemplate, commands)
}
func usage() {
printUsage(os.Stderr)
os.Exit(2)
}
// help implements the 'help' command.
func help(args []string) {
if len(args) == 0 {
printUsage(os.Stdout)
// not exit 2: succeeded at 'weed help'.
return
}
if len(args) != 1 {
fmt.Fprintf(os.Stderr, "usage: weed help command\n\nToo many arguments given.\n")
os.Exit(2) // failed at 'weed help'
}
arg := args[0]
for _, cmd := range commands {
if cmd.Name() == arg {
tmpl(os.Stdout, helpTemplate, cmd)
// not exit 2: succeeded at 'weed help cmd'.
return
}
}
fmt.Fprintf(os.Stderr, "Unknown help topic %#q. Run 'weed help'.\n", arg)
os.Exit(2) // failed at 'weed help cmd'
}
var atexitFuncs []func()
func atexit(f func()) {
atexitFuncs = append(atexitFuncs, f)
}
func exit() {
for _, f := range atexitFuncs {
f()
}
os.Exit(exitStatus)
}
func exitIfErrors() {
if exitStatus != 0 {
exit()
}
}
2013-02-27 06:54:22 +00:00
func writeJson(w http.ResponseWriter, r *http.Request, obj interface{}) (err error) {
2012-08-24 03:56:09 +00:00
w.Header().Set("Content-Type", "application/javascript")
2012-09-28 16:12:36 +00:00
var bytes []byte
if r.FormValue("pretty") != "" {
2013-02-27 06:54:22 +00:00
bytes, err = json.MarshalIndent(obj, "", " ")
2012-09-28 16:12:36 +00:00
} else {
2013-02-27 06:54:22 +00:00
bytes, err = json.Marshal(obj)
}
if err != nil {
return
2012-09-28 16:12:36 +00:00
}
2012-08-24 03:56:09 +00:00
callback := r.FormValue("callback")
if callback == "" {
2013-02-27 06:54:22 +00:00
_, err = w.Write(bytes)
2012-08-24 03:56:09 +00:00
} else {
2013-02-27 06:54:22 +00:00
if _, err = w.Write([]uint8(callback)); err != nil {
return
}
if _, err = w.Write([]uint8("(")); err != nil {
return
}
2012-08-24 03:56:09 +00:00
fmt.Fprint(w, string(bytes))
2013-02-27 06:54:22 +00:00
if _, err = w.Write([]uint8(")")); err != nil {
return
}
}
return
}
// wrapper for writeJson - just logs errors
func writeJsonQuiet(w http.ResponseWriter, r *http.Request, obj interface{}) {
if err := writeJson(w, r, obj); err != nil {
log.Printf("error writing JSON %s: %s", obj, err)
2012-08-24 03:56:09 +00:00
}
}
2012-09-27 19:17:27 +00:00
2012-09-28 16:12:36 +00:00
func debug(params ...interface{}) {
if *IsDebug {
fmt.Println(params)
}
2012-09-27 19:17:27 +00:00
}