compflag/gen/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

// Generates flags.go.
package main

import (
	"log"
	"strings"

	"github.com/posener/autogen"
)

//go:generate go run .

type flag struct {
	Name          string
	Type          string
	IsBool        bool
	CustomPredict bool
}

func (f flag) NewInternalTypeFuncName() string {
	return "new" + strings.Title(f.InternalTypeName())
}

func (f flag) InternalTypeName() string {
	return strings.ToLower(f.Name[:1]) + f.Name[1:] + "Value"
}

var flags = []flag{
	{Name: "String", Type: "string"},
	{Name: "Bool", Type: "bool", IsBool: true, CustomPredict: true},
	{Name: "Int", Type: "int"},
	{Name: "Duration", Type: "time.Duration"},
}

func main() {
	err := autogen.Execute(flags)
	if err != nil {
		log.Fatal(err)
	}
}