Function AutoFixGoCode

pkg/runtime/quality.go:494–534

(code string)

}

func AutoFixGoCode(code string) string {
	lines := strings.Split(code, "\n")
	var fixed []string

	for i := 0; i < len(lines); i++ {
		trimmed := strings.TrimSpace(lines[i])

		if i > 0 && trimmed == "}" {
			prevTrimmed := strings.TrimSpace(lines[i-1])
			if strings.Contains(prevTrimmed, ", err :=") || strings.Contains(prevTrimmed, ", err =") {
				// Detect indentation from the assignment line
				indent := getIndent(lines[i-1])
				fixed = append(fixed, indent+"if err != nil {")
				fixed = append(fixed, indent+indentUnit(indent)+"return nil, err")
				fixed = append(fixed, indent+"}")
				continue
			}
		}

		if strings.Contains(trimmed, "return") && strings.Contains(trimmed, "err") &&
			!strings.HasPrefix(trimmed, "if ") && !strings.HasPrefix(trimmed, "//") {
			if i > 0 {
				prevTrimmed := strings.TrimSpace(lines[i-1])
				if prevTrimmed == "}" && i >= 2 {
					prevPrev := strings.TrimSpace(lines[i-2])
					if strings.Contains(prevPrev, "io.EOF") || strings.Contains(prevPrev, "break") {
						indent := getIndent(lines[i])
						fixed = append(fixed, indent+"if err != nil {")
						fixed = append(fixed, indent+indentUnit(indent)+trimmed)
						fixed = append(fixed, indent+"}")
						continue
					}
				}
			}
		}

		fixed = append(fixed, lines[i])
	}

	return strings.Join(fixed, "\n")
}
func getIndent(line string) string {
	for i, c := range line {
		if c != ' ' && c != '\t' {