# -*- coding: utf-8 -*-
"""健康检查工具"""

import requests
import sqlite3
import threading
import time
from config import Config

def check_service_health(service_id):
    """检查单个服务健康状态"""
    conn = sqlite3.connect(Config.DATABASE_PATH)
    cursor = conn.cursor()

    cursor.execute('''
        SELECT name, url, health_check_url, health_check_enabled
        FROM services
        WHERE id = ?
    ''', (service_id,))

    service = cursor.fetchone()
    if not service:
        conn.close()
        return

    name, url, check_url, check_enabled = service
    check_url = check_url or url  # 如果没有单独配置，使用主url

    status = 'offline'
    status_code = None
    error = None

    if check_enabled:
        try:
            response = requests.get(check_url, timeout=Config.HEALTH_CHECK_TIMEOUT)
            status_code = response.status_code
            if response.status_code < 500:
                status = 'online'
        except requests.exceptions.Timeout:
            error = 'timeout'
        except requests.exceptions.ConnectionError:
            error = 'connection_error'
        except Exception as e:
            error = str(e)

    # 这里可以添加状态记录表，暂时只检查
    conn.close()

    return {
        'id': service_id,
        'name': name,
        'status': status,
        'status_code': status_code,
        'error': error
    }

def check_all_services():
    """检查所有启用健康检测的服务"""
    conn = sqlite3.connect(Config.DATABASE_PATH)
    cursor = conn.cursor()

    cursor.execute('''
        SELECT id FROM services
        WHERE is_enabled = 1 AND health_check_enabled = 1
        ORDER BY sort_order DESC
    ''')

    services = cursor.fetchall()
    conn.close()

    results = []
    for (service_id,) in services:
        result = check_service_health(service_id)
        if result:
            results.append(result)

    return results

class HealthCheckWorker:
    """后台健康检查工作线程"""

    def __init__(self):
        self.running = False
        self.thread = None

    def start(self):
        """启动后台检查"""
        if not self.running:
            self.running = True
            self.thread = threading.Thread(target=self._run, daemon=True)
            self.thread.start()
            print("健康检查线程已启动")

    def stop(self):
        """停止后台检查"""
        self.running = False
        if self.thread:
            self.thread.join()
            print("健康检查线程已停止")

    def _run(self):
        """检查循环"""
        while self.running:
            results = check_all_services()
            # 记录日志
            for result in results:
                print(f"[HealthCheck] {result['name']}: {result['status']}")
            time.sleep(Config.HEALTH_CHECK_INTERVAL)

# 全局工作线程
health_worker = HealthCheckWorker()