Golang在特定位置插入XML元素的实现方法

在Go中处理XML插入，可以使用标准库的encoding/xml包。以下是完整的实现示例：

package main

import (
    "encoding/xml"
    "fmt"
    "io/ioutil"
    "os"
    "strings"
)

type PropertyGroup struct {
    XMLName          xml.Name `xml:"PropertyGroup"`
    Condition        string   `xml:"Condition,attr,omitempty"`
    Label            string   `xml:"Label,attr,omitempty"`
    ConfigurationType string   `xml:"ConfigurationType,omitempty"`
    CharacterSet     string   `xml:"CharacterSet,omitempty"`
    PlatformToolset  string   `xml:"PlatformToolset,omitempty"`
    EnableASAN       string   `xml:"EnableASAN,omitempty"`
}

type Project struct {
    XMLName        xml.Name        `xml:"Project"`
    PropertyGroups []PropertyGroup `xml:"PropertyGroup"`
}

func insertASANElement(filePath string) error {
    // 读取文件内容
    content, err := ioutil.ReadFile(filePath)
    if err != nil {
        return err
    }

    // 解析XML
    var project Project
    err = xml.Unmarshal(content, &project)
    if err != nil {
        return err
    }

    // 查找目标PropertyGroup并插入EnableASAN
    targetCondition := "'$(Configuration)|$(Platform)'=='Debug|x64'"
    inserted := false
    
    for i := range project.PropertyGroups {
        if project.PropertyGroups[i].Condition == targetCondition && 
           project.PropertyGroups[i].Label == "Configuration" {
            project.PropertyGroups[i].EnableASAN = "true"
            inserted = true
            break
        }
    }

    if !inserted {
        return fmt.Errorf("target PropertyGroup not found")
    }

    // 重新编码为XML
    output, err := xml.MarshalIndent(project, "", "  ")
    if err != nil {
        return err
    }

    // 添加XML头部
    result := xml.Header + string(output)

    // 写回文件
    return ioutil.WriteFile(filePath, []byte(result), 0644)
}

// 如果需要在特定位置精确插入（保持原有结构）
func insertASANWithExactPosition(filePath string) error {
    content, err := ioutil.ReadFile(filePath)
    if err != nil {
        return err
    }

    // 将文件内容转换为字符串便于处理
    xmlContent := string(content)
    
    // 查找目标PropertyGroup的结束位置
    targetStart := `<PropertyGroup Condition="'$(Configuration)|$(Platform)'=='Debug|x64'" Label="Configuration">`
    targetEnd := `</PropertyGroup>`
    
    startIdx := strings.Index(xmlContent, targetStart)
    if startIdx == -1 {
        return fmt.Errorf("target PropertyGroup not found")
    }
    
    // 找到该PropertyGroup的结束标签
    groupStart := startIdx + len(targetStart)
    endIdx := strings.Index(xmlContent[groupStart:], targetEnd)
    if endIdx == -1 {
        return fmt.Errorf("PropertyGroup end tag not found")
    }
    
    // 在结束标签前插入EnableASAN元素
    insertPosition := groupStart + endIdx
    indent := "    " // 4空格缩进
    newElement := "\n" + indent + "<EnableASAN>true</EnableASAN>"
    
    // 构建新内容
    newContent := xmlContent[:insertPosition] + newElement + xmlContent[insertPosition:]
    
    return ioutil.WriteFile(filePath, []byte(newContent), 0644)
}

func main() {
    // 使用方法1：使用结构体解析和修改
    err := insertASANElement("project.vcxproj")
    if err != nil {
        fmt.Printf("Error: %v\n", err)
    }

    // 使用方法2：精确位置插入（保持原有格式）
    err = insertASANWithExactPosition("project.vcxproj")
    if err != nil {
        fmt.Printf("Error: %v\n", err)
    }
}

如果需要在现有元素之间插入，可以使用更精确的字符串操作方法：

func insertBetweenElements(filePath string) error {
    content, err := ioutil.ReadFile(filePath)
    if err != nil {
        return err
    }

    xmlContent := string(content)
    
    // 查找PlatformToolset元素的结束位置
    targetElement := "<PlatformToolset>v143</PlatformToolset>"
    idx := strings.Index(xmlContent, targetElement)
    if idx == -1 {
        return fmt.Errorf("PlatformToolset element not found")
    }
    
    // 在PlatformToolset后插入EnableASAN
    insertPos := idx + len(targetElement)
    newElement := "\n    <EnableASAN>true</EnableASAN>"
    
    newContent := xmlContent[:insertPos] + newElement + xmlContent[insertPos:]
    
    return ioutil.WriteFile(filePath, []byte(newContent), 0644)
}

对于批量处理多个vcxproj文件：

func processMultipleFiles(filePattern string) error {
    files, err := filepath.Glob(filePattern)
    if err != nil {
        return err
    }

    for _, file := range files {
        fmt.Printf("Processing: %s\n", file)
        err := insertASANElement(file)
        if err != nil {
            fmt.Printf("Failed to process %s: %v\n", file, err)
        }
    }
    return nil
}

这些方法提供了不同的插入策略：第一种通过结构体解析更规范，第二种通过字符串操作更精确控制插入位置。根据你的具体需求选择合适的方法。