qefro_backend_sdk/

lib.rs

//! Qefro backend SDK — mirrors `@qefro-ai/backend` (TypeScript).
//!
//! Organizations expose one signed webhook (typically `POST /qefro`).
//! Qefro Runtime calls `ping`, `tools.list`, `tool.invoke`, and `tool.resume`.

mod customer_hub;

pub use customer_hub::{
    env_flag_true, hub_call, hub_customer_from_person, is_customer_hub_enabled,
    is_customer_hub_optional, pick_identity, read_identity_phone, seed_from_person,
    ConsentContext, CustomerState, MembershipContext, PlatformCapabilities,
    PlatformCustomerBinding, PlatformCustomerContext, PlatformStorageBinding,
    PlatformStorageContext, TimelineContext,
};

use std::collections::HashMap;
use std::future::Future;
use std::pin::Pin;
use std::sync::{Arc, RwLock};

use anyhow::{anyhow, Result};
use async_trait::async_trait;
use axum::body::Bytes;
use axum::extract::State;
use axum::http::{HeaderMap, StatusCode};
use axum::response::IntoResponse;
use axum::routing::post;
use axum::{Json, Router};
use chrono::Utc;
use hmac::{Hmac, Mac};
use serde::{Deserialize, Serialize};
use serde_json::{json, Value};
use sha2::Sha256;
use subtle::ConstantTimeEq;
use tokio::sync::Mutex;
use uuid::Uuid;

type HmacSha256 = Hmac<Sha256>;

/// Package name reported to Qefro Runtime (`X-Qefro-SDK` / protocol payloads).
pub const SDK_NAME: &str = "qefro-backend-sdk";
/// Package version reported to Qefro Runtime (`sdk_version` / `X-Qefro-Version`).
pub const SDK_VERSION: &str = env!("CARGO_PKG_VERSION");

type ToolHandler = Arc<dyn Fn(ToolContext) -> ToolFuture + Send + Sync>;
type ToolFuture = Pin<Box<dyn Future<Output = Result<Value>> + Send>>;
type BeforeHook = Arc<dyn Fn(ToolContext) -> HookFuture + Send + Sync>;
type AfterHook = Arc<dyn Fn(ToolContext, Value) -> AfterFuture + Send + Sync>;
type HookFuture = Pin<Box<dyn Future<Output = Result<()>> + Send>>;
type AfterFuture = Pin<Box<dyn Future<Output = Result<Value>> + Send>>;
type MiddlewareFn = Arc<
    dyn Fn(ToolContext, NextFn) -> Pin<Box<dyn Future<Output = Result<Value>> + Send>>
        + Send
        + Sync,
>;
type NextFn = Box<dyn FnOnce(ToolContext) -> Pin<Box<dyn Future<Output = Result<Value>> + Send>> + Send>;

#[derive(Debug, Clone)]
pub struct QefroConfig {
    pub signing_secret: String,
    pub protocol_version: String,
    pub max_timestamp_skew_secs: i64,
    pub endpoint_path: String,
}

impl QefroConfig {
    pub fn new(signing_secret: impl Into<String>) -> Self {
        Self {
            signing_secret: signing_secret.into(),
            protocol_version: "1".into(),
            max_timestamp_skew_secs: 300,
            endpoint_path: "/qefro".into(),
        }
    }
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, Default)]
#[serde(rename_all = "snake_case")]
pub enum ToolAuthMode {
    None,
    #[default]
    Optional,
    Required,
}

/// Identity attributes the Qefro runtime must resolve before `tool.invoke`.
#[derive(Debug, Clone, Serialize, Deserialize, Default, PartialEq, Eq)]
pub struct ToolLookup {
    /// Shorthand for a single required attribute, e.g. `"email"` or `"phone"`.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub by: Option<String>,
    /// Explicit list, e.g. `["email"]` or `["phone", "customer_id"]`.
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub required: Vec<String>,
}

/// Normalize `lookup.by` / `lookup.required` into a deduped lowercase attribute list.
pub fn normalize_lookup(lookup: Option<&ToolLookup>) -> Vec<String> {
    let Some(lookup) = lookup else {
        return Vec::new();
    };
    let mut seen = std::collections::HashSet::new();
    let mut out = Vec::new();
    for item in lookup
        .required
        .iter()
        .cloned()
        .chain(lookup.by.iter().cloned())
    {
        let key = item.trim().to_ascii_lowercase();
        if key.is_empty() || !seen.insert(key.clone()) {
            continue;
        }
        out.push(key);
    }
    out
}

fn normalized_lookup_field(lookup: Option<&ToolLookup>) -> Option<ToolLookup> {
    let attrs = normalize_lookup(lookup);
    if attrs.is_empty() {
        None
    } else {
        Some(ToolLookup {
            by: None,
            required: attrs,
        })
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, Default)]
pub struct ToolMetadata {
    pub name: String,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub input_schema: Option<Value>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub authentication_methods: Vec<String>,
    #[serde(default)]
    pub auth: ToolAuthMode,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub permissions: Vec<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub timeout: Option<u64>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub default_auth_method: Option<String>,
    /// What identity the runtime must have before invoking this tool.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub lookup: Option<ToolLookup>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RegisteredTool {
    pub name: String,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub input_schema: Option<Value>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub authentication_methods: Vec<String>,
    #[serde(default)]
    pub auth: ToolAuthMode,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub permissions: Vec<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub timeout: Option<u64>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub lookup: Option<ToolLookup>,
}

// ---------------------------------------------------------------------------
// Business Flows (metadata only — the SDK advertises them, never executes them)
// ---------------------------------------------------------------------------

fn default_flow_version() -> u32 {
    1
}

/// Immutable identity + descriptive metadata for a Business Flow.
///
/// `id` is the identity key: renaming `name` never creates a new flow.
#[derive(Debug, Clone, Serialize, Deserialize, Default, PartialEq)]
pub struct BusinessFlowMetadata {
    pub id: String,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub name: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
    /// Integer flow version, defaults to 1. Bump when the definition changes.
    #[serde(default = "default_flow_version")]
    pub version: u32,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub category: Option<String>,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub tags: Vec<String>,
    /// Example utterances used by the runtime for AI flow selection.
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub intent: Vec<String>,
    /// Identity/context attributes this flow requires before it can run.
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub inputs: Vec<String>,
    /// Values this flow produces (for analytics and future flow chaining).
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub outputs: Vec<String>,
    /// Entry trigger. Conversation (default) keeps Phase 2 behaviour.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub trigger: Option<FlowTrigger>,
}

/// How a Business Flow is entered (Phase 3). Events are triggers into the
/// Qefro runtime — not a second execution engine.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum FlowTrigger {
    Conversation,
    Event {
        event: String,
        #[serde(default, skip_serializing_if = "Option::is_none")]
        when: Option<String>,
    },
    Schedule { cron: String },
    Webhook {
        #[serde(default, skip_serializing_if = "Option::is_none")]
        name: Option<String>,
        #[serde(default, skip_serializing_if = "Option::is_none")]
        when: Option<String>,
    },
}

impl FlowTrigger {
    pub fn normalize(self) -> Result<Self, FlowError> {
        match self {
            FlowTrigger::Conversation => Ok(FlowTrigger::Conversation),
            FlowTrigger::Event { event, when } => {
                let event = event.trim().to_string();
                if event.is_empty() {
                    return Err(FlowError::InvalidTrigger(
                        "trigger.type=event requires a non-empty event name".into(),
                    ));
                }
                if !event.contains('.') {
                    return Err(FlowError::InvalidTrigger(
                        "trigger.event must be namespaced (e.g. shopify.order.created)".into(),
                    ));
                }
                let when = when
                    .map(|w| w.trim().to_string())
                    .filter(|w| !w.is_empty());
                Ok(FlowTrigger::Event { event, when })
            }
            FlowTrigger::Schedule { cron } => {
                let cron = cron.trim().to_string();
                if cron.is_empty() {
                    return Err(FlowError::InvalidTrigger(
                        "trigger.type=schedule requires a non-empty cron expression".into(),
                    ));
                }
                Ok(FlowTrigger::Schedule { cron })
            }
            FlowTrigger::Webhook { name, when } => {
                let name = name
                    .map(|n| n.trim().to_string())
                    .filter(|n| !n.is_empty());
                let when = when
                    .map(|w| w.trim().to_string())
                    .filter(|w| !w.is_empty());
                Ok(FlowTrigger::Webhook { name, when })
            }
        }
    }
}

/// Standalone event / webhook / schedule handler advertised via capabilities.list.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct EventHandlerDefinition {
    pub name: String,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub cron: Option<String>,
}

/// Type-specific settings for a flow step. Serialized as `{ "type": ..., "config": {...} }`
/// so new settings (retry, timeout, permissions, parallel) extend `config` without
/// changing the wire schema.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
#[serde(tag = "type", content = "config", rename_all = "snake_case")]
pub enum FlowStepKind {
    Ask {
        field: String,
        prompt: String,
    },
    Tool {
        /// Name of an existing Business Tool. Namespaceable later (e.g. `CRM.lookup_customer`).
        tool_ref: String,
    },
    Challenge {
        #[serde(default, skip_serializing_if = "Option::is_none")]
        message: Option<String>,
    },
    Upload {
        #[serde(default, skip_serializing_if = "Option::is_none")]
        field: Option<String>,
        #[serde(default, skip_serializing_if = "Option::is_none")]
        prompt: Option<String>,
        #[serde(default, skip_serializing_if = "Vec::is_empty")]
        accept: Vec<String>,
    },
    Condition {
        when: String,
        #[serde(default, skip_serializing_if = "Option::is_none")]
        then: Option<String>,
        #[serde(rename = "else", default, skip_serializing_if = "Option::is_none")]
        else_step: Option<String>,
    },
    Delay {
        duration_seconds: u64,
    },
    Approval {
        #[serde(default, skip_serializing_if = "Option::is_none")]
        prompt: Option<String>,
    },
    Complete {
        #[serde(default, skip_serializing_if = "Option::is_none")]
        message: Option<String>,
    },
}

