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

package file

import (
	"context"
	"fmt"

	"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 GetFileUrlLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// 获取文件访问URL
func NewGetFileUrlLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetFileUrlLogic {
	return &GetFileUrlLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *GetFileUrlLogic) GetFileUrl(req *types.GetFileRequest) (resp *types.FileUrlResponse, 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: non-admin can only access own files + public files
	if userRole != model.RoleAdmin && userRole != model.RoleSuperAdmin {
		if fileRecord.UserId != userId && !fileRecord.IsPublic {
			return nil, fmt.Errorf("permission denied")
		}
	}

	// Get URL from storage backend
	// For local storage, this returns "local://" + key
	// The handler will detect the "local://" prefix and serve the file directly
	// For OSS/MinIO, this returns a signed URL
	url, err := l.svcCtx.Storage.GetURL(l.ctx, fileRecord.Key)
	if err != nil {
		return nil, fmt.Errorf("failed to get file URL: %v", err)
	}

	resp = &types.FileUrlResponse{
		Url: url,
	}

	return resp, nil
}