package classify

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/aisim/kb-cli/internal/llm"
)

// Document 文档信息
type Document struct {
	Path    string `json:"path"`
	Title   string `json:"title"`
	Content string `json:"content"`
}

// Classification 分类结果
type Classification struct {
	Path       string  `json:"path"`
	Platform   string  `json:"platform"`
	Device     string  `json:"device"`
	Confidence float64 `json:"confidence"`
}

// ClassificationResult 分类结果集合
type ClassificationResult struct {
	Total         int              `json:"total"`
	Classified    int              `json:"classified"`
	NeedsReview   int              `json:"needs_review"`
	Items         []Classification `json:"items"`
}

// ScanDocuments 扫描知识库文档
func ScanDocuments(vaultPath string) ([]Document, error) {
	var docs []Document

	// 转换为绝对路径
	absPath, err := filepath.Abs(vaultPath)
	if err != nil {
		return nil, fmt.Errorf("转换路径失败: %w", err)
	}

	// 扫描整个知识库的所有 .md 文件
	err = filepath.Walk(absPath, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		
		// 跳过隐藏目录和特殊目录
		if info.IsDir() {
			name := info.Name()
			if strings.HasPrefix(name, ".") || name == "node_modules" {
				return filepath.SkipDir
			}
			// 跳过待审阅目录
			if name == "待审阅" {
				return filepath.SkipDir
			}
			return nil
		}
		
		// 只处理 .md 文件
		if !strings.HasSuffix(path, ".md") {
			return nil
		}

		content, err := os.ReadFile(path)
		if err != nil {
			return err
		}

		relPath, err := filepath.Rel(absPath, path)
		if err != nil {
			relPath = path
		}

		title := extractTitle(string(content))
		summary := extractSummary(string(content), 500)

		docs = append(docs, Document{
			Path:    relPath,
			Title:   title,
			Content: summary,
		})
		return nil
	})
	
	if err != nil {
		return nil, fmt.Errorf("扫描知识库失败: %w", err)
	}

	return docs, nil
}

// extractTitle 从 frontmatter 或第一行 # 提取标题
func extractTitle(content string) string {
	lines := strings.Split(content, "\n")
	inFrontmatter := false

	for _, line := range lines {
		line = strings.TrimSpace(line)
		if line == "---" {
			inFrontmatter = !inFrontmatter
			continue
		}
		if inFrontmatter {
			if strings.HasPrefix(line, "title:") {
				title := strings.TrimPrefix(line, "title:")
				title = strings.TrimSpace(title)
				title = strings.Trim(title, "\"'")
				return title
			}
		} else if strings.HasPrefix(line, "# ") {
			return strings.TrimPrefix(line, "# ")
		}
	}
	return "未知标题"
}

// extractSummary 提取摘要（前 maxLen 字）
func extractSummary(content string, maxLen int) string {
	// 跳过 frontmatter
	lines := strings.Split(content, "\n")
	startIdx := 0
	inFrontmatter := false

	for i, line := range lines {
		if strings.TrimSpace(line) == "---" {
			if !inFrontmatter {
				inFrontmatter = true
			} else {
				startIdx = i + 1
				break
			}
		}
	}

	// 提取正文
	summary := strings.Join(lines[startIdx:], "\n")
	summary = strings.TrimSpace(summary)

	if len(summary) > maxLen {
		summary = summary[:maxLen] + "..."
	}

	return summary
}

// ClassifyBatch 批量分类文档
func ClassifyBatch(docs []Document, llmClient *llm.Client) ([]Classification, error) {
	// 构建 prompt
	prompt := buildClassifyPrompt(docs)

	// 调用 LLM
	response, err := llmClient.ClassifyDocuments(prompt)
	if err != nil {
		return nil, err
	}

	// 解析响应
	var results []Classification
	if err := json.Unmarshal([]byte(response), &results); err != nil {
		return nil, fmt.Errorf("解析 LLM 响应失败: %w", err)
	}

	return results, nil
}