/// Wire shape of a flow step: `{ id, type, config }`.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct FlowStep {
    pub id: String,
    #[serde(flatten)]
    pub kind: FlowStepKind,
}

/// A Business Flow as advertised through `capabilities.list`. Never executed by the SDK.
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct BusinessFlow {
    pub metadata: BusinessFlowMetadata,
    pub steps: Vec<FlowStep>,
}

/// Developer mistakes surfaced as explicit errors — the SDK never panics on a
/// malformed flow declaration.
#[derive(Debug, Clone, PartialEq, Eq)]
pub enum FlowError {
    EmptyFlowId,
    DuplicateFlowId(String),
    EmptyStepId { flow: String },
    DuplicateStepId { flow: String, step: String },
    InvalidTrigger(String),
    EmptyHandlerName(&'static str),
    DuplicateHandler { kind: &'static str, name: String },
}

impl std::fmt::Display for FlowError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            FlowError::EmptyFlowId => write!(f, "flow() requires a non-empty metadata.id"),
            FlowError::DuplicateFlowId(id) => write!(f, "flow \"{id}\" is already registered"),
            FlowError::EmptyStepId { flow } => {
                write!(f, "flow \"{flow}\": every step requires a non-empty id")
            }
            FlowError::DuplicateStepId { flow, step } => {
                write!(f, "flow \"{flow}\": duplicate step id \"{step}\"")
            }
            FlowError::InvalidTrigger(msg) => write!(f, "{msg}"),
            FlowError::EmptyHandlerName(kind) => {
                write!(f, "{kind}() requires a non-empty name")
            }
            FlowError::DuplicateHandler { kind, name } => {
                write!(f, "{kind} \"{name}\" is already registered")
            }
        }
    }
}

impl std::error::Error for FlowError {}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AuthenticationContextPayload {
    #[serde(rename = "type", skip_serializing_if = "Option::is_none")]
    pub credential_type: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub access_token: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub credential: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub refresh_token: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub expires_in: Option<i64>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub customer_id: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ChallengePayload {
    #[serde(rename = "type")]
    pub challenge_type: String,
    pub message: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub destination_hint: Option<String>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub login_url: Option<String>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct QefroRequest {
    pub protocol_version: String,
    pub request_id: Uuid,
    #[serde(rename = "type")]
    pub request_type: String,
    pub organization_id: Option<Uuid>,
    pub conversation_id: Option<Uuid>,
    pub channel: Option<String>,
    pub identity: Option<Value>,
    pub tool: Option<String>,
    pub parameters: Option<Value>,
    pub authentication: Option<Value>,
    pub resume_token: Option<String>,
    pub challenge_response: Option<String>,
    /// Customer Hub Person snapshot from Qefro memory (not connector customer).
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub person: Option<Value>,
    /// Managed storage / Customer Hub gateway for sdk.* bindings.
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub platform: Option<PlatformCapabilities>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
#[serde(tag = "type", rename_all = "snake_case")]
pub enum QefroResponse {
    Pong {
        protocol_version: String,
        sdk_version: String,
    },
    #[serde(rename = "tools.list")]
    ToolsList {
        tools: Vec<RegisteredTool>,
        protocol_version: String,
        sdk_version: String,
    },
    #[serde(rename = "capabilities.list")]
    CapabilitiesList {
        tools: Vec<RegisteredTool>,
        flows: Vec<BusinessFlow>,
        #[serde(default, skip_serializing_if = "Vec::is_empty")]
        events: Vec<EventHandlerDefinition>,
        #[serde(default, skip_serializing_if = "Vec::is_empty")]
        webhooks: Vec<EventHandlerDefinition>,
        #[serde(default, skip_serializing_if = "Vec::is_empty")]
        schedules: Vec<EventHandlerDefinition>,
        protocol_version: String,
        sdk_version: String,
        sdk_name: String,
    },
    Result {
        output: Value,
        #[serde(skip_serializing_if = "Option::is_none")]
        authentication_context: Option<AuthenticationContextPayload>,
    },
    Challenge {
        resume_token: String,
        challenge: ChallengePayload,
    },
    Error {
        code: String,
        message: String,
    },
}

#[derive(Debug, Clone)]
struct PendingInvocation {
    tool: String,
    conversation_id: Uuid,
    parameters: Value,
    identity: Option<Value>,
    channel: Option<String>,
    platform: Option<PlatformCapabilities>,
    person: Option<Value>,
}

#[derive(Debug, Clone)]
struct StoredAuth {
    customer: Value,
    auth: AuthenticationContextPayload,
    expires_at_epoch_ms: i64,
}

#[derive(Debug, Clone)]
pub struct Conversation {
    pub id: Uuid,
}

#[derive(Clone)]
pub struct ToolContext {
    pub identity: Value,
    pub parameters: Value,
    pub conversation: Conversation,
    pub channel: Option<String>,
    pub authentication: Option<Value>,
    pub auth_response: Option<String>,
    /// Customer resolved for `auth = required` (or via in-handler authorize).
    pub customer: Option<Value>,
    customer_api: Option<CustomerApi>,
    /// Append Customer Hub timeline activities.
    pub timeline: TimelineContext,
    /// Attach/detach solution membership on a Hub customer.
    pub membership: MembershipContext,
    /// Grant/revoke consent purposes on a Hub customer.
    pub consent: ConsentContext,
    /// Platform capabilities from `tool.invoke` (`platform.customer` / storage).
    pub platform: Option<PlatformCapabilities>,
}

impl ToolContext {
    /// In-handler customer helpers (mirrors JS `ctx.customer`).
    pub fn customer_api(&self) -> Option<&CustomerApi> {
        self.customer_api.as_ref()
    }

    /// Raise an auth challenge from inside a tool handler (mirrors JS `AuthBuilder.challenge`).
    pub fn raise_challenge(challenge: ChallengePayload) -> Result<Value> {
        Err(ChallengeSignal { challenge }.into())
    }
}

/// Signal an auth challenge from a tool handler (caught like JS `ChallengeSignal`).
#[derive(Debug, Clone)]
pub struct ChallengeSignal {
    pub challenge: ChallengePayload,
}

impl std::fmt::Display for ChallengeSignal {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.challenge.message)
    }
}

impl std::error::Error for ChallengeSignal {}

#[derive(Debug, Clone)]
pub enum AuthOutcome {
    Success {
        customer: Value,
        auth: AuthenticationContextPayload,
    },
    Challenge(ChallengePayload),
    Denied,
    NotFound,
}

/// Helpers matching JS `AuthBuilder`.
#[derive(Debug, Clone)]
pub struct AuthBuilder {
    pub response: Option<String>,
}

impl AuthBuilder {
    pub fn new(response: Option<String>) -> Self {
        Self { response }
    }

    pub fn success(
        &self,
        customer: Value,
        mut auth: AuthenticationContextPayload,
    ) -> AuthOutcome {
        if auth.customer_id.is_none() {
            auth.customer_id = customer
                .get("id")
                .and_then(|v| v.as_str())
                .map(str::to_string);
        }
        AuthOutcome::Success { customer, auth }
    }

    pub fn denied(&self) -> AuthOutcome {
        AuthOutcome::Denied
    }

    pub fn not_found(&self) -> AuthOutcome {
        AuthOutcome::NotFound
    }

    pub fn email_otp(&self, email: &str, message: Option<&str>) -> AuthOutcome {
        AuthOutcome::Challenge(ChallengePayload {
            challenge_type: "email_otp".into(),
            message: message
                .unwrap_or("Enter the OTP sent to your email.")
                .into(),
            destination_hint: Some(mask(email)),
            login_url: None,
        })
    }

    pub fn sms_otp(&self, phone: &str, message: Option<&str>) -> AuthOutcome {
        AuthOutcome::Challenge(ChallengePayload {
            challenge_type: "sms_otp".into(),
            message: message
                .unwrap_or("Enter the OTP sent to your phone.")
                .into(),
            destination_hint: Some(mask(phone)),
            login_url: None,
        })
    }

    pub fn login(&self, url: &str, message: Option<&str>) -> AuthOutcome {
        AuthOutcome::Challenge(ChallengePayload {
            challenge_type: "login".into(),
            message: message
                .unwrap_or("Please continue in your login page.")
                .into(),
            destination_hint: None,
            login_url: Some(url.into()),
        })
    }

