bb/tool/in/main.go

package in

import (
	"os"
	"io"
	"bufio"
	"fmt"
	"vultras.su/core/cli/mtool"
)

func Run(flagSet *mtool.Flags) {
	var (
		print bool
		not bool
	)
	flagSet.BoolVar(&print, "p", false, "print matching lines")
	flagSet.BoolVar(&not, "n", false, "find not matching lines")
	flagSet.Parse()
	args := flagSet.Args()

	if len(args) == 0 {
		//flagSet.Usage()
		if !not {
			os.Exit(1)
		}
	}

	mp := make(map[string] int)
	for _, v := range args {
		mp[v] = 0
	}

	status := 1
	r := bufio.NewReader(os.Stdin)

	if not {
		status = 0
	}
	for {
		l, err := r.ReadString('\n')
		if err == io.EOF {
			break
		}

		if len(l) != 0 && l[len(l)-1] == '\n' {
			l = l[:len(l)-1]
		}

		_, ok := mp[l]
		if not {
			if ok {
				status = 1
				ok = false
			} else {
				ok = true
			}
		} else {
			if ok {
				status = 0
			}
		}

		if print && ok {
			fmt.Println(l)
		}
	}
	os.Exit(status)
}