package core

import (
	"encoding/json"
	"os"
	"path/filepath"
	"scagnet/util"

	"go.uber.org/zap"
)

// 传入的参数
type Args struct {
	FilePath 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 {
	rootDir := args.FilePath
	logger := util.NewProductionLogger()
	defer logger.Sync()
	// 遍历目录下的文件
	var allFiles []string // 存储所有文件路径
	var flplist FlpList   // 文件列表

	// 递归遍历目录及其子目录
	err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
		// 处理遍历过程中的错误（如权限不足）
		if err != nil {
			logger.Error("访问路径出错", zap.Error(err))
			return filepath.SkipDir // 跳过当前目录，继续遍历其他目录
		}

		// 只收集文件（排除目录）
		if !d.IsDir() {
			allFiles = append(allFiles, path)
		}
		return nil
	})

	if err != nil {
		logger.Error("遍历目录出错", zap.Error(err))
		return 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))
		return err
	}
	// fmt.Printf("jsonStr: %s\n", jsonStr)
	*replay = string(jsonStr)

	//
	return nil
}

func (f *FileRpcService) GetFileHash(args *Args, replay *Reply) error {
	// 计算文件hash
	hash := util.CalacHash(args.FilePath)
	replay.Redata = hash
	return nil
}