Fix "fmt" import (#3)

thnxdev · Oct 26, 2023 · a851a26 · a851a26
1 parent 03e5a17
commit a851a26
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/happy/main.go b/cmd/happy/main.go
@@ -379,27 +379,30 @@ func genQueryDecoderFunc(gctx *genContext, paramType types.Type) (name string, e
 			w.L("out.%s = new(%s)", field.Name(), fieldType)
 			strctRef = "*" + strctRef
 		}
-		w.Import("fmt")
 		switch fieldType.String() {
 		case "time.Duration":
 			gctx.Import("time")
+			w.Import("fmt")
 			w.L("if %s.%s, err = time.ParseDuration(q[len(q)-1]); err != nil {", strctRef, field.Name())
 			w.L(`  return fmt.Errorf("failed to decode query parameter \"%s\" as %s: %%w", err)`, fieldName, fieldType)
 			w.L("}")
 		case "bool":
 			gctx.Import("strconv")
+			w.Import("fmt")
 			w.L("if %s.%s, err = strconv.ParseBool(q[len(q)-1]); err != nil {", strctRef, field.Name())
 			w.L(`  return fmt.Errorf("failed to decode query parameter \"%s\" as %s: %%w", err)`, fieldName, fieldType)
 			w.L("}")
 		case "int":
 			gctx.Import("strconv")
+			w.Import("fmt")
 			w.L("if %s.%s, err = strconv.Atoi(q[len(q)-1]); err != nil {", strctRef, field.Name())
 			w.L(`  return fmt.Errorf("failed to decode query parameter \"%s\" as %s: %%w", err)`, fieldName, fieldType)
 			w.L("}")
 		case "string":
 			w.L("%s.%s = q[len(q)-1]", strctRef, field.Name())
 		default:
 			if implements(field, textUnmarshalerInterface()) {
+				w.Import("fmt")
 				w.L("if err = %s.%s.UnmarshalText([]byte(q[len(q)-1])); err != nil {", strctRef, field.Name())
 				w.L(`  return fmt.Errorf("failed to decode query parameter \"%s\" as %s: %%w", err)`, fieldName, fieldType)
 				w.L("}")