Sign in

flagon-io/g1t

public

g1t/services/billing/src/sales.rs

758 lines33,136 bytesCodeBlame
1//! What g1t's team needs to sell and support: month-by-month figures, the
2//! signals that say a workspace is worth a call, and what was done about
3//! it. Staff only, through sudo.g1t.sh.
4
5use g1t_contracts::billing::{
6 AdminAddNoteArgs, AdminOverviewArgs, AdminOveragesArgs, AdminSalesArgs, AdminSetSalesArgs, AdminSignalsArgs, GivenFigures,
7 InternalUse, KindFigures, LimitState, MonthFigures, Overview, SalesNote, SalesRecord, Signal, SignalKind, TermsKind, Trust,
8};
9use g1t_contracts::time::rfc3339;
10use g1t_contracts::{FailureCode, Outcome, new_id};
11use g1t_kit::now_ms;
12use serde::Deserialize;
13use worker::Result;
14use worker::wasm_bindgen::JsValue;
15
16use crate::Billing;
17use crate::features::dollars;
18use crate::limits::previous_month;
19
20pub(crate) const STAGES: &[&str] = &["none", "lead", "contacted", "negotiating", "won", "lost", "churn_risk"];
21
22/// A workspace this much ahead of last month's pace is growing.
23const GROWING_FACTOR: f64 = 1.5;
24/// Below this last month, growth is noise.
25const GROWING_FROM_MICROS: i64 = 10_000_000;
26/// Spending this much a month may suit custom terms or an enterprise.
27const HIGH_SPEND_MICROS: i64 = 500_000_000;
28
29/// The six months ending with `month`, oldest first.
30pub(crate) fn last_months(month: &str, count: usize) -> Vec<String> {
31 let mut months = vec![month.to_owned()];
32 while months.len() < count {
33 let earlier = previous_month(months.last().unwrap());
34 months.push(earlier);
35 }
36 months.reverse();
37 months
38}
39
40/// How urgent a kind of signal is: lower first.
41fn urgency(kind: SignalKind) -> u8 {
42 match kind {
43 SignalKind::AtLimit => 0,
44 SignalKind::Declined => 1,
45 SignalKind::NearCeiling => 2,
46 SignalKind::HighSpend => 3,
47 SignalKind::Growing => 4,
48 SignalKind::Established => 5,
49 SignalKind::FirstPayment => 6,
50 }
51}
52
53/// Whether this month, at its pace so far, is well ahead of last month.
54pub(crate) fn growing(this_month: i64, last_month: i64, day: u32, days_in_month: u32) -> bool {
55 if last_month < GROWING_FROM_MICROS || day == 0 {
56 return false;
57 }
58 let pace = this_month as f64 * f64::from(days_in_month) / f64::from(day);
59 pace >= last_month as f64 * GROWING_FACTOR
60}
61
62fn days_in(month: &str) -> u32 {
63 let year: i32 = month[..4].parse().unwrap_or(1970);
64 match month[5..7].parse::<u32>().unwrap_or(1) {
65 2 if (year % 4 == 0 && year % 100 != 0) || year % 400 == 0 => 29,
66 2 => 28,
67 4 | 6 | 9 | 11 => 30,
68 _ => 31,
69 }
70}
71
72/// The workspaces g1t does not charge (comped terms): its own and
73/// Flagon's, and any enterprise on comped terms. Their usage is recorded at
74/// what it cost and shown as given, apart from margin.
75pub(crate) const INTERNAL_SQL: &str = "SELECT substr(id, 4) FROM billing_accounts WHERE kind = 'workspace' AND terms_kind = 'comped'
76 UNION SELECT m.workspace FROM account_members m JOIN billing_accounts b ON b.id = m.account_id WHERE b.terms_kind = 'comped'";
77
78/// One group of ledger rows in a month, as sudo's figures add them up.
79#[derive(Clone, Debug, Default, Deserialize)]
80pub(crate) struct LedgerGroup {
81 pub month: String,
82 /// `usage` or `top_up`.
83 pub kind: String,
84 /// `g1t`, or `workspace` for a run on the workspace's own provider.
85 pub billed_to: Option<String>,
86 /// A credit from g1t rather than a payment (`crd…`), and of those, a
87 /// goodwill credit.
88 pub credit: i64,
89 pub goodwill: i64,
90 /// One of g1t's own workspaces (comped).
91 pub internal: i64,
92 pub amount: Option<i64>,
93 pub cost: Option<i64>,
94 /// What the trial, the open-source pool and g1t itself paid.
95 pub pools: Option<i64>,
96}
97
98/// A month's figures from its ledger groups and the plan's price paid.
99///
100/// - **Cost** is only what g1t paid: a run on the workspace's own model
101/// provider was paid for there, so its cost is never g1t's.
102/// - **Given** is what g1t gave at price, apart from its margin: internal
103/// use (its cost plus the margin, since nothing was charged), what the
104/// trial, the open-source pool and g1t itself paid, and goodwill credits.
105/// - **Paid** is money in: payments, never credits from g1t.
106pub(crate) fn fold_month(month: &str, groups: &[LedgerGroup], plans: i64, margin_percent: u32) -> MonthFigures {
107 let mut figures = MonthFigures { month: month.to_owned(), plans_micros: plans, ..MonthFigures::default() };
108 for g in groups.iter().filter(|g| g.month == month) {
109 let amount = g.amount.unwrap_or(0);
110 let own_provider = g.billed_to.as_deref() == Some("workspace");
111 if g.kind == "usage" {
112 figures.charged_micros += -amount;
113 if !own_provider {
114 let cost = g.cost.unwrap_or(0);
115 figures.cost_micros += cost;
116 if g.internal == 1 {
117 figures.given_micros += crate::credits::with_margin(cost, margin_percent);
118 }
119 }
120 figures.given_micros += g.pools.unwrap_or(0);
121 } else if g.credit == 0 {
122 figures.paid_micros += amount;
123 } else if g.goodwill == 1 {
124 figures.given_micros += amount;
125 }
126 }
127 figures
128}
129
130#[derive(Deserialize)]
131struct RecordRow {
132 stage: String,
133 owner: Option<String>,
134 next_step: Option<String>,
135 next_at: Option<String>,
136 updated_at: String,
137}
138
139#[derive(Deserialize)]
140struct NoteRow {
141 id: String,
142 text: String,
143 by: String,
144 created_at: String,
145}
146
147impl Billing {
148 /// Month-by-month figures for some workspaces (all, when empty): usage
149 /// charged, what it cost g1t (never a workspace's own provider), money
150 /// paid, the plan's price paid, and what g1t gave.
151 pub(crate) async fn months_for(&self, workspaces: &[String], count: usize) -> Result<Vec<MonthFigures>> {
152 let months = last_months(&rfc3339(now_ms())[..7], count);
153 let since = format!("{}-01", months[0]);
154 let (filter, values): (String, Vec<JsValue>) = if workspaces.is_empty() {
155 (String::new(), vec![])
156 } else {
157 let marks = vec!["?"; workspaces.len()].join(", ");
158 (format!("AND workspace IN ({marks})"), workspaces.iter().map(|w| JsValue::from(w.as_str())).collect())
159 };
160 let groups = self
161 .db
162 .prepare(format!(
163 "SELECT substr(created_at, 1, 7) AS month, kind, COALESCE(billed_to, 'g1t') AS billed_to,
164 CASE WHEN reference LIKE 'crd%' THEN 1 ELSE 0 END AS credit,
165 CASE WHEN reference LIKE '{goodwill}%' THEN 1 ELSE 0 END AS goodwill,
166 CASE WHEN workspace IN ({INTERNAL_SQL}) THEN 1 ELSE 0 END AS internal,
167 SUM(amount_micros) AS amount, SUM(cost_micros) AS cost,
168 SUM(trial_micros + oss_micros + given_micros) AS pools
169 FROM ledger WHERE created_at >= '{since}' {filter} GROUP BY 1, 2, 3, 4, 5, 6",
170 goodwill = crate::overages::GOODWILL_PREFIX
171 ))
172 .bind(&values)?
173 .all()
174 .await?
175 .results::<LedgerGroup>()?;
176 #[derive(Deserialize)]
177 struct Plans {
178 month: String,
179 micros: Option<i64>,
180 }
181 let plans = self
182 .db
183 .prepare(format!(
184 "SELECT substr(paid_at, 1, 7) AS month, SUM(amount_micros) AS micros FROM plan_payments
185 WHERE paid_at >= '{since}' {filter} GROUP BY 1"
186 ))
187 .bind(&values)?
188 .all()
189 .await?
190 .results::<Plans>()?;
191 Ok(months
192 .into_iter()
193 .map(|month| {
194 let paid_plans = plans.iter().find(|p| p.month == month).and_then(|p| p.micros).unwrap_or(0);
195 fold_month(&month, &groups, paid_plans, self.margin_percent)
196 })
197 .collect())
198 }
199
200 async fn record_row(&self, workspace: &str) -> Result<Option<RecordRow>> {
201 self.db
202 .prepare("SELECT * FROM sales_records WHERE workspace = ?")
203 .bind(&[workspace.into()])?
204 .first::<RecordRow>(None)
205 .await
206 }
207
208 pub(crate) async fn admin_sales(&self, a: AdminSalesArgs) -> Result<SalesRecord> {
209 let workspace = a.workspace.trim().to_lowercase();
210 let row = self.record_row(&workspace).await?;
211 let notes = self
212 .db
213 .prepare("SELECT id, text, by, created_at FROM sales_notes WHERE workspace = ? ORDER BY created_at DESC LIMIT 100")
214 .bind(&[workspace.as_str().into()])?
215 .all()
216 .await?
217 .results::<NoteRow>()?
218 .into_iter()
219 .map(|n| SalesNote { id: n.id, text: n.text, by: n.by, created_at: n.created_at })
220 .collect();
221 Ok(match row {
222 Some(row) => SalesRecord {
223 workspace,
224 stage: row.stage,
225 owner: row.owner,
226 next_step: row.next_step,
227 next_at: row.next_at,
228 notes,
229 updated_at: Some(row.updated_at),
230 },
231 None => SalesRecord {
232 workspace,
233 stage: "none".to_owned(),
234 owner: None,
235 next_step: None,
236 next_at: None,
237 notes,
238 updated_at: None,
239 },
240 })
241 }
242
243 pub(crate) async fn admin_set_sales(&self, a: AdminSetSalesArgs) -> Result<Outcome<SalesRecord>> {
244 let workspace = a.workspace.trim().to_lowercase();
245 if workspace.is_empty() || a.by.trim().is_empty() {
246 return Ok(Outcome::fail(FailureCode::Invalid, "Name the workspace, and who is making the change."));
247 }
248 if !STAGES.contains(&a.stage.as_str()) {
249 return Ok(Outcome::fail(FailureCode::Invalid, format!("A stage is one of: {}.", STAGES.join(", "))));
250 }
251 let clean = |value: &Option<String>| value.as_deref().map(str::trim).filter(|v| !v.is_empty()).map(str::to_owned);
252 let (owner, next_step, next_at) = (clean(&a.owner), clean(&a.next_step), clean(&a.next_at));
253 let before = self.record_row(&workspace).await?;
254 self.db
255 .prepare(
256 "INSERT INTO sales_records (workspace, stage, owner, next_step, next_at, updated_by, updated_at)
257 VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7)
258 ON CONFLICT (workspace) DO UPDATE SET stage = ?2, owner = ?3, next_step = ?4, next_at = ?5,
259 updated_by = ?6, updated_at = ?7",
260 )
261 .bind(&[
262 workspace.as_str().into(),
263 a.stage.as_str().into(),
264 crate::optional(owner.as_deref()),
265 crate::optional(next_step.as_deref()),
266 crate::optional(next_at.as_deref()),
267 a.by.as_str().into(),
268 rfc3339(now_ms()).into(),
269 ])?
270 .run()
271 .await?;
272 let account = self.account_of(&workspace).await?;
273 let was = before.map_or_else(|| "none".to_owned(), |b| b.stage);
274 let detail = format!(
275 "{workspace}: stage {was} → {}{}{}",
276 a.stage,
277 owner.as_deref().map(|o| format!(", owner {o}")).unwrap_or_default(),
278 next_step.as_deref().map(|s| format!(", next: {s}")).unwrap_or_default(),
279 );
280 self.audit(&account.id, "sales", &detail, &a.by).await?;
281 Ok(Outcome::Ok(self.admin_sales(AdminSalesArgs { workspace }).await?))
282 }
283
284 pub(crate) async fn admin_add_note(&self, a: AdminAddNoteArgs) -> Result<Outcome<SalesRecord>> {
285 let workspace = a.workspace.trim().to_lowercase();
286 let text = a.text.trim();
287 if workspace.is_empty() || text.is_empty() || a.by.trim().is_empty() {
288 return Ok(Outcome::fail(FailureCode::Invalid, "A note needs a workspace, some words, and who wrote it."));
289 }
290 if text.chars().count() > 4000 {
291 return Ok(Outcome::fail(FailureCode::Invalid, "Keep a note under 4,000 characters."));
292 }
293 let now = now_ms();
294 self.db
295 .prepare("INSERT INTO sales_notes (id, workspace, text, by, created_at) VALUES (?, ?, ?, ?, ?)")
296 .bind(&[new_id("note", now).into(), workspace.as_str().into(), text.into(), a.by.as_str().into(), rfc3339(now).into()])?
297 .run()
298 .await?;
299 Ok(Outcome::Ok(self.admin_sales(AdminSalesArgs { workspace }).await?))
300 }
301
302 /// Every workspace worth reaching out to, most urgent first.
303 pub(crate) async fn admin_signals(&self, _: AdminSignalsArgs) -> Result<Vec<Signal>> {
304 let now = rfc3339(now_ms());
305 let month = &now[..7];
306 let last = previous_month(month);
307 #[derive(Deserialize)]
308 struct Active {
309 workspace: String,
310 this_month: Option<i64>,
311 last_month: Option<i64>,
312 first_paid: Option<String>,
313 }
314 let active = self
315 .db
316 .prepare(
317 "SELECT workspace,
318 -SUM(CASE WHEN kind = 'usage' AND created_at >= ?1 THEN amount_micros END) AS this_month,
319 -SUM(CASE WHEN kind = 'usage' AND created_at >= ?2 AND created_at < ?1 THEN amount_micros END) AS last_month,
320 MIN(CASE WHEN kind = 'top_up' AND amount_micros > 0 AND reference NOT LIKE 'crd%' THEN created_at END) AS first_paid
321 FROM ledger GROUP BY workspace
322 HAVING MAX(created_at) >= ?2
323 LIMIT 500",
324 )
325 .bind(&[format!("{month}-01").into(), format!("{last}-01").into()])?
326 .all()
327 .await?
328 .results::<Active>()?;
329 let day: u32 = now[8..10].parse().unwrap_or(1);
330 let fortnight_ago = rfc3339(now_ms() - 14 * 24 * 60 * 60 * 1000);
331 let mut signals = vec![];
332 for row in active {
333 let limit = self.limit_of(&row.workspace).await?;
334 if limit.trust == Trust::Internal {
335 continue;
336 }
337 let this_month = row.this_month.unwrap_or(0).max(0);
338 let last_month = row.last_month.unwrap_or(0).max(0);
339 let record = self.record_row(&row.workspace).await?;
340 let (stage, owner, next_step, next_at) =
341 record.map_or((None, None, None, None), |r| (Some(r.stage), r.owner, r.next_step, r.next_at));
342 let mut push = |kind: SignalKind, detail: String, value: i64| {
343 signals.push(Signal {
344 workspace: row.workspace.clone(),
345 kind,
346 detail,
347 value_micros: value,
348 stage: stage.clone(),
349 owner: owner.clone(),
350 next_step: next_step.clone(),
351 next_at: next_at.clone(),
352 });
353 };
354 let declined = limit.message.as_deref().is_some_and(|m| m.contains("could not be charged"));
355 if declined {
356 push(SignalKind::Declined, limit.message.clone().unwrap_or_default(), limit.exposure_micros);
357 } else if limit.state == LimitState::Stopped {
358 push(SignalKind::AtLimit, limit.message.clone().unwrap_or_default(), limit.exposure_micros.max(limit.spent_micros));
359 } else if let Some(available) = limit.available_micros.filter(|a| *a > 0) {
360 if limit.exposure_micros * 5 >= available * 4 {
361 push(
362 SignalKind::NearCeiling,
363 format!(
364 "{} unpaid of the {} g1t allows it ({:?}); a call could raise it before it stops.",
365 dollars(limit.exposure_micros),
366 dollars(available),
367 limit.trust
368 ),
369 limit.exposure_micros,
370 );
371 }
372 }
373 if last_month >= HIGH_SPEND_MICROS {
374 let terms = self.terms_of(&row.workspace).await?;
375 if terms.kind == TermsKind::Standard && !limit.account.starts_with("ent_") {
376 push(
377 SignalKind::HighSpend,
378 format!("Spent {} last month on standard terms: worth offering custom terms or an enterprise.", dollars(last_month)),
379 last_month,
380 );
381 }
382 }
383 if growing(this_month, last_month, day, days_in(month)) {
384 push(
385 SignalKind::Growing,
386 format!(
387 "{} so far this month, on pace for about {}, against {} last month.",
388 dollars(this_month),
389 dollars((this_month as f64 * f64::from(days_in(month)) / f64::from(day.max(1))) as i64),
390 dollars(last_month)
391 ),
392 this_month,
393 );
394 }
395 if limit.trust == Trust::Established {
396 push(
397 SignalKind::Established,
398 format!(
399 "A steady customer: its limit now follows its spend ({} available).",
400 limit.available_micros.map(dollars).unwrap_or_default()
401 ),
402 last_month,
403 );
404 }
405 if row.first_paid.as_deref().is_some_and(|at| at >= fortnight_ago.as_str()) {
406 push(SignalKind::FirstPayment, "Paid g1t for the first time in the last two weeks: say hello.".to_owned(), this_month);
407 }
408 }
409 signals.sort_by(|a, b| urgency(a.kind).cmp(&urgency(b.kind)).then(b.value_micros.cmp(&a.value_micros)));
410 Ok(signals)
411 }
412
413 /// Every invoice g1t has sent, workspaces' and enterprises'.
414 pub(crate) async fn admin_invoices(&self, a: g1t_contracts::billing::AdminInvoicesArgs) -> Result<Vec<g1t_contracts::billing::InvoiceSummary>> {
415 #[derive(Deserialize)]
416 struct Row {
417 invoice_id: String,
418 kind: String,
419 account: String,
420 name: String,
421 reason: String,
422 period: String,
423 amount_micros: i64,
424 status: String,
425 hosted_url: Option<String>,
426 created_at: String,
427 paid_at: Option<String>,
428 }
429 let status = a.status.filter(|s| ["paid", "open", "failed", "overdue", "void"].contains(&s.as_str()));
430 let month = a.month.filter(|m| m.len() == 7 && m.chars().all(|c| c.is_ascii_digit() || c == '-'));
431 let rows = self
432 .db
433 .prepare(
434 "SELECT * FROM (
435 SELECT invoice_id, 'workspace' AS kind, workspace AS account, workspace AS name, reason, period,
436 amount_micros, status, hosted_url, created_at, paid_at
437 FROM workspace_invoices
438 UNION ALL
439 SELECT i.invoice_id, 'enterprise', i.account_id, COALESCE(b.name, i.account_id), 'enterprise', i.period,
440 i.amount_micros, i.status, i.hosted_url, i.created_at, i.paid_at
441 FROM enterprise_invoices i LEFT JOIN billing_accounts b ON b.id = i.account_id
442 )
443 WHERE (?1 IS NULL OR status = ?1) AND (?2 IS NULL OR substr(created_at, 1, 7) = ?2)
444 ORDER BY created_at DESC LIMIT 200",
445 )
446 .bind(&[crate::optional(status.as_deref()), crate::optional(month.as_deref())])?
447 .all()
448 .await?
449 .results::<Row>()?;
450 Ok(rows
451 .into_iter()
452 .map(|r| g1t_contracts::billing::InvoiceSummary {
453 invoice_id: r.invoice_id,
454 kind: r.kind,
455 account: r.account,
456 name: r.name,
457 reason: r.reason,
458 period: r.period,
459 amount_micros: r.amount_micros,
460 status: r.status,
461 hosted_url: r.hosted_url,
462 created_at: r.created_at,
463 paid_at: r.paid_at,
464 })
465 .collect())
466 }
467
468 /// Every change made in sudo and by Stripe, newest first.
469 pub(crate) async fn admin_audit(&self, a: g1t_contracts::billing::AdminAuditArgs) -> Result<Vec<g1t_contracts::billing::AdminAction>> {
470 #[derive(Deserialize)]
471 struct Row {
472 id: String,
473 account: String,
474 action: String,
475 detail: String,
476 by: String,
477 created_at: String,
478 }
479 let rows = self
480 .db
481 .prepare(
482 "SELECT id, account, action, detail, by, created_at FROM admin_actions
483 WHERE (?1 IS NULL OR by = ?1) AND (?2 IS NULL OR action = ?2) AND (?3 IS NULL OR created_at < ?3)
484 ORDER BY created_at DESC LIMIT 100",
485 )
486 .bind(&[
487 crate::optional(a.by.as_deref().map(str::trim).filter(|s| !s.is_empty())),
488 crate::optional(a.action.as_deref().map(str::trim).filter(|s| !s.is_empty())),
489 crate::optional(a.before.as_deref()),
490 ])?
491 .all()
492 .await?
493 .results::<Row>()?;
494 Ok(rows
495 .into_iter()
496 .map(|r| g1t_contracts::billing::AdminAction {
497 id: r.id,
498 account: r.account,
499 action: r.action,
500 detail: r.detail,
501 by: r.by,
502 created_at: r.created_at,
503 })
504 .collect())
505 }
506
507 /// The business at a glance.
508 pub(crate) async fn admin_overview(&self, _: AdminOverviewArgs) -> Result<Overview> {
509 let now = rfc3339(now_ms());
510 let month = now[..7].to_owned();
511 let months = self.months_for(&[], 6).await?;
512 #[derive(Deserialize)]
513 struct KindRow {
514 kind: Option<String>,
515 charged: Option<i64>,
516 cost: Option<i64>,
517 }
518 let by_kind = self
519 .db
520 .prepare(
521 "SELECT CASE
522 WHEN task = 'sandbox' THEN 'Sandbox time'
523 WHEN task = 'self_hosted' THEN 'Self-hosted runner time'
524 WHEN task = 'deployments' THEN 'Deployments'
525 WHEN task = 'security' THEN 'Security scans'
526 WHEN task = 'context' THEN 'Search embeddings'
527 WHEN task = 'storage' THEN 'Private storage'
528 WHEN task = 'git' THEN 'Git operations'
529 WHEN billed_to = 'workspace' THEN 'Own-provider runs'
530 ELSE 'Models' END AS kind,
531 -SUM(amount_micros) AS charged,
532 SUM(CASE WHEN COALESCE(billed_to, 'g1t') = 'g1t' THEN cost_micros ELSE 0 END) AS cost
533 FROM ledger WHERE kind = 'usage' AND created_at >= ? GROUP BY 1 ORDER BY charged DESC",
534 )
535 .bind(&[format!("{month}-01").into()])?
536 .all()
537 .await?
538 .results::<KindRow>()?
539 .into_iter()
540 .map(|r| KindFigures {
541 kind: r.kind.unwrap_or_else(|| "Other".to_owned()),
542 charged_micros: r.charged.unwrap_or(0),
543 cost_micros: r.cost.unwrap_or(0),
544 })
545 .collect();
546 #[derive(Deserialize)]
547 struct Count {
548 n: Option<i64>,
549 }
550 let count = |sql: &'static str, args: Vec<JsValue>| {
551 let db = &self.db;
552 async move {
553 Ok::<i64, worker::Error>(db.prepare(sql).bind(&args)?.first::<Count>(None).await?.and_then(|c| c.n).unwrap_or(0))
554 }
555 };
556 let paying = count(
557 "SELECT COUNT(DISTINCT workspace) AS n FROM ledger
558 WHERE kind = 'top_up' AND amount_micros > 0 AND reference NOT LIKE 'crd%' AND created_at >= ?",
559 vec![rfc3339(now_ms() - 60 * 24 * 60 * 60 * 1000).into()],
560 )
561 .await?;
562 let open_invoices = count(
563 "SELECT (SELECT COALESCE(SUM(amount_micros), 0) FROM workspace_invoices WHERE status IN ('open', 'failed'))
564 + (SELECT COALESCE(SUM(amount_micros), 0) FROM enterprise_invoices WHERE status IN ('open', 'overdue')) AS n",
565 vec![],
566 )
567 .await?;
568 let follow_ups = count(
569 "SELECT COUNT(*) AS n FROM sales_records WHERE next_at IS NOT NULL AND next_at <= ? AND stage NOT IN ('won', 'lost')",
570 vec![now[..10].into()],
571 )
572 .await?;
573 let signals = self.admin_signals(AdminSignalsArgs {}).await?;
574 let tally = |kind: SignalKind| signals.iter().filter(|s| s.kind == kind).count() as u32;
575 let this = months.last().cloned().unwrap_or_default();
576 let active_plans = count(
577 "SELECT COUNT(*) AS n FROM subscriptions WHERE feature = 'plan' AND status IN ('active', 'canceling')",
578 vec![],
579 )
580 .await?;
581 let open_requests = count("SELECT COUNT(*) AS n FROM limit_requests WHERE status = 'open'", vec![]).await?;
582 let open_spikes = count("SELECT COUNT(*) AS n FROM spikes WHERE status = 'open'", vec![]).await?;
583 let overages = self.admin_overages(AdminOveragesArgs {}).await?.len() as u32;
584 Ok(Overview {
585 revenue_micros: this.charged_micros + this.plans_micros,
586 active_plans: active_plans as u32,
587 plan_mrr_micros: active_plans * i64::from(self.plans.plan_monthly_cents) * 10_000,
588 given: self.given_this_month(&month).await?,
589 internal: self.internal_use(&month).await?,
590 open_requests: open_requests as u32,
591 open_spikes: open_spikes as u32,
592 overages,
593 month,
594 months,
595 by_kind,
596 paying_workspaces: paying as u32,
597 stopped: tally(SignalKind::AtLimit),
598 near_ceiling: tally(SignalKind::NearCeiling),
599 declined: tally(SignalKind::Declined),
600 open_invoices_micros: open_invoices,
601 follow_ups_due: follow_ups as u32,
602 pools: Some(self.pools().await?),
603 })
604 }
605
606 /// What g1t gave in `month`, by source, at price and at cost.
607 async fn given_this_month(&self, month: &str) -> Result<Vec<GivenFigures>> {
608 #[derive(Deserialize)]
609 struct Row {
610 internal_cost: Option<i64>,
611 trial: Option<i64>,
612 oss: Option<i64>,
613 covered: Option<i64>,
614 goodwill: Option<i64>,
615 }
616 let row = self
617 .db
618 .prepare(format!(
619 "SELECT
620 SUM(CASE WHEN kind = 'usage' AND COALESCE(billed_to, 'g1t') = 'g1t' AND workspace IN ({INTERNAL_SQL})
621 THEN cost_micros END) AS internal_cost,
622 SUM(CASE WHEN kind = 'usage' THEN trial_micros END) AS trial,
623 SUM(CASE WHEN kind = 'usage' THEN oss_micros END) AS oss,
624 SUM(CASE WHEN kind = 'usage' THEN given_micros END) AS covered,
625 SUM(CASE WHEN kind = 'top_up' AND reference LIKE '{goodwill}%' THEN amount_micros END) AS goodwill
626 FROM ledger WHERE created_at >= ?",
627 goodwill = crate::overages::GOODWILL_PREFIX
628 ))
629 .bind(&[format!("{month}-01").into()])?
630 .first::<Row>(None)
631 .await?;
632 let Some(row) = row else { return Ok(vec![]) };
633 let at_cost = |price: i64| price * 100 / i64::from(100 + self.margin_percent);
634 let internal_cost = row.internal_cost.unwrap_or(0);
635 Ok([
636 ("internal", "g1t's and Flagon's own use", crate::credits::with_margin(internal_cost, self.margin_percent), internal_cost),
637 ("trial", "Trials", row.trial.unwrap_or(0), at_cost(row.trial.unwrap_or(0))),
638 ("oss_pool", "Open-source pool", row.oss.unwrap_or(0), at_cost(row.oss.unwrap_or(0))),
639 ("covered", "Covered past a trial's end", row.covered.unwrap_or(0), at_cost(row.covered.unwrap_or(0))),
640 ("goodwill", "Goodwill credits", row.goodwill.unwrap_or(0), row.goodwill.unwrap_or(0)),
641 ]
642 .into_iter()
643 .map(|(source, label, micros, cost)| GivenFigures { source: source.to_owned(), label: label.to_owned(), micros, cost_micros: cost })
644 .collect())
645 }
646
647 /// g1t's own workspaces' use in `month`, with why each is not charged.
648 async fn internal_use(&self, month: &str) -> Result<Vec<InternalUse>> {
649 #[derive(Deserialize)]
650 struct Row {
651 workspace: String,
652 cost: Option<i64>,
653 entries: Option<u32>,
654 }
655 let rows = self
656 .db
657 .prepare(format!(
658 "SELECT workspace, SUM(cost_micros) AS cost, COUNT(*) AS entries FROM ledger
659 WHERE kind = 'usage' AND COALESCE(billed_to, 'g1t') = 'g1t' AND created_at >= ? AND workspace IN ({INTERNAL_SQL})
660 GROUP BY workspace ORDER BY cost DESC"
661 ))
662 .bind(&[format!("{month}-01").into()])?
663 .all()
664 .await?
665 .results::<Row>()?;
666 let mut list = vec![];
667 for row in rows {
668 let terms = self.terms_of(&row.workspace).await?;
669 list.push(InternalUse {
670 reason: if terms.note.is_empty() { "Comped".to_owned() } else { terms.note },
671 cost_micros: row.cost.unwrap_or(0),
672 entries: row.entries.unwrap_or(0),
673 workspace: row.workspace,
674 });
675 }
676 Ok(list)
677 }
678}
679
680#[cfg(test)]
681mod tests {
682 use super::*;
683
684 fn group(kind: &str, billed_to: &str, amount: i64, cost: i64) -> LedgerGroup {
685 LedgerGroup {
686 month: "2026-10".into(),
687 kind: kind.into(),
688 billed_to: Some(billed_to.into()),
689 amount: Some(amount),
690 cost: Some(cost),
691 ..LedgerGroup::default()
692 }
693 }
694
695 #[test]
696 fn a_workspaces_own_provider_is_never_g1ts_cost() {
697 let groups = [
698 // Runs on g1t's models: charged $12, cost g1t $10.
699 group("usage", "g1t", -12_000_000, 10_000_000),
700 // A run on the workspace's own provider: it paid that $0.21 itself.
701 group("usage", "workspace", -100_000, 209_700),
702 ];
703 let month = fold_month("2026-10", &groups, 0, 20);
704 assert_eq!(month.cost_micros, 10_000_000);
705 assert_eq!(month.charged_micros, 12_100_000);
706 }
707
708 #[test]
709 fn plan_revenue_counts_and_what_g1t_gives_is_kept_apart() {
710 let groups = [
711 group("usage", "g1t", -12_000_000, 10_000_000),
712 // Paid for by the trial and the open-source pool: $3 at price.
713 LedgerGroup { pools: Some(3_000_000), ..group("usage", "g1t", 0, 2_500_000) },
714 // g1t's own workspace: nothing charged, $5 of cost.
715 LedgerGroup { internal: 1, ..group("usage", "g1t", 0, 5_000_000) },
716 // A payment, a goodwill credit, and another credit from g1t.
717 group("top_up", "g1t", 40_000_000, 0),
718 LedgerGroup { credit: 1, goodwill: 1, ..group("top_up", "g1t", 15_000_000, 0) },
719 LedgerGroup { credit: 1, ..group("top_up", "g1t", 2_000_000, 0) },
720 // Another month is not this one.
721 LedgerGroup { month: "2026-09".into(), ..group("usage", "g1t", -1, 1) },
722 ];
723 let month = fold_month("2026-10", &groups, 20_000_000, 20);
724 assert_eq!(month.plans_micros, 20_000_000);
725 assert_eq!(month.paid_micros, 40_000_000);
726 assert_eq!(month.charged_micros, 12_000_000);
727 assert_eq!(month.cost_micros, 17_500_000);
728 // $6 of internal use at price, $3 of pools, $15 of goodwill.
729 assert_eq!(month.given_micros, 6_000_000 + 3_000_000 + 15_000_000);
730 }
731
732 #[test]
733 fn six_months_end_with_this_one() {
734 assert_eq!(last_months("2026-02", 3), vec!["2025-12", "2026-01", "2026-02"]);
735 }
736
737 #[test]
738 fn growth_is_judged_on_pace_not_on_the_month_so_far() {
739 // Ten days in, $20 on a 30-day month is a $60 pace against $30.
740 assert!(growing(20_000_000, 30_000_000, 10, 30));
741 assert!(!growing(10_000_000, 30_000_000, 10, 30));
742 // Too small last month to say.
743 assert!(!growing(9_000_000, 1_000_000, 10, 30));
744 }
745
746 #[test]
747 fn the_most_urgent_comes_first() {
748 assert!(urgency(SignalKind::AtLimit) < urgency(SignalKind::NearCeiling));
749 assert!(urgency(SignalKind::Declined) < urgency(SignalKind::Growing));
750 }
751
752 #[test]
753 fn february_knows_its_leap_years() {
754 assert_eq!(days_in("2028-02"), 29);
755 assert_eq!(days_in("2026-02"), 28);
756 assert_eq!(days_in("2026-10"), 31);
757 }
758}