shoopea/gowebdav
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor getCmd

Browse Source
This commit is contained in:
Christoph Polcin 2018-05-25 10:15:26 +02:00
parent 1e4b7a5554
commit c16034adb0
1 changed files with 128 additions and 110 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

238
cmd/gowebdav/main.go
View File

@ -10,116 +10,6 @@ import (
"strings" "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() { func main() {
root := flag.String("root", os.Getenv("ROOT"), "WebDAV Endpoint [ENV.ROOT]") root := flag.String("root", os.Getenv("ROOT"), "WebDAV Endpoint [ENV.ROOT]")
usr := flag.String("user", os.Getenv("USER"), "User [ENV.USER]") 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) { func getStream(pathOrString string) (io.ReadCloser, error) {
fi, err := os.Stat(pathOrString) fi, err := os.Stat(pathOrString)
if err == nil { if err == nil {