#!/usr/bin/env python3
"""自動審核機器人 — 用 discli 自動偵測垃圾訊息並執行規則。"""

import json
import subprocess
import sys
from collections import defaultdict

# --- 設定 ---
BANNED_WORDS = {"spam", "scam", "phishing", "free nitro", "discord.gg/scam"}
WARN_THRESHOLD = 3  # 達到此次數後踢出


def run_discli(*args):
    """執行 discli 指令並回傳結果。"""
    result = subprocess.run(
        ["discli", "-y", *args],
        capture_output=True,
        text=True,
    )
    if result.returncode != 0:
        print(f"[錯誤] discli {' '.join(args)}: {result.stderr.strip()}", file=sys.stderr)
    return result


def run_discli_json(*args):
    """執行帶 JSON 輸出的 discli 指令並解析回傳。"""
    result = subprocess.run(
        ["discli", "--json", "-y", *args],
        capture_output=True,
        text=True,
    )
    if result.returncode != 0:
        return None
    try:
        return json.loads(result.stdout)
    except json.JSONDecodeError:
        return None


# 追蹤每位使用者違規次數
violations = defaultdict(int)

# 開始監聽所有訊息（可選是否包含 bot）
proc = subprocess.Popen(
    ["discli", "--json", "listen", "--events", "messages"],
    stdout=subprocess.PIPE,
    text=True,
)

print("自動審核機器人已啟動，按 Ctrl+C 可停止。", file=sys.stderr)

for line in proc.stdout:
    line = line.strip()
    if not line:
        continue

    try:
        event = json.loads(line)
    except json.JSONDecodeError:
        continue

    content = event.get("content", "").lower()
    author = event.get("author", "unknown")
    author_id = event.get("author_id", "")
    channel_id = event.get("channel_id", "")
    message_id = event.get("message_id", "")
    server = event.get("server", "")

    # 檢查違規字串
    found = [w for w in BANNED_WORDS if w in content]
    if not found:
        continue

    print(f"[違規] {author} ({author_id}): 命中 {found}", file=sys.stderr)

    # 步驟一：刪除違規訊息
    run_discli("message", "delete", channel_id, message_id)

    # 步驟二：累積違規並升級處理
    violations[author_id] += 1
    count = violations[author_id]

    if count >= WARN_THRESHOLD:
        # 踢除使用者
        run_discli(
            "member", "kick", server, author_id,
            "--reason", f"自動化處置：違規次數 {count} 次",
        )
        run_discli(
            "message", "send", channel_id,
            f"**{author}** 已因 {count} 次違規而被移除。",
        )
        del violations[author_id]
        print(f"[已踢除] {author} ({author_id})，已超過 {count} 次違規", file=sys.stderr)
    else:
        remaining = WARN_THRESHOLD - count
        run_discli(
            "message", "send", channel_id,
            f"{author}，你的訊息已違反伺服器規範而移除。"
            f"**{remaining}** 次警告後將啟動踢除。",
        )
        print(f"[警告] {author} ({author_id}): {count}/{WARN_THRESHOLD}", file=sys.stderr)