tsift-cli 0.1.64

//! Shared output infrastructure for tsift-cli command handlers.
//!
//! Phase 1a-i of `#split-tsift-cli-subcommands`:
//! - `OutputFormat` — per-command format flags (compact / pretty / terse / schema / envelope)
//! - `ResponseBudget` + `ResponseBudgetPreset` — adaptive item/byte budgets for envelope-wrapped previews
//! - `DEFAULT_BUDGET_*` constants
//!
//! Phase 1a-ii:
//! - `ToolEnvelope` + `ToolEnvelopeMetric` + `ToolEnvelopeSummary` — summary-wrapping JSON envelope
//! - `TranscriptArtifactRef` — session-transcript artifact reference
//!
//! Phase 1a-iii:
//! - `tagpath` submodule — `TagpathSearchOpts`, `TagpathAnnotationDiagnostic`,
//!   `CommunityMemberAmbiguityDiagnostic`, and the `annotate_*_with_tagpath` family.

pub(crate) mod tagpath;

use clap::ValueEnum;
use serde::Serialize;

pub(crate) const DEFAULT_BUDGET_ITEMS: usize = 5;
pub(crate) const DEFAULT_BUDGET_BYTES: usize = 160;
pub(crate) const DEFAULT_FOLLOW_UP_ITEMS: usize = 4;
pub(crate) const DEFAULT_BODY_TOKEN_CAP: usize = 1500;

#[derive(Clone, Copy, Default)]
pub(crate) struct OutputFormat {
    pub json_output: bool,
    pub compact: bool,
    pub pretty: bool,
    pub terse: bool,
    pub ultra_terse: bool,
    pub schema: bool,
    pub envelope: bool,
}

#[derive(Clone, Copy, Debug, Default)]
pub(crate) struct ResponseBudget {
    pub max_items: Option<usize>,
    pub max_bytes: Option<usize>,
    pub body_token_cap: Option<usize>,
}

#[derive(Clone, Copy, Debug, Eq, PartialEq, ValueEnum)]
pub(crate) enum ResponseBudgetPreset {
    Small,
    Normal,
    Deep,
    Auto,
}

impl ResponseBudget {
    pub fn new(max_items: Option<usize>, max_bytes: Option<usize>) -> Self {
        Self {
            max_items,
            max_bytes,
            body_token_cap: None,
        }
    }

    pub fn with_body_token_cap(mut self, cap: usize) -> Self {
        self.body_token_cap = Some(cap);
        self
    }

    pub fn from_cli(
        max_items: Option<usize>,
        max_bytes: Option<usize>,
        preset: Option<ResponseBudgetPreset>,
        envelope: bool,
    ) -> Self {
        let preset = preset.or_else(|| envelope.then_some(ResponseBudgetPreset::Auto));
        let Some(preset) = preset else {
            return Self::new(max_items, max_bytes);
        };

        let defaults = preset.resolve();
        Self::new(
            max_items.or(defaults.max_items),
            max_bytes.or(defaults.max_bytes),
        )
        .with_body_token_cap(defaults.body_token_cap.unwrap_or(DEFAULT_BODY_TOKEN_CAP))
    }

    pub fn is_active(self) -> bool {
        self.max_items.is_some() || self.max_bytes.is_some()
    }

    pub fn preview_items(self) -> usize {
        self.max_items.unwrap_or(DEFAULT_BUDGET_ITEMS)
    }

    pub fn preview_bytes(self) -> usize {
        self.max_bytes.unwrap_or(DEFAULT_BUDGET_BYTES)
    }

    pub fn follow_up_items(self) -> usize {
        self.preview_items().max(DEFAULT_FOLLOW_UP_ITEMS)
    }

    pub fn body_token_cap(self) -> usize {
        self.body_token_cap.unwrap_or(DEFAULT_BODY_TOKEN_CAP)
    }
}

impl ResponseBudgetPreset {
    pub fn resolve(self) -> ResponseBudget {
        match self {
            ResponseBudgetPreset::Small => {
                ResponseBudget::new(Some(3), Some(120)).with_body_token_cap(500)
            }
            ResponseBudgetPreset::Normal => ResponseBudget::new(Some(DEFAULT_BUDGET_ITEMS), Some(DEFAULT_BUDGET_BYTES))
                .with_body_token_cap(DEFAULT_BODY_TOKEN_CAP),
            ResponseBudgetPreset::Deep => {
                ResponseBudget::new(Some(10), Some(240)).with_body_token_cap(3000)
            }
            ResponseBudgetPreset::Auto => adaptive_response_budget(),
        }
    }
}

#[derive(Serialize)]
pub(crate) struct ToolEnvelopeMetric {
    pub label: String,
    pub value: String,
}

#[derive(Serialize)]
pub(crate) struct ToolEnvelopeSummary {
    pub text: String,
    pub metrics: Vec<ToolEnvelopeMetric>,
}

#[derive(Serialize)]
pub(crate) struct ToolEnvelope<'a, T: Serialize> {
    pub tool: &'a str,
    pub view: &'a str,
    pub summary: ToolEnvelopeSummary,
    pub truncated: bool,
    #[serde(skip_serializing_if = "Vec::is_empty")]
    pub follow_up: Vec<String>,
    pub report: &'a T,
}

#[derive(Serialize)]
pub(crate) struct TranscriptArtifactRef {
    pub handle: String,
    pub path: String,
    pub bytes: usize,
    pub lines: usize,
    pub expand: String,
}

fn adaptive_response_budget() -> ResponseBudget {
    let context_window = [
        "TSIFT_CONTEXT_WINDOW",
        "CODEX_CONTEXT_WINDOW",
        "CLAUDE_CONTEXT_WINDOW",
    ]
    .iter()
    .find_map(|key| {
        std::env::var(key)
            .ok()
            .and_then(|value| value.replace('_', "").parse::<usize>().ok())
    });

    match context_window {
        Some(window) if window <= 64_000 => ResponseBudgetPreset::Small.resolve(),
        Some(window) if window >= 200_000 => ResponseBudgetPreset::Deep.resolve(),
        _ => ResponseBudgetPreset::Normal.resolve(),
    }
}