// buildClassifyPrompt 构建分类 prompt
func buildClassifyPrompt(docs []Document) string {
	prompt := `你是知识库分类专家。分析以下文档，判断其所属平台、设备和内容类型。

## 实体定义（必须严格遵守）

### 硬件设备
| 设备 | 归属 | 说明 |
|------|------|------|
| 智能枪 | 运营管理平台配套 | 智能控制箱+智能枪头，独立4G网络，可独立工作或安装在电子秤中 |
| 电子秤 | 电子秤平台 | 含扫码枪（分防爆/非防爆），扫码枪是电子秤配件，**不是智能枪** |
| 智能阀 | 运营管理平台基础 | NFC识别芯片，是运营平台气瓶管理的基础 |
| 艾信盒子 | 两平台共用 | 4G通信设备，同步存储充装数据，控制上传第三方平台 |

### 软件平台
| 平台 | 域名 | 用户 | 核心功能 |
|------|------|------|----------|
| 运营管理平台 | rb.zhiheiot.com | 老板/管理层/客服/配送调度 | 经营管理、售后、配送、工单、会员 |
| 电子秤平台 | elc.zhiheiot.com | 气站站长/充装员/开票员 | 充装作业（扫码、充装、称重、开票/补单）+ 终端配送（简化版） |

### App终端
| App | 归属平台 | 功能 |
|-----|----------|------|
| 易配送App | 运营管理平台 | NFC芯片识别配送 |
| 安全用气App | 电子秤平台 | 二维码配送 |
| 艾信助手App | 融合两平台 | 建档 + 充前/充后检查 |

### 小程序
| 小程序 | 归属 | 用户 | 功能 |
|--------|------|------|------|
| 艾信LPG物联网小程序 | 电子秤平台 | 气站管理员/配送员 | 报表/迎检/配送 |
| 艾信发货小程序 | 内部工具 | 艾信内部员工 | 设备总览、客户资料、气站资料 |

## 关键区分点（必须遵守）

| 容易混淆 | 正确归属 |
|----------|----------|
| 智能枪 vs 扫码枪 | 智能枪是独立设备（4G网络）；扫码枪是电子秤配件 |
| 开票/补单/充装 | 电子秤平台功能 |
| 主板版本过低 | 智能枪问题 |
| 艾信发货小程序 | 内部工具，可控制所有智能枪、电子秤 |
| 艾信LPG物联网小程序 | 电子秤平台气站管理员用 |
| 配送功能 | 两平台都有，运营平台用易配送App，电子秤平台用安全用气App |

## 分类规则

1. **platform**（属于哪个平台，必须使用中文）:
   - 电子秤平台：充装作业、开票、电子秤设备、安全用气App、艾信LPG小程序
   - 运营管理平台：配送、档案、会员、工单、易配送App、智能枪、智能阀
   - 共有：两个平台都涉及（艾信盒子、艾信助手App、充装记录同步）
   - 第三方平台：祥康、监管平台等
   - 通用：不涉及具体平台/内部工具

2. **device**（涉及什么设备/App，必须使用中文）:
   - 电子秤：含扫码枪、电磁阀、称重传感器、显示屏、键盘
   - 智能枪：智能控制箱+智能枪头，独立4G网络
   - 智能阀：NFC识别智能阀的芯片
   - 艾信盒子：4G通信设备
   - 安全用气App：电子秤平台配送端
   - 易配送App：运营管理平台配送端
   - 艾信助手App：融合两平台建档
   - 艾信LPG小程序：电子秤平台管理+配送
   - 艾信发货小程序：内部工具
   - 无：不涉及具体设备

3. **content_type**（内容类型）:
   - FAQ / PRD / 知识 / 案例 / 实体 / 其他

## 输出格式

输出 JSON 数组，每项包含：
- path: 文件路径（字符串）
- platform: 平台名称（中文，必须是上述枚举值之一）
- device: 设备名称（中文，必须是上述枚举值之一）
- content_type: 内容类型（字符串，必须是上述枚举值之一）
- confidence: 置信度（浮点数，范围 0.0-1.0，例如 0.95）

示例输出：
[
  {
    "path": "FAQ/充装类/001-智能枪通气杆卡住漏气.md",
    "platform": "运营管理平台",
    "device": "智能枪",
    "content_type": "FAQ",
    "confidence": 0.95
  }
]

文档列表：
`

	for _, doc := range docs {
		prompt += fmt.Sprintf("\n文件: %s\n标题: %s\n摘要: %s\n", doc.Path, doc.Title, doc.Content)
	}

	prompt += "\n请输出 JSON 数组："

	return prompt
}

// SaveClassification 保存分类结果
func SaveClassification(vaultPath string, results []Classification) error {
	// 统计
	total := len(results)
	classified := 0
	needsReview := 0

	for _, r := range results {
		if r.Confidence >= 0.8 {
			classified++
		} else {
			needsReview++
		}
	}

	result := ClassificationResult{
		Total:       total,
		Classified:  classified,
		NeedsReview: needsReview,
		Items:       results,
	}

	path := filepath.Join(vaultPath, "classification.json")
	data, err := json.MarshalIndent(result, "", "  ")
	if err != nil {
		return err
	}

	return os.WriteFile(path, data, 0644)
}