"""
聊天记录数据库（SQLite）
每条消息独立存储，按客户ID分开，支持查询和展示。
"""

import sqlite3
import os
from datetime import datetime
from typing import List, Dict, Optional

_DB_PATH = os.path.join(os.path.dirname(__file__), "chat_log_db", "chats.db")
_DB_TYPE = os.getenv("DB_TYPE", "sqlite").lower()
_MYSQL_HOST = os.getenv("MYSQL_HOST", "127.0.0.1")
_MYSQL_PORT = int(os.getenv("MYSQL_PORT", "3306"))
_MYSQL_USER = os.getenv("MYSQL_USER", "root")
_MYSQL_PASSWORD = os.getenv("MYSQL_PASSWORD", "")
_MYSQL_DATABASE = os.getenv("MYSQL_DATABASE", "ai_cs")


class _CompatResult:
    def __init__(self, rows=None, rowcount: int = 0, lastrowid: int = 0):
        self._rows = rows or []
        self.rowcount = rowcount
        self.lastrowid = lastrowid

    def fetchall(self):
        return self._rows

    def fetchone(self):
        return self._rows[0] if self._rows else None


class _PyMySQLCompatConn:
    """让 pymysql 连接兼容 sqlite 的 conn.execute 用法。"""

    def __init__(self, conn):
        self._conn = conn

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc, tb):
        if exc_type:
            try:
                self._conn.rollback()
            except Exception:
                pass
        self._conn.close()

    def execute(self, query: str, args=None):
        cur = self._conn.cursor()
        cur.execute(query, args or ())
        rows = cur.fetchall() if cur.description else []
        res = _CompatResult(rows=rows, rowcount=cur.rowcount, lastrowid=getattr(cur, "lastrowid", 0))
        cur.close()
        return res

    def commit(self):
        self._conn.commit()

    def close(self):
        self._conn.close()

def _is_mysql() -> bool:
    return _DB_TYPE in ("mysql", "mariadb")

def _sql(query: str) -> str:
    return query.replace("?", "%s") if _is_mysql() else query


def _get_conn() -> sqlite3.Connection:
    if _is_mysql():
        import pymysql
        conn = pymysql.connect(
            host=_MYSQL_HOST,
            port=_MYSQL_PORT,
            user=_MYSQL_USER,
            password=_MYSQL_PASSWORD,
            database=_MYSQL_DATABASE,
            charset="utf8mb4",
            cursorclass=pymysql.cursors.DictCursor,
            autocommit=False,
        )
        return _PyMySQLCompatConn(conn)
    os.makedirs(os.path.dirname(_DB_PATH), exist_ok=True)
    conn = sqlite3.connect(_DB_PATH)
    conn.row_factory = sqlite3.Row
    return conn


def init_db():
    """建表（首次运行时自动调用）"""
    with _get_conn() as conn:
        if _is_mysql():
            conn.execute("""
                CREATE TABLE IF NOT EXISTS chat_logs (
                    id          INTEGER PRIMARY KEY AUTO_INCREMENT,
                    customer_id VARCHAR(128) NOT NULL,
                    customer_name VARCHAR(255) DEFAULT '',
                    acc_id      VARCHAR(128) DEFAULT '',
                    platform    VARCHAR(64) DEFAULT '',
                    direction   VARCHAR(8) NOT NULL,
                    message     TEXT NOT NULL,
                    msg_type    INTEGER DEFAULT 0,
                    timestamp   DATETIME NOT NULL
                ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4
            """)
            idx_rows = conn.execute("SHOW INDEX FROM chat_logs").fetchall()
            exists = {str(r.get("Key_name", "")) for r in idx_rows}
            if "idx_customer" not in exists:
                conn.execute("CREATE INDEX idx_customer ON chat_logs(customer_id)")
            if "idx_ts" not in exists:
                conn.execute("CREATE INDEX idx_ts ON chat_logs(timestamp)")
            if "idx_acc" not in exists:
                conn.execute("CREATE INDEX idx_acc ON chat_logs(acc_id)")
            # 添加 image_urls 列（如果不存在）
            try:
                conn.execute("ALTER TABLE chat_logs ADD COLUMN image_urls TEXT DEFAULT ''")
            except Exception:
                pass  # 列已存在
        else:
            conn.execute("""
                CREATE TABLE IF NOT EXISTS chat_logs (
                    id          INTEGER PRIMARY KEY AUTOINCREMENT,
                    customer_id TEXT    NOT NULL,
                    customer_name TEXT  DEFAULT '',
                    acc_id      TEXT    DEFAULT '',
                    platform    TEXT    DEFAULT '',
                    direction   TEXT    NOT NULL CHECK(direction IN ('in','out')),
                    message     TEXT    NOT NULL,
                    msg_type    INTEGER DEFAULT 0,
                    timestamp   TEXT    NOT NULL
                )
            """)
            conn.execute("CREATE INDEX IF NOT EXISTS idx_customer ON chat_logs(customer_id)")
            conn.execute("CREATE INDEX IF NOT EXISTS idx_ts ON chat_logs(timestamp)")
            try:
                conn.execute("ALTER TABLE chat_logs ADD COLUMN acc_id TEXT DEFAULT ''")
            except Exception:
                pass
            try:
                conn.execute("ALTER TABLE chat_logs ADD COLUMN image_urls TEXT DEFAULT ''")
            except Exception:
                pass
            conn.execute("CREATE INDEX IF NOT EXISTS idx_acc ON chat_logs(acc_id)")
        conn.commit()