    pub fn custom(&self, challenge: ChallengePayload) -> AuthOutcome {
        AuthOutcome::Challenge(challenge)
    }
}

fn mask(value: &str) -> String {
    if value.len() <= 4 {
        return value.to_string();
    }
    format!("{}***{}", &value[..2], &value[value.len() - 2..])
}

#[derive(Debug, Clone)]
pub struct CustomerLookupContext {
    pub identity: Value,
    pub parameters: Value,
    pub conversation: Conversation,
    pub channel: Option<String>,
}

#[derive(Debug, Clone)]
pub struct CustomerAuthorizeContext {
    pub customer: Value,
    pub method: Option<String>,
    pub response: Option<String>,
    pub identity: Value,
    pub parameters: Value,
    pub conversation: Conversation,
    pub channel: Option<String>,
}

#[async_trait]
pub trait CustomerProvider: Send + Sync {
    async fn lookup(&self, ctx: &CustomerLookupContext) -> Result<Option<Value>>;
    async fn authorize(&self, ctx: &CustomerAuthorizeContext) -> Result<AuthOutcome>;
}

#[derive(Clone)]
struct ToolRegistration {
    metadata: ToolMetadata,
    handler: ToolHandler,
}

#[derive(Clone)]
struct FlowRegistration {
    metadata: BusinessFlowMetadata,
    steps: Vec<FlowStep>,
    /// First builder violation recorded for this flow; if set the flow is
    /// excluded from `capabilities.list`.
    error: Option<FlowError>,
}

/// Fluent builder returned by [`Qefro::flow`]. Step methods append into the SDK's
/// flow registry as they are declared and never panic; the first step-id
/// violation is recorded and surfaced by [`FlowBuilder::complete`].
pub struct FlowBuilder {
    inner: Arc<Inner>,
    flow_id: String,
}

impl FlowBuilder {
    pub fn ask(self, id: impl Into<String>, field: impl Into<String>, prompt: impl Into<String>) -> Self {
        self.push(
            id.into(),
            FlowStepKind::Ask {
                field: field.into(),
                prompt: prompt.into(),
            },
        )
    }

    pub fn tool(self, id: impl Into<String>, tool_ref: impl Into<String>) -> Self {
        self.push(
            id.into(),
            FlowStepKind::Tool {
                tool_ref: tool_ref.into(),
            },
        )
    }

    pub fn challenge(self, id: impl Into<String>, message: Option<String>) -> Self {
        self.push(id.into(), FlowStepKind::Challenge { message })
    }

    pub fn upload(
        self,
        id: impl Into<String>,
        field: Option<String>,
        prompt: Option<String>,
        accept: Vec<String>,
    ) -> Self {
        self.push(
            id.into(),
            FlowStepKind::Upload {
                field,
                prompt,
                accept,
            },
        )
    }

    pub fn condition(
        self,
        id: impl Into<String>,
        when: impl Into<String>,
        then: Option<String>,
        else_step: Option<String>,
    ) -> Self {
        self.push(
            id.into(),
            FlowStepKind::Condition {
                when: when.into(),
                then,
                else_step,
            },
        )
    }

    pub fn delay(self, id: impl Into<String>, duration_seconds: u64) -> Self {
        self.push(id.into(), FlowStepKind::Delay { duration_seconds })
    }

    pub fn approval(self, id: impl Into<String>, prompt: Option<String>) -> Self {
        self.push(id.into(), FlowStepKind::Approval { prompt })
    }

    /// Append a non-final `complete` step and keep building. Use this for
    /// branch terminals (e.g. a `condition` else-target) when more steps
    /// follow; finish the chain with [`FlowBuilder::complete`].
    pub fn complete_step(self, id: impl Into<String>, message: Option<String>) -> Self {
        self.push(id.into(), FlowStepKind::Complete { message })
    }

    /// Append the terminal `complete` step and surface any recorded builder error.
    #[must_use = "handle the FlowError so malformed flows fail fast at startup"]
    pub fn complete(self, id: impl Into<String>, message: Option<String>) -> Result<(), FlowError> {
        let flow_id = self.flow_id.clone();
        let inner = self.inner.clone();
        let _ = self.push(id.into(), FlowStepKind::Complete { message });
        let flows = inner.flows.read().expect("flows");
        match flows.get(&flow_id).and_then(|r| r.error.clone()) {
            Some(err) => Err(err),
            None => Ok(()),
        }
    }

    fn push(self, id: String, kind: FlowStepKind) -> Self {
        let step_id = id.trim().to_string();
        let mut flows = self.inner.flows.write().expect("flows");
        if let Some(reg) = flows.get_mut(&self.flow_id) {
            if reg.error.is_none() {
                if step_id.is_empty() {
                    reg.error = Some(FlowError::EmptyStepId {
                        flow: self.flow_id.clone(),
                    });
                } else if reg.steps.iter().any(|s| s.id == step_id) {
                    reg.error = Some(FlowError::DuplicateStepId {
                        flow: self.flow_id.clone(),
                        step: step_id.clone(),
                    });
                } else {
                    reg.steps.push(FlowStep { id: step_id, kind });
                }
            }
        }
        drop(flows);
        self
    }
}

#[derive(Debug, Clone)]
pub struct ListenOptions {
    pub port: u16,
    pub host: Option<String>,
    pub path: Option<String>,
}

pub struct ListenHandle {
    pub url: String,
    shutdown: Option<tokio::sync::oneshot::Sender<()>>,
    join: Option<tokio::task::JoinHandle<()>>,
}

impl ListenHandle {
    pub async fn close(mut self) {
        if let Some(tx) = self.shutdown.take() {
            let _ = tx.send(());
        }
        if let Some(join) = self.join.take() {
            let _ = join.await;
        }
    }
}

struct Inner {
    config: QefroConfig,
    tools: RwLock<HashMap<String, ToolRegistration>>,
    flows: RwLock<HashMap<String, FlowRegistration>>,
    events: RwLock<HashMap<String, EventHandlerDefinition>>,
    webhooks: RwLock<HashMap<String, EventHandlerDefinition>>,
    schedules: RwLock<HashMap<String, EventHandlerDefinition>>,
    pending: Mutex<HashMap<String, PendingInvocation>>,
    auth_by_conversation: Mutex<HashMap<Uuid, StoredAuth>>,
    customer_provider: RwLock<Option<Arc<dyn CustomerProvider>>>,
    middlewares: RwLock<Vec<MiddlewareFn>>,
    before_hooks: RwLock<Vec<BeforeHook>>,
    after_hooks: RwLock<Vec<AfterHook>>,
}

/// In-handler customer API (mirrors JS `ctx.customer`).
///
/// Hub methods (`resolve` / `create` / `update` / `note` / `tag`) talk to the
/// platform Customer Hub via `platform.customer`. External `CustomerProvider`
/// auth (`authorize` / provider `lookup`) is unchanged for connector CRMs.
#[derive(Clone)]
pub struct CustomerApi {
    app: Qefro,
    identity: Value,
    parameters: Value,
    conversation_id: Uuid,
    channel: Option<String>,
    auth_response: Option<String>,
    state: Arc<Mutex<CustomerState>>,
    platform: Option<PlatformCapabilities>,
}

impl CustomerApi {
    async fn set_current(&self, customer: Option<Value>) -> Option<Value> {
        let mut state = self.state.lock().await;
        state.current = customer.clone();
        state.lookup_completed = true;
        customer
    }

    async fn require_current_id(&self) -> Result<String> {
        let state = self.state.lock().await;
        if let Some(id) = state
            .current
            .as_ref()
            .and_then(|v| v.get("id"))
            .and_then(|v| v.as_str())
            .filter(|s| !s.is_empty())
        {
            return Ok(id.to_string());
        }
        Err(anyhow!("customer_not_found"))
    }

    /// Resolve-or-create Customer Hub identity (preferred for apps).
    pub async fn resolve(&self, input: Option<Value>) -> Result<Option<Value>> {
        let identity = pick_identity(input.as_ref(), &self.identity);
        let mut body = Value::Object(identity);
        if let Some(obj) = body.as_object_mut() {
            if let Some(ch) = self.channel.as_ref() {
                obj.insert("channel".into(), json!(ch));
            }
            obj.insert("conversation_id".into(), json!(self.conversation_id.to_string()));
        }
        let out = hub_call(self.platform.as_ref(), "resolve", body).await?;
        let hub = hub_customer_from_person(out.as_ref());
        if hub.is_some() {
            self.set_current(hub.clone()).await;
        }
        Ok(hub)
    }

    /// Lookup only (no create). Hub when args / hub-only; else external provider.
    pub async fn lookup(&self, input: Option<Value>) -> Result<Option<Value>> {
        let provider = self
            .app
            .inner
            .customer_provider
            .read()
            .expect("customer_provider")
            .clone();

        if input.is_some() || provider.is_none() {
            {
                let state = self.state.lock().await;
                if state.lookup_completed && input.is_none() && state.current.is_some() {
                    return Ok(state.current.clone());
                }
            }
            let identity = pick_identity(input.as_ref(), &self.identity);
            let mut body = Value::Object(identity);
            if let Some(obj) = body.as_object_mut() {
                if let Some(ch) = self.channel.as_ref() {
                    obj.insert("channel".into(), json!(ch));
                }
                obj.insert(
                    "conversation_id".into(),
                    json!(self.conversation_id.to_string()),
                );
            }
            let out = hub_call(self.platform.as_ref(), "lookup", body).await?;
            let hub = hub_customer_from_person(out.as_ref());
            self.set_current(hub.clone()).await;
            return Ok(hub);
        }

        {
            let state = self.state.lock().await;
            if state.lookup_completed {
                return Ok(state.current.clone());
            }
        }

        let customer = provider
            .unwrap()
            .lookup(&CustomerLookupContext {
                identity: self.identity.clone(),
                parameters: self.parameters.clone(),
                conversation: Conversation {
                    id: self.conversation_id,
                },
                channel: self.channel.clone(),
            })
            .await?;

        Ok(self.set_current(customer).await)
    }

