格式化后的Golang修改AST出现奇怪缩进问题

在Go中修改AST后出现缩进问题通常是因为格式化时没有正确处理节点的位置信息。当直接替换AST节点而不更新其位置信息时，go/format包可能无法正确格式化代码。

在你的示例中，主要问题是替换后的节点缺少正确的位置信息。以下是修复后的代码：

package main

import (
	"fmt"
	"go/ast"
	"go/format"
	"go/parser"
	"go/token"
	"log"
	"strings"
)

func main() {
	src := `package main

type Foo struct {
	Bar string
}`

	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, "", src, parser.ParseComments)
	if err != nil {
		log.Fatal(err)
	}

	ast.Inspect(f, func(n ast.Node) bool {
		if ts, ok := n.(*ast.TypeSpec); ok && ts.Name.Name == "Foo" {
			if st, ok := ts.Type.(*ast.StructType); ok {
				for _, field := range st.Fields.List {
					if field.Names[0].Name == "Bar" {
						// 创建新的字段值并设置正确的位置
						newValue := &ast.BasicLit{
							Kind:  token.STRING,
							Value: `"new value"`,
						}
						
						// 复制原始值的位置信息
						if field.Tag != nil {
							newValue.ValuePos = field.Tag.ValuePos
						} else if field.Type != nil {
							// 如果没有tag，使用类型的位置
							newValue.ValuePos = field.Type.Pos()
						}
						
						// 将新值作为tag添加，而不是替换字段
						field.Tag = newValue
					}
				}
			}
		}
		return true
	})

	var buf strings.Builder
	if err := format.Node(&buf, fset, f); err != nil {
		log.Fatal(err)
	}

	fmt.Println(buf.String())
}

更健壮的解决方案是使用astutil包并确保正确设置位置信息：

package main

import (
	"fmt"
	"go/ast"
	"go/format"
	"go/parser"
	"go/token"
	"go/astutil"
	"log"
	"strings"
)

func main() {
	src := `package main

type Foo struct {
	Bar string
}`

	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, "", src, parser.ParseComments)
	if err != nil {
		log.Fatal(err)
	}

	// 使用astutil进行更安全的AST操作
	astutil.Apply(f, nil, func(c *astutil.Cursor) bool {
		if field, ok := c.Node().(*ast.Field); ok {
			if len(field.Names) > 0 && field.Names[0].Name == "Bar" {
				// 创建新的tag
				newTag := &ast.BasicLit{
					Kind:     token.STRING,
					Value:    `"new value"`,
					ValuePos: field.Type.Pos(), // 设置正确的位置
				}
				field.Tag = newTag
			}
		}
		return true
	})

	// 使用format.Node时指定fset
	var buf strings.Builder
	if err := format.Node(&buf, fset, f); err != nil {
		log.Fatal(err)
	}

	fmt.Println(buf.String())
}

如果你需要完全替换字段而不仅仅是添加tag，可以这样做：

package main

import (
	"fmt"
	"go/ast"
	"go/format"
	"go/parser"
	"go/token"
	"log"
	"strings"
)

func main() {
	src := `package main

type Foo struct {
	Bar string
	Baz int
}`

	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, "", src, parser.ParseComments)
	if err != nil {
		log.Fatal(err)
	}

	ast.Inspect(f, func(n ast.Node) bool {
		if ts, ok := n.(*ast.TypeSpec); ok && ts.Name.Name == "Foo" {
			if st, ok := ts.Type.(*ast.StructType); ok {
				for i, field := range st.Fields.List {
					if field.Names[0].Name == "Bar" {
						// 创建新字段
						newField := &ast.Field{
							Names: []*ast.Ident{ast.NewIdent("Bar")},
							Type:  &ast.Ident{Name: "string"},
							Tag:   &ast.BasicLit{Kind: token.STRING, Value: `"new value"`},
						}
						
						// 复制位置信息
						if field.Type != nil {
							newField.Type.SetPos(field.Type.Pos())
						}
						if field.Tag != nil {
							newField.Tag.SetPos(field.Tag.Pos())
						}
						
						// 替换字段
						st.Fields.List[i] = newField
					}
				}
			}
		}
		return true
	})

	var buf strings.Builder
	if err := format.Node(&buf, fset, f); err != nil {
		log.Fatal(err)
	}

	fmt.Println(buf.String())
}

关键点是确保新创建的AST节点具有正确的位置信息。如果位置信息不正确，格式化器就无法确定正确的缩进级别。