bb/tool/in/main.go
2024-05-15 23:07:35 +05:00

68 lines
881 B
Go

package in
import (
"os"
"io"
"bufio"
"fmt"
"surdeus.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)
}