import os
import sys
import argparse
import sqlite3
import pandas as pd
from datetime import datetime

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from config import DATA_DIR, DATE_TODAY, OUTPUT_DIR
from core.database import (
    get_connection, 
    get_available_snapshots, 
    get_all_rules_from_db,
    load_scud_from_db_by_snapshot,
    get_latest_snapshot_time
)

DB_PATH = os.path.join(DATA_DIR, "scud_orion_ai.db")


def print_stats():
    """Выводит общую статистику по записям в таблицах БД."""
    print("\n" + "=" * 60)
    print("📊 СТАТИСТИКА БАЗЫ ДАННЫХ SQLITE (scud_orion_ai.db):")
    print("=" * 60)
    with get_connection() as conn:
        cursor = conn.cursor()
        tables = ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base']
        for t in tables:
            cursor.execute(f"SELECT COUNT(*) FROM {t}")
            cnt = cursor.fetchone()[0]
            print(f"  • Таблица [{t:<18}]: {cnt:>6} записей")
    print("=" * 60 + "\n")


def print_snapshots_list(date_str=None):
    """Выводит реестр снапшотов с отображением даты и точного времени среза."""
    rows = get_available_snapshots(date_str)
    
    print("\n" + "=" * 105)
    print(f"📸 РЕЕСТР СОХРАНЕННЫХ СНАПШОТОВ (СВЕРХУ СВЕЖИЕ) {'ЗА ЛОГИ ' + date_str if date_str else ''}:")
    print("=" * 105)
    
    header = f"{'ID снапшота':<16} | {'Дата снапшота (создания)':<24} | {'Дата и время среза':<20} | {'Записей':<8}"
    print(header)
    print("-" * 105)

    if not rows:
        print("Снапшотов пока нет.")
        print("=" * 105 + "\n")
        return

    def snapshot_sort_key(row):
        snap_id = row[0] or ""
        snap_time = row[2] or ""
        
        seq_num = 0
        if "-" in snap_id:
            parts = snap_id.replace("Y", "").split("-")
            if len(parts) > 1 and parts[1].isdigit():
                seq_num = int(parts[1])
                
        return (snap_time, seq_num)

    sorted_rows = sorted(rows, key=snapshot_sort_key, reverse=True)

    for r in sorted_rows:
        snap_id = r[0] if r[0] else '----------'
        log_date = r[1] if r[1] else '—'
        snap_time = r[2] if r[2] else '—'
        count = r[3]

        # Извлекаем время (ЧЧ:ММ:СС) из snapshot_time, если оно есть
        time_part = "—"
        if snap_time and " " in snap_time:
            time_part = snap_time.split(" ")[1]

        slice_datetime_str = f"{log_date} {time_part}" if time_part != "—" else log_date

        if not snap_id.startswith("Y"):
            formatted_snap_id = f" {snap_id}"
        else:
            formatted_snap_id = snap_id
        
        print(f"{formatted_snap_id:<16} | {snap_time:<24} | {slice_datetime_str:<20} | {count:<8}")

    print("=" * 105 + "\n")


def delete_snapshot_by_id(snapshot_id: str):
    """Удаляет конкретный снапшот из таблицы scud_logs по его ID."""
    with get_connection() as conn:
        cursor = conn.cursor()
        cursor.execute("DELETE FROM scud_logs WHERE snapshot_id = ?", (snapshot_id,))
        deleted_count = cursor.rowcount
        conn.commit()
    print(f"\n[✓] Успешно удален снапшот [{snapshot_id}]. Удалено строк: {deleted_count}\n")
    return deleted_count


def delete_snapshots_by_date(date_str: str):
    """Удаляет все снапшоты за указанную дату (например, '04.08.2026')."""
    with get_connection() as conn:
        cursor = conn.cursor()
        cursor.execute("DELETE FROM scud_logs WHERE log_date = ? OR snapshot_id LIKE ?", (date_str, f"%{date_str.replace('.', '')}%"))
        deleted_count = cursor.rowcount
        conn.commit()
    print(f"\n[✓] Успешно удалены все снапшоты за дату [{date_str}]. Удалено строк: {deleted_count}\n")
    return deleted_count


