From e1b74fdc9e19e1bfa6a4d15c7427a0ccf22be203 Mon Sep 17 00:00:00 2001
From: Vasiliy Tolstov <v.tolstov@selfip.ru>
Date: Thu, 30 Apr 2015 17:26:59 +0300
Subject: [PATCH] add nlst command

closes #10

Signed-off-by: Vasiliy Tolstov <v.tolstov@selfip.ru>
---
 main.go | 71 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)

diff --git a/main.go b/main.go
index 41f452e..666c4e9 100644
--- a/main.go
+++ b/main.go
@@ -65,6 +65,8 @@ func handle(c net.Conn) {
 			s.cmdServerType(params)
 		case "LIST":
 			s.cmdServerList(params)
+		case "NLST":
+			s.cmdServerNlst(params)
 		case "SYST":
 			s.cmdServerSyst(params)
 		case "QUIT":
@@ -228,6 +230,75 @@ func (s *Conn) cmdServerList(args string) {
 	s.ctrl.PrintfLine(`226 Closing data connection`)
 }
 
+func (s *Conn) cmdServerNlst(args string) {
+	fmt.Printf("cmdServerNlst: %s\n", args)
+
+	var files []string
+	cnt := ""
+	p := ""
+	switch args {
+	case "-la", "-l", "-a", "-al":
+		args = ""
+	}
+	if len(args) > 0 && args[0] == '/' {
+		cnt = strings.Split(args, "/")[1]
+		p = strings.Join(strings.Split(args, "/")[2:], "/")
+	}
+
+	if cnt == "" && s.path == "/" {
+		cnts, err := s.sw.ContainersAll(nil)
+		if err != nil {
+			fmt.Printf(err.Error())
+			return
+		}
+		files = append(files, ".", "..")
+		for _, ct := range cnts {
+			files = append(files, ct.Name)
+		}
+	} else {
+		opts := &swift.ObjectsOpts{Delimiter: '/'}
+		cnt = strings.Split(s.path, "/")[1]
+		p = filepath.Clean(filepath.Join(strings.Join(strings.Split(s.path, "/")[2:], "/"), args))
+		fmt.Printf("cnt: %s p: %s\n", cnt, p)
+		if p != "." {
+			opts.Path = p
+		}
+		objs, err := s.sw.ObjectsAll(cnt, opts)
+		if err != nil {
+			fmt.Printf(err.Error())
+			return
+		}
+		fmt.Printf("%+v\n", objs)
+		files = append(files, ".", "..")
+
+		for _, obj := range objs {
+			if obj.PseudoDirectory || obj.ContentType == "application/directory" {
+				files = append(files, obj.Name)
+			} else {
+				files = append(files, obj.Name)
+			}
+		}
+	}
+
+	s.ctrl.PrintfLine(`150 Opening data channel for directory listing of "%s"`, args)
+	c, err := s.newSocket()
+	if err != nil {
+		fmt.Printf(err.Error())
+		s.ctrl.PrintfLine("425 Data connection failed")
+		return
+	}
+	s.data = c
+	defer c.Close()
+
+	_, err = c.Write([]byte(strings.Join(files, "\r\n")))
+	if err != nil {
+		s.ctrl.PrintfLine("425 Data connection failed")
+		fmt.Printf(err.Error())
+		return
+	}
+	s.ctrl.PrintfLine(`226 Closing data connection`)
+}
+
 func (s *Conn) cmdServerSyst(args string) {
 	fmt.Printf("cmdServerSyst: %s\n", args)
 	s.ctrl.PrintfLine("215 UNIX Type: L8")