    pub async fn lookup_by_phone(&self, phone: Option<&str>) -> Result<Option<Value>> {
        let source = phone
            .map(str::to_string)
            .or_else(|| read_identity_phone(&self.identity));

        let Some(source) = source else {
            let mut state = self.state.lock().await;
            state.lookup_completed = true;
            state.current = None;
            return Ok(None);
        };

        let provider = self
            .app
            .inner
            .customer_provider
            .read()
            .expect("customer_provider")
            .clone();

        if provider.is_none() || is_customer_hub_enabled() {
            return self
                .lookup(Some(json!({
                    "phone_number": source,
                    "whatsapp_number": source,
                })))
                .await;
        }

        let mut identity = self.identity.clone();
        if let Some(obj) = identity.as_object_mut() {
            obj.insert("phone".into(), json!(source));
        }

        let customer = provider
            .unwrap()
            .lookup(&CustomerLookupContext {
                identity,
                parameters: self.parameters.clone(),
                conversation: Conversation {
                    id: self.conversation_id,
                },
                channel: self.channel.clone(),
            })
            .await?;

        Ok(self.set_current(customer).await)
    }

    pub async fn create(&self, input: Value) -> Result<Option<Value>> {
        let identity = pick_identity(Some(&input), &self.identity);
        let mut body = Value::Object(identity);
        if let Some(obj) = body.as_object_mut() {
            if let Some(ch) = self.channel.as_ref() {
                obj.insert("channel".into(), json!(ch));
            }
            obj.insert(
                "conversation_id".into(),
                json!(self.conversation_id.to_string()),
            );
        }
        let out = hub_call(self.platform.as_ref(), "create", body).await?;
        let hub = hub_customer_from_person(out.as_ref());
        if hub.is_some() {
            self.set_current(hub.clone()).await;
        }
        Ok(hub)
    }

    pub async fn update(&self, input: Value) -> Result<Option<Value>> {
        let id = input
            .get("id")
            .and_then(|v| v.as_str())
            .map(str::to_string)
            .or_else(|| {
                // filled from state below
                None
            });
        let id = match id {
            Some(id) => id,
            None => {
                let state = self.state.lock().await;
                match state
                    .current
                    .as_ref()
                    .and_then(|v| v.get("id"))
                    .and_then(|v| v.as_str())
                    .map(str::to_string)
                {
                    Some(id) => id,
                    None if is_customer_hub_optional() => return Ok(None),
                    None => return Err(anyhow!("customer_not_found")),
                }
            }
        };
        let identity = pick_identity(Some(&input), &self.identity);
        let mut body = Value::Object(identity);
        if let Some(obj) = body.as_object_mut() {
            obj.insert("id".into(), json!(id));
        }
        let out = hub_call(self.platform.as_ref(), "update", body).await?;
        let hub = hub_customer_from_person(out.as_ref());
        if hub.is_some() {
            self.set_current(hub.clone()).await;
        }
        Ok(hub)
    }

    pub async fn note(&self, content: &str, options: Option<Value>) -> Result<()> {
        let trimmed = content.trim();
        if trimmed.is_empty() {
            return Err(anyhow!("customer_note_empty"));
        }
        let customer_id = match self.require_current_id().await {
            Ok(id) => id,
            Err(_err) if is_customer_hub_optional() => return Ok(()),
            Err(err) => return Err(err),
        };
        let author_id = options
            .as_ref()
            .and_then(|v| v.get("author_id"))
            .cloned()
            .unwrap_or(Value::Null);
        hub_call(
            self.platform.as_ref(),
            "note",
            json!({
                "customer_id": customer_id,
                "content": trimmed,
                "author_id": author_id,
            }),
        )
        .await?;
        Ok(())
    }

    pub async fn tag(&self, name: &str, options: Option<Value>) -> Result<()> {
        let trimmed = name.trim();
        if trimmed.is_empty() {
            return Err(anyhow!("customer_tag_empty"));
        }
        let customer_id = match self.require_current_id().await {
            Ok(id) => id,
            Err(_err) if is_customer_hub_optional() => return Ok(()),
            Err(err) => return Err(err),
        };
        let color = options
            .as_ref()
            .and_then(|v| v.get("color"))
            .cloned()
            .unwrap_or(Value::Null);
        hub_call(
            self.platform.as_ref(),
            "tag",
            json!({
                "customer_id": customer_id,
                "name": trimmed,
                "color": color,
            }),
        )
        .await?;
        Ok(())
    }

    pub async fn authorize(&self, method: Option<String>) -> Result<Value> {
        let provider = self
            .app
            .inner
            .customer_provider
            .read()
            .expect("customer_provider")
            .clone()
            .ok_or_else(|| anyhow!("customer_provider_missing"))?;

        {
            let auth = self.app.inner.auth_by_conversation.lock().await;
            if let Some(existing) = auth.get(&self.conversation_id) {
                if existing.expires_at_epoch_ms > Utc::now().timestamp_millis() {
                    let mut state = self.state.lock().await;
                    state.current = Some(existing.customer.clone());
                    state.lookup_completed = true;
                    return Ok(existing.customer.clone());
                }
            }
        }

        let customer = self
            .lookup(None)
            .await?
            .ok_or_else(|| anyhow!("customer_not_found"))?;

        let outcome = provider
            .authorize(&CustomerAuthorizeContext {
                customer: customer.clone(),
                method,
                response: self.auth_response.clone(),
                identity: self.identity.clone(),
                parameters: self.parameters.clone(),
                conversation: Conversation {
                    id: self.conversation_id,
                },
                channel: self.channel.clone(),
            })
            .await?;

        self.app
            .consume_auth_outcome(
                outcome,
                self.conversation_id,
                Some(self.state.clone()),
                None,
                None,
                None,
                None,
            )
            .await
    }

    pub async fn get(&self) -> Option<Value> {
        self.state.lock().await.current.clone()
    }

    pub async fn require(&self) -> Result<Value> {
        self.get()
            .await
            .ok_or_else(|| anyhow!("customer_not_found"))
    }

    /// Convenience: Hub / provider customer `id` when available.
    pub async fn id(&self) -> Option<String> {
        self.get()
            .await
            .and_then(|v| v.get("id")?.as_str().map(str::to_string))
    }

    pub async fn phone_number(&self) -> Option<String> {
        self.get().await.and_then(|v| {
            v.get("phone_number")?
                .as_str()
                .map(str::to_string)
                .or_else(|| v.get("phone")?.as_str().map(str::to_string))
        })
    }

    pub async fn whatsapp_number(&self) -> Option<String> {
        self.get()
            .await
            .and_then(|v| v.get("whatsapp_number")?.as_str().map(str::to_string))
    }

    pub async fn display_name(&self) -> Option<String> {
        self.get().await.and_then(|v| {
            v.get("display_name")?
                .as_str()
                .map(str::to_string)
                .or_else(|| v.get("name")?.as_str().map(str::to_string))
        })
    }
}

#[derive(Clone)]
pub struct Qefro {
    inner: Arc<Inner>,
}

impl Qefro {
    pub fn new(config: QefroConfig) -> Self {
        Self {
            inner: Arc::new(Inner {
                config,
                tools: RwLock::new(HashMap::new()),
                flows: RwLock::new(HashMap::new()),
                events: RwLock::new(HashMap::new()),
                webhooks: RwLock::new(HashMap::new()),
                schedules: RwLock::new(HashMap::new()),
                pending: Mutex::new(HashMap::new()),
                auth_by_conversation: Mutex::new(HashMap::new()),
                customer_provider: RwLock::new(None),
                middlewares: RwLock::new(Vec::new()),
                before_hooks: RwLock::new(Vec::new()),
                after_hooks: RwLock::new(Vec::new()),
            }),
        }
    }

    pub fn customer<P>(&self, provider: P) -> &Self
    where
        P: CustomerProvider + 'static,
    {
        *self.inner.customer_provider.write().expect("customer_provider") =
            Some(Arc::new(provider));
        self
    }

    pub fn tool<F, Fut>(&self, metadata: ToolMetadata, handler: F) -> &Self
    where
        F: Fn(ToolContext) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = Result<Value>> + Send + 'static,
    {
        let name = metadata.name.clone();
        let lookup = normalized_lookup_field(metadata.lookup.as_ref());
        let metadata = ToolMetadata {
            lookup,
            ..metadata
        };
        let registration = ToolRegistration {
            metadata,
            handler: Arc::new(move |ctx| Box::pin(handler(ctx))),
        };
        self.inner
            .tools
            .write()
            .expect("tools")
            .insert(name, registration);
        self
    }

    /// Register a Business Flow. Flows are metadata only: the SDK advertises them
    /// through `capabilities.list` and the Qefro Runtime orchestrates execution.
    ///
    /// Returns [`FlowError`] on a duplicate or empty flow id — the SDK never panics.
    pub fn flow(&self, metadata: BusinessFlowMetadata) -> std::result::Result<FlowBuilder, FlowError> {
        let id = metadata.id.trim().to_string();
        if id.is_empty() {
            return Err(FlowError::EmptyFlowId);
        }
        {
            let flows = self.inner.flows.read().expect("flows");
            if flows.contains_key(&id) {
                return Err(FlowError::DuplicateFlowId(id));
            }
        }
        let version = if metadata.version == 0 { 1 } else { metadata.version };
        let trigger = match metadata.trigger.clone() {
            Some(t) => Some(t.normalize()?),
            None => None,
        };
        let metadata = BusinessFlowMetadata {
            id: id.clone(),
            version,
            trigger,
            ..metadata
        };
        self.inner.flows.write().expect("flows").insert(
            id.clone(),
            FlowRegistration {
                metadata,
                steps: Vec::new(),
                error: None,
            },
        );
        Ok(FlowBuilder {
            inner: self.inner.clone(),
            flow_id: id,
        })
    }

