hecks 2.8.0 → 3.0.0

data/rust/host/src/lambda_client.rs

@@ -0,0 +1,378 @@
+//! Delivers a `PendingCrossDomainReaction` to another domain's Lambda function,
+//! mirroring `Adapters::Lambda::Client`'s invoke/dispatch shape.
+
+use serde_json::Value;
+
+// Decoupled from `aws_sdk_lambda`'s own response type so a mock never needs one.
+pub struct InvokeOutcome {
+    pub body: Value,
+    pub function_error: bool,
+}
+
+// `&dyn`, not a generic — the call chain reaching this is already deep, and making
+// every function in it generic over `L: LambdaInvoker` would be a much bigger diff.
+#[async_trait::async_trait]
+pub trait LambdaInvoker: Send + Sync {
+    async fn invoke(&self, function_name: &str, payload: &str) -> anyhow::Result<InvokeOutcome>;
+}
+
+// Mirrors Ruby's `"hecks-#{domain.downcase}"` — deployed stack names depend on
+// this exact casing.
+pub fn function_name_for(domain: &str) -> String {
+    format!("hecks-{}", domain.to_lowercase())
+}
+
+// Mirrors `PolicyInterpreter#deliver`'s record shape, minus `on:` — this crate
+// never sees the triggering event's own name, only its payload.
+#[derive(Debug)]
+pub struct CrossDomainDeliveryRecord {
+    pub policy: String,
+    pub target_domain: String,
+    pub target_verb: String,
+    pub delivered: bool,
+    pub reason: Option<String>,
+}
+
+impl CrossDomainDeliveryRecord {
+    pub fn to_json(&self) -> Value {
+        serde_json::json!({
+            "policy": self.policy,
+            "target_domain": self.target_domain,
+            "target_verb": self.target_verb,
+            "delivered": self.delivered,
+            "reason": self.reason,
+        })
+    }
+}
+
+// A target-side refusal returns `delivered: false`; a genuine invoke fault
+// (`function_error`, or a hard `Err` from `invoke`) returns `Err` instead.
+pub async fn deliver<L: LambdaInvoker + ?Sized>(invoker: &L, reaction: &Value) -> anyhow::Result<CrossDomainDeliveryRecord> {
+    let policy = reaction.get("policy").and_then(Value::as_str).unwrap_or_default().to_string();
+    let target_domain = reaction.get("target_domain").and_then(Value::as_str).unwrap_or_default().to_string();
+    let target_verb = reaction.get("target_verb").and_then(Value::as_str).unwrap_or_default().to_string();
+    let payload = reaction.get("payload").cloned().unwrap_or_else(|| serde_json::json!({}));
+
+    let function_name = function_name_for(&target_domain);
+    // No `"role"` — a policy reaction is system-triggered, the same reasoning
+    // `orchestrate.rs`'s local branch already applies via `Caller.without`.
+    let request_payload = serde_json::json!({ "verb": target_verb, "args": payload }).to_string();
+
+    let outcome = invoker.invoke(&function_name, &request_payload).await?;
+
+    if outcome.function_error {
+        let message = outcome.body.get("errorMessage").and_then(Value::as_str).map(str::to_string).unwrap_or_else(|| outcome.body.to_string());
+        anyhow::bail!("Lambda {function_name} ({}): {message}", target_verb);
+    }
+
+    let target_side_refusal = outcome
+        .body
+        .get("refusals")
+        .and_then(Value::as_array)
+        .and_then(|refusals| refusals.iter().find(|r| r.get("verb").and_then(Value::as_str) == Some(target_verb.as_str())));
+
+    let Some(refusal) = target_side_refusal else {
+        return Ok(CrossDomainDeliveryRecord { policy, target_domain, target_verb, delivered: true, reason: None });
+    };
+
+    Ok(CrossDomainDeliveryRecord {
+        policy,
+        target_domain,
+        target_verb,
+        delivered: false,
+        reason: refusal.get("error").and_then(Value::as_str).map(String::from),
+    })
+}
+
+// Kept small — shares the invoking Lambda's own tight execution-time budget
+// with every other cross-domain reaction still queued behind it.
+pub const MAX_DELIVERY_ATTEMPTS: u32 = 3;
+
+// Carries what `journal::record_dead_letter` needs; this file stays free of
+// Postgres itself.
+#[derive(Debug)]
+pub struct DeliveryFailure {
+    pub policy: String,
+    pub target_domain: String,
+    pub target_verb: String,
+    pub payload: Value,
+    pub error: anyhow::Error,
+    pub attempts: u32,
+}
+
+// Retries only a genuine invoke fault (`Err`); a clean `Ok(delivered: false)`
+// refusal is a business outcome and is never retried.
+pub async fn deliver_with_retry<L: LambdaInvoker + ?Sized>(
+    invoker: &L,
+    reaction: &Value,
+) -> Result<CrossDomainDeliveryRecord, DeliveryFailure> {
+    let mut last_error = None;
+
+    for attempt in 1..=MAX_DELIVERY_ATTEMPTS {
+        match deliver(invoker, reaction).await {
+            Ok(record) => return Ok(record),
+            Err(error) => {
+                last_error = Some(error);
+                if attempt < MAX_DELIVERY_ATTEMPTS {
+                    let backoff_ms = 100u64 * (1 << (attempt - 1));
+                    tokio::time::sleep(std::time::Duration::from_millis(backoff_ms)).await;
+                }
+            }
+        }
+    }
+
+    Err(DeliveryFailure {
+        policy: reaction.get("policy").and_then(Value::as_str).unwrap_or_default().to_string(),
+        target_domain: reaction.get("target_domain").and_then(Value::as_str).unwrap_or_default().to_string(),
+        target_verb: reaction.get("target_verb").and_then(Value::as_str).unwrap_or_default().to_string(),
+        payload: reaction.get("payload").cloned().unwrap_or_else(|| serde_json::json!({})),
+        // Safe: this arm only runs after the loop above executed at least once,
+        // and every iteration's `Err` arm sets `last_error` before falling through.
+        error: last_error.unwrap(),
+        attempts: MAX_DELIVERY_ATTEMPTS,
+    })
+}
+
+// Untestable here — needs a real deployed Lambda and live AWS credentials,
+// neither of which this repository's test environment has.
+pub struct AwsLambdaInvoker {
+    client: aws_sdk_lambda::Client,
+}
+
+impl AwsLambdaInvoker {
+    // Explicit rustls+ring HTTP client — neither crate's own default feature
+    // actually wires in `ring` (see Cargo.toml's note on this dependency).
+    pub async fn from_env() -> Self {
+        let http_client = aws_smithy_http_client::Builder::new()
+            .tls_provider(aws_smithy_http_client::tls::Provider::Rustls(aws_smithy_http_client::tls::rustls_provider::CryptoMode::Ring))
+            .build_https();
+        let config = aws_config::defaults(aws_config::BehaviorVersion::latest()).http_client(http_client).load().await;
+        Self { client: aws_sdk_lambda::Client::new(&config) }
+    }
+}
+
+// Panics if invoked — asserts that same-domain-only test suites never
+// trigger a cross-domain delivery.
+#[cfg(test)]
+pub struct NeverInvoker;
+
+#[cfg(test)]
+#[async_trait::async_trait]
+impl LambdaInvoker for NeverInvoker {
+    async fn invoke(&self, function_name: &str, _payload: &str) -> anyhow::Result<InvokeOutcome> {
+        unreachable!("NeverInvoker: unexpected cross-domain Lambda invoke attempted against {function_name:?}")
+    }
+}
+
+#[async_trait::async_trait]
+impl LambdaInvoker for AwsLambdaInvoker {
+    async fn invoke(&self, function_name: &str, payload: &str) -> anyhow::Result<InvokeOutcome> {
+        let response = self
+            .client
+            .invoke()
+            .function_name(function_name)
+            .payload(aws_sdk_lambda::primitives::Blob::new(payload.as_bytes()))
+            .send()
+            .await
+            .map_err(|e| anyhow::anyhow!("invoking Lambda {function_name}: {e:?}"))?;
+
+        let function_error = response.function_error().is_some();
+        let bytes = response.payload().map(|blob| blob.as_ref().to_vec()).unwrap_or_default();
+        // A non-JSON/empty body reads as `{}` rather than failing the call;
+        // `function_error` (checked above) is what actually signals a fault.
+        let body: Value = serde_json::from_slice(&bytes).unwrap_or_else(|_| serde_json::json!({}));
+
+        Ok(InvokeOutcome { body, function_error })
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use std::sync::Mutex;
+
+    #[test]
+    fn function_name_matches_rubys_own_hecks_dash_domain_convention() {
+        assert_eq!(function_name_for("Compliance"), "hecks-compliance");
+        assert_eq!(function_name_for("Notifications"), "hecks-notifications");
+        // Ruby's `.downcase`, not snake_case — an internally-capitalized domain
+        // stays one lowercase word.
+        assert_eq!(function_name_for("Banking"), "hecks-banking");
+    }
+
+    // Records each call and answers with whatever canned outcome the test configured.
+    struct MockLambdaInvoker {
+        response: Result<InvokeOutcome, String>,
+        calls: Mutex<Vec<(String, String)>>,
+    }
+
+    impl MockLambdaInvoker {
+        fn answering(body: Value, function_error: bool) -> Self {
+            Self { response: Ok(InvokeOutcome { body, function_error }), calls: Mutex::new(Vec::new()) }
+        }
+
+        fn failing(message: &str) -> Self {
+            Self { response: Err(message.to_string()), calls: Mutex::new(Vec::new()) }
+        }
+    }
+
+    #[async_trait::async_trait]
+    impl LambdaInvoker for MockLambdaInvoker {
+        async fn invoke(&self, function_name: &str, payload: &str) -> anyhow::Result<InvokeOutcome> {
+            self.calls.lock().unwrap().push((function_name.to_string(), payload.to_string()));
+            match &self.response {
+                Ok(outcome) => Ok(InvokeOutcome { body: outcome.body.clone(), function_error: outcome.function_error }),
+                Err(message) => anyhow::bail!("{message}"),
+            }
+        }
+    }
+
+    fn reaction(target_domain: &str, target_verb: &str) -> Value {
+        serde_json::json!({
+            "policy": "ReviewOnFreeze",
+            "target_domain": target_domain,
+            "target_verb": target_verb,
+            "payload": { "number": { "value": "acct-1" } },
+        })
+    }
+
+    #[tokio::test]
+    async fn delivers_and_invokes_the_computed_function_name_with_the_verb_args_payload_shape() {
+        let invoker = MockLambdaInvoker::answering(serde_json::json!({ "refusals": [] }), false);
+
+        let record = deliver(&invoker, &reaction("Compliance", "Compliance::AccountFreezeReview.Open")).await.unwrap();
+
+        assert!(record.delivered);
+        assert_eq!(record.reason, None);
+        assert_eq!(record.target_domain, "Compliance");
+
+        let calls = invoker.calls.lock().unwrap();
+        assert_eq!(calls.len(), 1);
+        let (function_name, payload) = &calls[0];
+        assert_eq!(function_name, "hecks-compliance");
+        let sent: Value = serde_json::from_str(payload).unwrap();
+        assert_eq!(sent["verb"], "Compliance::AccountFreezeReview.Open");
+        assert_eq!(sent["args"]["number"]["value"], "acct-1");
+    }
+
+    #[tokio::test]
+    async fn a_target_side_refusal_is_recorded_non_fatally_not_raised() {
+        let invoker = MockLambdaInvoker::answering(
+            serde_json::json!({ "refusals": [ { "verb": "Compliance::AccountFreezeReview.Open", "error": "already under review" } ] }),
+            false,
+        );
+
+        let record = deliver(&invoker, &reaction("Compliance", "Compliance::AccountFreezeReview.Open")).await.unwrap();
+
+        assert!(!record.delivered);
+        assert_eq!(record.reason.as_deref(), Some("already under review"));
+    }
+
+    #[tokio::test]
+    async fn a_refusal_for_a_different_verb_in_the_same_response_does_not_count_as_this_ones() {
+        // A response can legitimately carry refusals from other steps; only a
+        // refusal naming this verb counts as this delivery's outcome.
+        let invoker = MockLambdaInvoker::answering(
+            serde_json::json!({ "refusals": [ { "verb": "Compliance.SomethingElse", "error": "unrelated" } ] }),
+            false,
+        );
+
+        let record = deliver(&invoker, &reaction("Compliance", "Compliance::AccountFreezeReview.Open")).await.unwrap();
+
+        assert!(record.delivered);
+    }
+
+    #[tokio::test]
+    async fn a_function_error_response_is_a_hard_failure_not_a_delivery_record() {
+        // Mirrors Ruby's `raise WiringError` on `function_error` — not a domain
+        // refusal, so it propagates rather than becoming a `delivered: false` record.
+        let invoker = MockLambdaInvoker::answering(serde_json::json!({ "errorMessage": "unhandled panic" }), true);
+
+        let outcome = deliver(&invoker, &reaction("Compliance", "Compliance::AccountFreezeReview.Open")).await;
+
+        assert!(outcome.is_err(), "a function_error response must propagate as Err, not a delivered:false record");
+    }
+
+    #[tokio::test]
+    async fn an_invoke_that_never_reaches_the_target_at_all_is_also_a_hard_failure() {
+        // The function doesn't exist, or the call never completes (network,
+        // throttling, AccessDenied) — never even reaches `Ok(InvokeOutcome)`.
+        let invoker = MockLambdaInvoker::failing("ResourceNotFoundException: function not found");
+
+        let outcome = deliver(&invoker, &reaction("Notifications", "Notifications.Send")).await;
+
+        assert!(outcome.is_err());
+    }
+
+    // Fails the first `fail_count` calls, then succeeds — the shape of a
+    // transient throttle that clears before a retry limit is hit.
+    struct FlakyLambdaInvoker {
+        fail_count: std::sync::atomic::AtomicU32,
+        calls: Mutex<Vec<String>>,
+    }
+
+    impl FlakyLambdaInvoker {
+        fn failing_then_succeeding(fail_count: u32) -> Self {
+            Self { fail_count: std::sync::atomic::AtomicU32::new(fail_count), calls: Mutex::new(Vec::new()) }
+        }
+    }
+
+    #[async_trait::async_trait]
+    impl LambdaInvoker for FlakyLambdaInvoker {
+        async fn invoke(&self, function_name: &str, _payload: &str) -> anyhow::Result<InvokeOutcome> {
+            self.calls.lock().unwrap().push(function_name.to_string());
+            let remaining = self.fail_count.load(std::sync::atomic::Ordering::SeqCst);
+            if remaining > 0 {
+                self.fail_count.store(remaining - 1, std::sync::atomic::Ordering::SeqCst);
+                anyhow::bail!("ThrottlingException: rate exceeded");
+            }
+            Ok(InvokeOutcome { body: serde_json::json!({ "refusals": [] }), function_error: false })
+        }
+    }
+
+    #[tokio::test]
+    async fn deliver_with_retry_rides_out_a_transient_fault_that_clears_before_attempts_run_out() {
+        let invoker = FlakyLambdaInvoker::failing_then_succeeding(MAX_DELIVERY_ATTEMPTS - 1);
+
+        let record = deliver_with_retry(&invoker, &reaction("Compliance", "Compliance::AccountFreezeReview.Open"))
+            .await
+            .expect("should succeed once the fault clears, within MAX_DELIVERY_ATTEMPTS");
+
+        assert!(record.delivered);
+        assert_eq!(invoker.calls.lock().unwrap().len(), MAX_DELIVERY_ATTEMPTS as usize, "should have retried exactly up to the successful attempt");
+    }
+
+    #[tokio::test]
+    async fn deliver_with_retry_never_retries_a_clean_domain_side_refusal() {
+        // A clean refusal is `Ok(delivered: false)`, a business decision — retrying
+        // it would not change what the target domain decided.
+        let invoker = MockLambdaInvoker::answering(
+            serde_json::json!({ "refusals": [ { "verb": "Compliance::AccountFreezeReview.Open", "error": "already under review" } ] }),
+            false,
+        );
+
+        let record = deliver_with_retry(&invoker, &reaction("Compliance", "Compliance::AccountFreezeReview.Open"))
+            .await
+            .expect("a domain-side refusal is Ok, not a DeliveryFailure");
+
+        assert!(!record.delivered);
+        assert_eq!(invoker.calls.lock().unwrap().len(), 1, "a clean refusal should never be retried");
+    }
+
+    #[tokio::test]
+    async fn deliver_with_retry_gives_up_after_max_attempts_and_carries_everything_a_dead_letter_needs() {
+        let invoker = MockLambdaInvoker::failing("ResourceNotFoundException: function not found");
+
+        let failure = deliver_with_retry(&invoker, &reaction("Notifications", "Notifications.Send"))
+            .await
+            .expect_err("a persistent fault should exhaust every attempt and fail");
+
+        assert_eq!(failure.policy, "ReviewOnFreeze");
+        assert_eq!(failure.target_domain, "Notifications");
+        assert_eq!(failure.target_verb, "Notifications.Send");
+        assert_eq!(failure.payload, serde_json::json!({ "number": { "value": "acct-1" } }));
+        assert_eq!(failure.attempts, MAX_DELIVERY_ATTEMPTS);
+        assert!(format!("{:#}", failure.error).contains("ResourceNotFoundException"));
+        assert_eq!(invoker.calls.lock().unwrap().len(), MAX_DELIVERY_ATTEMPTS as usize);
+    }
+}

data/rust/host/src/log.rs

@@ -0,0 +1,147 @@
+// Structured logging: one JSON object per line on stdout, shipped by the `awslogs` driver.
+// Never put secrets, cookies, query strings or command facts in `fields`; log identifiers only.
+
+use serde_json::{json, Map, Value};
+use std::io::Write;
+
+/// Logs an ordinary event.
+pub fn info(msg: &str, fields: Value) {
+    emit("info", msg, fields);
+}
+
+/// Logs a failure worth an operator's attention.
+pub fn error(msg: &str, fields: Value) {
+    emit("error", msg, fields);
+}
+
+/// A boot phase in flight: logs `boot_phase` start and end lines, so a hang shows as a start
+/// with no end. A failed phase never logs its end.
+pub struct Phase {
+    name: &'static str,
+    fields: Value,
+    started: std::time::Instant,
+}
+
+/// Starts a boot phase with no extra fields.
+pub fn phase(name: &'static str) -> Phase {
+    phase_with(name, json!({}))
+}
+
+/// Starts a boot phase and logs its `start` line.
+pub fn phase_with(name: &'static str, fields: Value) -> Phase {
+    emit("info", "boot_phase", phase_fields(name, "start", &fields, None));
+    Phase { name, fields, started: std::time::Instant::now() }
+}
+
+impl Phase {
+    /// Logs the `end` line and returns the elapsed milliseconds.
+    pub fn end(self) -> u64 {
+        self.end_with(json!({}))
+    }
+
+    /// Logs the `end` line with extra fields describing the outcome.
+    pub fn end_with(self, outcome: Value) -> u64 {
+        let elapsed_ms = self.started.elapsed().as_millis() as u64;
+        let mut fields = self.fields.clone();
+        if let (Value::Object(base), Value::Object(extra)) = (&mut fields, outcome) {
+            base.extend(extra);
+        }
+        emit("info", "boot_phase", phase_fields(self.name, "end", &fields, Some(elapsed_ms)));
+        elapsed_ms
+    }
+}
+
+/// The fields of one phase line: `phase`, `event`, `elapsed_ms` on an end line,
+/// then the phase's own identifiers.
+fn phase_fields(name: &str, event: &str, fields: &Value, elapsed_ms: Option<u64>) -> Value {
+    let mut object = Map::new();
+    object.insert("phase".to_string(), json!(name));
+    object.insert("event".to_string(), json!(event));
+    if let Some(ms) = elapsed_ms {
+        object.insert("elapsed_ms".to_string(), json!(ms));
+    }
+    if let Value::Object(fields) = fields {
+        for (key, value) in fields {
+            object.entry(key.clone()).or_insert(value.clone());
+        }
+    }
+    Value::Object(object)
+}
+
+fn emit(level: &str, msg: &str, fields: Value) {
+    let line = render(level, msg, fields);
+    // A closed stdout must never take a request down with it.
+    let _ = writeln!(std::io::stdout().lock(), "{line}");
+}
+
+/// The line as text — `level` and `msg` first, then `fields` (an object;
+/// anything else is ignored).
+fn render(level: &str, msg: &str, fields: Value) -> String {
+    let mut object = Map::new();
+    object.insert("level".to_string(), json!(level));
+    object.insert("msg".to_string(), json!(msg));
+    if let Value::Object(fields) = fields {
+        for (key, value) in fields {
+            object.entry(key).or_insert(value);
+        }
+    }
+    Value::Object(object).to_string()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn renders_one_json_object_on_a_single_line() {
+        let line = render("info", "request", json!({"method": "GET", "status": 200}));
+        assert!(!line.contains('\n'));
+        let parsed: Value = serde_json::from_str(&line).unwrap();
+        assert_eq!(parsed["level"], "info");
+        assert_eq!(parsed["msg"], "request");
+        assert_eq!(parsed["method"], "GET");
+        assert_eq!(parsed["status"], 200);
+    }
+
+    #[test]
+    fn a_field_cannot_overwrite_level_or_msg() {
+        let line = render("error", "boom", json!({"level": "info", "msg": "fine"}));
+        let parsed: Value = serde_json::from_str(&line).unwrap();
+        assert_eq!(parsed["level"], "error");
+        assert_eq!(parsed["msg"], "boom");
+    }
+
+    #[test]
+    fn a_phase_line_names_the_phase_and_an_end_line_carries_elapsed_ms() {
+        let start = render("info", "boot_phase", phase_fields("audit_aggregate", "start", &json!({"aggregate": "registration"}), None));
+        let parsed: Value = serde_json::from_str(&start).unwrap();
+        assert_eq!(parsed["msg"], "boot_phase");
+        assert_eq!((parsed["phase"].as_str(), parsed["event"].as_str()), (Some("audit_aggregate"), Some("start")));
+        assert_eq!(parsed["aggregate"], "registration");
+        assert!(parsed.get("elapsed_ms").is_none(), "a start line has no duration yet");
+
+        let end = render("info", "boot_phase", phase_fields("audit_aggregate", "end", &json!({"aggregate": "registration"}), Some(42)));
+        let parsed: Value = serde_json::from_str(&end).unwrap();
+        assert_eq!((parsed["event"].as_str(), parsed["elapsed_ms"].as_u64()), (Some("end"), Some(42)));
+    }
+
+    #[test]
+    fn a_phase_field_cannot_overwrite_the_phase_or_event() {
+        let fields = phase_fields("connect", "start", &json!({"phase": "other", "event": "end", "era": 7}), None);
+        assert_eq!((fields["phase"].as_str(), fields["event"].as_str(), fields["era"].as_u64()), (Some("connect"), Some("start"), Some(7)));
+    }
+
+    #[test]
+    fn ending_a_phase_reports_elapsed_time_and_merges_the_outcome() {
+        let phase = phase_with("mint_era", json!({"era": 7}));
+        std::thread::sleep(std::time::Duration::from_millis(15));
+        let elapsed = phase.end_with(json!({"aggregates": 5}));
+        assert!(elapsed >= 15, "{elapsed}");
+    }
+
+    #[test]
+    fn newlines_inside_a_value_stay_escaped() {
+        let line = render("error", "boom", json!({"error": "a\nb"}));
+        assert!(!line.contains('\n'));
+    }
+}