summaryrefslogtreecommitdiff
path: root/flags.go
blob: 4fe55161ef5c77255e130742c327a2b51e2e8ba2 (plain)
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package log

import 	(
	"errors"
)

var INFO bool = true
var WARN bool = true
var ERROR bool = true

var VERBOSE bool = false
// var SPEW bool = false

var SPEW LogFlag

type LogFlag struct {
	B	bool
	Name	string
	Subsystem	string
	Desc	string
}

var registered map[string][]string
var flags []*LogFlag

func init() {
	registered = make(map[string][]string)

	SPEW.B = false
	SPEW.Name = "SPEW"
	SPEW.Subsystem = "log"
	SPEW.Desc = "Enable log.Spew()"

	// register the default flags used by this log package
	registered["log"] = []string{"SPEW","INFO", "WARN", "ERROR", "VERBOSE"}
}

func All(b bool) {
	Set("SPEW", b)
	Set("INFO", b)
	Set("WARN", b)
	Set("ERROR", b)
	Set("VERBOSE", b)
	for _, f := range flags {
		Warn("All() ", "(" + f.Subsystem + ")", f.Name, "=", f.B, ":", f.Desc)
		f.B = b
		Warn("All() f.B is now", f.B)
	}
}

// this bypasses all checks and _always_ logs the info to STDOUT
// is this a bad idea? Probably not....
func ListFlags() map[string][]string {
	Log(true, "ListFlags() registered =", registered)
	for _, f := range flags {
		Log(true, "ListFlags() ", "(" + f.Subsystem + ")", f.Name, "=", f.B, ":", f.Desc)
	}

	return registered
}

func Set(flag string, b bool) {
	switch flag {
	case "INFO":
		INFO = b
	case "WARN":
		WARN = b
	case "SPEW":
		SPEW.B = b
	case "ERROR":
		ERROR = b
	case "VERBOSE":
		VERBOSE = b
	default:
		Error(errors.New("unknown flag"), "Flag name sent:", flag)
	}
}

func Get(flag string) bool {
	switch flag {
	case "INFO":
		return INFO
	case "WARN":
		return WARN
	case "SPEW":
		return SPEW.B
	case "ERROR":
		return ERROR
	case "VERBOSE":
		return VERBOSE
	default:
		Error(errors.New("unknown flag"), "Flag name sent:", flag)
	}
	return false
}

// register a variable name from a subsystem
// inspired by Alex Flint
func (f *LogFlag) Register() {
	Info("log.Register() ", f)
	flags = append(flags,f)
}

// register a variable name from a subsystem
// this is used for custom log flags
func Register(subsystem string, name string, b *bool) {
	Info("log.Register() got subsystem", subsystem, "with name =", name, "bool value =", b)
	registered[subsystem] = append(registered[subsystem], name)
}