package utils

import (
	"bufio"
	"crypto/md5"
	"crypto/sha256"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
	"time"
)

// FileInfo 文件信息结构体
type FileInfo struct {
	Name    string    `json:"name"`
	Path    string    `json:"path"`
	Size    int64     `json:"size"`
	ModTime time.Time `json:"mod_time"`
	IsDir   bool      `json:"is_dir"`
	Ext     string    `json:"ext"`
}

// 1. 检查文件或目录是否存在
func Exists(path string) bool {
	_, err := os.Stat(path)
	return !os.IsNotExist(err)
}

// 2. 检查是否为目录
func IsDir(path string) bool {
	info, err := os.Stat(path)
	if err != nil {
		return false
	}
	return info.IsDir()
}

// 3. 检查是否为文件
func IsFile(path string) bool {
	info, err := os.Stat(path)
	if err != nil {
		return false
	}
	return !info.IsDir()
}

// 4. 获取指定目录下的所有文件夹
func GetDirectories(dirPath string) ([]string, error) {
	var dirs []string

	entries, err := os.ReadDir(dirPath)
	if err != nil {
		return nil, fmt.Errorf("读取目录失败: %w", err)
	}

	for _, entry := range entries {
		if entry.IsDir() {
			dirs = append(dirs, entry.Name())
		}
	}

	return dirs, nil
}

// 5. 获取指定目录下指定格式的文件
func GetFilesByExtension(dirPath string, extensions ...string) ([]string, error) {
	var files []string

	// 标准化扩展名（确保以.开头）
	for i, ext := range extensions {
		if !strings.HasPrefix(ext, ".") {
			extensions[i] = "." + ext
		}
		extensions[i] = strings.ToLower(extensions[i])
	}

	entries, err := os.ReadDir(dirPath)
	if err != nil {
		return nil, fmt.Errorf("读取目录失败: %w", err)
	}

	for _, entry := range entries {
		if !entry.IsDir() {
			fileExt := strings.ToLower(filepath.Ext(entry.Name()))
			for _, ext := range extensions {
				if fileExt == ext {
					files = append(files, entry.Name())
					break
				}
			}
		}
	}

	return files, nil
}

// 6. 找到指定目录下第一个符合格式的文件
func GetFirstFileByExtension(dirPath string, extensions ...string) (string, error) {
	files, err := GetFilesByExtension(dirPath, extensions...)
	if err != nil {
		return "", err
	}

	if len(files) == 0 {
		return "", fmt.Errorf("未找到符合扩展名的文件")
	}

	return files[0], nil
}

// 7. 递归获取目录下所有文件
func GetAllFiles(dirPath string, recursive bool) ([]FileInfo, error) {
	var files []FileInfo

	if recursive {
		err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}

			if !info.IsDir() {
				files = append(files, FileInfo{
					Name:    info.Name(),
					Path:    path,
					Size:    info.Size(),
					ModTime: info.ModTime(),
					IsDir:   false,
					Ext:     filepath.Ext(info.Name()),
				})
			}
			return nil
		})
		return files, err
	} else {
		entries, err := os.ReadDir(dirPath)
		if err != nil {
			return nil, fmt.Errorf("读取目录失败: %w", err)
		}

		for _, entry := range entries {
			if !entry.IsDir() {
				info, err := entry.Info()
				if err != nil {
					continue
				}

				files = append(files, FileInfo{
					Name:    info.Name(),
					Path:    filepath.Join(dirPath, info.Name()),
					Size:    info.Size(),
					ModTime: info.ModTime(),
					IsDir:   false,
					Ext:     filepath.Ext(info.Name()),
				})
			}
		}
		return files, nil
	}
}

// 递归获取目录下所有文件夹
func GetAllDirs(dirPath string, recursive bool) ([]FileInfo, error) {
	var dirs []FileInfo

	if recursive {
		err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}

			// 只收集目录，排除根目录本身
			if info.IsDir() && path != dirPath {
				dirs = append(dirs, FileInfo{
					Name:    info.Name(),
					Path:    path,
					Size:    0, // 目录大小通常为0或不适用
					ModTime: info.ModTime(),
					IsDir:   true,
					Ext:     "", // 目录没有扩展名
				})
			}
			return nil
		})
		return dirs, err
	} else {
		entries, err := os.ReadDir(dirPath)
		if err != nil {
			return nil, fmt.Errorf("读取目录失败: %w", err)
		}

		for _, entry := range entries {
			if entry.IsDir() {
				info, err := entry.Info()
				if err != nil {
					continue
				}

				dirs = append(dirs, FileInfo{
					Name:    info.Name(),
					Path:    filepath.Join(dirPath, info.Name()),
					Size:    0, // 目录大小通常为0或不适用
					ModTime: info.ModTime(),
					IsDir:   true,
					Ext:     "", // 目录没有扩展名
				})
			}
		}
		return dirs, nil
	}
}

// 8. 删除文件或目录
func Delete(path string) error {
	if !Exists(path) {
		return fmt.Errorf("路径不存在: %s", path)
	}

	return os.RemoveAll(path)
}

// 9. 创建目录（递归创建）
func CreateDir(dirPath string) error {
	return os.MkdirAll(dirPath, 0755)
}

// 10. 复制文件
func CopyFile(src, dst string) error {
	sourceFile, err := os.Open(src)
	if err != nil {
		return fmt.Errorf("打开源文件失败: %w", err)
	}
	defer sourceFile.Close()

	// 确保目标目录存在
	dstDir := filepath.Dir(dst)
	if err := CreateDir(dstDir); err != nil {
		return fmt.Errorf("创建目标目录失败: %w", err)
	}

	destFile, err := os.Create(dst)
	if err != nil {
		return fmt.Errorf("创建目标文件失败: %w", err)
	}
	defer destFile.Close()

	_, err = io.Copy(destFile, sourceFile)
	if err != nil {
		return fmt.Errorf("复制文件失败: %w", err)
	}

	return nil
}

// 11. 移动/重命名文件
func MoveFile(src, dst string) error {
	// 确保目标目录存在
	dstDir := filepath.Dir(dst)
	if err := CreateDir(dstDir); err != nil {
		return fmt.Errorf("创建目标目录失败: %w", err)
	}

	return os.Rename(src, dst)
}

// 12. 读取文件全部内容
func ReadFile(filePath string) (string, error) {
	content, err := os.ReadFile(filePath)
	if err != nil {
		return "", fmt.Errorf("读取文件失败: %w", err)
	}
	return string(content), nil
}

// 13. 按行读取文件
func ReadFileLines(filePath string) ([]string, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return nil, fmt.Errorf("打开文件失败: %w", err)
	}
	defer file.Close()

	var lines []string
	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		lines = append(lines, scanner.Text())
	}

	if err := scanner.Err(); err != nil {
		return nil, fmt.Errorf("读取文件失败: %w", err)
	}

	return lines, nil
}

// 14. 写入文件内容
func WriteFile(filePath, content string) error {
	// 确保目录存在
	dir := filepath.Dir(filePath)
	if err := CreateDir(dir); err != nil {
		return fmt.Errorf("创建目录失败: %w", err)
	}

	return os.WriteFile(filePath, []byte(content), 0644)
}

// 15. 追加写入文件
func AppendFile(filePath, content string) error {
	// 确保目录存在
	dir := filepath.Dir(filePath)
	if err := CreateDir(dir); err != nil {
		return fmt.Errorf("创建目录失败: %w", err)
	}

	file, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
	if err != nil {
		return fmt.Errorf("打开文件失败: %w", err)
	}
	defer file.Close()

	_, err = file.WriteString(content)
	return err
}

// 16. 获取文件大小
func GetFileSize(filePath string) (int64, error) {
	info, err := os.Stat(filePath)
	if err != nil {
		return 0, fmt.Errorf("获取文件信息失败: %w", err)
	}
	return info.Size(), nil
}

