#!/usr/bin/env python3
"""
Schedule Research Script
Runs market research and upgrade suggestions on a schedule.
"""

import schedule
import time
import subprocess
import sys
from datetime import datetime, timedelta

# Use shared utilities to reduce code duplication
from common.logger_config import setup_logger
from common.paths import REPO_ROOT, LOGS_DIR, ensure_dirs
from common.config_loader import load_env

# Setup logging with file output
ensure_dirs(LOGS_DIR)
log_file = LOGS_DIR / "scheduler.log"
logger = setup_logger(__name__, log_file=log_file, console=True)

# Load environment variables
load_env()

def job():
    logger.info("Running scheduled research task...")

    # Run market research
    research_script = REPO_ROOT / "scripts" / "market_research.py"
    try:
        subprocess.run([sys.executable, str(research_script)], check=True)
    except subprocess.CalledProcessError as e:
        logger.error(f"Market research failed: {e}")
        return

    # Run upgrade repo
    upgrade_script = REPO_ROOT / "scripts" / "upgrade_repo.py"
    try:
        subprocess.run([sys.executable, str(upgrade_script)], check=True)
    except subprocess.CalledProcessError as e:
        logger.error(f"Upgrade repo failed: {e}")
        return

    logger.info("Scheduled task completed successfully.")

    # Calculate next run
    next_run = datetime.now() + timedelta(hours=4)
    logger.info(f"Next scheduled run at: {next_run.strftime('%Y-%m-%d %H:%M:%S')}")

def main():
    # Load env vars
    load_dotenv()

    if not os.environ.get("GEMINI_API_KEY") and not os.environ.get("GOOGLE_API_KEY"):
        logger.warning("Missing GEMINI_API_KEY or GOOGLE_API_KEY in environment.")

    if not os.environ.get("JULES_API_KEY"):
        logger.warning("Missing JULES_API_KEY in environment.")

    logger.info("Starting Schedule Research Service...")
    logger.info(f"Logs will be written to {log_file}")

    # Run immediately on start
    job()

    # Schedule every 4 hours
    schedule.every(4).hours.do(job)

    while True:
        schedule.run_pending()
        time.sleep(60)

if __name__ == "__main__":
    main()