Update server.py

server.py

@@ -1,184 +1,198 @@
-# summary_daily.py
+# server.py — tambah scheduler rekap harian 06:00 WIB (H-1) + endpoint force summary
 import os
-import sqlite3
-from datetime import datetime, timezone, timedelta, date
-from collections import defaultdict
-
-DB_PATH = os.environ.get("BOTSIGNAL_DB", "/tmp/botsignal.db")
-OUT_DIR = os.environ.get("SUMMARY_OUT_DIR", "summaries")
-
-def _db():
-    conn = sqlite3.connect(DB_PATH)
-    conn.execute("PRAGMA journal_mode=WAL;")
-    return conn
-
-def _ensure_tables():
-    conn = _db()
-    conn.executescript("""
-    CREATE TABLE IF NOT EXISTS summary_calls (
-        keyword TEXT PRIMARY KEY,
-        posted_at INTEGER NOT NULL,
-        tier TEXT NOT NULL,
-        msg_id INTEGER NOT NULL
-    );
-    CREATE TABLE IF NOT EXISTS summary_milestones (
-        keyword TEXT NOT NULL,
-        reply_msg_id INTEGER,
-        multiple REAL,
-        replied_at INTEGER NOT NULL
-    );
-    CREATE TABLE IF NOT EXISTS summary_outcomes (
-        keyword TEXT PRIMARY KEY,
-        is_deleted INTEGER DEFAULT 0
-    );
-    """)
-    conn.commit()
-    conn.close()
-
-def _day_bounds_utc(d: date):
-    # ambil batas harian [00:00, 23:59:59] UTC untuk tanggal d (UTC)
-    sod = datetime(d.year, d.month, d.day, tzinfo=timezone.utc)
-    eod = datetime(d.year, d.month, d.day, 23, 59, 59, tzinfo=timezone.utc)
-    return int(sod.timestamp()), int(eod.timestamp())
-
-def _fmt(num):
-    return f"{num:,}".replace(",", ".")
-
-def compute_summary_for_date(d: date):
-    _ensure_tables()
-    sod, eod = _day_bounds_utc(d)
-    conn = _db(); cur = conn.cursor()
-
-    cur.execute("""
-        SELECT keyword, tier, posted_at, msg_id
-        FROM summary_calls
-        WHERE posted_at BETWEEN ? AND ?
-    """, (sod, eod))
-    rows = cur.fetchall()
-    total_calls = len(rows)
-    calls = {r[0]: {"tier": r[1], "posted_at": r[2], "msg_id": r[3]} for r in rows}
-
-    max_mult_per_kw = {}
-    have_reply = set()
-    if calls:
-        q = ",".join("?" for _ in calls)
-        cur.execute(f"""
-            SELECT keyword, MAX(multiple)
-            FROM summary_milestones
-            WHERE keyword IN ({q})
-              AND replied_at BETWEEN ? AND ?
-            GROUP BY keyword
-        """, (*calls.keys(), sod, eod))
-        for kw, mx in cur.fetchall():
-            if mx is not None:
-                have_reply.add(kw)
-                max_mult_per_kw[kw] = mx
-
-    deleted = set()
-    if calls:
-        q = ",".join("?" for _ in calls)
-        cur.execute(f"""
-            SELECT keyword
-            FROM summary_outcomes
-            WHERE keyword IN ({q})
-              AND is_deleted = 1
-        """, (*calls.keys(),))
-        for (kw,) in cur.fetchall():
-            deleted.add(kw)
-
-    conn.close()
-
-    wins = have_reply
-    loses = (deleted | (set(calls.keys()) - have_reply))
-
-    calls_per_tier = defaultdict(int)
-    for _, meta in calls.items():
-        calls_per_tier[meta["tier"]] += 1
-
-    win_per_tier = defaultdict(int)
-    lose_per_tier = defaultdict(int)
-    for kw, meta in calls.items():
-        if kw in wins:
-            win_per_tier[meta["tier"]] += 1
-        if kw in loses:
-            lose_per_tier[meta["tier"]] += 1
-
-    top_kw, top_mult = None, None
-    for kw, mult in max_mult_per_kw.items():
-        if top_mult is None or mult > top_mult:
-            top_kw, top_mult = kw, mult
-
-    return {
-        "total_calls": total_calls,
-        "win_rate": (len(wins) / total_calls) * 100 if total_calls else 0.0,
-        "calls_per_tier": dict(sorted(calls_per_tier.items())),
-        "total_win": len(wins),
-        "total_lose": len(loses),
-        "win_per_tier": dict(sorted(win_per_tier.items())),
-        "lose_per_tier": dict(sorted(lose_per_tier.items())),
-        "top_call": {"keyword": top_kw, "max_multiple": top_mult} if top_kw else None,
-    }
-
-def render_telegram_html(summary: dict, label_date: str) -> str:
-    # Pakai HTML biar rapi (parse_mode='html' di send_message)
-    lines = []
-    lines.append(f"<b>Daily Summary — {label_date}</b>")
-    lines.append("")
-    lines.append(f"<b>Total Call:</b> {_fmt(summary['total_calls'])}")
-    lines.append(f"<b>Win Rate:</b> {summary['win_rate']:.2f}%")
-    lines.append("")
-    lines.append("<b>Total Call per Tier</b>")
-    if summary["calls_per_tier"]:
-        for tier, cnt in summary["calls_per_tier"].items():
-            lines.append(f"• {tier}: {_fmt(cnt)}")
-    else:
-        lines.append("• (kosong)")
-    lines.append("")
-    lines.append("<b>Total Win & Lose</b>")
-    lines.append(f"• Win: {_fmt(summary['total_win'])}")
-    lines.append(f"• Lose: {_fmt(summary['total_lose'])}")
-    lines.append("")
-    lines.append("<b>Detail per Tier</b>")
-    lines.append("<u>Win</u>")
-    if summary["win_per_tier"]:
-        for tier, cnt in summary["win_per_tier"].items():
-            lines.append(f"• {tier}: {_fmt(cnt)}")
-    else:
-        lines.append("• (kosong)")
-    lines.append("<u>Lose</u>")
-    if summary["lose_per_tier"]:
-        for tier, cnt in summary["lose_per_tier"].items():
-            lines.append(f"• {tier}: {_fmt(cnt)}")
-    else:
-        lines.append("• (kosong)")
-    lines.append("")
-    lines.append("<b>Top Call (Max Reply Multiple)</b>")
-    if summary["top_call"]:
-        lines.append(f"• CA: <code>{summary['top_call']['keyword']}</code>")
-        lines.append(f"• Max: {summary['top_call']['max_multiple']}×")
-    else:
-        lines.append("• (belum ada reply)")
-    return "\n".join(lines)
-
-if __name__ == "__main__":
-    from datetime import date
-    from telethon import TelegramClient
-    import asyncio
-    import os
-
-    # ======== CONFIG TELEGRAM =========
-    API_ID = int(os.environ.get("API_ID", "0"))             
-    API_HASH = os.environ.get("API_HASH", "")                
-    STRING_SESSION = os.environ.get("STRING_SESSION", "") 
-    TARGET_CHANNEL = os.environ.get("TARGET_CHANNEL", "@MidasTouchsignalll")  # contoh
-
-    # ======== GENERATE SUMMARY HARINI =========
-    summary = compute_summary_for_date(date.today())
-    html_text = render_telegram_html(summary, date.today().strftime("%Y-%m-%d"))
-
-    async def main():
-        async with TelegramClient( STRING_SESSION, API_ID, API_HASH) as client:
-            await client.send_message(TARGET_CHANNEL, html_text, parse_mode="html")
-            print("[SUMMARY] posted to Telegram")
-
-    asyncio.run(main())
+import asyncio
+from datetime import datetime, timedelta, date
+from zoneinfo import ZoneInfo
+from fastapi import FastAPI, HTTPException
+
+# Core yang sudah ada
+from botsignal import start_bot_background, client  # ← client dipakai untuk kirim summary
+
+# Ringkasan harian
+from summary_daily import compute_summary_for_date, render_telegram_html
+
+app = FastAPI(title="Telegram Curator Health")
+
+# ====== ENV ======
+SUMMARY_CHAT_ID = os.environ.get("SUMMARY_CHAT_ID")  # contoh: -1001234567890 atau @nama_channel
+JAKARTA_TZ = ZoneInfo("Asia/Jakarta")
+
+# ====== Scheduler util ======
+def _seconds_until_6am_wib() -> float:
+    now = datetime.now(JAKARTA_TZ)
+    target = now.replace(hour=6, minute=0, second=0, microsecond=0)
+    if now >= target:
+        target = target + timedelta(days=1)
+    return (target - now).total_seconds()
+
+async def _send_daily_summary_loop():
+    """
+    Kirim rekap tiap hari jam 06:00 WIB untuk H-1.
+    Jika mau rekap 'hari ini', ubah target_day = today_local.
+    """
+    while True:
+        try:
+            # Tunggu sampai 06:00 WIB berikutnya
+            sleep_s = _seconds_until_6am_wib()
+            await asyncio.sleep(sleep_s)
+
+            # Rekap H-1 (kemarin)
+            today_local = datetime.now(JAKARTA_TZ).date()
+            target_day = today_local - timedelta(days=1)
+
+            # Hitung dan kirim
+            data = compute_summary_for_date(target_day)
+            text = render_telegram_html(data, label_date=target_day.strftime("%Y-%m-%d"))
+            if not SUMMARY_CHAT_ID:
+                print("[SUMMARY] SUMMARY_CHAT_ID not set; skip send.")
+            else:
+                await client.send_message(entity=SUMMARY_CHAT_ID, message=text, parse_mode="html")
+                print(f"[SUMMARY] sent for {target_day}")
+        except Exception as e:
+            print(f"[SUMMARY] loop error: {e}")
+            await asyncio.sleep(10)  # retry kecil lalu lanjut loop
+
+# Simpan task supaya bisa di-cancel saat shutdown (kalau kamu tambahkan event shutdown nanti)
+_summary_task: asyncio.Task | None = None
+
+@app.on_event("startup")
+async def startup():
+    # (Core) mulai bot di background — TETAP seperti semula
+    await start_bot_background()
+
+    # (Baru) mulai scheduler summary harian
+    global _summary_task
+    if _summary_task is None or _summary_task.done():
+        _summary_task = asyncio.create_task(_send_daily_summary_loop())
+    print("[SERVER] startup complete (scheduler armed at 06:00 WIB)")
+
+@app.get("/")
+async def root():
+    # (Core) tetap
+    return {"status": "ok"}
+
+@app.get("/health")
+async def health():
+    # (Core) tetap — bisa ditambah info simple kalau mau
+    return {"healthy": True}
+
+@app.post("/summary/force")
+async def force_summary(day: str | None = None):
+    """
+    Trigger kirim summary manual.
+    - Tanpa 'day' -> rekap H-1 lokal WIB
+    - Format 'day' = 'YYYY-MM-DD' (lokal WIB)
+    """
+    if not SUMMARY_CHAT_ID:
+        raise HTTPException(status_code=400, detail="SUMMARY_CHAT_ID not set in environment.")
+    try:
+        if day:
+            target = date.fromisoformat(day)
+        else:
+            target = datetime.now(JAKARTA_TZ).date() - timedelta(days=1)
+
+        data = compute_summary_for_date(target)
+        text = render_telegram_html(data, label_date=target.strftime("%Y-%m-%d"))
+        await client.send_message(entity=SUMMARY_CHAT_ID, message=text, parse_mode="html")
+        return {"ok": True, "sent_for": target.isoformat()}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+# server.py — tambah scheduler rekap harian 06:00 WIB (H-1) + endpoint force summary
+import os
+import asyncio
+from datetime import datetime, timedelta, date
+from zoneinfo import ZoneInfo
+from fastapi import FastAPI, HTTPException
+
+# Core yang sudah ada
+from botsignal import start_bot_background, client  # ← client dipakai untuk kirim summary
+
+# Ringkasan harian
+from summary_daily import compute_summary_for_date, render_telegram_html
+
+app = FastAPI(title="Telegram Curator Health")
+
+# ====== ENV ======
+SUMMARY_CHAT_ID = os.environ.get("SUMMARY_CHAT_ID")  # contoh: -1001234567890 atau @nama_channel
+JAKARTA_TZ = ZoneInfo("Asia/Jakarta")
+
+# ====== Scheduler util ======
+def _seconds_until_6am_wib() -> float:
+    now = datetime.now(JAKARTA_TZ)
+    target = now.replace(hour=6, minute=0, second=0, microsecond=0)
+    if now >= target:
+        target = target + timedelta(days=1)
+    return (target - now).total_seconds()
+
+async def _send_daily_summary_loop():
+    """
+    Kirim rekap tiap hari jam 06:00 WIB untuk H-1.
+    Jika mau rekap 'hari ini', ubah target_day = today_local.
+    """
+    while True:
+        try:
+            # Tunggu sampai 06:00 WIB berikutnya
+            sleep_s = _seconds_until_6am_wib()
+            await asyncio.sleep(sleep_s)
+
+            # Rekap H-1 (kemarin)
+            today_local = datetime.now(JAKARTA_TZ).date()
+            target_day = today_local - timedelta(days=1)
+
+            # Hitung dan kirim
+            data = compute_summary_for_date(target_day)
+            text = render_telegram_html(data, label_date=target_day.strftime("%Y-%m-%d"))
+            if not SUMMARY_CHAT_ID:
+                print("[SUMMARY] SUMMARY_CHAT_ID not set; skip send.")
+            else:
+                await client.send_message(entity=SUMMARY_CHAT_ID, message=text, parse_mode="html")
+                print(f"[SUMMARY] sent for {target_day}")
+        except Exception as e:
+            print(f"[SUMMARY] loop error: {e}")
+            await asyncio.sleep(10)  # retry kecil lalu lanjut loop
+
+# Simpan task supaya bisa di-cancel saat shutdown (kalau kamu tambahkan event shutdown nanti)
+_summary_task: asyncio.Task | None = None
+
+@app.on_event("startup")
+async def startup():
+    # (Core) mulai bot di background — TETAP seperti semula
+    await start_bot_background()
+
+    # (Baru) mulai scheduler summary harian
+    global _summary_task
+    if _summary_task is None or _summary_task.done():
+        _summary_task = asyncio.create_task(_send_daily_summary_loop())
+    print("[SERVER] startup complete (scheduler armed at 06:00 WIB)")
+
+@app.get("/")
+async def root():
+    # (Core) tetap
+    return {"status": "ok"}
+
+@app.get("/health")
+async def health():
+    # (Core) tetap — bisa ditambah info simple kalau mau
+    return {"healthy": True}
+
+@app.post("/summary/force")
+async def force_summary(day: str | None = None):
+    """
+    Trigger kirim summary manual.
+    - Tanpa 'day' -> rekap H-1 lokal WIB
+    - Format 'day' = 'YYYY-MM-DD' (lokal WIB)
+    """
+    if not SUMMARY_CHAT_ID:
+        raise HTTPException(status_code=400, detail="SUMMARY_CHAT_ID not set in environment.")
+    try:
+        if day:
+            target = date.fromisoformat(day)
+        else:
+            target = datetime.now(JAKARTA_TZ).date() - timedelta(days=1)
+
+        data = compute_summary_for_date(target)
+        text = render_telegram_html(data, label_date=target.strftime("%Y-%m-%d"))
+        await client.send_message(entity=SUMMARY_CHAT_ID, message=text, parse_mode="html")
+        return {"ok": True, "sent_for": target.isoformat()}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))