// Code scaffolded by goctl. Safe to edit.
// goctl 1.9.2

package file

import (
	"context"
	"fmt"
	"strings"

	"github.com/youruser/base/internal/svc"
	"github.com/youruser/base/internal/types"
	"github.com/youruser/base/model"

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

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

// 更新文件信息
func NewUpdateFileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateFileLogic {
	return &UpdateFileLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UpdateFileLogic) UpdateFile(req *types.UpdateFileRequest) (resp *types.FileInfo, err error) {
	// Get userId and role from context
	userId, _ := l.ctx.Value("userId").(int64)
	if userId == 0 {
		if num, ok := l.ctx.Value("userId").(interface{ Int64() (int64, error) }); ok {
			userId, _ = num.Int64()
		}
	}
	userRole, _ := l.ctx.Value("role").(string)

	// Find file by ID
	fileRecord, err := model.FileFindOne(l.ctx, l.svcCtx.DB, req.Id)
	if err != nil {
		if err == model.ErrNotFound {
			return nil, fmt.Errorf("file not found")
		}
		return nil, fmt.Errorf("failed to query file: %v", err)
	}

	// Permission check: user can only edit own files, admin can edit all
	if userRole != model.RoleAdmin && userRole != model.RoleSuperAdmin {
		if fileRecord.UserId != userId {
			return nil, fmt.Errorf("permission denied")
		}
	}

	// Apply optional updates
	if req.Name != "" {
		fileRecord.Name = req.Name
	}
	if req.Category != "" {
		fileRecord.Category = req.Category
	}
	if req.IsPublic != nil {
		fileRecord.IsPublic = *req.IsPublic
	}

	// Save updates
	if err := model.FileUpdate(l.ctx, l.svcCtx.DB, fileRecord); err != nil {
		return nil, fmt.Errorf("failed to update file: %v", err)
	}

	// Build URL
	url := l.buildFileURL(fileRecord.Id, fileRecord.Key)

	resp = &types.FileInfo{
		Id:          fileRecord.Id,
		Name:        fileRecord.Name,
		Key:         fileRecord.Key,
		Size:        fileRecord.Size,
		MimeType:    fileRecord.MimeType,
		Category:    fileRecord.Category,
		IsPublic:    fileRecord.IsPublic,
		UserId:      fileRecord.UserId,
		StorageType: fileRecord.StorageType,
		Url:         url,
		CreatedAt:   fileRecord.CreatedAt.Format("2006-01-02 15:04:05"),
		UpdatedAt:   fileRecord.UpdatedAt.Format("2006-01-02 15:04:05"),
	}

	return resp, nil
}

// buildFileURL generates the appropriate URL for the file
func (l *UpdateFileLogic) buildFileURL(fileId int64, key string) string {
	rawURL, err := l.svcCtx.Storage.GetURL(l.ctx, key)
	if err != nil {
		l.Errorf("failed to get URL for file %d: %v", fileId, err)
		return ""
	}

	// For local storage, return API endpoint URL
	if strings.HasPrefix(rawURL, "local://") {
		return fmt.Sprintf("/api/v1/file/%d/url", fileId)
	}

	return rawURL
}