package photo

import (
	"context"
	"database/sql"
	"errors"
	"mime/multipart"
	"time"

	"photography-backend/internal/model"
	"photography-backend/internal/svc"
	"photography-backend/internal/types"
	fileUtil "photography-backend/pkg/utils/file"

	"github.com/zeromicro/go-zero/core/logx"
)

type UploadPhotoLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// 上传照片
func NewUploadPhotoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UploadPhotoLogic {
	return &UploadPhotoLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UploadPhotoLogic) UploadPhoto(req *types.UploadPhotoRequest, file multipart.File, header *multipart.FileHeader) (resp *types.UploadPhotoResponse, err error) {
	// 1. 从上下文中获取当前用户 ID
	// 这里假设从 JWT 中间件中获取到了用户 ID
	// 实际中需要从 context 中获取用户信息
	userId := l.ctx.Value("userId")
	if userId == nil {
		// 临时解决方案：如果没有用户ID，使用默认值1
		// 后续需要实现JWT中间件
		userId = int64(1)
	}
	
	// 2. 验证分类是否存在
	_, err = l.svcCtx.CategoryModel.FindOne(l.ctx, req.CategoryId)
	if err != nil {
		return nil, errors.New("分类不存在")
	}
	
	// 3. 处理文件上传
	fileConfig := fileUtil.Config{
		MaxSize:      l.svcCtx.Config.FileUpload.MaxSize,
		UploadDir:    l.svcCtx.Config.FileUpload.UploadDir,
		AllowedTypes: l.svcCtx.Config.FileUpload.AllowedTypes,
	}
	
	uploadResult, err := fileUtil.UploadPhoto(file, header, fileConfig)
	if err != nil {
		return nil, err
	}
	
	// 4. 创建照片记录
	photo := &model.Photo{
		Title:         req.Title,
		Description:   sql.NullString{String: req.Description, Valid: req.Description != ""},
		FilePath:      uploadResult.Original.FilePath,
		ThumbnailPath: uploadResult.Thumbnail.FilePath,
		UserId:        userId.(int64),
		CategoryId:    req.CategoryId,
		CreatedAt:     time.Now(),
		UpdatedAt:     time.Now(),
	}
	
	_, err = l.svcCtx.PhotoModel.Insert(l.ctx, photo)
	if err != nil {
		// 如果数据库保存失败，删除已上传的文件
		fileUtil.DeleteFile(uploadResult.Original.FilePath)
		fileUtil.DeleteFile(uploadResult.Thumbnail.FilePath)
		return nil, err
	}
	
	// 5. 返回上传结果
	return &types.UploadPhotoResponse{
		BaseResponse: types.BaseResponse{
			Code:    200,
			Message: "上传成功",
		},
		Data: types.Photo{
			Id:            photo.Id,
			Title:         photo.Title,
			Description:   photo.Description.String,
			FilePath:      uploadResult.Original.URL,
			ThumbnailPath: uploadResult.Thumbnail.URL,
			UserId:        photo.UserId,
			CategoryId:    photo.CategoryId,
			CreatedAt:     photo.CreatedAt.Unix(),
			UpdatedAt:     photo.UpdatedAt.Unix(),
		},
	}, nil
}