Skip to content

Instantly share code, notes, and snippets.

@ShawnMilo
Created Jan 6, 2015
Embed
What would you like to do?
foe. A file search that opens the files found in vim.
package main
import (
"bufio"
"flag"
"log"
"net/http"
"os"
"os/exec"
"path/filepath"
"strings"
)
var cFiles = make(chan string)
var path string // path to search
var args []string
// isText determines whether this is the kind of file we
// want to open in vim.
func isText(filename string) bool {
file, err := os.Open(filename)
if err != nil {
log.Println("Unable to check type of", filename, ":", err)
return false
}
defer file.Close()
scanner := bufio.NewScanner(file)
scanner.Scan()
fileType := http.DetectContentType(scanner.Bytes())
if fileType[:4] == "text" {
return true
}
return false
}
// isDir accepts a string (file path) and returns
// a boolean which indicates if the path is
// a valid directory.
func isDir(path string) bool {
stat, err := os.Stat(path)
if err != nil {
log.Println("Error checking whether", path, "was a directory:", err)
return false
}
return stat.IsDir()
}
func init() {
// Set up command-line flags.
flag.StringVar(&path, "p", ".", "path")
flag.Parse()
// Validate flags.
if !isDir(path) {
log.Fatal(path, "is not a valid path.")
}
args = flag.Args()
if len(args) == 0 {
log.Fatal("no arguments passed")
}
}
// walker implements filepath.WalkFunc.
// Search filename for arguments passed in.
func walker(path string, info os.FileInfo, err error) error {
if err != nil {
log.Println(err, "in file", path)
return nil
}
if isDir(path) {
return nil
}
for _, arg := range args {
if !strings.Contains(strings.ToLower(info.Name()), strings.ToLower(arg)) {
return nil
}
}
if !isText(path) {
return nil
}
cFiles <- path
return nil
}
func main() {
go func() {
filepath.Walk(path, walker)
close(cFiles)
}()
paths := make([]string, 0, 50)
for path := range cFiles {
paths = append(paths, path)
}
if len(paths) < 1 {
log.Println("No results.")
return
}
if len(paths) > 3 {
log.Println("Too many results:")
for _, x := range paths {
log.Println(x)
}
return
}
cmd := exec.Command("vim", paths...)
cmd.Stdout = os.Stdout
cmd.Stdin = os.Stdin
cmd.Run()
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment