57 lines
1.1 KiB
Go
Executable File
57 lines
1.1 KiB
Go
Executable File
package notes
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"local/notes-server/filetree"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
)
|
|
|
|
func (n *Notes) Search(phrase string) (string, error) {
|
|
files := filetree.NewFiles()
|
|
err := filepath.Walk(n.root,
|
|
func(walked string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
if size := info.Size(); size < 1 || size > (5*1024*1024) {
|
|
return nil
|
|
}
|
|
ok, err := grepFile(walked, []byte(phrase))
|
|
if err != nil && err.Error() == "bufio.Scanner: token too long" {
|
|
err = nil
|
|
}
|
|
if err == nil && ok {
|
|
p := filetree.NewPathFromLocal(path.Dir(walked))
|
|
files.Push(p, info)
|
|
}
|
|
if err != nil {
|
|
log.Printf("failed to scan %v: %v", walked, err)
|
|
}
|
|
return err
|
|
},
|
|
)
|
|
return filetree.Paths(*files).List(true), err
|
|
}
|
|
|
|
func grepFile(file string, phrase []byte) (bool, error) {
|
|
f, err := os.Open(file)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer f.Close()
|
|
scanner := bufio.NewScanner(f)
|
|
for scanner.Scan() {
|
|
if bytes.Contains(scanner.Bytes(), phrase) {
|
|
return true, scanner.Err()
|
|
}
|
|
}
|
|
return false, scanner.Err()
|
|
}
|