package osT

import (
	"io"
	"os"
	"path/filepath"
	"strings"
	
	"gitee.com/Jiudan0905/go-tools/baseT/logT"
	"gitee.com/Jiudan0905/go-tools/baseT/osT/fileT"
)

//  获取目录下所有文件
func GetDirAllFile(root string, getDir bool, justLevel1 bool) []string {
	var rst []string
	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			logT.Error(err.Error())
			return err
		}
		// 如果只采取一层目录
		if justLevel1 && strings.ContainsAny(path, `\/`) {
			return nil
		}
		// 如果是目录的情况下并且不允许采集目录则直接返回
		if !getDir && info.IsDir() {
			return nil
		}
		// 其他情况则采集
		rst = append(rst, path)
		return nil
	})
	if err != nil {
		logT.Error(err.Error())
	}
	return rst
}

// 获取一级目录下的文件列表
func GetFileListLeval1(root string, getDir ...bool) []string {
	if len(getDir) > 0 && getDir[0] {
		return GetDirAllFile(root, true, true)
	}
	return GetDirAllFile(root, false, true)
}

// 文件复制
func CopyFile(src string, dst ...string) {
	srcFile, err := os.Open(src)
	if err != nil {
		logT.Error(err.Error())
		return
	}
	defer srcFile.Close()
	copyFile := fileT.GetFileDir(src) + "/" + fileT.GetFileName(src) + ".copy" + fileT.GetFileSuffix(src)
	if len(dst) > 0 && fileT.FileIsExists(dst[0]) {
		copyFile = dst[0]
	}
	dstFile, err := os.Create(copyFile)
	if err != nil {
		logT.Error(err.Error())
		return
	}
	defer dstFile.Close()
	_, err = io.Copy(dstFile, srcFile)
	if err != nil {
		logT.Error(err.Error())
		return
	}
	return
}

// 文件重命名
func RenameFile(old, new string) {
	// 替换为实际的旧文件名和新文件名
	err := os.Rename(old, new)
	if err != nil {
		logT.Error(err.Error())
	}
}