    /// Register a standalone event handler (advertised via capabilities.list).
    /// The Qefro runtime owns delivery; connectors only emit into the bus.
    pub fn event(&self, def: EventHandlerDefinition) -> std::result::Result<&Self, FlowError> {
        self.register_named_handler("event", &self.inner.events, def)
    }

    /// Register a webhook alias (normalized to an orchestration event at ingest).
    pub fn webhook(&self, def: EventHandlerDefinition) -> std::result::Result<&Self, FlowError> {
        self.register_named_handler("webhook", &self.inner.webhooks, def)
    }

    /// Register a cron schedule. The runtime scheduler emits the named event.
    pub fn schedule(&self, def: EventHandlerDefinition) -> std::result::Result<&Self, FlowError> {
        let cron = def.cron.as_deref().unwrap_or("").trim();
        if cron.is_empty() {
            return Err(FlowError::InvalidTrigger(
                "schedule() requires a non-empty cron expression".into(),
            ));
        }
        self.register_named_handler("schedule", &self.inner.schedules, def)
    }

    fn register_named_handler(
        &self,
        kind: &'static str,
        lock: &RwLock<HashMap<String, EventHandlerDefinition>>,
        def: EventHandlerDefinition,
    ) -> std::result::Result<&Self, FlowError> {
        let name = def.name.trim().to_string();
        if name.is_empty() {
            return Err(FlowError::EmptyHandlerName(kind));
        }
        let mut map = lock.write().expect(kind);
        if map.contains_key(&name) {
            return Err(FlowError::DuplicateHandler { kind, name });
        }
        map.insert(
            name.clone(),
            EventHandlerDefinition {
                name,
                description: def.description,
                cron: def.cron,
            },
        );
        Ok(self)
    }

    fn list_named_handlers(
        lock: &RwLock<HashMap<String, EventHandlerDefinition>>,
    ) -> Vec<EventHandlerDefinition> {
        lock.read()
            .expect("handlers")
            .values()
            .cloned()
            .collect()
    }

    /// Snapshot the valid registered flows for `capabilities.list`. Flows with a
    /// recorded builder error are excluded and logged.
    fn list_registered_flows(&self) -> Vec<BusinessFlow> {
        let flows = self.inner.flows.read().expect("flows");
        flows
            .values()
            .filter_map(|reg| {
                if let Some(err) = &reg.error {
                    eprintln!("[qefro] skipping invalid flow \"{}\": {err}", reg.metadata.id);
                    None
                } else {
                    Some(BusinessFlow {
                        metadata: reg.metadata.clone(),
                        steps: reg.steps.clone(),
                    })
                }
            })
            .collect()
    }

    pub fn before<F, Fut>(&self, hook: F) -> &Self
    where
        F: Fn(ToolContext) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = Result<()>> + Send + 'static,
    {
        let hook: BeforeHook = Arc::new(move |ctx| Box::pin(hook(ctx)));
        self.inner
            .before_hooks
            .write()
            .expect("before_hooks")
            .push(hook);
        self
    }

    pub fn after<F, Fut>(&self, hook: F) -> &Self
    where
        F: Fn(ToolContext, Value) -> Fut + Send + Sync + 'static,
        Fut: Future<Output = Result<Value>> + Send + 'static,
    {
        let hook: AfterHook = Arc::new(move |ctx, out| Box::pin(hook(ctx, out)));
        self.inner
            .after_hooks
            .write()
            .expect("after_hooks")
            .push(hook);
        self
    }

    /// Onion middleware (mirrors JS `app.use`).
    pub fn use_middleware<F>(&self, middleware: F) -> &Self
    where
        F: Fn(ToolContext, NextFn) -> Pin<Box<dyn Future<Output = Result<Value>> + Send>>
            + Send
            + Sync
            + 'static,
    {
        let mw: MiddlewareFn = Arc::new(middleware);
        self.inner.middlewares.write().expect("middlewares").push(mw);
        self
    }

    pub fn verify_signature(&self, signature: &str, timestamp: i64, body: &str) -> bool {
        let now = Utc::now().timestamp();
        if (now - timestamp).abs() > self.inner.config.max_timestamp_skew_secs {
            return false;
        }
        let payload = format!("v1:{timestamp}:{body}");
        let mut mac = HmacSha256::new_from_slice(self.inner.config.signing_secret.as_bytes())
            .expect("HMAC accepts any key length");
        mac.update(payload.as_bytes());
        let expected = format!("v1={}", hex::encode(mac.finalize().into_bytes()));
        let a = expected.as_bytes();
        let b = signature.as_bytes();
        a.len() == b.len() && bool::from(a.ct_eq(b))
    }

    /// Handle a verified protocol request (after signature check).
    pub async fn handle(&self, request: QefroRequest) -> QefroResponse {
        if request.protocol_version != self.inner.config.protocol_version {
            return QefroResponse::Error {
                code: "protocol_mismatch".into(),
                message: "Unsupported protocol version".into(),
            };
        }

        match request.request_type.as_str() {
            "ping" => QefroResponse::Pong {
                protocol_version: self.inner.config.protocol_version.clone(),
                sdk_version: SDK_VERSION.into(),
            },
            "tools.list" => {
                let tools = self.inner.tools.read().expect("tools");
                QefroResponse::ToolsList {
                    tools: tools
                        .values()
                        .map(|r| RegisteredTool {
                            name: r.metadata.name.clone(),
                            description: r.metadata.description.clone(),
                            input_schema: r.metadata.input_schema.clone(),
                            authentication_methods: r.metadata.authentication_methods.clone(),
                            auth: r.metadata.auth,
                            permissions: r.metadata.permissions.clone(),
                            timeout: r.metadata.timeout,
                            lookup: r.metadata.lookup.clone(),
                        })
                        .collect(),
                    protocol_version: self.inner.config.protocol_version.clone(),
                    sdk_version: SDK_VERSION.into(),
                }
            }
            "capabilities.list" => {
                let tools = {
                    let tools = self.inner.tools.read().expect("tools");
                    tools
                        .values()
                        .map(|r| RegisteredTool {
                            name: r.metadata.name.clone(),
                            description: r.metadata.description.clone(),
                            input_schema: r.metadata.input_schema.clone(),
                            authentication_methods: r.metadata.authentication_methods.clone(),
                            auth: r.metadata.auth,
                            permissions: r.metadata.permissions.clone(),
                            timeout: r.metadata.timeout,
                            lookup: r.metadata.lookup.clone(),
                        })
                        .collect()
                };
                QefroResponse::CapabilitiesList {
                    tools,
                    flows: self.list_registered_flows(),
                    events: Self::list_named_handlers(&self.inner.events),
                    webhooks: Self::list_named_handlers(&self.inner.webhooks),
                    schedules: Self::list_named_handlers(&self.inner.schedules),
                    protocol_version: self.inner.config.protocol_version.clone(),
                    sdk_version: SDK_VERSION.into(),
                    sdk_name: SDK_NAME.into(),
                }
            }
            "tool.invoke" => {
                self.invoke(
                    request.tool,
                    request.parameters.unwrap_or_else(|| json!({})),
                    request.conversation_id.unwrap_or_else(Uuid::new_v4),
                    request.identity,
                    request.channel,
                    request.authentication,
                    None,
                    request.platform,
                    request.person,
                )
                .await
            }
            "tool.resume" => {
                let Some(resume_token) = request.resume_token else {
                    return QefroResponse::Error {
                        code: "invalid_request".into(),
                        message: "resume_token is required".into(),
                    };
                };
                let Some(challenge_response) = request.challenge_response else {
                    return QefroResponse::Error {
                        code: "invalid_request".into(),
                        message: "challenge_response is required".into(),
                    };
                };
                let pending = {
                    let mut map = self.inner.pending.lock().await;
                    map.remove(&resume_token)
                };
                let Some(pending) = pending else {
                    return QefroResponse::Error {
                        code: "not_found".into(),
                        message: "resume token not found or expired".into(),
                    };
                };
                self.invoke(
                    Some(pending.tool),
                    pending.parameters,
                    pending.conversation_id,
                    pending.identity,
                    pending.channel,
                    request.authentication,
                    Some(challenge_response),
                    request.platform.or(pending.platform),
                    request.person.or(pending.person),
                )
                .await
            }
            _ => QefroResponse::Error {
                code: "invalid_request".into(),
                message: "Unsupported request type".into(),
            },
        }
    }

