import argparse
import os
import signal
import subprocess
import sys
from datetime import datetime

import requests

#!/usr/bin/env python3

# ── 설정 부분 ──
PYTHON_SCRIPT = "./src/main.py"
DOMAIN_FILE = "./data/domains.txt"
# ─────────────

def download_domains():
    """도메인 파일 다운로드"""
    try:
        print("도메인 파일 다운로드 중...")
        response = requests.get("https://f.imnya.ng/.whs/tp-domains/data/domains/latest.txt")
        response.raise_for_status()
        
        # 디렉토리가 없으면 생성
        os.makedirs(os.path.dirname("./data"), exist_ok=True)
        
        with open(DOMAIN_FILE, 'w', encoding='utf-8') as f:
            f.write(response.text)
        print("도메인 파일 다운로드 완료")
    except requests.RequestException as e:
        print(f"도메인 파일 다운로드 실패: {e}")
        sys.exit(1)

def run_script(start_line, end_line, skh_option):
    """Python 스크립트 실행"""
    current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    print(f"[{current_time}] Processing lines {start_line} to {end_line}...")

    process = None
    try:
        command = [
            "uv", "run", PYTHON_SCRIPT,
            "-f", DOMAIN_FILE,
            "-s", str(start_line),
            "-e", str(end_line),
        ]
        if skh_option:
            command.append("--skip-html-check")

        # KeyboardInterrupt를 subprocess에 전달하도록 수정
        process = subprocess.Popen(command)
        process.wait()
        
        if process.returncode != 0:
            print("Python 스크립트 실행 실패")
            sys.exit(1)
    except subprocess.CalledProcessError:
        print("Python 스크립트 실행 실패")
        sys.exit(1)

def main():
    parser = argparse.ArgumentParser(
        description="도메인 처리 스크립트 실행기",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
사용 예시:
  uv run run.py 10000 11000           # 10000~11000 라인 처리
  uv run run.py 10000 11000 --skh     # SKH 옵션 활성화
  uv run run.py 10000 11000 --no-download  # 다운로드 생략
        """
    )
    
    parser.add_argument("start_line", type=int, help="시작 라인 번호")
    parser.add_argument("end_line", type=int, help="종료 라인 번호")
    parser.add_argument("--skh", action="store_true", help="SKH 옵션 활성화")
    parser.add_argument("--no-download", action="store_true", help="도메인 파일 다운로드 생략")
    
    args = parser.parse_args()
    
    # 라인 범위 검증
    if args.start_line < 0 or args.end_line < 0:
        print("라인 번호는 0 이상이어야 합니다.")
        sys.exit(1)
    
    if args.start_line > args.end_line:
        print("시작 라인은 종료 라인보다 크거나 같아야 합니다.")
        sys.exit(1)
    
    # 도메인 파일 다운로드
    if not args.no_download:
        download_domains()
    elif not os.path.exists(DOMAIN_FILE):
        print(f"도메인 파일({DOMAIN_FILE})이 존재하지 않습니다. --no-download 옵션을 제거하거나 파일을 준비해주세요.")
        sys.exit(1)
    
    # 스크립트 실행
    run_script(args.start_line, args.end_line, args.skh)
    
    print("처리 완료.")

if __name__ == "__main__":
    main()