/
cli.go
55 lines (47 loc) · 1.04 KB
/
cli.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
package cksumfly
import (
"fmt"
"io"
"log"
"github.com/jessevdk/go-flags"
)
const (
exitCodeOK = iota
exitCodeParseFlagError
exitCodeErr
)
type CLI struct {
OutStream, ErrStream io.Writer
}
func (cli *CLI) Run(argv []string) int {
log.SetOutput(cli.ErrStream)
p, cksum, err := cli.parseArgs(argv)
if err != nil {
if ferr, ok := err.(*flags.Error); !ok || ferr.Type != flags.ErrHelp {
p.WriteHelp(cli.ErrStream)
}
return exitCodeParseFlagError
}
if err := cksum.Run(); err != nil {
log.Println(err)
return exitCodeErr
}
return exitCodeOK
}
var usage = fmt.Sprintf(`[OPTIONS]
or: cksumfly [FILE]
Version: %s (rev: %s)`, version, revision)
var description = "Print CRC checksum for flyway of each FILE."
func (cli *CLI) parseArgs(args []string) (*flags.Parser, *Cksumfly, error) {
cksum := &Cksumfly{
OutStream: cli.OutStream,
}
p := flags.NewParser(cksum, flags.Default)
p.Usage = usage
p.LongDescription = description
rest, err := p.ParseArgs(args)
if len(rest) > 0 {
cksum.FilePath = rest[0]
}
return p, cksum, err
}