    /// Verify signature + protocol headers, then handle (mirrors JS `handleRaw`).
    pub async fn handle_raw(
        &self,
        body: &str,
        headers: &HeaderMap,
    ) -> (StatusCode, QefroResponse) {
        let signature = header_str(headers, "x-qefro-signature");
        let timestamp = header_str(headers, "x-qefro-timestamp")
            .and_then(|t| t.parse::<i64>().ok());

        let protocol_header = header_str(headers, "x-qefro-protocol")
            .or_else(|| header_str(headers, "x-qefro-protocol-version"));
        if let Some(proto) = protocol_header {
            if proto != self.inner.config.protocol_version {
                return (
                    StatusCode::BAD_REQUEST,
                    QefroResponse::Error {
                        code: "protocol_mismatch".into(),
                        message: format!("Unsupported protocol version {proto}"),
                    },
                );
            }
        }

        match (signature, timestamp) {
            (Some(sig), Some(ts)) if self.verify_signature(sig, ts, body) => {}
            _ => {
                return (
                    StatusCode::UNAUTHORIZED,
                    QefroResponse::Error {
                        code: "invalid_signature".into(),
                        message: "Invalid Qefro signature".into(),
                    },
                );
            }
        }

        let request: QefroRequest = match serde_json::from_str(body) {
            Ok(r) => r,
            Err(e) => {
                return (
                    StatusCode::BAD_REQUEST,
                    QefroResponse::Error {
                        code: "invalid_request".into(),
                        message: e.to_string(),
                    },
                );
            }
        };

        (StatusCode::OK, self.handle(request).await)
    }

    /// Start an HTTP server (mirrors JS `listen`).
    pub async fn listen(&self, options: ListenOptions) -> Result<ListenHandle> {
        let host = options
            .host
            .unwrap_or_else(|| "0.0.0.0".to_string());
        let path = options
            .path
            .unwrap_or_else(|| self.inner.config.endpoint_path.clone());
        let path = if path.starts_with('/') {
            path
        } else {
            format!("/{path}")
        };

        let app_state = self.clone();
        let router = Router::new()
            .route(&path, post(http_handler))
            .with_state(app_state);

        let addr = format!("{host}:{}", options.port);
        let listener = tokio::net::TcpListener::bind(&addr).await?;
        let url = format!("http://{host}:{}{path}", options.port);

        let (tx, rx) = tokio::sync::oneshot::channel::<()>();
        let join = tokio::spawn(async move {
            let _ = axum::serve(listener, router)
                .with_graceful_shutdown(async {
                    let _ = rx.await;
                })
                .await;
        });

        Ok(ListenHandle {
            url,
            shutdown: Some(tx),
            join: Some(join),
        })
    }

    async fn invoke(
        &self,
        tool: Option<String>,
        parameters: Value,
        conversation_id: Uuid,
        identity: Option<Value>,
        channel: Option<String>,
        authentication: Option<Value>,
        auth_response: Option<String>,
        platform: Option<PlatformCapabilities>,
        person: Option<Value>,
    ) -> QefroResponse {
        let Some(tool_name) = tool else {
            return QefroResponse::Error {
                code: "invalid_request".into(),
                message: "tool is required".into(),
            };
        };

        let registration = {
            let tools = self.inner.tools.read().expect("tools");
            tools.get(&tool_name).cloned()
        };
        let Some(registration) = registration else {
            return QefroResponse::Error {
                code: "not_found".into(),
                message: format!("Unknown tool: {tool_name}"),
            };
        };

        let identity_value = identity.clone().unwrap_or_else(|| json!({}));
        let customer_state = Arc::new(Mutex::new(CustomerState::default()));

        {
            let auth = self.inner.auth_by_conversation.lock().await;
            if let Some(stored) = auth.get(&conversation_id) {
                if stored.expires_at_epoch_ms > Utc::now().timestamp_millis() {
                    let mut state = customer_state.lock().await;
                    state.current = Some(stored.customer.clone());
                    state.lookup_completed = true;
                }
            }
        }

        // Seed hub customer from Person snapshot when present (native chat path).
        if let Some(ref person_val) = person {
            if let Some(seeded) = seed_from_person(person_val) {
                let mut state = customer_state.lock().await;
                state.current = Some(seeded);
                state.lookup_completed = true;
            }
        }

        let customer_api = CustomerApi {
            app: self.clone(),
            identity: identity_value.clone(),
            parameters: parameters.clone(),
            conversation_id,
            channel: channel.clone(),
            auth_response: auth_response.clone(),
            state: customer_state.clone(),
            platform: platform.clone(),
        };

        let mut current_customer = customer_state.lock().await.current.clone();

        if registration.metadata.auth == ToolAuthMode::Required {
            match customer_api
                .authorize(registration.metadata.default_auth_method.clone())
                .await
            {
                Ok(customer) => current_customer = Some(customer),
                Err(e) => {
                    return map_invoke_error(
                        e,
                        self,
                        &tool_name,
                        &parameters,
                        conversation_id,
                        identity.clone(),
                        channel.clone(),
                        platform.clone(),
                        person.clone(),
                    )
                    .await
                }
            }
        }

        let solution_id = platform
            .as_ref()
            .and_then(|p| p.customer.as_ref())
            .and_then(|c| c.context.as_ref())
            .and_then(|c| c.solution_id.clone())
            .or_else(|| {
                platform
                    .as_ref()
                    .and_then(|p| p.storage.as_ref())
                    .and_then(|s| s.context.as_ref())
                    .map(|c| c.solution_id.clone())
            });

        let timeline = TimelineContext {
            platform: platform.clone(),
            state: customer_state.clone(),
        };
        let membership = MembershipContext {
            platform: platform.clone(),
            state: customer_state.clone(),
            solution_id,
        };
        let consent = ConsentContext {
            platform: platform.clone(),
            state: customer_state.clone(),
        };

        let ctx = ToolContext {
            identity: identity_value,
            parameters: parameters.clone(),
            conversation: Conversation {
                id: conversation_id,
            },
            channel: channel.clone(),
            authentication,
            auth_response,
            customer: current_customer,
            customer_api: Some(customer_api),
            timeline,
            membership,
            consent,
            platform: platform.clone(),
        };

        let before_hooks = self.inner.before_hooks.read().expect("before_hooks").clone();
        for hook in &before_hooks {
            if let Err(e) = hook(ctx.clone()).await {
                return map_invoke_error(
                    e,
                    self,
                    &tool_name,
                    &parameters,
                    conversation_id,
                    identity.clone(),
                    channel.clone(),
                    platform.clone(),
                    person.clone(),
                )
                .await;
            }
        }

        let handler = registration.handler.clone();
        let middlewares = self.inner.middlewares.read().expect("middlewares").clone();
        let run_result = run_middlewares(middlewares, ctx.clone(), handler).await;

        let output = match run_result {
            Ok(v) => v,
            Err(e) => {
                return map_invoke_error(
                    e,
                    self,
                    &tool_name,
                    &parameters,
                    conversation_id,
                    identity,
                    channel,
                    platform,
                    person,
                )
                .await;
            }
        };

        let after_hooks = self.inner.after_hooks.read().expect("after_hooks").clone();
        let mut output = output;
        for hook in &after_hooks {
            match hook(ctx.clone(), output).await {
                Ok(v) => output = v,
                Err(e) => {
                    return map_invoke_error(
                        e,
                        self,
                        &tool_name,
                        &parameters,
                        conversation_id,
                        identity,
                        channel,
                        platform,
                        person,
                    )
                    .await;
                }
            }
        }

        let auth = {
            let map = self.inner.auth_by_conversation.lock().await;
            map.get(&conversation_id)
                .filter(|v| v.expires_at_epoch_ms > Utc::now().timestamp_millis())
                .map(|v| v.auth.clone())
        };

        QefroResponse::Result {
            output,
            authentication_context: auth,
        }
    }

    async fn consume_auth_outcome(
        &self,
        outcome: AuthOutcome,
        conversation_id: Uuid,
        customer_state: Option<Arc<Mutex<CustomerState>>>,
        // When challenge: stash pending invoke
        pending_tool: Option<&str>,
        pending_parameters: Option<Value>,
        pending_identity: Option<Value>,
        pending_channel: Option<String>,
    ) -> Result<Value> {
        match outcome {
            AuthOutcome::Success { customer, auth } => {
                let ttl = auth.expires_in.unwrap_or(900).max(1);
                self.inner.auth_by_conversation.lock().await.insert(
                    conversation_id,
                    StoredAuth {
                        customer: customer.clone(),
                        auth,
                        expires_at_epoch_ms: Utc::now().timestamp_millis() + ttl * 1000,
                    },
                );
                if let Some(state) = customer_state {
                    let mut s = state.lock().await;
                    s.current = Some(customer.clone());
                    s.lookup_completed = true;
                }
                Ok(customer)
            }
            AuthOutcome::Challenge(challenge) => {
                if let (Some(tool), Some(parameters)) = (pending_tool, pending_parameters) {
                    let resume_token = Uuid::new_v4().to_string();
                    self.inner.pending.lock().await.insert(
                        resume_token.clone(),
                        PendingInvocation {
                            tool: tool.to_string(),
                            conversation_id,
                            parameters,
                            identity: pending_identity,
                            channel: pending_channel,
                            platform: None,
                            person: None,
                        },
                    );
                    // Encode resume in error path via ChallengeSignal — callers for required-auth
                    // use map_invoke_error. For CustomerApi.authorize we raise ChallengeSignal.
                    let _ = resume_token;
                }
                Err(ChallengeSignal { challenge }.into())
            }
            AuthOutcome::Denied => Err(anyhow!("denied")),
            AuthOutcome::NotFound => Err(anyhow!("customer_not_found")),
        }
    }

