Golang 基于正则表达式的文件内容及名称查找替换

在 Golang 中，我们可以使用正则表达式来高效地处理文件内容和文件名的查找替换操作。下面我将介绍如何使用标准库和第三方库 github.com/bmatcuk/doublestar 来实现这些功能。

1. 文件内容查找替换

使用标准库实现

package main

import (
	"bufio"
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"regexp"
)

// ReplaceInFile 在单个文件中进行正则替换
func ReplaceInFile(filePath string, re *regexp.Regexp, replacement string) error {
	// 读取文件内容
	content, err := ioutil.ReadFile(filePath)
	if err != nil {
		return err
	}

	// 执行正则替换
	newContent := re.ReplaceAllString(string(content), replacement)

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

// ReplaceInFiles 在多个文件中进行正则替换
func ReplaceInFiles(rootDir, pattern string, re *regexp.Regexp, replacement string) error {
	return filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		// 跳过目录
		if info.IsDir() {
			return nil
		}

		// 匹配文件名模式
		matched, err := filepath.Match(pattern, info.Name())
		if err != nil {
			return err
		}

		if matched {
			return ReplaceInFile(path, re, replacement)
		}

		return nil
	})
}

func main() {
	// 示例：将所有.go文件中的"oldText"替换为"newText"
	re := regexp.MustCompile(`oldText`)
	err := ReplaceInFiles(".", "*.go", re, "newText")
	if err != nil {
		fmt.Println("Error:", err)
	}
}

2. 文件名查找替换

使用标准库实现

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"regexp"
)

// RenameFiles 使用正则表达式重命名文件
func RenameFiles(rootDir string, re *regexp.Regexp, replacement string) error {
	return filepath.Walk(rootDir, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}

		// 跳过目录
		if info.IsDir() {
			return nil
		}

		// 获取目录和文件名
		dir := filepath.Dir(path)
		filename := info.Name()

		// 执行正则替换
		newName := re.ReplaceAllString(filename, replacement)

		// 如果文件名有变化，则重命名
		if newName != filename {
			newPath := filepath.Join(dir, newName)
			fmt.Printf("Renaming %s to %s\n", path, newPath)
			return os.Rename(path, newPath)
		}

		return nil
	})
}

func main() {
	// 示例：将所有.txt文件中的"old"替换为"new"
	re := regexp.MustCompile(`old`)
	err := RenameFiles(".", re, "new")
	if err != nil {
		fmt.Println("Error:", err)
	}
}

3. 使用 Jet 插件

Jet 是一个强大的模板引擎，也可以用于文件处理。以下是使用 Jet 结合正则表达式处理文件的示例：

package main

import (
	"fmt"
	"github.com/CloudyKit/jet"
	"regexp"
)

func main() {
	// 创建 Jet 模板集
	set := jet.NewSet(jet.NewOSFileSystemLoader("."))

	// 注册正则替换函数
	set.AddGlobalFunc("regexReplace", func(args jet.Arguments) reflect.Value {
		str := args.Get(0).String()
		pattern := args.Get(1).String()
		replacement := args.Get(2).String()
		
		re := regexp.MustCompile(pattern)
		return reflect.ValueOf(re.ReplaceAllString(str, replacement))
	})

	// 解析模板
	tpl, err := set.GetTemplate("example.jet")
	if err != nil {
		panic(err)
	}

	// 执行模板
	err = tpl.Execute(os.Stdout, nil, map[string]interface{}{
		"content": "This is oldText in the content",
	})
	if err != nil {
		panic(err)
	}
}

对应的 example.jet 文件内容：

{{ regexReplace(content, "oldText", "newText") }}

4. 使用 doublestar 进行更强大的文件匹配

对于更复杂的文件匹配模式，可以使用 doublestar 库：

package main

import (
	"fmt"
	"github.com/bmatcuk/doublestar"
	"os"
	"path/filepath"
	"regexp"
)

func main() {
	// 使用双星号匹配模式
	matches, err := doublestar.Glob("./**/*.{go,txt}")
	if err != nil {
		panic(err)
	}

	re := regexp.MustCompile(`oldPattern`)

	for _, match := range matches {
		// 处理文件内容替换
		content, err := os.ReadFile(match)
		if err != nil {
			fmt.Printf("Error reading %s: %v\n", match, err)
			continue
		}

		newContent := re.ReplaceAllString(string(content), "newPattern")
		err = os.WriteFile(match, []byte(newContent), 0644)
		if err != nil {
			fmt.Printf("Error writing %s: %v\n", match, err)
		}

		// 处理文件名替换
		dir, file := filepath.Split(match)
		newFile := re.ReplaceAllString(file, "newPattern")
		if newFile != file {
			newPath := filepath.Join(dir, newFile)
			err = os.Rename(match, newPath)
			if err != nil {
				fmt.Printf("Error renaming %s to %s: %v\n", match, newPath, err)
			}
		}
	}
}

总结

1. 对于简单的文件内容替换，可以使用标准库的 regexp 和 filepath 包
2. 对于复杂的文件匹配模式，可以使用 doublestar 库
3. Jet 模板引擎适合需要复杂模板处理的场景
4. 在生产环境中，建议添加更多的错误处理和日志记录

这些方法可以灵活组合使用，根据实际需求选择最适合的方案。