Put search and command in the same file
Henrique Dias
parent
2ca408d14f
commit
2d82967b44
116
search.go
116
search.go
|
|
@ -1,116 +0,0 @@
|
||||||
package filemanager
|
|
||||||
|
|
||||||
import (
|
|
||||||
"net/http"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/gorilla/websocket"
|
|
||||||
)
|
|
||||||
|
|
||||||
type searchOptions struct {
|
|
||||||
CaseInsensitive bool
|
|
||||||
Terms []string
|
|
||||||
}
|
|
||||||
|
|
||||||
func parseSearch(value string) *searchOptions {
|
|
||||||
opts := &searchOptions{
|
|
||||||
CaseInsensitive: strings.Contains(value, "case:insensitive"),
|
|
||||||
}
|
|
||||||
|
|
||||||
// removes the options from the value
|
|
||||||
value = strings.Replace(value, "case:insensitive", "", -1)
|
|
||||||
value = strings.Replace(value, "case:sensitive", "", -1)
|
|
||||||
value = strings.TrimSpace(value)
|
|
||||||
|
|
||||||
if opts.CaseInsensitive {
|
|
||||||
value = strings.ToLower(value)
|
|
||||||
}
|
|
||||||
|
|
||||||
// if the value starts with " and finishes what that character, we will
|
|
||||||
// only search for that term
|
|
||||||
if value[0] == '"' && value[len(value)-1] == '"' {
|
|
||||||
unique := strings.TrimPrefix(value, "\"")
|
|
||||||
unique = strings.TrimSuffix(unique, "\"")
|
|
||||||
|
|
||||||
opts.Terms = []string{unique}
|
|
||||||
return opts
|
|
||||||
}
|
|
||||||
|
|
||||||
opts.Terms = strings.Split(value, " ")
|
|
||||||
return opts
|
|
||||||
}
|
|
||||||
|
|
||||||
// search searches for a file or directory.
|
|
||||||
func search(c *RequestContext, w http.ResponseWriter, r *http.Request) (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 (
|
|
||||||
value string
|
|
||||||
search *searchOptions
|
|
||||||
message []byte
|
|
||||||
)
|
|
||||||
|
|
||||||
// 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 {
|
|
||||||
value = string(message)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
search = parseSearch(value)
|
|
||||||
scope := strings.TrimPrefix(r.URL.Path, "/")
|
|
||||||
scope = "/" + scope
|
|
||||||
scope = string(c.User.FileSystem) + scope
|
|
||||||
scope = strings.Replace(scope, "\\", "/", -1)
|
|
||||||
scope = filepath.Clean(scope)
|
|
||||||
|
|
||||||
err = filepath.Walk(scope, func(path string, f os.FileInfo, err error) error {
|
|
||||||
if search.CaseInsensitive {
|
|
||||||
path = strings.ToLower(path)
|
|
||||||
}
|
|
||||||
|
|
||||||
path = strings.TrimPrefix(path, scope)
|
|
||||||
path = strings.TrimPrefix(path, "/")
|
|
||||||
path = strings.Replace(path, "\\", "/", -1)
|
|
||||||
is := false
|
|
||||||
|
|
||||||
for _, term := range search.Terms {
|
|
||||||
if is {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
|
|
||||||
if strings.Contains(path, term) {
|
|
||||||
if !c.User.Allowed(path) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
is = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !is {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return conn.WriteMessage(websocket.TextMessage, []byte(path))
|
|
||||||
})
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return http.StatusInternalServerError, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0, nil
|
|
||||||
}
|
|
||||||
|
|
@ -3,6 +3,7 @@ package filemanager
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
@ -133,3 +134,109 @@ func command(c *RequestContext, w http.ResponseWriter, r *http.Request) (int, er
|
||||||
|
|
||||||
return 0, nil
|
return 0, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type searchOptions struct {
|
||||||
|
CaseInsensitive bool
|
||||||
|
Terms []string
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseSearch(value string) *searchOptions {
|
||||||
|
opts := &searchOptions{
|
||||||
|
CaseInsensitive: strings.Contains(value, "case:insensitive"),
|
||||||
|
}
|
||||||
|
|
||||||
|
// removes the options from the value
|
||||||
|
value = strings.Replace(value, "case:insensitive", "", -1)
|
||||||
|
value = strings.Replace(value, "case:sensitive", "", -1)
|
||||||
|
value = strings.TrimSpace(value)
|
||||||
|
|
||||||
|
if opts.CaseInsensitive {
|
||||||
|
value = strings.ToLower(value)
|
||||||
|
}
|
||||||
|
|
||||||
|
// if the value starts with " and finishes what that character, we will
|
||||||
|
// only search for that term
|
||||||
|
if value[0] == '"' && value[len(value)-1] == '"' {
|
||||||
|
unique := strings.TrimPrefix(value, "\"")
|
||||||
|
unique = strings.TrimSuffix(unique, "\"")
|
||||||
|
|
||||||
|
opts.Terms = []string{unique}
|
||||||
|
return opts
|
||||||
|
}
|
||||||
|
|
||||||
|
opts.Terms = strings.Split(value, " ")
|
||||||
|
return opts
|
||||||
|
}
|
||||||
|
|
||||||
|
// search searches for a file or directory.
|
||||||
|
func search(c *RequestContext, w http.ResponseWriter, r *http.Request) (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 (
|
||||||
|
value string
|
||||||
|
search *searchOptions
|
||||||
|
message []byte
|
||||||
|
)
|
||||||
|
|
||||||
|
// 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 {
|
||||||
|
value = string(message)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
search = parseSearch(value)
|
||||||
|
scope := strings.TrimPrefix(r.URL.Path, "/")
|
||||||
|
scope = "/" + scope
|
||||||
|
scope = string(c.User.FileSystem) + scope
|
||||||
|
scope = strings.Replace(scope, "\\", "/", -1)
|
||||||
|
scope = filepath.Clean(scope)
|
||||||
|
|
||||||
|
err = filepath.Walk(scope, func(path string, f os.FileInfo, err error) error {
|
||||||
|
if search.CaseInsensitive {
|
||||||
|
path = strings.ToLower(path)
|
||||||
|
}
|
||||||
|
|
||||||
|
path = strings.TrimPrefix(path, scope)
|
||||||
|
path = strings.TrimPrefix(path, "/")
|
||||||
|
path = strings.Replace(path, "\\", "/", -1)
|
||||||
|
is := false
|
||||||
|
|
||||||
|
for _, term := range search.Terms {
|
||||||
|
if is {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
if strings.Contains(path, term) {
|
||||||
|
if !c.User.Allowed(path) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
is = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !is {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return conn.WriteMessage(websocket.TextMessage, []byte(path))
|
||||||
|
})
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return http.StatusInternalServerError, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0, nil
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue