package models

import (
	"strings"
	"time"

	"github.com/beego/beego/v2/client/orm"
)

// FileInfo 文件信息模型
// 对应 yz_files 表结构
type FileInfo struct {
	ID       int64  `orm:"column(id);auto" json:"id"`
	TenantID string `orm:"column(tenant_id);size(64)" json:"tenant_id"`

	// 用户关联信息（通过JWT认证获取）
	UserID int `orm:"column(user_id);default(0)" json:"user_id"`

	// 文件基础信息
	FileName     string `orm:"column(file_name);size(255)" json:"file_name"`
	OriginalName string `orm:"column(original_name);size(255)" json:"original_name"`
	FilePath     string `orm:"column(file_path);size(500)" json:"file_path"`
	FileURL      string `orm:"column(file_url);size(500);null" json:"file_url"`
	FileSize     int64  `orm:"column(file_size);default(0)" json:"file_size"`
	FileType     string `orm:"column(file_type);size(50)" json:"file_type"`
	FileExt      string `orm:"column(file_ext);size(20)" json:"file_ext"`
	MD5          string `orm:"column(md5);size(32);null" json:"md5"`

	// 分类信息
	Category    string `orm:"column(category);size(100)" json:"category"`
	SubCategory string `orm:"column(sub_category);size(100);null" json:"sub_category"`

	// 状态信息
	Status   int8 `orm:"column(status);default(1)" json:"status"`
	IsPublic int8 `orm:"column(is_public);default(0)" json:"is_public"`

	// 上传信息
	UploadBy   string     `orm:"column(upload_by);size(100)" json:"upload_by"`
	UploadTime time.Time  `orm:"column(upload_time);type(datetime);auto_now_add" json:"upload_time"`
	DeleteTime *time.Time `orm:"column(delete_time);type(datetime);null" json:"delete_time,omitempty"`

	// 关联的用户信息（非数据库字段）
	User *User `orm:"-" json:"user,omitempty"`
}

// TableName 设置表名
func (f *FileInfo) TableName() string {
	return "yz_files"
}

func init() {
	// Removed duplicate ORM registration
}

// CanPreview 判断文件是否可以在线预览
func (f *FileInfo) CanPreview() bool {
	previewableExts := map[string]bool{
		// 图片格式
		".jpg":  true,
		".jpeg": true,
		".png":  true,
		".gif":  true,
		".bmp":  true,
		".webp": true,
		".svg":  true,
		// 文档格式（仅支持 .docx，不支持旧的 .doc、Excel、PPT）
		".pdf":  true,
		".txt":  true,
		".docx": true,
		// 视频格式
		".mp4":  true,
		".webm": true,
		// 音频格式
		".mp3": true,
		".wav": true,
	}
	return previewableExts[strings.ToLower(f.FileExt)]
}

// GetAllFiles 获取所有文件信息
func GetAllFiles() ([]*FileInfo, error) {
	o := orm.NewOrm()
	var files []*FileInfo
	_, err := o.QueryTable("yz_files").Filter("status", 1).OrderBy("-upload_time").All(&files)
	return files, err
}

// GetFilesByUserID 根据用户ID获取文件列表
func GetFilesByUserID(userID int) ([]*FileInfo, error) {
	o := orm.NewOrm()
	var files []*FileInfo
	_, err := o.QueryTable("yz_files").Filter("user_id", userID).Filter("status", 1).OrderBy("-upload_time").All(&files)
	return files, err
}

// GetFileById 根据ID获取文件信息
func GetFileById(id int64) (*FileInfo, error) {
	o := orm.NewOrm()
	file := &FileInfo{ID: id}
	err := o.Read(file, "ID")
	if err == orm.ErrNoRows {
		return nil, err
	}
	return file, nil
}

// GetFilesByTenant 根据租户ID获取文件信息
func GetFilesByTenant(tenantID string) ([]*FileInfo, error) {
	o := orm.NewOrm()
	var files []*FileInfo
	_, err := o.QueryTable("yz_files").Filter("tenant_id", tenantID).OrderBy("-upload_time").All(&files)
	return files, err
}

