import os
from fastapi import FastAPI
from apscheduler.schedulers.background import BackgroundScheduler

from .db import init_db, save_articles, get_latest_articles
from .scraper import fetch_market_news, fetch_major_indices

app = FastAPI()
scheduler = BackgroundScheduler(timezone=os.getenv("TZ", "Asia/Seoul"))

@app.on_event("startup")
def on_startup():
    init_db()
    
    # 매일 아침 8시 뉴스 스크랩
    scheduler.add_job(run_scraping_job, "cron", hour="8", minute="0")
    
    # 앱 시작 시에도 한 번 실행 (데이터 없으면)
    if not get_latest_articles(1):
        run_scraping_job()
        
    scheduler.start()

def run_scraping_job():
    print("[StockLab] Starting news scraping...")
    articles = fetch_market_news()
    count = save_articles(articles)
    print(f"[StockLab] Saved {count} new articles.")

@app.get("/health")
def health():
    return {"ok": True}

@app.get("/api/stock/news")
def get_news(limit: int = 20):
    """최신 주식 뉴스 조회"""
    return get_latest_articles(limit)

@app.get("/api/stock/indices")
def get_indices():
    """주요 지표(KOSPI 등) 실시간 크롤링 조회"""
    return fetch_major_indices()

@app.post("/api/stock/scrap")
def trigger_scrap():
    """수동 스크랩 트리거"""
    run_scraping_job()
    return {"ok": True}

@app.get("/api/version")
def version():
    return {"version": os.getenv("APP_VERSION", "dev")}