pug.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"runtime"
	"strings"
	"sync"

	"github.com/CrowdSurge/banner"
	"github.com/fatih/color"
)

var wg sync.WaitGroup
var search_string string = ""

var cyan = color.New(color.FgCyan)
var boldCyan = cyan.Add(color.Bold)

func walk_r(dir string) {
	f, err := os.Open(dir)
	if err != nil {
		fmt.Println(err)
	}
	list, err := f.Readdir(-1)
	f.Close()
	if err != nil {
		fmt.Println(err)
	}
	for _, v := range list {
		if strings.HasPrefix(v.Name(), ".") == false &&
			strings.HasSuffix(v.Name(), ".log") == false {
			path := fmt.Sprintf("%s%s%s", dir, string(os.PathSeparator), v.Name())
			if v.IsDir() {
				wg.Add(1)
				go walk_r(path)
			} else {
				file, err := ioutil.ReadFile(path)
				if err != nil {
					fmt.Println(err)
				}
				str := string(file)
				line := 0
				temp := strings.Split(str, "\n")
				matched_lines := []string{}
				for _, item := range temp {
					if strings.Contains(item, search_string) {
						white := color.New(color.BgRed, color.FgYellow, color.Bold).SprintFunc()

						h_item := strings.Replace(item, search_string, white(search_string), -1)

						t1 := fmt.Sprintf("%d \t %s", line, h_item)
						t2 := len(t1)
						if len(t1) > 255 {
							t2 = 255
						}
						matched_lines = append(matched_lines, t1[:t2])
					}
					line++
				}
				if len(matched_lines) > 0 {

					boldCyan.Println(path)
					for _, str := range matched_lines {
						color.Yellow(str)
					}
				}

			}
		}
	}
	wg.Done()
}

func main() {
	runtime.GOMAXPROCS(runtime.NumCPU())

	args := os.Args[1:]

	banner.Print("  pug  ")
	color.Red("\n- by Siddhartha Mukherjee <mukherjee.siddhartha@gmail.com>")
	color.Yellow("\nUsage: pug [search string] [optional: /path/to/search]\n")

	dir := "."
	search_string = args[0]

	if len(args) > 1 {
		dir = args[1]
	}

	wg.Add(1)
	go walk_r(dir)
	wg.Wait()
}