// 17. 获取文件修改时间
func GetModTime(filePath string) (time.Time, error) {
	info, err := os.Stat(filePath)
	if err != nil {
		return time.Time{}, fmt.Errorf("获取文件信息失败: %w", err)
	}
	return info.ModTime(), nil
}

// 18. 获取文件扩展名
func GetExtension(filePath string) string {
	return filepath.Ext(filePath)
}

// 19. 获取文件名（不含扩展名）
func GetFileName(filePath string) string {
	base := filepath.Base(filePath)
	ext := filepath.Ext(base)
	return strings.TrimSuffix(base, ext)
}

// 20. 获取文件基本名称（含扩展名）
func GetBaseName(filePath string) string {
	return filepath.Base(filePath)
}

// 21. 获取文件目录
func GetDir(filePath string) string {
	return filepath.Dir(filePath)
}

// 22. 计算文件MD5哈希值
func GetFileMD5(filePath string) (string, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return "", fmt.Errorf("打开文件失败: %w", err)
	}
	defer file.Close()

	hash := md5.New()
	if _, err := io.Copy(hash, file); err != nil {
		return "", fmt.Errorf("计算哈希失败: %w", err)
	}

	return fmt.Sprintf("%x", hash.Sum(nil)), nil
}

// 23. 计算文件SHA256哈希值
func GetFileSHA256(filePath string) (string, error) {
	file, err := os.Open(filePath)
	if err != nil {
		return "", fmt.Errorf("打开文件失败: %w", err)
	}
	defer file.Close()

	hash := sha256.New()
	if _, err := io.Copy(hash, file); err != nil {
		return "", fmt.Errorf("计算哈希失败: %w", err)
	}

	return fmt.Sprintf("%x", hash.Sum(nil)), nil
}

// 24. 清空目录内容（保留目录本身）
func CleanDir(dirPath string) error {
	entries, err := os.ReadDir(dirPath)
	if err != nil {
		return fmt.Errorf("读取目录失败: %w", err)
	}

	for _, entry := range entries {
		path := filepath.Join(dirPath, entry.Name())
		if err := os.RemoveAll(path); err != nil {
			return fmt.Errorf("删除 %s 失败: %w", path, err)
		}
	}

	return nil
}

// 25. 获取目录大小
func GetDirSize(dirPath string) (int64, error) {
	var size int64

	err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if !info.IsDir() {
			size += info.Size()
		}
		return nil
	})

	return size, err
}

// 26. 格式化文件大小
func FormatFileSize(bytes int64) string {
	const unit = 1024
	if bytes < unit {
		return fmt.Sprintf("%d B", bytes)
	}
	div, exp := int64(unit), 0
	for n := bytes / unit; n >= unit; n /= unit {
		div *= unit
		exp++
	}
	return fmt.Sprintf("%.1f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
}

// 27. 搜索文件（按文件名模式）
func SearchFiles(dirPath, pattern string, recursive bool) ([]string, error) {
	var matches []string

	if recursive {
		err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
			if err != nil {
				return err
			}

			if !info.IsDir() {
				matched, err := filepath.Match(pattern, info.Name())
				if err != nil {
					return err
				}
				if matched {
					matches = append(matches, path)
				}
			}
			return nil
		})
		return matches, err
	} else {
		entries, err := os.ReadDir(dirPath)
		if err != nil {
			return nil, fmt.Errorf("读取目录失败: %w", err)
		}

		for _, entry := range entries {
			if !entry.IsDir() {
				matched, err := filepath.Match(pattern, entry.Name())
				if err != nil {
					return nil, err
				}
				if matched {
					matches = append(matches, filepath.Join(dirPath, entry.Name()))
				}
			}
		}
		return matches, nil
	}
}

// 28. 获取当前工作目录
func GetCurrentDir() (string, error) {
	return os.Getwd()
}

// 29. 改变工作目录
func ChangeDir(dirPath string) error {
	return os.Chdir(dirPath)
}