refactor getCmd
This commit is contained in:
Christoph Polcin
parent
1e4b7a5554
commit
c16034adb0
@ -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 {
|
||||
|
||||
Loading…
Reference in New Issue
Block a user