init_db()


# ========== 写入 ==========

def log_message(
    customer_id: str,
    message: str,
    direction: str,              # "in" = 客户发来，"out" = 客服回复
    customer_name: str = "",
    acc_id: str = "",            # 店铺账号ID
    platform: str = "",
    msg_type: int = 0,
    image_urls: str = "",        # 图片URL列表，用\n分隔
):
    """记录一条聊天消息"""
    ts = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    with _get_conn() as conn:
        conn.execute(
            _sql("INSERT INTO chat_logs "
                 "(customer_id, customer_name, acc_id, platform, direction, message, msg_type, timestamp, image_urls) "
                 "VALUES (?,?,?,?,?,?,?,?,?)"),
            (customer_id, customer_name, acc_id, platform, direction, message, msg_type, ts, image_urls),
        )
        conn.commit()


# ========== 查询 ==========

def get_customers(limit: int = 100) -> List[Dict]:
    """返回所有有记录的客户列表（按最新消息时间排序）"""
    with _get_conn() as conn:
        rows = conn.execute("""
            SELECT
                customer_id,
                MAX(customer_name) AS customer_name,
                MAX(platform)      AS platform,
                COUNT(*)           AS total_msgs,
                SUM(direction='in')  AS recv,
                SUM(direction='out') AS sent,
                MAX(timestamp)     AS last_time
            FROM chat_logs
            GROUP BY customer_id
            ORDER BY last_time DESC
            LIMIT %s
        """ if _is_mysql() else """
            SELECT
                customer_id,
                MAX(customer_name) AS customer_name,
                MAX(platform)      AS platform,
                COUNT(*)           AS total_msgs,
                SUM(direction='in')  AS recv,
                SUM(direction='out') AS sent,
                MAX(timestamp)     AS last_time
            FROM chat_logs
            GROUP BY customer_id
            ORDER BY last_time DESC
            LIMIT ?
        """, (limit,)).fetchall()
    return [dict(r) for r in rows]


def get_conversation(customer_id: str, limit: int = 200, acc_id: str = "") -> List[Dict]:
    """返回某客户的最近对话记录（按时间升序）"""
    # 忽略 acc_id 过滤，实现全店铺记忆
    with _get_conn() as conn:
        rows = conn.execute(_sql("""
            SELECT * FROM (
                SELECT id, direction, message, msg_type, timestamp, acc_id
                FROM chat_logs
                WHERE customer_id = ?
                ORDER BY timestamp DESC, id DESC
                LIMIT ?
            ) AS recent
            ORDER BY timestamp ASC, id ASC
        """), (customer_id, limit)).fetchall()
    return [dict(r) for r in rows]


def get_recent_conversation(customer_id: str, acc_id: str = "", limit: int = 10) -> List[Dict]:
    """返回某客户近期对话，忽略 acc_id 过滤"""
    with _get_conn() as conn:
        rows = conn.execute(_sql("""
            SELECT id, direction, message, timestamp, acc_id
            FROM chat_logs
            WHERE customer_id = ?
            ORDER BY id DESC
            LIMIT ?
        """), (customer_id, limit)).fetchall()
        out = [dict(r) for r in reversed(rows)]
    return out


