bitrouter 1.0.0-alpha.28

use crate::workflow_state::extractors::generic::GenericPromptExtractor;
use crate::workflow_state::extractors::{ExtractorInput, WorkflowStateExtractor};
use crate::workflow_state::ir::{HarnessId, WorkflowStateIR};

pub struct SmithersExtractor;

impl WorkflowStateExtractor for SmithersExtractor {
    fn detect(
        &self,
        input: &ExtractorInput<'_>,
    ) -> Option<crate::workflow_state::extractors::TraceAdapterMatch> {
        let has_native_metadata = ["x-smithers-workflow-id", "x-smithers-node-id"]
            .into_iter()
            .any(|name| non_empty_header(input, name).is_some());
        has_native_metadata.then_some(crate::workflow_state::extractors::TraceAdapterMatch {
            source: HarnessId::Smithers,
            confidence: 0.9,
            evidence_kind: "smithers_metadata",
            native: true,
        })
    }

    fn extract(&self, input: &ExtractorInput<'_>) -> WorkflowStateIR {
        let mut ir = GenericPromptExtractor.extract(input);
        ir.harness_id = HarnessId::Smithers;
        ir.active_workflow = non_empty_header(input, "x-smithers-workflow-id");
        ir.subagent_role = non_empty_header(input, "x-smithers-node-id");
        ir
    }
}

fn non_empty_header(input: &ExtractorInput<'_>, name: &str) -> Option<String> {
    input
        .headers
        .get(name)
        .and_then(|value| value.to_str().ok())
        .map(str::trim)
        .filter(|value| !value.is_empty())
        .map(ToString::to_string)
}