批量替换模版文件代码分享

package main

import (
"bufio"
"bytes"
"flag"
"fmt"
"io"
"io/ioutil"
"os"
)

func main() {
config := flag.String("f", "replations.ini", "-f replations.ini")
dir := flag.String("d", "config", "-d configdir")
flag.Parse()
M, err := ParseConfig(*config)
if err != nil {
fmt.Println(err)
return
}
err = AutoReplace(*dir, M)
if err != nil {
fmt.Println(err)
return
}
}

func ParseConfig(configpath string) (map[string]string, error) {
file, err := os.Open(configpath)
if err != nil {
return nil, err
}
defer file.Close()
M := make(map[string]string)
buf := bufio.NewReader(file)
var num int
for {
line, _, err := buf.ReadLine()
if err != nil {
if err.Error() == "EOF" {
break
}
return nil, err
}
list := bytes.Split(line, []byte("="))
if len(list) != 2 {
return nil, fmt.Errorf("第%d行 出现多次'='", num)
}
M[string(bytes.TrimSpace(list[0]))] = string(bytes.TrimSpace(list[1]))
num++
}
return M, nil
}

func AutoReplace(dirpath string, variablesMap map[string]string) error {
Files, err := getFiles(dirpath)
if err != nil {
return err
}
for fileName, body := range Files {
fmt.Println(string(body))
for k, v := range variablesMap {
body = bytes.Replace(body, []byte(k), []byte(fmt.Sprintf("{{.%s}}", v)), -1)
}
F, err := os.OpenFile(fileName, os.O_CREATE|os.O_RDWR|os.O_TRUNC, 0644)
if err != nil {
F.Close()
return err
}
F.Write(body)
F.Close()
}
return nil
}

func getFiles(path string) (map[string][]byte, error) {
if !bytes.HasSuffix([]byte(path), []byte("/")) {
path = path + "/"
}
files, err := ioutil.ReadDir(path)
if err != nil || len(files) <= 0 {
return nil, err
}
var fileInfo map[string][]byte = make(map[string][]byte)
for _, file := range files {
if file.IsDir() {
continue
}
F, err := os.Open(path + file.Name())
if err != nil {
return nil, err
}
buf := make([]byte, file.Size())
n, err := io.ReadFull(F, buf)
if err != nil {
return nil, err
}
if bytes.Contains(buf[:n], []byte{0}) {
continue
}
fileInfo[path+file.Name()] = buf[:n]
}
return fileInfo, nil
}