def get_conversation_today(customer_id: str) -> List[Dict]:
    """返回某客户今天的对话"""
    today = datetime.now().strftime("%Y-%m-%d")
    with _get_conn() as conn:
        rows = conn.execute(_sql("""
            SELECT id, direction, message, msg_type, timestamp
            FROM chat_logs
            WHERE customer_id = ? AND timestamp LIKE ?
            ORDER BY timestamp ASC, id ASC
        """), (customer_id, f"{today}%")).fetchall()
    return [dict(r) for r in rows]


def get_daily_stats(date: str = "") -> List[Dict]:
    """
    返回指定日期各店铺的统计数据。
    date 格式 'YYYY-MM-DD'，默认今天。
    每条记录对应一个 acc_id（店铺）。
    """
    if not date:
        date = datetime.now().strftime("%Y-%m-%d")
    with _get_conn() as conn:
        rows = conn.execute(_sql("""
            SELECT
                acc_id,
                platform,
                COUNT(DISTINCT customer_id)          AS unique_customers,
                COUNT(*)                              AS total_msgs,
                SUM(direction='in')                   AS recv,
                SUM(direction='out')                  AS sent,
                MIN(timestamp)                        AS first_msg,
                MAX(timestamp)                        AS last_msg
            FROM chat_logs
            WHERE timestamp LIKE ?
            GROUP BY acc_id
            ORDER BY unique_customers DESC
        """), (f"{date}%",)).fetchall()
    return [dict(r) for r in rows]


def get_daily_conversations(date: str = "") -> List[Dict]:
    """
    返回指定日期每个客户的对话摘要（每人最多取前5条消息用于 AI 摘要）。
    """
    if not date:
        date = datetime.now().strftime("%Y-%m-%d")
    with _get_conn() as conn:
        if _is_mysql():
            rows = conn.execute(_sql("""
                SELECT
                    acc_id,
                    customer_id,
                    MAX(customer_name)  AS customer_name,
                    COUNT(*)            AS msg_count,
                    GROUP_CONCAT(
                        CONCAT(CASE WHEN direction='in' THEN '买:' ELSE '客:' END, LEFT(message,40))
                        SEPARATOR ' | '
                    ) AS snippet
                FROM chat_logs
                WHERE timestamp LIKE ?
                GROUP BY acc_id, customer_id
                ORDER BY acc_id, MAX(timestamp) DESC
            """), (f"{date}%",)).fetchall()
        else:
            rows = conn.execute(_sql("""
                SELECT
                    acc_id,
                    customer_id,
                    MAX(customer_name)  AS customer_name,
                    COUNT(*)            AS msg_count,
                    GROUP_CONCAT(
                        CASE WHEN direction='in'  THEN '买:' || SUBSTR(message,1,40)
                             ELSE '客:' || SUBSTR(message,1,40) END,
                        ' | '
                    ) AS snippet
                FROM chat_logs
                WHERE timestamp LIKE ?
                GROUP BY acc_id, customer_id
                ORDER BY acc_id, MAX(timestamp) DESC
            """), (f"{date}%",)).fetchall()
    return [dict(r) for r in rows]


def search_messages(keyword: str, customer_id: Optional[str] = None, limit: int = 50) -> List[Dict]:
    """全文搜索消息"""
    if customer_id:
        with _get_conn() as conn:
            rows = conn.execute(_sql("""
                SELECT customer_id, customer_name, direction, message, timestamp
                FROM chat_logs
                WHERE customer_id = ? AND message LIKE ?
                ORDER BY timestamp DESC LIMIT ?
            """), (customer_id, f"%{keyword}%", limit)).fetchall()
    else:
        with _get_conn() as conn:
            rows = conn.execute(_sql("""
                SELECT customer_id, customer_name, direction, message, timestamp
                FROM chat_logs
                WHERE message LIKE ?
                ORDER BY timestamp DESC LIMIT ?
            """), (f"%{keyword}%", limit)).fetchall()
    return [dict(r) for r in rows]


def get_latest_messages(limit: int = 20) -> List[Dict]:
    with _get_conn() as conn:
        rows = conn.execute(_sql("""
            SELECT id, customer_id, customer_name, direction, message, timestamp
            FROM chat_logs
            ORDER BY id DESC LIMIT ?
        """), (limit,)).fetchall()
    return [dict(r) for r in rows]