def inspect_scud(snapshot_id=None, date_str=None, export_xlsx=None):
    """Инспектирует логи СКУД за выбранный снапшот или дату и опционально сохраняет XLSX."""
    target_date = date_str if date_str else DATE_TODAY
    
    print("\n" + "=" * 90)
    if snapshot_id:
        print(f"🔍 ИНСПЕКЦИЯ СКУД ПО СНАПШОТУ [{snapshot_id}] (Дата: {target_date}):")
    else:
        print(f"🔍 ИНСПЕКЦИЯ СКУД ЗА ТЕКУЩУЮ ДАТУ [{target_date}] (ПОСЛЕДНИЙ СРЕЗ):")
    print("=" * 90)

    df = load_scud_from_db_by_snapshot(target_date, snapshot_param=snapshot_id)

    if df.empty:
        print(f"Записи СКУД не найдены.")
        print("=" * 90 + "\n")
        return

    total = len(df)
    present_cnt = len(df[df['Пришел'] == True]) if 'Пришел' in df.columns else 0
    absent_cnt = total - present_cnt

    print(f"Всего записей: {total} | Пришли: {present_cnt} | Не пришли: {absent_cnt}")
    print("-" * 90)

    cols_to_show = ['fio', 'department', 'position', 'time_in', 'first_activity', 'time_out', 'is_present', 'anomaly_flag', 'snapshot_id']
    existing_cols = [c for c in cols_to_show if c in df.columns]
    
    print(df[existing_cols].head(30).to_string(index=False))
    if len(df) > 30:
        print(f"\n... и ещё {len(df) - 30} строк.")

    if export_xlsx:
        out_path = export_xlsx if export_xlsx.endswith('.xlsx') else f"{export_xlsx}.xlsx"
        if not os.path.isabs(out_path):
            out_path = os.path.join(OUTPUT_DIR, out_path)
        
        df.to_excel(out_path, index=False)
        print("\n" + "*" * 90)
        print(f"[✓] УСПЕШНЫЙ ЭКСПОРТ ДЕБАГ-ФАЙЛА В EXCEL: {out_path}")
        print("*" * 90)

    print("=" * 90 + "\n")


def print_anomalies():
    """Выводит список аномалий СКУД из БД."""
    print("\n" + "=" * 80)
    print("🚨 ИСТОРИЯ НАЙДЕННЫХ АНОМАЛИЙ СКУД ⟷ 1С:")
    print("=" * 80)
    with get_connection() as conn:
        df = pd.read_sql_query("SELECT anomaly_date, fio, anomaly_type, details FROM anomalies_history ORDER BY id DESC LIMIT 50", conn)
        if df.empty:
            print("Аномалии не найдены.")
        else:
            print(df.to_string(index=False))
    print("=" * 80 + "\n")


def print_rules():
    """Выводит правила базы знаний ИИ."""
    rules = get_all_rules_from_db()
    print("\n" + "=" * 80)
    print("🧠 ПРАВИЛА БАЗЫ ЗНАНИЙ ИИ:")
    print("=" * 80)
    if not rules:
        print("База знаний пуста.")
    else:
        for idx, r in enumerate(rules, 1):
            print(f"  {idx}. {r}")
    print("=" * 80 + "\n")


def dump_all_to_excel(out_filename="db_dump_full.xlsx"):
    """Дампит всю базу SQLite во многостраничный Excel."""
    out_path = os.path.join(OUTPUT_DIR, out_filename)
    print(f"\n[🔄] Создание полного дампа БД в файл: {out_path} ...")
    with get_connection() as conn, pd.ExcelWriter(out_path, engine='openpyxl') as writer:
        for table in ['scud_logs', 'zup_staff', 'zup_absences', 'anomalies_history', 'ai_knowledge_base']:
            df = pd.read_sql_query(f"SELECT * FROM {table}", conn)
            df.to_excel(writer, sheet_name=table[:31], index=False)
    print(f"[✓] Дамп успешно сохранен: {out_path}\n")

def print_system_prompts():
    """Выводит все системные промпты из базы данных."""
    print("\n" + "=" * 80)
    print("📝 СИСТЕМНЫЕ ПРОМПТЫ (system_prompts):")
    print("=" * 80)
    with get_connection() as conn:
        cursor = conn.cursor()
        cursor.execute("SELECT id, name, is_active, updated_at, prompt_text FROM system_prompts ORDER BY id DESC")
        rows = cursor.fetchall()
        if not rows:
            print("Таблица system_prompts пуста.")
        else:
            for r in rows:
                print(f"ID: {r[0]} | Name: {r[1]} | Active: {r[2]} | Updated: {r[3]}")
                print("-" * 80)
                print(f"{r[4]}\n")
    print("=" * 80 + "\n")