    pub async fn require_authentication(
        &self,
        conversation_id: Uuid,
        outcome: AuthOutcome,
        tool: &str,
        parameters: Value,
        identity: Option<Value>,
        channel: Option<String>,
    ) -> std::result::Result<Value, QefroResponse> {
        match outcome {
            AuthOutcome::Success { customer, auth } => {
                let ttl = auth.expires_in.unwrap_or(900).max(1);
                self.inner.auth_by_conversation.lock().await.insert(
                    conversation_id,
                    StoredAuth {
                        customer: customer.clone(),
                        auth,
                        expires_at_epoch_ms: Utc::now().timestamp_millis() + ttl * 1000,
                    },
                );
                Ok(customer)
            }
            AuthOutcome::Challenge(challenge) => {
                let resume_token = Uuid::new_v4().to_string();
                self.inner.pending.lock().await.insert(
                    resume_token.clone(),
                    PendingInvocation {
                        tool: tool.to_string(),
                        conversation_id,
                        parameters,
                        identity,
                        channel,
                        platform: None,
                        person: None,
                    },
                );
                Err(QefroResponse::Challenge {
                    resume_token,
                    challenge,
                })
            }
            AuthOutcome::Denied => Err(QefroResponse::Error {
                code: "denied".into(),
                message: "Authentication denied".into(),
            }),
            AuthOutcome::NotFound => Err(QefroResponse::Error {
                code: "customer_not_found".into(),
                message: "Customer not found".into(),
            }),
        }
    }
}

async fn map_invoke_error(
    e: anyhow::Error,
    app: &Qefro,
    tool_name: &str,
    parameters: &Value,
    conversation_id: Uuid,
    identity: Option<Value>,
    channel: Option<String>,
    platform: Option<PlatformCapabilities>,
    person: Option<Value>,
) -> QefroResponse {
    if let Some(signal) = e.downcast_ref::<ChallengeSignal>() {
        let resume_token = Uuid::new_v4().to_string();
        app.inner.pending.lock().await.insert(
            resume_token.clone(),
            PendingInvocation {
                tool: tool_name.to_string(),
                conversation_id,
                parameters: parameters.clone(),
                identity,
                channel,
                platform,
                person,
            },
        );
        return QefroResponse::Challenge {
            resume_token,
            challenge: signal.challenge.clone(),
        };
    }

    let message = e.to_string();
    if message == "denied" {
        return QefroResponse::Error {
            code: "denied".into(),
            message: "Authentication denied".into(),
        };
    }
    if message == "customer_not_found" {
        return QefroResponse::Error {
            code: "customer_not_found".into(),
            message: "Customer not found".into(),
        };
    }
    if message == "customer_provider_missing" {
        return QefroResponse::Error {
            code: "configuration_error".into(),
            message: "Tool requires customer provider. Configure app.customer(...) first.".into(),
        };
    }

    QefroResponse::Error {
        code: "internal_error".into(),
        message,
    }
}

async fn run_middlewares(
    middlewares: Vec<MiddlewareFn>,
    ctx: ToolContext,
    handler: ToolHandler,
) -> Result<Value> {
    fn dispatch(
        i: usize,
        middlewares: Arc<Vec<MiddlewareFn>>,
        ctx: ToolContext,
        handler: ToolHandler,
    ) -> Pin<Box<dyn Future<Output = Result<Value>> + Send>> {
        Box::pin(async move {
            if i == middlewares.len() {
                return handler(ctx).await;
            }
            let mw = middlewares[i].clone();
            let mws = middlewares.clone();
            let h = handler.clone();
            let next: NextFn = Box::new(move |c| dispatch(i + 1, mws, c, h));
            mw(ctx, next).await
        })
    }

    dispatch(0, Arc::new(middlewares), ctx, handler).await
}

fn header_str<'a>(headers: &'a HeaderMap, key: &str) -> Option<&'a str> {
    headers.get(key).and_then(|v| v.to_str().ok())
}

fn protocol_response_headers(app: &Qefro) -> HeaderMap {
    use axum::http::HeaderValue;
    let mut headers = HeaderMap::new();
    let proto = HeaderValue::from_str(&app.inner.config.protocol_version)
        .unwrap_or_else(|_| HeaderValue::from_static("1"));
    headers.insert("X-Qefro-Protocol", proto.clone());
    headers.insert("X-Qefro-Protocol-Version", proto);
    headers.insert("X-Qefro-SDK", HeaderValue::from_static(SDK_NAME));
    headers.insert(
        "X-Qefro-Version",
        HeaderValue::from_static(SDK_VERSION),
    );
    headers
}

