import time
import threading
import sys
import schedule
import logging
from datetime import datetime, date
from pathlib import Path

# Setup logging
logger = logging.getLogger("coordinator")
logger.setLevel(logging.INFO)

# Add src and root to path to ensure imports work correctly
SRC_PATH = Path(__file__).parent.parent
ROOT_PATH = SRC_PATH.parent
sys.path.append(str(SRC_PATH))
sys.path.append(str(ROOT_PATH))

# Import modules
try:
    from news_scraper.main import main as run_news_scraper_main
    from calendar_scraper.get_calendar_events import get_events, save_events_to_db
    # Import run_saturday_analysis dynamically or add root to path
    import run_saturday_analysis
except ImportError as e:
    logger.error(f"❌ Import Error: {e}")
    logger.error("Ensure you are running from the project root or src directory.")
    sys.exit(1)

class Coordinator:
    def __init__(self):
        self.is_running = False
        self.news_thread = None
        self.scheduler_thread = None

    def start_news_scraper(self):
        """Runs the news scraper in a separate daemon thread"""
        logger.info("📰 Starting News Scraper...")
        try:
            # Run news scraper in a separate thread since it runs constantly
            self.news_thread = threading.Thread(target=run_news_scraper_main, daemon=True)
            self.news_thread.start()
            logger.info("✅ News Scraper started")
        except Exception as e:
            logger.error(f"❌ Error starting News Scraper: {e}")

    def run_calendar_scraper(self):
        """Runs the calendar scraper for today"""
        logger.info(f"\n📅 Running Calendar Scraper for {date.today()}...")
        try:
            today = date.today()
            events = get_events(today)
            save_events_to_db(events, today)
            logger.info("✅ Calendar Scraper finished")
        except Exception as e:
            logger.error(f"❌ Error running Calendar Scraper: {e}")

    def run_saturday_analysis_task(self):
        """Runs the Saturday analysis"""
        logger.info("\n📊 Running Saturday Fundamental Analysis...")
        try:
            run_saturday_analysis.run_saturday_analysis()
            logger.info("✅ Saturday Analysis finished")
        except Exception as e:
            logger.error(f"❌ Error running Saturday Analysis: {e}")

    def scheduler_loop(self):
        """Main scheduler loop"""
        logger.info("⏰ Scheduler started")
        
        # Schedule Daily Calendar Scraper at 08:00
        schedule.every().day.at("08:00").do(self.run_calendar_scraper)
        logger.info("   - Scheduled Calendar Scraper daily at 08:00")
        
        # Schedule Saturday Analysis every Saturday
        schedule.every().saturday.at("09:00").do(self.run_saturday_analysis_task)
        logger.info("   - Scheduled Saturday Analysis every Saturday at 09:00")

        while self.is_running:
            schedule.run_pending()
            time.sleep(60) # Check every minute

    def start(self):
        """Start the coordinator"""
        self.is_running = True
        logger.info("🚀 Stock Alchemist Coordinator Starting...")

        # 1. Start News Scraper (Continuous)
        self.start_news_scraper()

        # 2. Start Scheduler (Blocking or Threaded?)
        # Let's run scheduler in main thread for simplicity as it just loops.
        
        try:
            self.scheduler_loop()
        except KeyboardInterrupt:
            self.stop()

    def stop(self):
        """Stop the coordinator"""
        logger.info("\n🛑 Stopping Coordinator...")
        self.is_running = False
        logger.info("✅ Coordinator stopped")

if __name__ == "__main__":
    coordinator = Coordinator()
    coordinator.start()