-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
108 lines (95 loc) · 2.32 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package main
import (
"flag"
"fmt"
"io"
"log"
"git.lenzplace.org/lenzj/chunkio"
"os"
"os/exec"
"path/filepath"
"strings"
)
func Parse(in io.Reader, out io.Writer, start, end []byte, filtSlice []string) error {
cio := chunkio.NewReader(in)
for cio.GetErr() == nil {
var cmd *exec.Cmd
if len(filtSlice) > 1 {
cmd = exec.Command(filtSlice[0], filtSlice[1:]...)
} else {
cmd = exec.Command(filtSlice[0])
}
cmd.Stdin = cio
cmd.Stdout = out
cio.SetKey(start)
_, err := io.Copy(out, cio)
if err == io.ErrUnexpectedEOF {
break
} else if err != nil {
return err
}
cio.Reset()
cio.SetKey(end)
if err = cmd.Run(); err != nil {
return err
}
cio.Reset()
}
return nil
}
var Version string
func main() {
// Get application name as executed from command prompt
appName := filepath.Base(os.Args[0])
// Set up formatting for error messages
log.SetFlags(0)
log.SetPrefix(appName + ": ")
// Parse command line
flag.Usage = func() {
fmt.Fprintf(flag.CommandLine.Output(),
"Usage: %s BEGIN END FILTER\n"+
"Transform sections of stdin by passing through a filter program.\n"+
"Sections are defined by user selected BEGIN and END identifiers.\n"+
"Options:\n", appName)
flag.PrintDefaults()
}
var vflag = flag.Bool("v", false, "display " + appName + " version")
flag.Parse()
// Display application version if requested
if *vflag {
fmt.Println(appName + " " + Version)
os.Exit(0)
}
var (
input io.Reader
output io.Writer
err error
filtStart, filtEnd string
filtSlice []string
)
switch flag.NArg() {
case 3:
filtStart = flag.Arg(0)
if len(filtStart) == 0 {
log.Fatalln("BEGIN chunk identifier must have length greater than zero")
}
filtEnd = flag.Arg(1)
if len(filtEnd) == 0 {
log.Fatalln("END chunk identifier must have length greater than zero")
}
filtSlice = strings.Split(flag.Arg(2), " ")
if _, err = exec.LookPath(filtSlice[0]); err != nil {
log.Fatalln(err)
}
default:
log.Println("incorrect number of arguments")
flag.Usage()
os.Exit(1)
}
input = os.Stdin
output = os.Stdout
err = Parse(input, output, []byte(filtStart), []byte(filtEnd), filtSlice)
if err != nil {
log.Fatalln(err)
}
}