ChingTech OS：整合 Printer MCP 與 ERPNext MCP

# 需透過 LibreOffice 轉 PDF 的格式
OFFICE_EXTENSIONS = {
    ".docx", ".xlsx", ".pptx", ".doc", ".xls", ".ppt",
    ".odt", ".ods", ".odp",
}

# printer-mcp 可直接列印的格式
PRINTABLE_EXTENSIONS = {
    ".pdf", ".txt", ".log", ".csv",
    ".png", ".jpg", ".jpeg", ".gif", ".bmp", ".tiff", ".webp",
}

# 允許存取的路徑前綴
ALLOWED_PRINT_PATHS = ("/mnt/nas/", "/tmp/ctos/")

@mcp.tool()
async def prepare_print_file(
    file_path: str,
    ctos_user_id: int | None = None,
) -> str:
    """將虛擬路徑轉換為可列印的絕對路徑，Office 文件會自動轉為 PDF"""
    # 路徑轉換：虛擬路徑 → 絕對路徑
    if "://" in file_path:
        actual_path = Path(path_manager.to_filesystem(file_path))
    else:
        actual_path = Path(file_path)

    # 安全檢查
    if not any(str(actual_path).startswith(prefix) for prefix in ALLOWED_PRINT_PATHS):
        return "不允許存取此路徑的檔案。"

    ext = actual_path.suffix.lower()

    if ext in PRINTABLE_EXTENSIONS:
        # 直接可印，回傳絕對路徑
        return f"請使用 printer-mcp 的 print_file 工具列印：\n絕對路徑：{actual_path}"

    if ext in OFFICE_EXTENSIONS:
        # Office 文件，先轉 PDF
        proc = await asyncio.create_subprocess_exec(
            "libreoffice", "--headless", "--convert-to", "pdf",
            "--outdir", "/tmp/ctos/print", str(actual_path),
        )
        await proc.communicate()
        pdf_path = f"/tmp/ctos/print/{actual_path.stem}.pdf"
        return f"已轉換為 PDF：\n絕對路徑：{pdf_path}"

PRINTER_TOOLS_PROMPT = """【列印功能】
...
步驟 2 - 實際列印（printer-mcp 工具）：
- mcp__printer__print_file: 將檔案送至印表機列印
  · color_mode: 色彩模式（可選，gray/color，預設 gray。
    除非用戶要求彩色列印，否則一律用 gray）
..."""

@mcp.tool()
async def send_nas_file(
    file_path: str,
    line_user_id: str | None = None,
    line_group_id: str | None = None,
    telegram_chat_id: str | None = None,   # 新增
    ctos_user_id: int | None = None,
) -> str:
    """直接發送 NAS 檔案給用戶"""
    if not line_user_id and not line_group_id and not telegram_chat_id:
        return "錯誤：請從【對話識別】區塊取得 ID"
    ...

# 加入 ERPNext MCP 工具（廠商/客戶/庫存/專案管理）
erpnext_tools = [
    "mcp__erpnext__list_documents",
    "mcp__erpnext__get_document",
    "mcp__erpnext__create_document",
    "mcp__erpnext__update_document",
    "mcp__erpnext__delete_document",
    "mcp__erpnext__submit_document",
    "mcp__erpnext__cancel_document",
    "mcp__erpnext__run_report",
    "mcp__erpnext__get_stock_balance",
    "mcp__erpnext__get_stock_ledger",
    "mcp__erpnext__get_count",
    "mcp__erpnext__get_list_with_summary",
    "mcp__erpnext__run_method",
    # ... 共 19 個工具
]

# 所有工具合併
all_tools = (
    builtin_tools + mcp_tools + nanobanana_tools
    + printer_tools + erpnext_tools + ["Read"]
)

printer_tools = [
    "mcp__printer__print_file",
    "mcp__printer__list_printers",
    "mcp__printer__printer_status",
    "mcp__printer__cancel_job",
    "mcp__printer__print_test_page",
]

