|
|
|
|
@ -2,10 +2,16 @@ package main
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"encoding/json"
|
|
|
|
|
"fmt"
|
|
|
|
|
"github.com/shurcooL/httpfs/union"
|
|
|
|
|
"github.com/shurcooL/vfsgen"
|
|
|
|
|
"github.com/urfave/cli"
|
|
|
|
|
"io/ioutil"
|
|
|
|
|
"log"
|
|
|
|
|
"net/http"
|
|
|
|
|
"os"
|
|
|
|
|
"text/template"
|
|
|
|
|
"time"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
type CliFlagInfo struct {
|
|
|
|
|
@ -25,178 +31,133 @@ type FlagType struct {
|
|
|
|
|
ParserCast string `json:"parser_cast"`
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var flagTemplate = `// Code generated by fg; DO NOT EDIT.
|
|
|
|
|
|
|
|
|
|
package {{ .PackageName }}
|
|
|
|
|
{{ if eq .PackageName "cli" }}
|
|
|
|
|
import (
|
|
|
|
|
"flag"
|
|
|
|
|
"strconv"
|
|
|
|
|
"time"
|
|
|
|
|
)
|
|
|
|
|
{{ range $i, $flag := .Flags }}
|
|
|
|
|
// {{ $flag.Name }}Flag is a flag with type {{ $flag.Type }}{{ $flag.Doctail }}
|
|
|
|
|
type {{ $flag.Name }}Flag struct {
|
|
|
|
|
Name string
|
|
|
|
|
Usage string
|
|
|
|
|
EnvVar string
|
|
|
|
|
FilePath string
|
|
|
|
|
Required bool
|
|
|
|
|
Hidden bool
|
|
|
|
|
{{- if eq $flag.Value true }}
|
|
|
|
|
Value {{ $flag.Type }}
|
|
|
|
|
{{- end }}
|
|
|
|
|
{{- if eq $flag.Destination true }}
|
|
|
|
|
Destination *{{ $flag.Type }}
|
|
|
|
|
{{- end }}
|
|
|
|
|
// zeroModTimeFileSystem is an http.FileSystem wrapper.
|
|
|
|
|
// It exposes a filesystem exactly like Source, except
|
|
|
|
|
// all file modification times are changed to zero.
|
|
|
|
|
// See https://github.com/shurcooL/vfsgen/pull/40#issuecomment-355416103
|
|
|
|
|
type zeroModTimeFileSystem struct {
|
|
|
|
|
Source http.FileSystem
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// String returns a readable representation of this value
|
|
|
|
|
// (for usage defaults)
|
|
|
|
|
func (f {{ $flag.Name }}Flag) String() string {
|
|
|
|
|
return FlagStringer(f)
|
|
|
|
|
func (fs zeroModTimeFileSystem) Open(name string) (http.File, error) {
|
|
|
|
|
f, err := fs.Source.Open(name)
|
|
|
|
|
return file{f}, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// GetName returns the name of the flag
|
|
|
|
|
func (f {{ $flag.Name }}Flag) GetName() string {
|
|
|
|
|
return f.Name
|
|
|
|
|
type file struct {
|
|
|
|
|
http.File
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// IsRequired returns whether or not the flag is required
|
|
|
|
|
func (f {{ $flag.Name }}Flag) IsRequired() bool {
|
|
|
|
|
return f.Required
|
|
|
|
|
func (f file) Stat() (os.FileInfo, error) {
|
|
|
|
|
fi, err := f.File.Stat()
|
|
|
|
|
return fileInfo{fi}, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// {{ $flag.Name }} looks up the value of a local {{ $flag.Name }}Flag, returns
|
|
|
|
|
// {{ $flag.ContextDefault }} if not found
|
|
|
|
|
func (c *Context) {{ $flag.Name }}(name string) {{ if ne .ContextType "" }} {{ $flag.ContextType }} {{ else }} {{ $flag.Type }} {{- end }} {
|
|
|
|
|
return lookup{{ $flag.Name }}(name, c.flagSet)
|
|
|
|
|
type fileInfo struct {
|
|
|
|
|
os.FileInfo
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Global{{ $flag.Name }} looks up the value of a global {{ $flag.Name }}Flag, returns
|
|
|
|
|
// {{ $flag.ContextDefault }} if not found
|
|
|
|
|
func (c *Context) Global{{ $flag.Name }}(name string) {{ if ne .ContextType "" }} {{ $flag.ContextType }} {{ else }} {{ $flag.Type }} {{- end }} {
|
|
|
|
|
if fs := lookupGlobalFlagSet(name, c); fs != nil {
|
|
|
|
|
return lookup{{ $flag.Name }}(name, fs)
|
|
|
|
|
}
|
|
|
|
|
return {{ $flag.ContextDefault }}
|
|
|
|
|
}
|
|
|
|
|
func (fi fileInfo) ModTime() time.Time { return time.Time{} }
|
|
|
|
|
|
|
|
|
|
func lookup{{ $flag.Name }}(name string, set *flag.FlagSet) {{ if ne .ContextType "" }} {{ $flag.ContextType }} {{ else }} {{ $flag.Type }} {{- end }} {
|
|
|
|
|
f := set.Lookup(name)
|
|
|
|
|
if f != nil {
|
|
|
|
|
{{ if ne .Parser "" }}parsed, err := {{ $flag.Parser }}{{ else }}parsed, err := f.Value, error(nil){{ end }}
|
|
|
|
|
if err != nil {
|
|
|
|
|
return {{ $flag.ContextDefault }}
|
|
|
|
|
}
|
|
|
|
|
{{ if ne .ParserCast "" }}return {{ $flag.ParserCast }}{{ else }}return parsed{{ end }}
|
|
|
|
|
}
|
|
|
|
|
return {{ $flag.ContextDefault }}
|
|
|
|
|
}
|
|
|
|
|
{{- end }}
|
|
|
|
|
{{ else }}
|
|
|
|
|
import (
|
|
|
|
|
"flag"
|
|
|
|
|
"github.com/urfave/cli"
|
|
|
|
|
)
|
|
|
|
|
func main() {
|
|
|
|
|
app := cli.NewApp()
|
|
|
|
|
|
|
|
|
|
{{ range $i, $flag := .Flags }}
|
|
|
|
|
// {{ $flag.Name }}Flag is the flag type that wraps cli.{{ $flag.Name }}Flag to allow
|
|
|
|
|
// for other values to be specified
|
|
|
|
|
type {{ $flag.Name }}Flag struct {
|
|
|
|
|
cli.{{ $flag.Name }}Flag
|
|
|
|
|
set *flag.FlagSet
|
|
|
|
|
}
|
|
|
|
|
app.Name = "fg"
|
|
|
|
|
app.Usage = "Generate flag type code!"
|
|
|
|
|
app.Version = "v0.1.0"
|
|
|
|
|
|
|
|
|
|
// New{{ $flag.Name }}Flag creates a new {{ $flag.Name }}Flag
|
|
|
|
|
func New{{ $flag.Name }}Flag(fl cli.{{ $flag.Name }}Flag) *{{ $flag.Name }}Flag {
|
|
|
|
|
return &{{ $flag.Name }}Flag{ {{ $flag.Name }}Flag: fl, set: nil }
|
|
|
|
|
}
|
|
|
|
|
app.Action = ActionFunc
|
|
|
|
|
|
|
|
|
|
// Apply saves the flagSet for later usage calls, then calls
|
|
|
|
|
// the wrapped {{ $flag.Name }}Flag.Apply
|
|
|
|
|
func (f *{{ $flag.Name }}Flag) Apply(set *flag.FlagSet) {
|
|
|
|
|
f.set = set
|
|
|
|
|
f.{{ $flag.Name }}Flag.Apply(set)
|
|
|
|
|
}
|
|
|
|
|
err := GenerateAssets()
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// ApplyWithError saves the flagSet for later usage calls, then calls
|
|
|
|
|
// the wrapped {{ $flag.Name }}Flag.ApplyWithError
|
|
|
|
|
func (f *{{ $flag.Name }}Flag) ApplyWithError(set *flag.FlagSet) error {
|
|
|
|
|
f.set = set
|
|
|
|
|
return f.{{ $flag.Name }}Flag.ApplyWithError(set)
|
|
|
|
|
err = app.Run(os.Args)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
{{- end }}
|
|
|
|
|
{{- end }}
|
|
|
|
|
`
|
|
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
|
var packageName, inputPath, outputPath string
|
|
|
|
|
func GenerateAssets() error {
|
|
|
|
|
fs := zeroModTimeFileSystem{
|
|
|
|
|
Source: union.New(map[string]http.FileSystem{
|
|
|
|
|
"/templates": http.Dir("templates"),
|
|
|
|
|
"/source": http.Dir("source"),
|
|
|
|
|
}),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
app := cli.NewApp()
|
|
|
|
|
return vfsgen.Generate(fs, vfsgen.Options{
|
|
|
|
|
PackageName: "main",
|
|
|
|
|
VariableName: "fs",
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
app.Name = "fg"
|
|
|
|
|
app.Usage = "Generate flag type code!"
|
|
|
|
|
app.Version = "v0.1.0"
|
|
|
|
|
func ActionFunc(_ *cli.Context) error {
|
|
|
|
|
var info CliFlagInfo
|
|
|
|
|
var tpl *template.Template
|
|
|
|
|
|
|
|
|
|
app.Flags = []cli.Flag{
|
|
|
|
|
cli.StringFlag{
|
|
|
|
|
Name: "package, p",
|
|
|
|
|
Value: "cli",
|
|
|
|
|
Usage: "`PACKAGE` for which the flag types will be generated",
|
|
|
|
|
Destination: &packageName,
|
|
|
|
|
},
|
|
|
|
|
cli.StringFlag{
|
|
|
|
|
Name: "input, i",
|
|
|
|
|
Usage: "path to the `INPUT JSON FILE` which defines each type to be generated",
|
|
|
|
|
Destination: &inputPath,
|
|
|
|
|
},
|
|
|
|
|
cli.StringFlag{
|
|
|
|
|
Name: "output, o",
|
|
|
|
|
Usage: "path to the `OUTPUT GO FILE` which will contain the flag types",
|
|
|
|
|
Destination: &outputPath,
|
|
|
|
|
},
|
|
|
|
|
inFile, err := fs.Open("/source/flag-types.json")
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
app.Action = func(c *cli.Context) error {
|
|
|
|
|
var info CliFlagInfo
|
|
|
|
|
var tpl *template.Template
|
|
|
|
|
decoder := json.NewDecoder(inFile)
|
|
|
|
|
err = decoder.Decode(&info.Flags)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
err = inFile.Close()
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for _, packageName := range []string{"cli", "altsrc"} {
|
|
|
|
|
info.PackageName = packageName
|
|
|
|
|
|
|
|
|
|
inFile, err := os.Open(inputPath)
|
|
|
|
|
bytes, err := ReadTemplate(packageName)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
defer inFile.Close()
|
|
|
|
|
tpl = template.Must(template.New("").Parse(string(bytes)))
|
|
|
|
|
|
|
|
|
|
decoder := json.NewDecoder(inFile)
|
|
|
|
|
var outFile *os.File
|
|
|
|
|
|
|
|
|
|
err = decoder.Decode(&info.Flags)
|
|
|
|
|
if packageName == "cli" {
|
|
|
|
|
outFile, err = os.Create("flag_generated.go")
|
|
|
|
|
} else {
|
|
|
|
|
outFile, err = os.Create("altsrc/flag_generated.go")
|
|
|
|
|
}
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
tpl = template.Must(template.New("").Parse(flagTemplate))
|
|
|
|
|
|
|
|
|
|
outFile, err := os.Create(outputPath)
|
|
|
|
|
err = tpl.Execute(outFile, info)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
defer outFile.Close()
|
|
|
|
|
|
|
|
|
|
err = tpl.Execute(outFile, info)
|
|
|
|
|
err = outFile.Close()
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return nil
|
|
|
|
|
func ReadTemplate(packageName string) ([]byte, error) {
|
|
|
|
|
templateFile, err := fs.Open(fmt.Sprintf("/templates/%s_flags_generated.gotpl", packageName))
|
|
|
|
|
if err != nil {
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
err := app.Run(os.Args)
|
|
|
|
|
templateFileBytes, err := ioutil.ReadAll(templateFile)
|
|
|
|
|
if err != nil {
|
|
|
|
|
log.Fatal(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return templateFileBytes, nil
|
|
|
|
|
}
|
|
|
|
|
|
Ajitem Sahasrabuddhe
5 years ago