naza/pkg/filebatch/filebatch.go

// Copyright 2019, Chef.  All rights reserved.
// https://github.com/q191201771/naza
//
// Use of this source code is governed by a MIT-style license
// that can be found in the License file.
//
// Author: Chef (191201771@qq.com)

package filebatch

import (
	"bytes"
	"errors"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
)

// @param path 带路径的文件名
// @param info 文件的 os.FileInfo 信息
// @param content 文件内容
// @return 返回nil或者content原始内容，则不修改文件内容，返回其他内容，则会覆盖重写文件
type WalkFunc func(path string, info os.FileInfo, content []byte, err error) []byte

// 遍历访问指定文件夹下的文件
// @param root 需要遍历访问的文件夹
// @param recursive 是否递归访问子文件夹
// @param suffix 指定文件名后缀进行过滤，如果为""，则不过滤
func Walk(root string, recursive bool, suffix string, walkFn WalkFunc) error {
	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			walkFn(path, info, nil, err)
			return nil
		}
		if !recursive && info.IsDir() && path != root {
			return filepath.SkipDir
		}
		if info.IsDir() {
			return nil
		}
		if suffix != "" && !strings.HasSuffix(info.Name(), suffix) {
			return nil
		}

		content, err := ioutil.ReadFile(path)
		if err != nil {
			walkFn(path, info, content, err)
			return nil
		}
		newContent := walkFn(path, info, content, nil)
		if newContent != nil && bytes.Compare(content, newContent) != 0 {
			if err = ioutil.WriteFile(path, newContent, 0755); err != nil {
				return err
			}
		}
		return nil
	})
	return err
}

// 文件尾部添加内容
func AddTailContent(content []byte, tail []byte) []byte {
	if !bytes.HasSuffix(content, []byte{'\n'}) {
		content = append(content, '\n')
	}
	return append(content, tail...)
}

// 文件头部添加内容
func AddHeadContent(content []byte, head []byte) []byte {
	if !bytes.HasSuffix(head, []byte{'\n'}) {
		head = append(head, '\n')
	}
	return append(head, content...)
}

// 行号范围
// 1表示首行，-1表示最后一行
type LineRange struct {
	From int
	To   int
}

var ErrLineRange = errors.New("naza.filebatch: line range error")

func calcLineRange(len int, lr LineRange) (LineRange, error) {
	// 换算成从0开始的下标
	if lr.From < 0 {
		lr.From = len + lr.From
	} else if lr.From > 0 {
		lr.From = lr.From - 1
	} else {
		return lr, ErrLineRange
	}
	if lr.To < 0 {
		lr.To = len + lr.To
	} else if lr.To > 0 {
		lr.To = lr.To - 1
	} else {
		return lr, ErrLineRange
	}

	// 排序交换
	if lr.From > lr.To {
		lr.From, lr.To = lr.To, lr.From
	}

	if lr.From < 0 || lr.From >= len || lr.To < 0 || lr.To >= len {
		return lr, ErrLineRange
	}

	return lr, nil
}

func DeleteLines(content []byte, lr LineRange) ([]byte, error) {
	lines := bytes.Split(content, []byte{'\n'})
	length := len(lines)
	nlr, err := calcLineRange(length, lr)
	if err != nil {
		return content, err
	}
	var nlines [][]byte
	if nlr.From > 0 {
		nlines = append(nlines, lines[:nlr.From]...)
	}
	if nlr.To < length-1 {
		nlines = append(nlines, lines[nlr.To+1:]...)
	}
	return bytes.Join(nlines, []byte{'\n'}), nil
}