async fn http_handler(
    State(app): State<Qefro>,
    headers: HeaderMap,
    body: Bytes,
) -> impl IntoResponse {
    let mut response_headers = protocol_response_headers(&app);
    let body_str = String::from_utf8_lossy(&body);
    let (status, resp) = app.handle_raw(&body_str, &headers).await;
    response_headers.insert(
        axum::http::header::CONTENT_TYPE,
        "application/json".parse().unwrap(),
    );
    (status, response_headers, Json(resp))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn normalize_lookup_dedupes() {
        let lookup = ToolLookup {
            by: Some("Email".into()),
            required: vec!["phone".into(), "email".into()],
        };
        assert_eq!(
            normalize_lookup(Some(&lookup)),
            vec!["phone".to_string(), "email".to_string()]
        );
    }

    #[test]
    fn signature_roundtrip() {
        let app = Qefro::new(QefroConfig::new("secret"));
        let body = r#"{"protocol_version":"1","request_id":"00000000-0000-0000-0000-000000000001","type":"ping"}"#;
        let ts = Utc::now().timestamp();
        let payload = format!("v1:{ts}:{body}");
        let mut mac = HmacSha256::new_from_slice(b"secret").unwrap();
        mac.update(payload.as_bytes());
        let sig = format!("v1={}", hex::encode(mac.finalize().into_bytes()));
        assert!(app.verify_signature(&sig, ts, body));
        assert!(!app.verify_signature("v1=deadbeef", ts, body));
    }

    #[tokio::test]
    async fn tools_list_includes_lookup() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.tool(
            ToolMetadata {
                name: "orders".into(),
                lookup: Some(ToolLookup {
                    by: Some("email".into()),
                    required: vec![],
                }),
                ..Default::default()
            },
            |_ctx| async move { Ok(json!({})) },
        );

        let resp = app
            .handle(QefroRequest {
                protocol_version: "1".into(),
                request_id: Uuid::new_v4(),
                request_type: "tools.list".into(),
                organization_id: None,
                conversation_id: None,
                channel: None,
                identity: None,
                tool: None,
                parameters: None,
                authentication: None,
                resume_token: None,
                challenge_response: None,
                person: None,
                platform: None,
            })
            .await;

        match resp {
            QefroResponse::ToolsList { tools, .. } => {
                assert_eq!(tools.len(), 1);
                assert_eq!(
                    tools[0].lookup.as_ref().unwrap().required,
                    vec!["email".to_string()]
                );
            }
            other => panic!("unexpected {other:?}"),
        }
    }

    fn capabilities_request() -> QefroRequest {
        QefroRequest {
            protocol_version: "1".into(),
            request_id: Uuid::new_v4(),
            request_type: "capabilities.list".into(),
            organization_id: None,
            conversation_id: None,
            channel: None,
            identity: None,
            tool: None,
            parameters: None,
            authentication: None,
            resume_token: None,
            challenge_response: None,
            person: None,
            platform: None,
        }
    }

    fn order_lookup_metadata() -> BusinessFlowMetadata {
        BusinessFlowMetadata {
            id: "order_lookup".into(),
            name: Some("Order Lookup".into()),
            description: Some("Lookup customer orders".into()),
            category: Some("crm".into()),
            tags: vec!["customer".into(), "orders".into()],
            intent: vec!["track order".into(), "where is my order".into()],
            inputs: vec!["email".into()],
            outputs: vec!["customer".into(), "orders".into()],
            ..Default::default()
        }
    }

    #[tokio::test]
    async fn capabilities_list_advertises_flows() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.tool(
            ToolMetadata {
                name: "lookup_customer".into(),
                ..Default::default()
            },
            |_ctx| async move { Ok(json!({})) },
        );
        app.flow(order_lookup_metadata())
            .expect("flow registers")
            .ask("email", "email", "Please enter your email.")
            .tool("lookup", "lookup_customer")
            .complete("done", None)
            .expect("flow builds without error");

        let resp = app.handle(capabilities_request()).await;
        let value = serde_json::to_value(&resp).expect("serialize");
        assert_eq!(value["type"], "capabilities.list");
        assert_eq!(value["flows"].as_array().unwrap().len(), 1);
        let flow = &value["flows"][0];
        // Wrapped { metadata, steps } shape with integer version.
        assert_eq!(flow["metadata"]["id"], "order_lookup");
        assert_eq!(flow["metadata"]["version"], 1);
        assert!(flow["metadata"]["version"].is_number());
        // { id, type, config } step model with tool_ref inside config.
        assert_eq!(flow["steps"][0]["id"], "email");
        assert_eq!(flow["steps"][0]["type"], "ask");
        assert_eq!(flow["steps"][0]["config"]["field"], "email");
        assert_eq!(flow["steps"][1]["type"], "tool");
        assert_eq!(flow["steps"][1]["config"]["tool_ref"], "lookup_customer");
        assert_eq!(flow["steps"][2]["type"], "complete");
        assert!(flow["steps"][2]["config"].is_object());
    }

    #[test]
    fn duplicate_flow_id_returns_error_no_panic() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.flow(order_lookup_metadata()).expect("first registers");
        let err = app.flow(order_lookup_metadata()).err().unwrap();
        assert_eq!(err, FlowError::DuplicateFlowId("order_lookup".into()));
    }

    #[test]
    fn empty_flow_id_returns_error() {
        let app = Qefro::new(QefroConfig::new("secret"));
        let err = app
            .flow(BusinessFlowMetadata {
                id: "   ".into(),
                ..Default::default()
            })
            .err()
            .unwrap();
        assert_eq!(err, FlowError::EmptyFlowId);
    }

    #[tokio::test]
    async fn capabilities_list_includes_event_triggers_and_handlers() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.flow(BusinessFlowMetadata {
            id: "abandoned_cart".into(),
            name: Some("Abandoned cart".into()),
            trigger: Some(FlowTrigger::Event {
                event: "shopify.cart.abandoned".into(),
                when: None,
            }),
            ..Default::default()
        })
        .expect("flow")
        .delay("wait", 60)
        .complete("done", None)
        .expect("build");

        app.event(EventHandlerDefinition {
            name: "shopify.cart.abandoned".into(),
            description: Some("cart abandoned".into()),
            cron: None,
        })
        .expect("event");
        app.schedule(EventHandlerDefinition {
            name: "nightly_sync".into(),
            description: None,
            cron: Some("0 2 * * *".into()),
        })
        .expect("schedule");

        let resp = app.handle(capabilities_request()).await;
        let value = serde_json::to_value(&resp).expect("serialize");
        assert_eq!(value["type"], "capabilities.list");
        assert_eq!(
            value["flows"][0]["metadata"]["trigger"]["type"],
            "event"
        );
        assert_eq!(
            value["flows"][0]["metadata"]["trigger"]["event"],
            "shopify.cart.abandoned"
        );
        assert_eq!(value["events"][0]["name"], "shopify.cart.abandoned");
        assert_eq!(value["schedules"][0]["cron"], "0 2 * * *");
    }

    #[tokio::test]
    async fn duplicate_step_id_surfaced_and_flow_excluded() {
        let app = Qefro::new(QefroConfig::new("secret"));
        let result = app
            .flow(order_lookup_metadata())
            .expect("flow registers")
            .ask("email", "email", "Please enter your email.")
            .tool("email", "lookup_customer") // duplicate step id
            .complete("done", None);
        assert_eq!(
            result.unwrap_err(),
            FlowError::DuplicateStepId {
                flow: "order_lookup".into(),
                step: "email".into()
            }
        );

        // Invalid flow is excluded from capabilities.list (never crashes the server).
        let resp = app.handle(capabilities_request()).await;
        let value = serde_json::to_value(&resp).expect("serialize");
        assert_eq!(value["flows"].as_array().unwrap().len(), 0);
    }

    #[tokio::test]
    async fn complete_step_allows_branch_terminal_before_complete() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.flow(order_lookup_metadata())
            .expect("flow registers")
            .ask("ask", "order_id", "Order id?")
            .tool("lookup", "order_status_check")
            .condition(
                "branch",
                "order_status_check.found == true",
                Some("ok".into()),
                Some("missing".into()),
            )
            .complete_step("missing", Some("Not found.".into()))
            .complete("ok", Some("Found.".into()))
            .expect("flow builds with a mid-chain complete_step");

        let resp = app.handle(capabilities_request()).await;
        let value = serde_json::to_value(&resp).expect("serialize");
        let steps = value["flows"][0]["steps"].as_array().unwrap();
        let kinds: Vec<&str> = steps.iter().map(|s| s["type"].as_str().unwrap()).collect();
        assert_eq!(kinds, ["ask", "tool", "condition", "complete", "complete"]);
    }

    #[tokio::test]
    async fn tools_list_unchanged_alongside_flows() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.tool(
            ToolMetadata {
                name: "lookup_customer".into(),
                ..Default::default()
            },
            |_ctx| async move { Ok(json!({})) },
        );
        app.flow(order_lookup_metadata())
            .expect("flow registers")
            .tool("lookup", "lookup_customer")
            .complete("done", None)
            .expect("flow builds");

        let resp = app
            .handle(QefroRequest {
                protocol_version: "1".into(),
                request_id: Uuid::new_v4(),
                request_type: "tools.list".into(),
                organization_id: None,
                conversation_id: None,
                channel: None,
                identity: None,
                tool: None,
                parameters: None,
                authentication: None,
                resume_token: None,
                challenge_response: None,
                person: None,
                platform: None,
            })
            .await;
        let value = serde_json::to_value(&resp).expect("serialize");
        // Legacy tools.list response carries no `flows` field.
        assert_eq!(value["type"], "tools.list");
        assert!(value.get("flows").is_none());
        assert_eq!(value["tools"].as_array().unwrap().len(), 1);
    }

    #[test]
    fn flow_step_roundtrips_through_wire_shape() {
        let step = FlowStep {
            id: "lookup".into(),
            kind: FlowStepKind::Tool {
                tool_ref: "lookup_customer".into(),
            },
        };
        let value = serde_json::to_value(&step).unwrap();
        assert_eq!(value["id"], "lookup");
        assert_eq!(value["type"], "tool");
        assert_eq!(value["config"]["tool_ref"], "lookup_customer");
        let back: FlowStep = serde_json::from_value(value).unwrap();
        assert_eq!(back, step);
    }

    #[tokio::test]
    async fn customer_hub_resolve_soft_skips_when_disabled() {
        let _guard = customer_hub::HUB_ENV_LOCK
            .lock()
            .unwrap_or_else(|e| e.into_inner());
        std::env::set_var("QEFRO_CUSTOMER_HUB_ENABLED", "false");
        std::env::set_var("QEFRO_CUSTOMER_HUB_OPTIONAL", "true");
        let app = Qefro::new(QefroConfig::new("secret"));
        app.tool(
            ToolMetadata {
                name: "hub_probe".into(),
                auth: ToolAuthMode::None,
                ..Default::default()
            },
            |ctx| async move {
                let api = ctx.customer_api().unwrap();
                let out = api.resolve(Some(json!({"phone_number": "+1"}))).await?;
                Ok(json!({ "customer": out }))
            },
        );
        let resp = app
            .handle(QefroRequest {
                protocol_version: "1".into(),
                request_id: Uuid::new_v4(),
                request_type: "tool.invoke".into(),
                organization_id: None,
                conversation_id: None,
                channel: None,
                identity: None,
                tool: Some("hub_probe".into()),
                parameters: Some(json!({})),
                authentication: None,
                resume_token: None,
                challenge_response: None,
                person: None,
                platform: None,
            })
            .await;
        let value = serde_json::to_value(&resp).unwrap();
        assert_eq!(value["type"], "result");
        assert!(value["output"]["customer"].is_null());
        std::env::remove_var("QEFRO_CUSTOMER_HUB_ENABLED");
        std::env::remove_var("QEFRO_CUSTOMER_HUB_OPTIONAL");
    }

    #[tokio::test]
    async fn customer_hub_person_seed_exposes_properties() {
        let app = Qefro::new(QefroConfig::new("secret"));
        app.tool(
            ToolMetadata {
                name: "who".into(),
                auth: ToolAuthMode::None,
                ..Default::default()
            },
            |ctx| async move {
                let api = ctx.customer_api().unwrap();
                Ok(json!({
                    "id": api.id().await,
                    "phone_number": api.phone_number().await,
                    "display_name": api.display_name().await,
                }))
            },
        );
        let resp = app
            .handle(QefroRequest {
                protocol_version: "1".into(),
                request_id: Uuid::new_v4(),
                request_type: "tool.invoke".into(),
                organization_id: None,
                conversation_id: None,
                channel: None,
                identity: None,
                tool: Some("who".into()),
                parameters: Some(json!({})),
                authentication: None,
                resume_token: None,
                challenge_response: None,
                person: Some(json!({
                    "id": "cust-1",
                    "phone": "+1999",
                    "name": "Sam",
                })),
                platform: None,
            })
            .await;
        let value = serde_json::to_value(&resp).unwrap();
        assert_eq!(value["output"]["id"], "cust-1");
        assert_eq!(value["output"]["phone_number"], "+1999");
        assert_eq!(value["output"]["display_name"], "Sam");
    }

    #[tokio::test]
    async fn customer_hub_timeline_noop_when_optional_no_customer() {
        let _guard = customer_hub::HUB_ENV_LOCK
            .lock()
            .unwrap_or_else(|e| e.into_inner());
        std::env::set_var("QEFRO_CUSTOMER_HUB_ENABLED", "true");
        std::env::set_var("QEFRO_CUSTOMER_HUB_OPTIONAL", "true");
        let app = Qefro::new(QefroConfig::new("secret"));
        app.tool(
            ToolMetadata {
                name: "hub_side".into(),
                auth: ToolAuthMode::None,
                ..Default::default()
            },
            |ctx| async move {
                ctx.timeline
                    .append(json!({"event_type": "x.y"}))
                    .await?;
                ctx.membership.attach(None).await?;
                ctx.consent
                    .grant(json!({"purpose": "marketing"}))
                    .await?;
                Ok(json!({"ok": true}))
            },
        );
        let resp = app
            .handle(QefroRequest {
                protocol_version: "1".into(),
                request_id: Uuid::new_v4(),
                request_type: "tool.invoke".into(),
                organization_id: None,
                conversation_id: None,
                channel: None,
                identity: None,
                tool: Some("hub_side".into()),
                parameters: Some(json!({})),
                authentication: None,
                resume_token: None,
                challenge_response: None,
                person: None,
                platform: None,
            })
            .await;
        let value = serde_json::to_value(&resp).unwrap();
        assert_eq!(value["type"], "result");
        assert_eq!(value["output"]["ok"], true);
        std::env::remove_var("QEFRO_CUSTOMER_HUB_ENABLED");
        std::env::remove_var("QEFRO_CUSTOMER_HUB_OPTIONAL");
    }
}