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
|
package main
import (
"encoding/json"
"flag"
"fmt"
"go/parser"
"go/token"
"os"
"github.com/onsi/ginkgo/ginkgo/outline"
)
const (
// indentWidth is the width used by the 'indent' output
indentWidth = 4
// stdinAlias is a portable alias for stdin. This convention is used in
// other CLIs, e.g., kubectl.
stdinAlias = "-"
usageCommand = "ginkgo outline <filename>"
)
func BuildOutlineCommand() *Command {
const defaultFormat = "csv"
var format string
flagSet := flag.NewFlagSet("outline", flag.ExitOnError)
flagSet.StringVar(&format, "format", defaultFormat, "Format of outline. Accepted: 'csv', 'indent', 'json'")
return &Command{
Name: "outline",
FlagSet: flagSet,
UsageCommand: usageCommand,
Usage: []string{
"Create an outline of Ginkgo symbols for a file",
"To read from stdin, use: `ginkgo outline -`",
"Accepts the following flags:",
},
Command: func(args []string, additionalArgs []string) {
outlineFile(args, format)
},
}
}
func outlineFile(args []string, format string) {
if len(args) != 1 {
println(fmt.Sprintf("usage: %s", usageCommand))
os.Exit(1)
}
filename := args[0]
var src *os.File
if filename == stdinAlias {
src = os.Stdin
} else {
var err error
src, err = os.Open(filename)
if err != nil {
println(fmt.Sprintf("error opening file: %s", err))
os.Exit(1)
}
}
fset := token.NewFileSet()
parsedSrc, err := parser.ParseFile(fset, filename, src, 0)
if err != nil {
println(fmt.Sprintf("error parsing source: %s", err))
os.Exit(1)
}
o, err := outline.FromASTFile(fset, parsedSrc)
if err != nil {
println(fmt.Sprintf("error creating outline: %s", err))
os.Exit(1)
}
var oerr error
switch format {
case "csv":
_, oerr = fmt.Print(o)
case "indent":
_, oerr = fmt.Print(o.StringIndent(indentWidth))
case "json":
b, err := json.Marshal(o)
if err != nil {
println(fmt.Sprintf("error marshalling to json: %s", err))
}
_, oerr = fmt.Println(string(b))
default:
complainAndQuit(fmt.Sprintf("format %s not accepted", format))
}
if oerr != nil {
println(fmt.Sprintf("error writing outline: %s", oerr))
os.Exit(1)
}
}
|