# 專案管理（遷移至 ERPNext）
PROJECT_TOOLS_PROMPT = """【專案管理】（使用 ERPNext）
專案管理功能已遷移至 ERPNext 系統，請使用 ERPNext MCP 工具操作：

【查詢專案】
- mcp__erpnext__list_documents: 查詢專案列表
  · doctype: "Project"
  · filters: 可依狀態過濾，如 '{"status": "Open"}'

【任務管理】
- mcp__erpnext__list_documents: 查詢專案任務
  · doctype: "Task"
  · filters: '{"project": "專案名稱"}'

【操作範例】
1. 查詢進行中的專案：
   mcp__erpnext__list_documents(doctype="Project", filters='{"status":"Open"}')
2. 更新任務狀態：
   mcp__erpnext__update_document(doctype="Task", name="TASK-00001",
     data='{"status":"Completed"}')
"""

INVENTORY_TOOLS_PROMPT = """【物料/庫存管理】（使用 ERPNext）

【廠商/客戶管理】
使用通用查詢工具：
- mcp__erpnext__list_documents: 查詢廠商/客戶列表
  · doctype: "Supplier" 或 "Customer"
  · filters: 可依名稱、群組等條件過濾
- mcp__erpnext__get_document: 查詢廠商/客戶完整資料

【查詢庫存】
- mcp__erpnext__get_stock_balance: 查詢即時庫存
- mcp__erpnext__get_stock_ledger: 查詢庫存異動記錄
"""

APP_PROMPT_MAPPING: dict[str, str] = {
    "project-management": PROJECT_TOOLS_PROMPT,
    "inventory-management": INVENTORY_TOOLS_PROMPT,
    "knowledge-base": KNOWLEDGE_TOOLS_PROMPT,
    "file-manager": FILE_TOOLS_PROMPT,
    "ai-assistant": AI_IMAGE_TOOLS_PROMPT + "\n\n" + AI_DOCUMENT_TOOLS_PROMPT,
    "printer": PRINTER_TOOLS_PROMPT,
}

def generate_tools_prompt(app_permissions: dict[str, bool]) -> str:
    sections = [BASE_TOOLS_PROMPT]  # 基礎工具永遠包含
    for app_id, prompt_section in APP_PROMPT_MAPPING.items():
        if app_permissions.get(app_id, False):
            sections.append(prompt_section)
    return "\n\n".join(sections)

# 原有的專案/廠商/物料管理工具已完全移除（遷移至 ERPNext）
DEPRECATED_TOOLS: dict[str, str] = {}

# ERPNext 工具對應指引（已整合至 AI Agent Prompt）
ERPNEXT_GUIDANCE: dict[str, str] = {}

@mcp.tool()
async def create_share_link(resource_type: str, resource_id: str, ...) -> str:
    # 驗證資源類型（專案相關類型已移除）
    valid_types = ("knowledge", "nas_file")
    if resource_type not in valid_types:
        if resource_type in ("project", "project_attachment"):
            return "錯誤：專案分享功能已遷移至 ERPNext"
        return f"錯誤：資源類型必須是 {', '.join(valid_types)}"

def upgrade() -> None:
    connection = op.get_bind()
    raw_conn = connection.connection.dbapi_connection
    cur = raw_conn.cursor()

    # 1. 執行 schema SQL（建立表格結構）
    schema_path = os.path.join(base_path, 'clean_schema.sql')
    with open(schema_path, 'r', encoding='utf-8') as f:
        schema_sql = f.read()
    cur.execute(schema_sql)

    # 2. 建立分區表
    cur.execute("SELECT create_ai_logs_partition()")
    cur.execute("SELECT create_next_month_partitions()")

DEFAULT_APP_PERMISSIONS: dict[str, bool] = {
    "file-manager": True,
    "project-management": True,
    "inventory-management": True,
    "knowledge-base": True,
    "ai-assistant": True,
    "md2ppt": True,
    "md2doc": True,
    "printer": True,       # 列印功能，預設開放
    # ...
}

TOOL_APP_MAPPING: dict[str, str | None] = {
    # 列印前置處理工具
    "prepare_print_file": "printer",
    # ...
}

@mcp.tool()
async def prepare_print_file(file_path: str, ctos_user_id: int | None = None) -> str:
    if ctos_user_id:
        allowed, error_msg = await check_mcp_tool_permission(
            "prepare_print_file", ctos_user_id
        )
        if not allowed:
            return f"❌ {error_msg}"
    ...