package core

import (
	"encoding/json"
	"os"
	"path/filepath"
	"runtime"
	"scagent/util"

	"go.uber.org/zap"
)

// 传入的参数
// scope: all 所有文件
// scope: dir 目录下的文件
type Args struct {
	FilePath, Scope string
}

// 返回的参数
type Reply struct {
	Redata string
}

// 返回的json数据
type FileRpcService struct{}

// 返回的文件json结构体
type FloopJson struct {
	Path string `json:"path"`
	Size int64  `json:"size"`
	Hash string `json:"hash"`
}

// 文件列表
type FlpList struct {
	Flist []FloopJson `json:"list"`
}

// 获取文件路径
func (f *FileRpcService) GetFilePath(args *Args, replay *string) error {

	// 启用日志
	logger := util.NewProductionLogger()
	defer logger.Sync()
	// 并发处理
	go scanFileUnderPath(args.FilePath, args.Scope, logger, replay)
	runtime.Gosched()
	//
	return nil
}

// 扫描指定目录下的文件，可以全部，也可以局部
func scanFileUnderPath(rootDir string, scope string, logger *zap.Logger, replay *string) {
	var flplist FlpList // 文件列表

	// 遍历目录下的文件
	var allFiles []string // 存储所有文件路径

	// 递归遍历目录及其子目录
	err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
		// 处理遍历过程中的错误（如权限不足）
		if err != nil {
			logger.Error("访问路径出错", zap.Error(err))
			// 判断扫描范围
			if scope == "all" {
				return filepath.SkipDir
			} else {
				return nil
			}
		}

		// 只收集文件（排除目录）
		if !d.IsDir() {
			allFiles = append(allFiles, path)
		}
		return nil
	})
	// 打印错误日志
	if err != nil {
		logger.Error("遍历目录出错", zap.Error(err))
	}

	//
	// 打印所有文件路径
	for _, file := range allFiles {
		// 打印所有文件路径
		// fmt.Printf("%d. %s\n", i+1, file)
		d, _ := os.Stat(file)
		// 加入到列表
		hash := util.CalacHash(file)
		flplist.Flist = append(flplist.Flist, FloopJson{Path: file, Size: d.Size(), Hash: hash})
	}

	// 将文件列表转换为json字符串
	jsonStr, err := json.Marshal(flplist)
	if err != nil {
		logger.Error("转换为json字符串出错", zap.Error(err))
	}
	// 执行结果存到replay
	*replay = string(jsonStr)

}