Clock_out_Time_Calculator/utils/crash_handler.py

"""
전역 예외 후킹 + Gitea Issues 자동 보고.

sys.excepthook을 등록해 처리되지 않은 예외를 가로채:
1. crash_log 테이블에 저장
2. 사용자에게 다이얼로그로 알림 + "Gitea에 보고" / "복사" 옵션
"""
from __future__ import annotations
import json
import sys
import traceback
import urllib.request
import urllib.error
from datetime import datetime
from typing import Optional

# 자체 호스팅 Gitea (updater_client와 동일)
GITEA_API = 'https://kindnick-git.duckdns.org/api/v1/repos/kindnick/Clock_out_Time_Calculator'
USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) ClockOutCalculator/2.4'


def install_global_handler(db, app_version: str = 'unknown') -> None:
    """sys.excepthook 등록. db는 crash_log 저장용."""
    original = sys.excepthook

    def hook(exc_type, exc_value, exc_tb):
        if exc_type is KeyboardInterrupt:
            original(exc_type, exc_value, exc_tb)
            return
        try:
            tb_str = ''.join(traceback.format_exception(exc_type, exc_value, exc_tb))
            _log_crash(db, exc_type.__name__, str(exc_value), tb_str, app_version)
            _show_dialog(db, exc_type.__name__, str(exc_value), tb_str, app_version)
        except Exception:
            pass  # 후킹 자체 실패는 silent
        original(exc_type, exc_value, exc_tb)

    sys.excepthook = hook


def _log_crash(db, exc_type: str, message: str, tb: str, version: str) -> None:
    """crash_log 테이블에 기록."""
    try:
        conn = db.get_connection()
        cursor = conn.cursor()
        # 테이블 자동 생성 (마이그레이션 누락 대비)
        cursor.execute('''
            CREATE TABLE IF NOT EXISTS crash_log (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                exception_type TEXT,
                message TEXT,
                traceback TEXT,
                app_version TEXT,
                reported BOOLEAN DEFAULT 0
            )
        ''')
        cursor.execute('''
            INSERT INTO crash_log (exception_type, message, traceback, app_version)
            VALUES (?, ?, ?, ?)
        ''', (exc_type, message, tb, version))
        conn.commit()
        conn.close()
    except Exception:
        pass


def _show_dialog(db, exc_type: str, message: str, tb: str, version: str) -> None:
    """크래시 알림 + Gitea 보고/복사 버튼."""
    try:
        from PyQt5.QtWidgets import (QApplication, QMessageBox, QDialog,
                                      QVBoxLayout, QLabel, QTextEdit,
                                      QHBoxLayout, QPushButton, QLineEdit)
    except ImportError:
        return

    app = QApplication.instance()
    if app is None:
        return

    dlg = QDialog()
    dlg.setWindowTitle("⚠️ 오류 발생")
    dlg.setMinimumSize(560, 420)
    layout = QVBoxLayout()

    title = QLabel(f"<b>{exc_type}</b>: {message[:200]}")
    title.setWordWrap(True)
    layout.addWidget(title)

    layout.addWidget(QLabel("무엇을 하다가 발생했나요? (선택)"))
    user_note = QLineEdit()
    user_note.setPlaceholderText("예: 출근 버튼 누른 직후")
    layout.addWidget(user_note)

    layout.addWidget(QLabel("기술 정보 (자동 보고에 포함):"))
    tb_view = QTextEdit()
    tb_view.setReadOnly(True)
    tb_view.setFont(__import__('PyQt5.QtGui', fromlist=['QFont']).QFont('Consolas', 9))
    tb_view.setText(tb[-3000:])  # 너무 긴 traceback 제한
    layout.addWidget(tb_view, 1)

    btn_row = QHBoxLayout()
    cancel_btn = QPushButton("닫기")
    copy_btn = QPushButton("📋 복사")
    report_btn = QPushButton("📤 Gitea에 보고")

    has_token = bool(db.get_setting('gitea_feedback_token', '') or '')
    enabled = db.get_setting('gitea_feedback_enabled', 'false').lower() == 'true'
    if not (has_token and enabled):
        report_btn.setEnabled(False)
        report_btn.setToolTip("설정 → 데이터 관리 → Gitea 피드백 토큰 입력 후 활성화 필요")

    def do_copy():
        clipboard = QApplication.clipboard()
        text = (
            f"## {exc_type}\n\n{message}\n\n"
            f"**Version**: {version}\n**Note**: {user_note.text()}\n\n"
            f"```\n{tb}\n```"
        )
        clipboard.setText(text)
        copy_btn.setText("✓ 복사됨")

    def do_report():
        token = db.get_setting('gitea_feedback_token', '') or ''
        if not token:
            QMessageBox.warning(dlg, "토큰 없음", "Gitea PAT를 설정에서 먼저 입력하세요.")
            return
        title_str = f"[Auto] {exc_type}: {message[:80]}"
        body = (
            f"**Version**: `{version}`\n"
            f"**Time**: {datetime.now().isoformat(timespec='seconds')}\n"
            f"**User note**: {user_note.text() or '(none)'}\n\n"
            f"### Traceback\n```\n{tb[-3000:]}\n```"
        )
        ok = _send_to_gitea(token, title_str, body)
        if ok:
            QMessageBox.information(dlg, "보고 완료", "Gitea Issues에 등록되었습니다.")
            report_btn.setText("✓ 보고됨")
            report_btn.setEnabled(False)
        else:
            QMessageBox.warning(dlg, "보고 실패", "네트워크 또는 토큰 권한 문제일 수 있습니다.")

    cancel_btn.clicked.connect(dlg.reject)
    copy_btn.clicked.connect(do_copy)
    report_btn.clicked.connect(do_report)
    btn_row.addWidget(cancel_btn)
    btn_row.addStretch()
    btn_row.addWidget(copy_btn)
    btn_row.addWidget(report_btn)
    layout.addLayout(btn_row)

    dlg.setLayout(layout)
    dlg.exec_()


def _send_to_gitea(token: str, title: str, body: str) -> bool:
    """Gitea Issues API로 issue 생성."""
    payload = json.dumps({'title': title, 'body': body}).encode('utf-8')
    req = urllib.request.Request(
        f"{GITEA_API}/issues",
        data=payload,
        headers={
            'Authorization': f'token {token}',
            'Content-Type': 'application/json',
            'User-Agent': USER_AGENT,
        },
        method='POST',
    )
    try:
        with urllib.request.urlopen(req, timeout=10) as resp:
            return 200 <= resp.status < 300
    except (urllib.error.URLError, urllib.error.HTTPError, OSError, TimeoutError):
        return False