#!/usr/bin/env python3
"""Moderation Bot — auto-detect spam and enforce rules with discli."""

import json
import subprocess
import sys
from collections import defaultdict

# --- Configuration ---
BANNED_WORDS = {"spam", "scam", "phishing", "free nitro", "discord.gg/scam"}
WARN_THRESHOLD = 3  # Kicks after this many violations

def run_discli(*args):
    """Run a discli command and return the result."""
    result = subprocess.run(
        ["discli", "-y", *args],
        capture_output=True,
        text=True,
    )
    if result.returncode != 0:
        print(f"[error] discli {' '.join(args)}: {result.stderr.strip()}", file=sys.stderr)
    return result

def run_discli_json(*args):
    """Run a discli command with JSON output."""
    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

# Track violations per user
violations = defaultdict(int)

# Start listening for all messages (including from bots is optional)
proc = subprocess.Popen(
    ["discli", "--json", "listen", "--events", "messages"],
    stdout=subprocess.PIPE,
    text=True,
)

print("Moderation bot running. Ctrl+C to stop.", 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", "")

    # Check for banned content
    found = [w for w in BANNED_WORDS if w in content]
    if not found:
        continue

    print(f"[violation] {author} ({author_id}): matched {found}", file=sys.stderr)

    # Step 1: Delete the offending message
    run_discli("message", "delete", channel_id, message_id)

    # Step 2: Track and escalate
    violations[author_id] += 1
    count = violations[author_id]

    if count >= WARN_THRESHOLD:
        # Kick the user
        run_discli(
            "member", "kick", server, author_id,
            "--reason", f"Automated: {count} content violations",
        )
        run_discli(
            "message", "send", channel_id,
            f"**{author}** has been kicked after {count} violations.",
        )
        del violations[author_id]
        print(f"[kicked] {author} ({author_id}) after {count} violations", file=sys.stderr)
    else:
        remaining = WARN_THRESHOLD - count
        run_discli(
            "message", "send", channel_id,
            f"{author}, your message was removed for violating server rules. "
            f"**{remaining}** warning(s) remaining before a kick.",
        )
        print(f"[warned] {author} ({author_id}): {count}/{WARN_THRESHOLD}", file=sys.stderr)