main.py

import json
import os
import subprocess
from datetime import date
from pathlib import Path

import requests
from dotenv import load_dotenv
load_dotenv()

from src.collectors import public_data, aihub
from src.generators.idea import build_public_data_prompt, build_aihub_prompt
from src.senders import discord

PAYLOAD_PATH = Path("data/today_discord_payload.json")
_GH_REPO = "makersfarm/daily-discord-bot"
_GH_WORKFLOW = "discord_send.yml"


def _trigger_github_actions(payloads: list[dict]) -> None:
    token = os.environ.get("GITHUB_TOKEN", "")
    if not token:
        raise RuntimeError("GITHUB_TOKEN not set")
    resp = requests.post(
        f"https://api.github.com/repos/{_GH_REPO}/actions/workflows/{_GH_WORKFLOW}/dispatches",
        headers={"Authorization": f"Bearer {token}", "Accept": "application/vnd.github+json"},
        json={"ref": "main", "inputs": {"payload": json.dumps(payloads)}},
        timeout=15,
    )
    resp.raise_for_status()


def ask_claude(prompt: str) -> str:
    env = {**os.environ, "CLAUDECODE": ""}
    result = subprocess.run(
        [
            "claude", "-p", prompt,
            "--model", "claude-haiku-4-5-20251001",
            "--allowed-tools", "WebSearch",
        ],
        capture_output=True,
        text=True,
        timeout=300,
        env=env,
    )
    if result.returncode != 0:
        raise RuntimeError(f"claude CLI 오류: {result.stderr}")
    return result.stdout.strip()


def main():
    day_of_year = date.today().timetuple().tm_yday

    print("📊 공공데이터포털 수집 중...")
    pub_data = public_data.collect(day_of_year)
    print(f"  분야: {pub_data['category']}, {len(pub_data['datasets'])}개 선택")

    print("🤖 AI Hub 수집 중...")
    hub_data = aihub.collect()
    print(f"  데이터셋: {hub_data['title']}")

    print("💡 아이디어 생성 중...")
    pub_idea = ask_claude(build_public_data_prompt(pub_data))
    hub_idea = ask_claude(build_aihub_prompt(hub_data))

    payloads = discord.build_payloads(pub_data, hub_data, pub_idea, hub_idea)
    PAYLOAD_PATH.write_text(json.dumps(payloads, ensure_ascii=False, indent=2), encoding="utf-8")
    print(f"💾 payloads({len(payloads)}건) 저장: {PAYLOAD_PATH}")

    try:
        discord.send(pub_data, hub_data, pub_idea, hub_idea)
        print(f"📨 Discord 직접 전송 성공 ({len(payloads)}건)")
    except Exception as e:
        print(f"⚠️ Discord 직접 전송 실패: {e}")
        try:
            _trigger_github_actions(payloads)
            print("🔁 GitHub Actions로 Discord 전송 요청 완료")
        except Exception as e2:
            print(f"❌ GitHub Actions 트리거 실패: {e2}")

    print("\n--- 공공데이터 아이디어 ---")
    print(pub_idea)
    print("\n--- AI Hub 아이디어 ---")
    print(hub_idea)


if __name__ == "__main__":
    main()