package usage

import (
	"context"
	"strconv"

	"apps/backend/internal/middleware"
	"apps/backend/internal/module/permission"
	"apps/backend/internal/response"
	"apps/backend/internal/svc"
	"apps/backend/internal/types"

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

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

func NewSetUsagePrefsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SetUsagePrefsLogic {
	return &SetUsagePrefsLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
}

func (l *SetUsagePrefsLogic) SetUsagePrefs(req *types.UsageSetPrefsReq) (*types.UsagePrefsData, error) {
	if l.svcCtx.Usage == nil {
		return nil, response.Biz(503, 503001, "usage not configured")
	}
	actor, ok := middleware.UIDFrom(l.ctx)
	if !ok {
		return nil, response.Biz(401, 401001, "missing authorization")
	}
	if !permission.IsAdmin(middleware.RolesFrom(l.ctx)) {
		return nil, response.Biz(403, 403002, "permission denied")
	}
	target, err := strconv.ParseInt(req.Uid, 10, 64)
	if err != nil {
		return nil, response.Biz(400, 400001, "invalid uid")
	}
	p, err := l.svcCtx.Usage.SetPrefs(l.ctx, actor, target, true, req.PlanId, req.Unlimited)
	if err != nil {
		return nil, err
	}
	return &types.UsagePrefsData{PlanId: p.PlanID, Unlimited: p.Unlimited}, nil
}
