From c16034adb0dfd1dcc8f8e7af77ddaafd13cafab0 Mon Sep 17 00:00:00 2001 From: Christoph Polcin Date: Fri, 25 May 2018 10:15:26 +0200 Subject: [PATCH] refactor getCmd --- cmd/gowebdav/main.go | 238 +++++++++++++++++++++++-------------------- 1 file changed, 128 insertions(+), 110 deletions(-) diff --git a/cmd/gowebdav/main.go b/cmd/gowebdav/main.go index b8e5bdc..ff6b313 100644 --- a/cmd/gowebdav/main.go +++ b/cmd/gowebdav/main.go @@ -10,116 +10,6 @@ import ( "strings" ) -func fail(err interface{}) { - if err != nil { - fmt.Println(err) - } - os.Exit(-1) -} - -func writeFile(path string, bytes []byte, mode os.FileMode) error { - f, err := os.Create(path) - defer f.Close() - if err != nil { - return err - } - - _, err = f.Write(bytes) - return err -} - -func getCmd(method string) func(c *d.Client, p0, p1 string) error { - switch method { - case "LS", "LIST", "PROPFIND": - return func(c *d.Client, p0, p1 string) (err error) { - files, err := c.ReadDir(p0) - if err == nil { - fmt.Println(fmt.Sprintf("ReadDir: '%s' entries: %d ", p0, len(files))) - for _, f := range files { - fmt.Println(f) - } - } - return - } - - case "STAT": - return func(c *d.Client, p0, p1 string) (err error) { - file, err := c.Stat(p0) - if err == nil { - fmt.Println(file) - } - return - } - - case "GET", "PULL", "READ": - return func(c *d.Client, p0, p1 string) (err error) { - bytes, err := c.Read(p0) - if err == nil { - if err = writeFile(p1, bytes, 0644); err == nil { - fmt.Println(fmt.Sprintf("Written %d bytes to: %s", len(bytes), p1)) - } - } - return - } - - case "DELETE", "RM", "DEL": - return func(c *d.Client, p0, p1 string) (err error) { - if err = c.Remove(p0); err == nil { - fmt.Println("RM: " + p0) - } - return - } - - case "MKCOL", "MKDIR": - return func(c *d.Client, p0, p1 string) (err error) { - if err = c.Mkdir(p0, 0755); err == nil { - fmt.Println("MkDir: " + p0) - } - return - } - - case "MKCOLALL", "MKDIRALL": - return func(c *d.Client, p0, p1 string) (err error) { - if err = c.MkdirAll(p0, 0755); err == nil { - fmt.Println("MkDirAll: " + p0) - } - return - } - - case "RENAME", "MV", "MOVE": - return func(c *d.Client, p0, p1 string) (err error) { - if err = c.Rename(p0, p1, true); err == nil { - fmt.Println("Rename: " + p0 + " -> " + p1) - } - return - } - - case "COPY", "CP": - return func(c *d.Client, p0, p1 string) (err error) { - if err = c.Copy(p0, p1, true); err == nil { - fmt.Println("Copy: " + p0 + " -> " + p1) - } - return - } - - case "PUT", "PUSH", "WRITE": - return func(c *d.Client, p0, p1 string) (err error) { - stream, err := getStream(p1) - if err == nil { - if err = c.WriteStream(p0, stream, 0644); err == nil { - fmt.Println(fmt.Sprintf("Put: '%s' -> %s", p1, p0)) - } - } - return - } - - default: - return func(c *d.Client, p0, p1 string) (err error) { - return errors.New("Unsupported method: " + method) - } - } -} - func main() { root := flag.String("root", os.Getenv("ROOT"), "WebDAV Endpoint [ENV.ROOT]") usr := flag.String("user", os.Getenv("USER"), "User [ENV.USER]") @@ -153,6 +43,134 @@ func main() { } } +func fail(err interface{}) { + if err != nil { + fmt.Println(err) + } + os.Exit(-1) +} + +func getCmd(method string) func(c *d.Client, p0, p1 string) error { + switch method { + case "LS", "LIST", "PROPFIND": + return cmdLs + + case "STAT": + return cmdStat + + case "GET", "PULL", "READ": + return cmdGet + + case "DELETE", "RM", "DEL": + return cmdRm + + case "MKCOL", "MKDIR": + return cmdMkdir + + case "MKCOLALL", "MKDIRALL": + return cmdMkdirAll + + case "RENAME", "MV", "MOVE": + return cmdMv + + case "COPY", "CP": + return cmdCp + + case "PUT", "PUSH", "WRITE": + return cmdPut + + default: + return func(c *d.Client, p0, p1 string) (err error) { + return errors.New("Unsupported method: " + method) + } + } +} + +func cmdLs(c *d.Client, p0, p1 string) (err error) { + files, err := c.ReadDir(p0) + if err == nil { + fmt.Println(fmt.Sprintf("ReadDir: '%s' entries: %d ", p0, len(files))) + for _, f := range files { + fmt.Println(f) + } + } + return +} + +func cmdStat(c *d.Client, p0, p1 string) (err error) { + file, err := c.Stat(p0) + if err == nil { + fmt.Println(file) + } + return +} + +func cmdGet(c *d.Client, p0, p1 string) (err error) { + bytes, err := c.Read(p0) + if err == nil { + if err = writeFile(p1, bytes, 0644); err == nil { + fmt.Println(fmt.Sprintf("Written %d bytes to: %s", len(bytes), p1)) + } + } + return +} + +func cmdRm(c *d.Client, p0, p1 string) (err error) { + if err = c.Remove(p0); err == nil { + fmt.Println("RM: " + p0) + } + return +} + +func cmdMkdir(c *d.Client, p0, p1 string) (err error) { + if err = c.Mkdir(p0, 0755); err == nil { + fmt.Println("MkDir: " + p0) + } + return +} + +func cmdMkdirAll(c *d.Client, p0, p1 string) (err error) { + if err = c.MkdirAll(p0, 0755); err == nil { + fmt.Println("MkDirAll: " + p0) + } + return +} + +func cmdMv(c *d.Client, p0, p1 string) (err error) { + if err = c.Rename(p0, p1, true); err == nil { + fmt.Println("Rename: " + p0 + " -> " + p1) + } + return +} + +func cmdCp(c *d.Client, p0, p1 string) (err error) { + if err = c.Copy(p0, p1, true); err == nil { + fmt.Println("Copy: " + p0 + " -> " + p1) + } + return +} + +func cmdPut(c *d.Client, p0, p1 string) (err error) { + stream, err := getStream(p1) + if err == nil { + if err = c.WriteStream(p0, stream, 0644); err == nil { + fmt.Println(fmt.Sprintf("Put: '%s' -> %s", p1, p0)) + } + } + return +} + +func writeFile(path string, bytes []byte, mode os.FileMode) error { + f, err := os.Create(path) + defer f.Close() + if err != nil { + return err + } + + _, err = f.Write(bytes) + return err +} + func getStream(pathOrString string) (io.ReadCloser, error) { fi, err := os.Stat(pathOrString) if err == nil {