Working #1

handlers/search.go

@@ -0,0 +1,65 @@
+package handlers
+
+import (
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"github.com/gorilla/websocket"
+	"github.com/hacdias/caddy-filemanager/config"
+)
+
+func Search(w http.ResponseWriter, r *http.Request, c *config.Config, u *config.User) (int, error) {
+	// Upgrades the connection to a websocket and checks for errors.
+	conn, err := upgrader.Upgrade(w, r, nil)
+	if err != nil {
+		return 0, err
+	}
+	defer conn.Close()
+
+	var search string
+
+	// Starts an infinite loop until a valid command is captured.
+	for {
+		_, message, err := conn.ReadMessage()
+		if err != nil {
+			return http.StatusInternalServerError, err
+		}
+
+		if len(message) != 0 {
+			search = string(message)
+			break
+		}
+	}
+
+	scope := strings.Replace(r.URL.Path, c.BaseURL, "", 1)
+	scope = strings.TrimPrefix(scope, "/")
+	scope = "/" + scope
+	scope = u.Scope + scope
+	scope = strings.Replace(scope, "\\", "/", -1)
+	scope = filepath.Clean(scope)
+
+	err = filepath.Walk(scope, func(path string, f os.FileInfo, err error) error {
+		// TODO: check user permissions?
+
+		if strings.Contains(path, search) {
+			path = strings.TrimPrefix(path, scope)
+			path = strings.Replace(path, "\\", "/", -1)
+			path = strings.TrimPrefix(path, "/")
+
+			err = conn.WriteMessage(websocket.TextMessage, []byte(path))
+			if err != nil {
+				return err
+			}
+		}
+
+		return nil
+	})
+
+	if err != nil {
+		return http.StatusInternalServerError, err
+	}
+
+	return http.StatusOK, nil
+}