75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package usecase
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"apps/backend/internal/module/radar/domain"
|
||
usageDomain "apps/backend/internal/module/usage/domain"
|
||
)
|
||
|
||
// GetCostPreview is deliberately read-only: it resolves the likely key mode
|
||
// and reads the current balance, but never calls PrepareCall or reserves a
|
||
// credit. The actual operation remains the only place that meters usage.
|
||
func (s *Service) GetCostPreview(ctx context.Context, ownerUID int64, action, watchID, productID string, candidateLimit int) (*domain.CostPreview, error) {
|
||
if ownerUID <= 0 {
|
||
return nil, fmt.Errorf("%w: owner_uid required", domain.ErrValidation)
|
||
}
|
||
action = strings.ToLower(strings.TrimSpace(action))
|
||
if action != "sweep" && action != "explore" && action != "demand_map_enrich" && action != "reply" {
|
||
return nil, fmt.Errorf("%w: unknown cost preview action %q", domain.ErrValidation, action)
|
||
}
|
||
if candidateLimit <= 0 {
|
||
candidateLimit = 20
|
||
}
|
||
if candidateLimit > 100 {
|
||
candidateLimit = 100
|
||
}
|
||
searchCalls := 0
|
||
if action == "sweep" || action == "explore" {
|
||
searchCalls = 1
|
||
if productID != "" {
|
||
if plan, err := s.BuildProductQueryPlan(ctx, ownerUID, productID); err == nil && len(plan.Groups) > 0 {
|
||
searchCalls = len(plan.Groups)
|
||
}
|
||
}
|
||
}
|
||
maxAI := 0
|
||
fixed := 0
|
||
min := searchCalls * usageDomain.MeterCost(usageDomain.MeterWebSearch)
|
||
max := min
|
||
switch action {
|
||
case "sweep", "explore":
|
||
maxAI = candidateLimit
|
||
max += maxAI * usageDomain.MeterCost(usageDomain.MeterAIResearch)
|
||
case "demand_map_enrich", "reply":
|
||
fixed = usageDomain.MeterCost(usageDomain.MeterAICopy)
|
||
min, max = fixed, fixed
|
||
}
|
||
mode := usageDomain.KeyModePlatform
|
||
meter := usageDomain.MeterWebSearch
|
||
if action == "demand_map_enrich" || action == "reply" {
|
||
meter = usageDomain.MeterAICopy
|
||
}
|
||
if s.ResolveKey != nil {
|
||
if resolved, _, err := s.ResolveKey(ctx, ownerUID, meter); err == nil && resolved != "" {
|
||
mode = resolved
|
||
}
|
||
}
|
||
remaining := 0
|
||
if s.Usage != nil {
|
||
if summary, err := s.Usage.GetSummary(ctx, ownerUID, ""); err == nil && summary != nil {
|
||
remaining = summary.Platform.CreditsRemaining
|
||
}
|
||
}
|
||
return &domain.CostPreview{
|
||
PreviewID: domain.NewID(), Action: action, KeyMode: mode,
|
||
FixedCredits: fixed, MinCredits: min, MaxCredits: max,
|
||
SearchCalls: searchCalls, MaxAICandidates: maxAI, RemainingCredits: remaining,
|
||
EstimateBasis: "免費前處理；provider 成功後依 web_search／ai_research／ai_copy meter 計點",
|
||
ExpiresAt: time.Now().UTC().Add(5 * time.Minute).UnixNano(),
|
||
}, nil
|
||
}
|