package search

import (
	"b612.me/stario"
	"b612.me/starlog"
	"b612.me/startext"
	"bufio"
	"fmt"
	"github.com/spf13/cobra"
	"io/ioutil"
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

var stFolder string
var stNum, stMax, stMin int
var stautoGBK bool

func init() {
	Cmd.Flags().StringVarP(&stFolder, "folder", "f", "./", "搜索的文件夹")
	Cmd.Flags().IntVarP(&stNum, "thread-num", "n", 5, "并发搜寻协程数")
	Cmd.Flags().BoolVarP(&stautoGBK, "autogbk", "g", true, "自动GBK识别")
	Cmd.Flags().IntVar(&stMax, "max", 0, "行最大字数")
	Cmd.Flags().IntVar(&stMin, "min", 0, "行最小字数")
}

var Cmd = &cobra.Command{
	Use:   "st",
	Short: "搜索文件中特定字符串",
	Long:  "搜索文件中特定字符串",
	Run: func(this *cobra.Command, args []string) {
		if len(args) != 2 {
			starlog.Errorln("应当传入两个参数，搜寻文件后缀和搜寻文本")
			os.Exit(1)
		}
		err := searchText(stFolder, args[0], args[1], stNum, stautoGBK, stMax, stMin)
		if err != nil {
			os.Exit(2)
		}
		return
	},
}

func searchText(folder string, filematch string, text string, thread int, autoGBK bool, max, min int) error {
	data, err := ioutil.ReadDir(folder)
	if err != nil {
		starlog.Errorln("read folder failed", folder, err)
		return err
	}
	wg := stario.NewWaitGroup(thread)
	searchFn := func(filepath string, text string) {
		//starlog.Debugln("searching", filepath, text)
		defer wg.Done()
		fp, err := os.Open(filepath)
		if err != nil {
			starlog.Errorln("open file failed", filepath, err)
			return
		}
		defer fp.Close()
		reader := bufio.NewReader(fp)
		count := 0
		for {
			origin, err := reader.ReadString('\n')
			count++
			if stautoGBK && startext.IsGBK([]byte(origin)) {
				originByte, _ := startext.GBK2UTF8([]byte(origin))
				origin = string(originByte)
			}
			origin = strings.TrimSpace(origin)
			if max != 0 && len(origin) > max {
				continue
			}
			if min != 0 && len(origin) < min {
				continue
			}
			if strings.Contains(origin, text) {
				fmt.Printf("file:%s line:%d matched:%s\n", filepath, count, origin)
			}
			if err != nil {
				break
			}
		}
	}
	for _, v := range data {
		if v.IsDir() {
			searchText(filepath.Join(folder, v.Name()), filematch, text, thread, autoGBK, stMax, stMin)
		}
		filepath := filepath.Join(folder, v.Name())
		if matched, _ := regexp.MatchString(filematch, filepath); matched {
			wg.Add(1)
			go searchFn(filepath, text)
		}
	}
	wg.Wait()
	return nil
}