68 lines
885 B
Go
68 lines
885 B
Go
package in
|
|
|
|
import (
|
|
"os"
|
|
"io"
|
|
"bufio"
|
|
"fmt"
|
|
"github.com/omnipunk/cli/mtool"
|
|
)
|
|
|
|
func Run(flagSet *mtool.Flags) {
|
|
var (
|
|
print bool
|
|
not bool
|
|
)
|
|
flagSet.BoolVar(&print, "p", false, "print matching lines")
|
|
flagSet.BoolVar(¬, "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)
|
|
}
|