| 1 |
|
| 2 |
|
| 3 |
|
| 4 |
|
| 5 |
|
| 6 |
|
| 7 |
|
| 8 |
use axum::{ |
| 9 |
body::Bytes, |
| 10 |
extract::State, |
| 11 |
http::{StatusCode, header::HeaderMap}, |
| 12 |
response::IntoResponse, |
| 13 |
}; |
| 14 |
use sqlx::PgPool; |
| 15 |
|
| 16 |
use crate::{ |
| 17 |
Billing, Integrations, db, |
| 18 |
error::{AppError, Result}, |
| 19 |
payments::{self, ThinEvent}, |
| 20 |
wam_client::WamClient, |
| 21 |
}; |
| 22 |
|
| 23 |
|
| 24 |
#[tracing::instrument(skip_all, name = "stripe::webhook_v2")] |
| 25 |
pub(super) async fn webhook_v2( |
| 26 |
State(db): State<PgPool>, |
| 27 |
State(integrations): State<Integrations>, |
| 28 |
State(payments): State<Billing>, |
| 29 |
State(config): State<crate::config::Config>, |
| 30 |
headers: HeaderMap, |
| 31 |
body: Bytes, |
| 32 |
) -> Result<impl IntoResponse> { |
| 33 |
let stripe = payments |
| 34 |
.stripe |
| 35 |
.as_ref() |
| 36 |
.ok_or_else(|| AppError::BadRequest("Stripe is not configured".to_string()))?; |
| 37 |
|
| 38 |
let signature = headers |
| 39 |
.get("stripe-signature") |
| 40 |
.and_then(|v| v.to_str().ok()) |
| 41 |
.ok_or_else(|| AppError::BadRequest("Missing Stripe signature".to_string()))?; |
| 42 |
|
| 43 |
let payload = std::str::from_utf8(&body) |
| 44 |
.map_err(|_| AppError::BadRequest("Invalid payload encoding".to_string()))?; |
| 45 |
|
| 46 |
|
| 47 |
|
| 48 |
|
| 49 |
let body_json = stripe |
| 50 |
.verify_webhook_v2(payload, signature) |
| 51 |
.inspect_err(|_| { |
| 52 |
crate::security_signals::note_webhook_signature_failure("stripe"); |
| 53 |
})?; |
| 54 |
|
| 55 |
|
| 56 |
let thin: ThinEvent = serde_json::from_value(body_json).map_err(|e| { |
| 57 |
tracing::warn!(error = ?e, "failed to parse v2 thin event"); |
| 58 |
AppError::BadRequest("Invalid v2 event format".to_string()) |
| 59 |
})?; |
| 60 |
|
| 61 |
tracing::info!(event_type = %thin.event_type, event_id = %thin.id, "received v2 thin event"); |
| 62 |
|
| 63 |
|
| 64 |
|
| 65 |
|
| 66 |
|
| 67 |
|
| 68 |
|
| 69 |
let _event_lock = match db::webhook_events::try_lock_event(&db, &thin.id).await { |
| 70 |
Ok(Some(tx)) => tx, |
| 71 |
Ok(None) => { |
| 72 |
tracing::info!(event_id = %thin.id, "concurrent delivery of this v2 event is in flight; returning 503 for redelivery"); |
| 73 |
return Ok(StatusCode::SERVICE_UNAVAILABLE); |
| 74 |
} |
| 75 |
Err(e) => { |
| 76 |
tracing::error!(event_id = %thin.id, error = ?e, "failed to acquire v2 webhook event lock, returning 503 for retry"); |
| 77 |
return Ok(StatusCode::SERVICE_UNAVAILABLE); |
| 78 |
} |
| 79 |
}; |
| 80 |
|
| 81 |
|
| 82 |
|
| 83 |
|
| 84 |
|
| 85 |
match db::webhook_events::is_event_processed(&db, &thin.id).await { |
| 86 |
Ok(true) => { |
| 87 |
tracing::debug!(event_id = %thin.id, "v2 event already processed, skipping"); |
| 88 |
return Ok(StatusCode::OK); |
| 89 |
} |
| 90 |
Err(e) => { |
| 91 |
|
| 92 |
tracing::error!(event_id = %thin.id, error = ?e, "v2 dedup check failed, returning 503 for retry"); |
| 93 |
return Ok(StatusCode::SERVICE_UNAVAILABLE); |
| 94 |
} |
| 95 |
Ok(false) => {} |
| 96 |
} |
| 97 |
|
| 98 |
if let Err(e) = process_v2_thin_event( |
| 99 |
&db, |
| 100 |
integrations.wam.as_ref(), |
| 101 |
stripe.as_ref(), |
| 102 |
&config.signing_secret, |
| 103 |
&thin, |
| 104 |
) |
| 105 |
.await |
| 106 |
{ |
| 107 |
|
| 108 |
|
| 109 |
|
| 110 |
|
| 111 |
|
| 112 |
|
| 113 |
|
| 114 |
|
| 115 |
|
| 116 |
|
| 117 |
|
| 118 |
|
| 119 |
|
| 120 |
|
| 121 |
tracing::warn!(event_id = %thin.id, error = ?e, "v2 event processing failed; queueing for retry"); |
| 122 |
if let Err(queue_err) = db::webhook_events::insert_failed_event( |
| 123 |
&db, |
| 124 |
"stripe_v2", |
| 125 |
&thin.event_type, |
| 126 |
payload, |
| 127 |
Some(signature), |
| 128 |
&format!("{e:?}"), |
| 129 |
) |
| 130 |
.await |
| 131 |
{ |
| 132 |
tracing::error!(event_id = %thin.id, error = ?queue_err, "failed to queue v2 event for retry; returning 503 for Stripe redelivery"); |
| 133 |
return Ok(StatusCode::SERVICE_UNAVAILABLE); |
| 134 |
} |
| 135 |
return Ok(StatusCode::OK); |
| 136 |
} |
| 137 |
|
| 138 |
|
| 139 |
if let Err(e) = db::webhook_events::mark_event_processed(&db, &thin.id).await { |
| 140 |
tracing::error!(event_id = %thin.id, error = ?e, "failed to record processed v2 event; returning 503 for redelivery"); |
| 141 |
return Ok(StatusCode::SERVICE_UNAVAILABLE); |
| 142 |
} |
| 143 |
|
| 144 |
Ok(StatusCode::OK) |
| 145 |
} |
| 146 |
|
| 147 |
|
| 148 |
|
| 149 |
|
| 150 |
pub(crate) async fn process_v2_thin_event( |
| 151 |
db: &PgPool, |
| 152 |
wam: Option<&WamClient>, |
| 153 |
stripe: &dyn payments::PaymentProvider, |
| 154 |
signing_secret: &str, |
| 155 |
thin: &ThinEvent, |
| 156 |
) -> Result<()> { |
| 157 |
if thin.event_type.starts_with("v2.core.account") { |
| 158 |
handle_account_thin_event(db, wam, stripe, signing_secret, thin).await |
| 159 |
} else { |
| 160 |
tracing::debug!(event_type = %thin.event_type, "unhandled v2 event type"); |
| 161 |
Ok(()) |
| 162 |
} |
| 163 |
} |
| 164 |
|
| 165 |
|
| 166 |
async fn handle_account_thin_event( |
| 167 |
db: &PgPool, |
| 168 |
wam: Option<&WamClient>, |
| 169 |
stripe: &dyn payments::PaymentProvider, |
| 170 |
signing_secret: &str, |
| 171 |
thin: &ThinEvent, |
| 172 |
) -> Result<()> { |
| 173 |
let account_id = match &thin.related_object { |
| 174 |
Some(obj) => &obj.id, |
| 175 |
None => { |
| 176 |
tracing::warn!(event_id = %thin.id, "v2 account event missing related_object"); |
| 177 |
return Ok(()); |
| 178 |
} |
| 179 |
}; |
| 180 |
|
| 181 |
let update = stripe.fetch_account(account_id).await.map_err(|e| { |
| 182 |
tracing::warn!(account_id = %account_id, error = ?e, "failed to fetch account for v2 event"); |
| 183 |
e |
| 184 |
})?; |
| 185 |
|
| 186 |
super::webhook::handle_account_updated_from_v2(db, wam, signing_secret, &update).await.map_err(|e| { |
| 187 |
tracing::warn!(account_id = %account_id, error = ?e, "failed to process account update from v2 event"); |
| 188 |
e |
| 189 |
})?; |
| 190 |
|
| 191 |
Ok(()) |
| 192 |
} |
| 193 |
|