Function FieldLoadCode

codegen/cli/cli.go:352–419

FieldLoadCode returns the code used in the build payload function that initializes one of the payload object fields. It returns the initialization code and a boolean indicating whether the code requires an "err" variable.

(f *FlagData, argName, argTypeName, validate string, defaultValue any, payload expr.DataType, payloadRef string)

// initializes one of the payload object fields. It returns the initialization
// code and a boolean indicating whether the code requires an "err" variable.
func FieldLoadCode(f *FlagData, argName, argTypeName, validate string, defaultValue any, payload expr.DataType, payloadRef string) (string, bool) {
	var (
		code    string
		declErr bool
		startIf string
		endIf   string
	)
	if !f.Required {
		startIf = fmt.Sprintf("if %s != \"\" {\n", f.FullName)
		endIf = "\n}"
	}
	if argTypeName == codegen.GoNativeTypeName(expr.String) {
		ref := "&"
		if f.Required || defaultValue != nil {
			ref = ""
		}
		code = argName + " = " + ref + f.FullName
		declErr = validate != ""
	} else {
		var checkErr bool
		code, declErr, checkErr = conversionCode(f.FullName, argName, argTypeName, !f.Required && defaultValue == nil)
		if checkErr {
			code += "\nif err != nil {\n"
			nilVal := "nil"
			if expr.IsPrimitive(payload) {
				code += fmt.Sprintf("var zero %s\n", payloadRef)
				nilVal = "zero"
			}
			if flagType(argTypeName) == "JSON" {
				code += fmt.Sprintf(`return %s, fmt.Errorf("invalid JSON for %s, \nerror: %%s, \nexample of valid JSON:\n%%s", err, %q)`,
					nilVal, argName, f.Example)
			} else {
				code += fmt.Sprintf(`return %s, fmt.Errorf("invalid value for %s, must be %s")`,
					nilVal, argName, f.Type)
			}
			code += "\n}"
		}
	}
	if validate != "" {
		nilCheck := "if " + argName + " != nil {"
		if strings.HasPrefix(validate, nilCheck) {
			// hackety hack... the validation code is generated for the client and needs to
			// account for the fact that the field could be nil in this case. We are reusing
			// that code to validate a CLI flag which can never be nil.  Lint tools complain
			// about that so remove the if statements. Ideally we'd have a better way to do
			// this but that requires a lot of changes and the added complexity might not be
			// worth it.
			var lines []string
			ls := strings.Split(validate, "\n")
			for i := 1; i < len(ls)-1; i++ {
				if ls[i+1] == nilCheck {
					i++ // skip both closing brace on previous line and check
					continue
				}
				lines = append(lines, ls[i])
			}
			validate = strings.Join(lines, "\n")
		}