// GetFilesByCategory 根据分类获取文件信息
func GetFilesByCategory(category string) ([]*FileInfo, error) {
	o := orm.NewOrm()
	var files []*FileInfo
	_, err := o.QueryTable("yz_files").Filter("category", category).OrderBy("-upload_time").All(&files)
	return files, err
}

// GetFilesByStatus 根据状态获取文件信息
func GetFilesByStatus(status int8) ([]*FileInfo, error) {
	o := orm.NewOrm()
	var files []*FileInfo
	_, err := o.QueryTable("yz_files").Filter("status", status).OrderBy("-upload_time").All(&files)
	return files, err
}

// AddFile 添加新文件信息
func AddFile(file *FileInfo) (int64, error) {
	o := orm.NewOrm()
	id, err := o.Insert(file)
	return id, err
}

// UpdateFile 更新文件信息
func UpdateFile(file *FileInfo) error {
	o := orm.NewOrm()
	_, err := o.Update(file)
	return err
}

// DeleteFile 删除文件信息（软删除，设置状态为0并记录删除时间）
func DeleteFile(id int64) error {
	o := orm.NewOrm()
	file := &FileInfo{ID: id}
	if err := o.Read(file, "ID"); err != nil {
		return err
	}
	file.Status = 0
	now := time.Now()
	file.DeleteTime = &now
	_, err := o.Update(file, "Status", "DeleteTime")
	return err
}

// HardDeleteFile 硬删除文件信息
func HardDeleteFile(id int64) error {
	o := orm.NewOrm()
	_, err := o.Delete(&FileInfo{ID: id})
	return err
}

// GetFileStatistics 获取文件统计信息
func GetFileStatistics(tenantID string) (map[string]interface{}, error) {
	o := orm.NewOrm()

	// 总文件数
	totalCount, err := o.QueryTable("yz_files").Filter("tenant_id", tenantID).Filter("status", 1).Count()
	if err != nil {
		return nil, err
	}

	// 总文件大小
	var totalSize int64
	err = o.Raw("SELECT COALESCE(SUM(file_size), 0) FROM yz_files WHERE tenant_id = ? AND status = 1", tenantID).QueryRow(&totalSize)
	if err != nil {
		return nil, err
	}

	// 按分类统计
	var categoryStats []orm.Params
	_, err = o.Raw("SELECT category, COUNT(*) as count, COALESCE(SUM(file_size), 0) as size FROM yz_files WHERE tenant_id = ? AND status = 1 GROUP BY category", tenantID).Values(&categoryStats)
	if err != nil {
		return nil, err
	}

	return map[string]interface{}{
		"total_count":    totalCount,
		"total_size":     totalSize,
		"category_stats": categoryStats,
	}, nil
}

// SearchFiles 搜索文件（通过原始文件名搜索）
func SearchFiles(keyword string, tenantID string) ([]*FileInfo, error) {
	o := orm.NewOrm()
	var files []*FileInfo

	// 构建查询条件 - 只通过原始文件名搜索
	qs := o.QueryTable("yz_files").Filter("tenant_id", tenantID).Filter("status", 1)

	// 搜索原始文件名
	qs = qs.Filter("original_name__icontains", keyword)

	_, err := qs.OrderBy("-upload_time").All(&files)
	return files, err
}

// GetFileByMD5 根据MD5获取文件信息（获取唯一的文件记录）
func GetFileByMD5(md5 string) (*FileInfo, error) {
	o := orm.NewOrm()
	var file FileInfo
	err := o.QueryTable("yz_files").Filter("md5", md5).Filter("status", 1).OrderBy("upload_time").One(&file)
	if err == orm.ErrNoRows {
		return nil, nil
	}
	if err != nil {
		return nil, err
	}
	return &file, nil
}

// GetFileByMD5AndTenant 根据MD5和租户ID获取文件信息
func GetFileByMD5AndTenant(md5 string, tenantID string) (*FileInfo, error) {
	o := orm.NewOrm()
	var file FileInfo
	err := o.QueryTable("yz_files").Filter("md5", md5).Filter("tenant_id", tenantID).Filter("status", 1).OrderBy("upload_time").One(&file)
	if err == orm.ErrNoRows {
		return nil, nil
	}
	if err != nil {
		return nil, err
	}
	return &file, nil
}