96 lines
2.1 KiB
Go
96 lines
2.1 KiB
Go
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)
|
|
}
|
|
}
|