Janady13/missionmatch / internal/modules/commandcenter/service.go

1package commandcenter 2 3import ( 4 "context" 5 "errors" 6 "fmt" 7 "time" 8 9 "github.com/jackc/pgx/v5" 10 "github.com/shopspring/decimal" 11 12 "github.com/missionworkforce/missionmatch/internal/modules/matching/engine" 13 "github.com/missionworkforce/missionmatch/internal/platform/audit" 14 "github.com/missionworkforce/missionmatch/internal/platform/tenant" 15) 16 17// ErrNoDatabase is returned when no pool is configured. 18var ErrNoDatabase = errors.New("commandcenter: no database configured") 19 20// ErrForecastNotFound is returned when an adopt targets a record that does not exist, and 21// ErrNotAForecast when it targets a record that is not a forecast (only forecasts are adoptable). 22var ( 23 ErrForecastNotFound = errors.New("commandcenter: forecast not found") 24 ErrNotAForecast = errors.New("commandcenter: record is not a forecast") 25 ErrBadRequest = errors.New("commandcenter: bad request") 26) 27 28var hundred = decimal.NewFromInt(100) 29 30// WCC composite weights (DISTINCT from the engine's per-candidate weights). Sum == 1.00 (tested). 31var weights = map[string]decimal.Decimal{ 32 "clearance_coverage": decimal.RequireFromString("0.30"), 33 "skill_coverage": decimal.RequireFromString("0.25"), 34 "labor_category_coverage": decimal.RequireFromString("0.20"), 35 "proposal_readiness": decimal.RequireFromString("0.15"), 36 "availability_coverage": decimal.RequireFromString("0.10"), 37} 38var driverOrder = []struct{ key, label string }{ 39 {"clearance_coverage", "Clearance Coverage"}, 40 {"skill_coverage", "Skill Coverage"}, 41 {"labor_category_coverage", "Labor Category Coverage"}, 42 {"proposal_readiness", "Proposal Readiness"}, 43 {"availability_coverage", "Availability Coverage"}, 44} 45 46var levelLabels = map[string]string{ 47 "unknown": "Unknown", "none": "None", "confidential": "Confidential", 48 "secret": "Secret", "top_secret": "Top Secret", "ts_sci": "TS/SCI", 49} 50 51// Service computes the Workforce Command Center view. 52type Service struct { 53 db *tenant.DB 54 audit *audit.Writer // optional; used to record an adopt as requirement (a tenant record create) 55 clock func() time.Time 56} 57 58func NewService(db *tenant.DB) *Service { return &Service{db: db, clock: time.Now} } 59 60// NewServiceWithAudit builds a Service that records the adopt as requirement action to the append only 61// audit trail. The read only dashboard paths do not need the writer, so NewService stays valid. 62func NewServiceWithAudit(db *tenant.DB, aud *audit.Writer) *Service { 63 return &Service{db: db, audit: aud, clock: time.Now} 64} 65 66func levelLabel(code string) string { 67 if l, ok := levelLabels[code]; ok { 68 return l 69 } 70 return code 71} 72 73func pct(num, den int64) decimal.Decimal { 74 if den <= 0 { 75 return decimal.Zero 76 } 77 return decimal.NewFromInt(num).Div(decimal.NewFromInt(den)).Mul(hundred).Round(2) 78} 79 80func driverState(p decimal.Decimal) string { 81 switch { 82 case p.GreaterThanOrEqual(engine.BandGreenMin): 83 return "ready" 84 case p.GreaterThanOrEqual(engine.BandYellowMin): 85 return "partial" 86 default: 87 return "gap" 88 } 89} 90 91func colorFor(p decimal.Decimal) string { 92 switch driverState(p) { 93 case "ready": 94 return "green" 95 case "partial": 96 return "yellow" 97 default: 98 return "red" 99 } 100} 101 102// CommandCenter computes the full dashboard for the caller's org as-of the given date. 103func (s *Service) CommandCenter(ctx context.Context, asOf time.Time) (View, error) { 104 if s.db == nil { 105 return View{}, ErrNoDatabase 106 } 107 asOfStr := asOf.Format("2006-01-02") 108 v := View{AsOf: asOfStr, GeneratedAt: asOf.UTC().Format(time.RFC3339), Meta: Meta{Notes: []string{}}} 109 if org, ok := tenant.OrgFromContext(ctx); ok { 110 v.OrganizationID = org 111 } 112 113 err := s.db.RunInTenantTx(ctx, func(ctx context.Context, tx pgx.Tx) error { 114 if err := buildTemps(ctx, tx, asOfStr); err != nil { 115 return err 116 } 117 118 // ---- opportunity metadata (RLS-scoped) ---- 119 oppMeta := map[string]oppInfo{} 120 rows, err := tx.Query(ctx, `SELECT id::text, name, COALESCE(customer_agency,''), COALESCE(solicitation_number,''), COALESCE(to_char(due_date,'YYYY-MM-DD'),''), value, value_source FROM opportunities WHERE status='open'`) 121 if err != nil { 122 return fmt.Errorf("opp map: %w", err) 123 } 124 for rows.Next() { 125 var id, name, cust, sol, due, src string 126 var val *decimal.Decimal 127 if err := rows.Scan(&id, &name, &cust, &sol, &due, &val, &src); err != nil { 128 rows.Close() 129 return err 130 } 131 oppMeta[id] = oppInfo{name: name, customer: cust, solicitation: sol, dueDate: due, value: val, valueSource: src} 132 } 133 rows.Close() 134 if err := rows.Err(); err != nil { 135 return err 136 } 137 138 // ---- assignment-based panels (each available candidate consumed at most once) ---- 139 assigned, qualified, reqs, _, err := fetchAssignment(ctx, tx) 140 if err != nil { 141 return err 142 } 143 opps, highRisk := opportunitiesFromAssignment(reqs, assigned, oppMeta) 144 v.ActiveOpportunities = opps 145 gaps, suggestReq, totalGaps := gapsFromAssignment(reqs, assigned, oppMeta) 146 v.StaffingGaps = gaps 147 v.LaborCategoryCoverage = lcatCoverageFromAssignment(reqs, assigned) 148 var proposalPct decimal.Decimal 149 v.ProposalReadiness, proposalPct = proposalFromAssignment(reqs, assigned, qualified, oppMeta) 150 151 // ---- drivers (clearance/skill/lcat coverage + availability via SQL; proposal from assignment) ---- 152 drivers, availReqCov, err := s.drivers(ctx, tx) 153 if err != nil { 154 return err 155 } 156 drivers["proposal_readiness"] = proposalPct 157 158 var totalCleared, availableAll, availCleared int64 159 if err := tx.QueryRow(ctx, 160 `SELECT count(*) FILTER (WHERE effective AND lvl>=2), count(*) FILTER (WHERE available), 161 count(*) FILTER (WHERE available AND effective AND lvl>=2) FROM cc_cand`). 162 Scan(&totalCleared, &availableAll, &availCleared); err != nil { 163 return fmt.Errorf("banner cand: %w", err) 164 } 165 // Proposal-ready = green slate members on slates tied to an OPEN (or standalone) opportunity, 166 // excluding closed-opportunity slates [review EN-2]. 167 var proposalReady, activeReqs int64 168 if err := tx.QueryRow(ctx, 169 `SELECT count(DISTINCT sm.candidate_id) 170 FROM slate_members sm 171 JOIN slates s ON s.id = sm.slate_id 172 JOIN requirements r ON r.id = s.requirement_id 173 LEFT JOIN opportunities o ON o.id = r.opportunity_id 174 WHERE sm.band='green' AND (o.id IS NULL OR o.status='open')`).Scan(&proposalReady); err != nil { 175 return fmt.Errorf("proposal ready: %w", err) 176 } 177 if err := tx.QueryRow(ctx, `SELECT count(*) FROM cc_active_req`).Scan(&activeReqs); err != nil { 178 return fmt.Errorf("active reqs: %w", err) 179 } 180 181 // ---- clearance intelligence ---- 182 if v.ClearanceIntelligence, err = s.clearance(ctx, tx, asOfStr); err != nil { 183 return err 184 } 185 // ---- win impact ---- 186 if v.WinImpact, err = s.winImpact(ctx, tx); err != nil { 187 return err 188 } 189 // ---- heat map ---- 190 if v.HeatMap, err = s.heatMap(ctx, tx); err != nil { 191 return err 192 } 193 // ---- upcoming opportunities (forecasts; planning data, does NOT alter current readiness or RaR) ---- 194 if v.UpcomingOpportunities, err = s.upcomingOpportunities(ctx, tx); err != nil { 195 return err 196 } 197 // ---- incumbent and award context (from the contract award feed) ---- 198 if v.IncumbentLandscape, err = s.incumbentLandscape(ctx, tx); err != nil { 199 return err 200 } 201 // ---- action center badge counts ---- 202 var openNoSlate, reqsUnscreened int64 203 if err := tx.QueryRow(ctx, 204 `SELECT count(*) FROM opportunities o WHERE o.status='open' 205 AND NOT EXISTS (SELECT 1 FROM slates s WHERE s.requirement_id IN (SELECT id FROM requirements r WHERE r.opportunity_id=o.id))`). 206 Scan(&openNoSlate); err != nil { 207 return fmt.Errorf("open no slate: %w", err) 208 } 209 if err := tx.QueryRow(ctx, 210 `SELECT count(*) FROM cc_active_req ar WHERE NOT EXISTS (SELECT 1 FROM slates s WHERE s.requirement_id=ar.id)`). 211 Scan(&reqsUnscreened); err != nil { 212 return fmt.Errorf("reqs unscreened: %w", err) 213 } 214 215 // ---- assemble mission readiness ---- 216 score := decimal.Zero 217 drvViews := make([]Driver, 0, len(driverOrder)) 218 for _, d := range driverOrder { 219 p := drivers[d.key] 220 score = score.Add(p.Mul(weights[d.key])) 221 drvViews = append(drvViews, Driver{ 222 Key: d.key, Label: d.label, Pct: p.StringFixed(2), State: driverState(p), 223 Weight: weights[d.key].String(), 224 }) 225 } 226 score = score.Round(2) 227 band, status := readinessBand(score, totalCleared == 0 && activeReqs == 0) 228 v.MissionReadiness = MissionReadiness{ 229 Score: score.StringFixed(2), Band: band, Status: status, Drivers: drvViews, 230 AvailabilityReqCoverage: availReqCov.StringFixed(2), Flags: []string{}, 231 } 232 233 v.Banner = Banner{ 234 TotalClearedWorkforce: int(totalCleared), AvailableCandidates: int(availableAll), 235 ProposalReadyCandidates: int(proposalReady), ActiveRequirements: int(activeReqs), 236 StaffingGaps: int(totalGaps), HighRiskOpportunities: highRisk, 237 } 238 239 v.ActionCenter = buildActionCenter(suggestReq, opps, int(openNoSlate), int(reqsUnscreened)) 240 v.Meta.Empty = totalCleared == 0 && activeReqs == 0 && availableAll == 0 241 return nil 242 }) 243 if err != nil { 244 return View{}, err 245 } 246 247 v.AIRecommendations = buildRecommendations(v) 248 if v.AIRecommendations == nil { 249 v.AIRecommendations = []Recommendation{} 250 } 251 return v, nil 252} 253 254type oppInfo struct { 255 name string 256 customer string 257 solicitation string 258 dueDate string 259 value *decimal.Decimal 260 valueSource string // representative, operator, sam, usaspending (W2 Revenue at Risk provenance) 261} 262 263// isRealValueSource reports whether an opportunity value came from a cached federal source (SAM or 264// USASpending). A representative or operator entered dollar is not a real federal figure, so it can 265// never clear the representative label on Revenue at Risk. 266func isRealValueSource(s string) bool { return s == "sam" || s == "usaspending" } 267 268func readinessBand(score decimal.Decimal, empty bool) (band, status string) { 269 if empty { 270 return "red", "No Data" 271 } 272 switch { 273 case score.GreaterThanOrEqual(engine.BandGreenMin): 274 return "green", "Mission Ready" 275 case score.GreaterThanOrEqual(engine.BandYellowMin): 276 return "yellow", "Watch" 277 default: 278 return "red", "At Risk" 279 } 280} 281