def print_session_states():
    """Выводит текущие активные сессии и превью (session_states)."""
    print("\n" + "=" * 80)
    print("🔄 АКТИВНЫЕ СЕССИИ И ПРЕВЬЮ (session_states):")
    print("=" * 80)
    with get_connection() as conn:
        cursor = conn.cursor()
        cursor.execute("SELECT session_id, state_type, updated_at, pending_data FROM session_states")
        rows = cursor.fetchall()
        if not rows:
            print("Таблица session_states пуста (нет активных превью).")
        else:
            for r in rows:
                print(f"Session: {r[0]} | Type: {r[1]} | Updated: {r[2]}")
                print("-" * 80)
                print(f"Pending Data:\n{r[3]}\n")
    print("=" * 80 + "\n")    


def main():
    help_text = """
CLI-утилита инспекции и управления SQLite базой данных СКУД (scud_orion_ai.db)

ДОСТУПНЫЕ КОМАНДЫ:
  stats                                                      -- Общая статистика строк по всем таблицам БД
  snapshots [ДД.ММ.ГГГГ]                                     -- Посмотреть список всех снапшотов (опционально за дату)
  scud [--snapshot ID] [--export-xlsx NAME]                 -- Инспекция логов СКУД и опциональный экспорт в Excel
  anomalies                                                  -- Посмотреть все найденные аномалии СКУД ⟷ 1С
  rules                                                      -- Посмотреть правила Базы Знаний ИИ из SQLite
  prompts                                                    -- Посмотреть системные промпты (system_prompts)
  sessions                                                   -- Посмотреть активные сессии и превью (session_states)
  dump [output.xlsx]                                         -- Полный дамп всех таблиц БД в многостраничный Excel
  snapshot del [ID] или [--day ДД.ММ.ГГГГ]                   -- Удаление снапшотов по ID или за весь день

ПРИМЕРЫ ЗАПУСКА:
  python scripts/db_cli.py stats
  python scripts/db_cli.py prompts
  python scripts/db_cli.py sessions
  python scripts/db_cli.py snapshots 04.08.2026
  python scripts/db_cli.py scud --snapshot Y20260805-007 --export-xlsx вчера_1
  python scripts/db_cli.py snapshot del Y20260805-007
  python scripts/db_cli.py snapshot del --day 04.08.2026
"""
    parser = argparse.ArgumentParser(
        description=help_text,
        formatter_class=argparse.RawDescriptionHelpFormatter,
        add_help=True
    )
    parser.add_argument('command', nargs='?', default=None, choices=['stats', 'snapshots', 'scud', 'anomalies', 'rules', 'prompts', 'sessions', 'dump', 'snapshot'], help="Основная команда")
    parser.add_argument('action', nargs='?', default=None, help="Дополнительное действие (например, 'del')")
    parser.add_argument('param', nargs='?', default=None, help="Параметр (дата, ID снапшота или имя файла)")
    parser.add_argument('--snapshot', type=str, default=None, help="ID конкретного снапшота для инспекции")
    parser.add_argument('--export-xlsx', type=str, default=None, help="Экспорт среза СКУД в Excel-файл")
    parser.add_argument('--day', type=str, default=None, help="Удалить снапшоты за конкретный день (ДД.ММ.ГГГГ)")

    # Если аргументы вообще не переданы, выводим статистику по умолчанию (или справку)
    if len(sys.argv) == 1:
        print_stats()
        return

    args = parser.parse_args()

    # Валидация команд
    if args.command == 'stats':
        print_stats()
    elif args.command == 'snapshots':
        print_snapshots_list(date_str=args.param)
    elif args.command == 'scud':
        inspect_scud(snapshot_id=args.snapshot, date_str=args.param, export_xlsx=args.export_xlsx)
    elif args.command == 'anomalies':
        print_anomalies()
    elif args.command == 'rules':
        print_rules()
    elif args.command == 'prompts':
        print_system_prompts()
    elif args.command == 'sessions':
        print_session_states()
    elif args.command == 'dump':
        filename = args.param if args.param else "db_dump_full.xlsx"
        dump_all_to_excel(filename)
    elif args.command == 'snapshot':
        if args.action == 'del':
            if args.day:
                delete_snapshots_by_date(args.day)
            elif args.param:
                delete_snapshot_by_id(args.param)
            else:
                print("\n[❌] Ошибка: Не указан ID снапшота или параметр --day для удаления.")
                print("Пример: python scripts/db_cli.py snapshot del Y20260805-007\n")
        else:
            print(f"\n[❌] Ошибка: Неизвестное действие '{args.action}' для команды snapshot.")
            print("Используйте: python scripts/db_cli.py snapshot del [ID или --day 'ДД.ММ.ГГГГ']\n")
    else:
        print(f"\n[❌] Ошибка: Неизвестная команда.")
        parser.print_help()

if __name__ == "__main__":
    main()