Merge pull request #30 from dmarkham/use-constant-names

Use constant names, Help the compiler detect value changes

enumer.go

@@ -61,7 +61,7 @@ func (g *Generator) buildBasicExtras(runs [][]Value, typeName string, runsThresh
 	g.Printf("\nvar _%sValues = []%s{", typeName, typeName)
 	for _, values := range runs {
 		for _, value := range values {
-			g.Printf("\t%s, ", value.str)
+			g.Printf("\t%s, ", value.originalName)
 		}
 	}
 	g.Printf("}\n\n")
@@ -98,8 +98,8 @@ func (g *Generator) printValueMap(runs [][]Value, typeName string, runsThreshold
 		}
 
 		for _, value := range values {
-			g.Printf("\t_%sName%s[%d:%d]: %s,\n", typeName, runID, n, n+len(value.name), &value)
-			g.Printf("\t_%sLowerName%s[%d:%d]: %s,\n", typeName, runID, n, n+len(value.name), &value)
+			g.Printf("\t_%sName%s[%d:%d]: %s,\n", typeName, runID, n, n+len(value.name), value.originalName)
+			g.Printf("\t_%sLowerName%s[%d:%d]: %s,\n", typeName, runID, n, n+len(value.name), value.originalName)
 			n += len(value.name)
 		}
 	}

stringer.go

@@ -445,6 +445,7 @@ func (g *Generator) generate(typeName string, includeJSON, includeYAML, includeS
 	default:
 		g.buildMap(runs, typeName)
 	}
+	g.buildNoOpOrderChangeDetect(runs, typeName)
 
 	g.buildBasicExtras(runs, typeName, runsThreshold)
 	if includeJSON {
@@ -508,6 +509,7 @@ func (g *Generator) format() []byte {
 
 // Value represents a declared constant.
 type Value struct {
+	originalName string // The name of the constant before transformation
 	name         string // The name of the constant after transformation (i.e. camel case => snake case)
 	// The value is stored as a bit pattern alone. The boolean tells us
 	// whether to interpret it as an int64 or a uint64; the only place
@@ -602,6 +604,7 @@ func (f *File) genDecl(node ast.Node) bool {
 				u64 = uint64(i64)
 			}
 			v := Value{
+				originalName: n.Name,
 				name:         n.Name,
 				value:        u64,
 				signed:       info&types.IsUnsigned == 0,
@@ -817,6 +820,24 @@ func (g *Generator) buildMap(runs [][]Value, typeName string) {
 	g.Printf(stringMap, typeName)
 }
 
+// buildNoOpOrderChangeDetect try to let the compiler and the user know if the order/value of the ENUMS have changed.
+func (g *Generator) buildNoOpOrderChangeDetect(runs [][]Value, typeName string) {
+	g.Printf("\n")
+
+	g.Printf(`
+	// An "invalid array index" compiler error signifies that the constant values have changed.
+	// Re-run the stringer command to generate them again.
+	`)
+	g.Printf("func _%sNoOp (){ ", typeName)
+	g.Printf("\n var x [1]struct{}\n")
+	for _, values := range runs {
+		for _, value := range values {
+			g.Printf("\t_ = x[%s-(%s)]\n", value.originalName, value.str)
+		}
+	}
+	g.Printf("}\n\n")
+}
+
 // Argument to format is the type name.
 const stringMap = `func (i %[1]s) String() string {
 	if str, ok := _%[1]sMap[i]; ok {

util_test.go

@@ -54,7 +54,7 @@ Outer:
 	for n, test := range splitTests {
 		values := make([]Value, len(test.input))
 		for i, v := range test.input {
-			values[i] = Value{"", v, test.signed, fmt.Sprint(v)}
+			values[i] = Value{"", "", v, test.signed, fmt.Sprint(v)}
 		}
 		runs := splitIntoRuns(values)
 		if len(runs) != len(test.output) {