//! F-15 上下文压缩 — LLM 摘要公共函数
//!
//! 阶段1 基础(本文件):仅暴露 `compress_via_llm` 函数,供阶段2 IPC
//! (`ai_chat_compress_context`)与阶段3 agentic loop 自动压缩共用。
//!
//! 模式严格对齐 [`super::title`] 的 `generate_title_via_llm`:非流式
//! `provider.complete()` + `LlmConcurrency` 双层 Semaphore 限流 + select_model_id
//! 路由(Lite 智力, 无 tool_use, 标准成本上限)兜底 `default_model`。失败返 `Err`,
//! 调用方按场景降级(手动压缩报错给用户;自动压缩降级为原裁剪行为)。

use df_ai::provider::{ChatMessage, CompletionRequest, LlmProvider};
// F-15: 压缩路由 — TaskRequirements(Standard 智力即可,无需工具)。压缩是结构化
// 摘要任务,选 Standard 智力保摘要质量;不传 tools(纯文本出,避免 LLM 调工具跑偏)。
// 池空/无匹配兜底 default_model(与 title.rs 一致,行为可预期)。
use df_ai::router::{
    select_model_id, Modality, TaskRequirements,
};
use df_storage::models::AiProviderRecord;

use crate::state::LlmConcurrency;

use super::prompt::compress_prompt;

/// 调 LLM 非流式把 active 消息压缩为四段式摘要。
///
/// - 输入:`provider`(经 [`super::secret::build_provider_for`] 构造,含 keyring 解析)
///   + `provider_config`(取 `default_model` 兜底 + `model_configs` 路由池)
///   + `active_msgs`(待压缩的 active 消息,来自 [`ContextManager::compress_old_messages`]
///     的返回或阶段2 IPC 取全段 active)
///   + `lang`("zh"/"en",与系统提示词语言一致,影响 `compress_prompt` 选模板)
/// - 输出:`Result<String, String>` — `Ok` 为 LLM 生成的摘要文本(已 trim);
///   `Err` 为 provider 调用失败(底层 `anyhow::Error` 转 String,调用方降级不阻塞)。
///
/// 模型选择:`select_model_id(TaskRequirements{无 tool_use})`,
/// 池空/无匹配兜底 `default_model`(与 `generate_title_via_llm` 一致)。
///
/// 限流:复用 `LlmConcurrency` 双层 Semaphore(压缩属独立 LLM 调用,纳入全局 + 单对话限流,
/// 不与主 stream_llm 抢资源失控)。
///
/// 注意:本函数只负责"喂 LLM 出摘要文本",不改 ContextManager 状态(标 compressed +
/// 插摘要 system 由调用方做,职责分离,便于单元测试与多调用点复用)。
//
// dead_code:阶段1 基础函数,阶段2 IPC(ai_chat_compress_context)/ 阶段3 agentic
// loop 自动压缩会调用。本阶段只暴露 + 单测,不接调用点(零行为变化原则)。
#[allow(dead_code)]
pub(crate) async fn compress_via_llm(
    provider: &dyn LlmProvider,
    provider_config: &AiProviderRecord,
    active_msgs: Vec<ChatMessage>,
    lang: &str,
    conv_id: &str,
    llm_concurrency: &LlmConcurrency,
) -> Result<String, String> {
    if active_msgs.is_empty() {
        return Err("无可压缩消息(active 消息为空)".to_string());
    }

    // 路由选模型:无 tool_use(纯文本摘要)。
    // None(池空/无匹配)→ 兜底 default_model(行为不变)。
    let compress_req = TaskRequirements {
        modalities: vec![Modality::Text],
        needs_tool_use: false,
        estimated_context: 0,
    };
    let model = select_model_id(&compress_req, &provider_config.model_configs)
        .unwrap_or_else(|| provider_config.default_model.clone());

    // 四段式 system prompt + active 消息原文,喂 complete() 非流式出摘要。
    let mut prompt = vec![ChatMessage::system(compress_prompt(lang))];
    prompt.extend(active_msgs);
    let request = CompletionRequest {
        model,
        messages: prompt,
        temperature: Some(0.2),
        max_tokens: Some(1024),
        stream: false,
        tools: None,
        tool_choice: None,
        reasoning_content: None,
    };

    // LLM 并发限流(压缩属独立调用,纳入双层 Semaphore)
    // F-09 B 批5: per_conv 改 HashMap<conv_id>,压缩在 loop 内针对本对话,用 conv_id 共享限流槽。
    let _global_permit = llm_concurrency.acquire_global().await;
    let _per_conv_permit = llm_concurrency.acquire_per_conv(conv_id).await;

    let resp = provider
        .complete(request)
        .await
        .map_err(|e| format!("LLM 压缩调用失败: {}", e))?;
    let summary = clean_summary(&resp.text);
    if summary.is_empty() {
        return Err("LLM 压缩返回空摘要".to_string());
    }
    Ok(summary)
}

/// 清理 LLM 返回的摘要:去首尾空白 / 包裹性 markdown 代码围栏,空则原样返回
/// (调用方据空值报错)。保留内部 markdown 结构(## 标题 / 列表),摘要本身就是结构化文本。
#[allow(dead_code)]
fn clean_summary(raw: &str) -> String {
    let t = raw.trim();
    // 去掉可能的整体代码围栏(LLM 偶尔把整段包成 ```markdown ... ```)
    let t = t
        .strip_prefix("```markdown")
        .or_else(|| t.strip_prefix("```"))
        .unwrap_or(t)
        .trim_start_matches('\n');
    let t = t.strip_suffix("```").unwrap_or(t).trim();
    t.to_string()
}

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

    #[test]
    fn clean_summary_trims_and_strips_fences() {
        assert_eq!(clean_summary("  hello  "), "hello");
        assert_eq!(clean_summary("```markdown\n## 意图\nfoo\n```"), "## 意图\nfoo");
        assert_eq!(clean_summary("```\n## 决策\nbar\n```"), "## 决策\nbar");
        // 保留内部 markdown 结构(摘要本就是结构化)
        assert_eq!(clean_summary("## 意图\n- a\n- b"), "## 意图\n- a\n- b");
    }

    #[test]
    fn clean_summary_empty_stays_empty() {
        assert_eq!(clean_summary("   "), "");
        assert_eq!(clean_summary(""), "");
    }

    #[test]
    fn compress_prompt_returns_template_per_lang() {
        // 中文模板(默认)
        let zh = compress_prompt("zh");
        assert!(zh.contains("意图"));
        assert!(zh.contains("决策"));
        assert!(zh.contains("文件"));
        assert!(zh.contains("约束"));
        // 英文模板
        let en = compress_prompt("en");
        assert!(en.contains("Intent"));
        assert!(en.contains("Decisions"));
        assert!(en.contains("Files"));
        assert!(en.contains("Constraints"));
        // 未知语言降级中文
        assert_eq!(compress_prompt("fr"), compress_prompt("zh"));
    }
}
