main.py

import discord
from discord.ext import commands
import asyncio
import logging
from flask import Flask, request
import threading
import os
import sqlite3

# -----------------------------
# 환경변수
TOKEN = os.environ.get("DISCORD_BOT_TOKEN")
LOG_CHANNEL_ID = 1417052732019310652
PORT = 10000
DB_PATH = "db.sqlite"

# -----------------------------
# 로깅 설정
logger = logging.getLogger("takkari")
logger.setLevel(logging.INFO)
console_handler = logging.StreamHandler()
console_handler.setFormatter(logging.Formatter("[%(asctime)s] [%(levelname)s] %(message)s"))
logger.addHandler(console_handler)

# -----------------------------
# Flask 서버
app = Flask(__name__)

@app.route("/")
def home():
    return "Takkari Bot Server is running!"

    # DB 저장
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute("INSERT INTO riot_codes (code) VALUES (?)", (code,))
    conn.commit()
    conn.close()

def run_flask():
    logger.info(f"Flask 서버 시작 (포트 {PORT})")
    app.run(host="0.0.0.0", port=PORT)

# -----------------------------
# Discord Bot 로그 핸들러
class DiscordLogHandler(logging.Handler):
    def __init__(self, bot, channel_id):
        super().__init__()
        self.bot = bot
        self.channel_id = channel_id
        self.queue = asyncio.Queue()
        self.worker_started = False

    async def log_worker(self):
        await self.bot.wait_until_ready()
        channel = self.bot.get_channel(self.channel_id)
        while True:
            log_entry = await self.queue.get()
            try:
                if channel:
                    await channel.send(f"```{log_entry}```")
            except Exception as e:
                print("Log send error:", e)

    def emit(self, record):
        log_entry = self.format(record)
        try:
            self.queue.put_nowait(log_entry)
        except Exception as e:
            print("Queue put error:", e)

# -----------------------------
# Bot 초기화
intents = discord.Intents.default()
intents.members = True
bot = commands.Bot(command_prefix="/", intents=intents)

# setup_hook에서 worker 등록
@bot.event
async def setup_hook():
    if not hasattr(bot, "log_worker_started"):
        bot.log_worker_started = True
        handler_task = bot.loop.create_task(discord_handler.log_worker())

# Discord 로그 채널 핸들러 연결
discord_handler = DiscordLogHandler(bot, LOG_CHANNEL_ID)
formatter = logging.Formatter("[%(asctime)s] [%(levelname)s] %(message)s")
discord_handler.setFormatter(formatter)
logger.addHandler(discord_handler)

# -----------------------------
# DB 초기화
def init_db():
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()
    cursor.execute("""
    CREATE TABLE IF NOT EXISTS users(
        id INTEGER PRIMARY KEY,
        name TEXT
    )
    """)
    cursor.execute("""
    CREATE TABLE IF NOT EXISTS riot_codes(
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        code TEXT NOT NULL
    )
    """)
    conn.commit()
    conn.close()
    logger.info("✅ DB 초기화 완료")

# -----------------------------
# 코그 로딩
async def load_cogs():
    cogs = [
        "takkari_bot.cogs.userinfo",
        "takkari_bot.cogs.announce",
        "takkari_bot.cogs.dm_feature",
        "takkari_bot.cogs.support",
        "takkari_bot.cogs.help",
        "takkari_bot.cogs.patchnote",
        "takkari_bot.cogs.dblookup",
        "takkari_bot.cogs.updown",
        "takkari_bot.cogs.schedule",
        "takkari_bot.cogs.loglookup",
        "takkari_bot.cogs.riot"
    ]
    for cog in cogs:
        try:
            await bot.load_extension(cog)
            logger.info(f"✅ {cog} 로드 완료")
        except Exception as e:
            logger.exception(f"❌ {cog} 로드 실패: {e}")

# -----------------------------
# 봇 이벤트
@bot.event
async def on_ready():
    logger.info(f"봇 시작 완료: {bot.user}")

    async def status_task():
        while True:
            try:
                guild_count = len(bot.guilds)
                member_count = sum(g.member_count for g in bot.guilds)
                statuses = [
                    discord.Game(f"{guild_count}개의 서버에서 활동 ✨"),
                    discord.Game(f"{member_count}명의 유저와 함께 👥"),
                    discord.Game("🔥 따까리봇 업데이트 진행"),
                ]
                for status in statuses:
                    await bot.change_presence(status=discord.Status.online, activity=status)
                    await asyncio.sleep(10)
            except Exception as e:
                logger.error(f"Presence 업데이트 중 오류: {e}")
                await asyncio.sleep(10)

    bot.loop.create_task(status_task())

# -----------------------------
# 메인 실행
async def main():
    init_db()
    await load_cogs()
    threading.Thread(target=run_flask, daemon=True).start()
    await bot.start(TOKEN)

if __name__ == "__main__":
    asyncio.run(main())