113 lines
2.3 KiB
Go
113 lines
2.3 KiB
Go
package client
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"reflect"
|
|
"strings"
|
|
"text/template"
|
|
|
|
flag "github.com/dotcloud/docker/pkg/mflag"
|
|
"github.com/dotcloud/docker/pkg/term"
|
|
"github.com/dotcloud/docker/registry"
|
|
)
|
|
|
|
var funcMap = template.FuncMap{
|
|
"json": func(v interface{}) string {
|
|
a, _ := json.Marshal(v)
|
|
return string(a)
|
|
},
|
|
}
|
|
|
|
func (cli *DockerCli) getMethod(name string) (func(...string) error, bool) {
|
|
if len(name) == 0 {
|
|
return nil, false
|
|
}
|
|
methodName := "Cmd" + strings.ToUpper(name[:1]) + strings.ToLower(name[1:])
|
|
method := reflect.ValueOf(cli).MethodByName(methodName)
|
|
if !method.IsValid() {
|
|
return nil, false
|
|
}
|
|
return method.Interface().(func(...string) error), true
|
|
}
|
|
|
|
func (cli *DockerCli) ParseCommands(args ...string) error {
|
|
if len(args) > 0 {
|
|
method, exists := cli.getMethod(args[0])
|
|
if !exists {
|
|
fmt.Println("Error: Command not found:", args[0])
|
|
return cli.CmdHelp(args[1:]...)
|
|
}
|
|
return method(args[1:]...)
|
|
}
|
|
return cli.CmdHelp(args...)
|
|
}
|
|
|
|
func (cli *DockerCli) Subcmd(name, signature, description string) *flag.FlagSet {
|
|
flags := flag.NewFlagSet(name, flag.ContinueOnError)
|
|
flags.Usage = func() {
|
|
fmt.Fprintf(cli.err, "\nUsage: docker %s %s\n\n%s\n\n", name, signature, description)
|
|
flags.PrintDefaults()
|
|
os.Exit(2)
|
|
}
|
|
return flags
|
|
}
|
|
|
|
func (cli *DockerCli) LoadConfigFile() (err error) {
|
|
cli.configFile, err = registry.LoadConfig(os.Getenv("HOME"))
|
|
if err != nil {
|
|
fmt.Fprintf(cli.err, "WARNING: %s\n", err)
|
|
}
|
|
return err
|
|
}
|
|
|
|
func NewDockerCli(in io.ReadCloser, out, err io.Writer, proto, addr string, tlsConfig *tls.Config) *DockerCli {
|
|
var (
|
|
isTerminal = false
|
|
terminalFd uintptr
|
|
scheme = "http"
|
|
)
|
|
|
|
if tlsConfig != nil {
|
|
scheme = "https"
|
|
}
|
|
|
|
if in != nil {
|
|
if file, ok := out.(*os.File); ok {
|
|
terminalFd = file.Fd()
|
|
isTerminal = term.IsTerminal(terminalFd)
|
|
}
|
|
}
|
|
|
|
if err == nil {
|
|
err = out
|
|
}
|
|
return &DockerCli{
|
|
proto: proto,
|
|
addr: addr,
|
|
in: in,
|
|
out: out,
|
|
err: err,
|
|
isTerminal: isTerminal,
|
|
terminalFd: terminalFd,
|
|
tlsConfig: tlsConfig,
|
|
scheme: scheme,
|
|
}
|
|
}
|
|
|
|
type DockerCli struct {
|
|
proto string
|
|
addr string
|
|
configFile *registry.ConfigFile
|
|
in io.ReadCloser
|
|
out io.Writer
|
|
err io.Writer
|
|
isTerminal bool
|
|
terminalFd uintptr
|
|
tlsConfig *tls.Config
|
|
scheme string
|
|
}
|