#!/usr/bin/env python3
"""Admin CLI for Arbitrage Engine"""
import sys, os, sqlite3, secrets, json

DB_PATH = os.path.join(os.path.dirname(__file__), "..", "arb.db")

def get_conn():
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    return conn

def gen_invite(count=1, max_uses=1):
    conn = get_conn()
    codes = []
    for _ in range(count):
        code = secrets.token_urlsafe(6)[:8].upper()
        conn.execute(
            "INSERT INTO invite_codes (code, created_by, max_uses) VALUES (?, 1, ?)",
            (code, max_uses)
        )
        codes.append(code)
    conn.commit()
    conn.close()
    for c in codes:
        print(f"  {c}")
    print(f"\nGenerated {len(codes)} invite code(s)")

def list_invites():
    conn = get_conn()
    rows = conn.execute("SELECT id, code, max_uses, used_count, status, created_at FROM invite_codes ORDER BY id DESC").fetchall()
    conn.close()
    if not rows:
        print("No invite codes found")
        return
    print(f"{'ID':>4} {'CODE':>10} {'MAX':>4} {'USED':>5} {'STATUS':>10} {'CREATED':>20}")
    print("-" * 60)
    for r in rows:
        print(f"{r['id']:>4} {r['code']:>10} {r['max_uses']:>4} {r['used_count']:>5} {r['status']:>10} {r['created_at']:>20}")

def disable_invite(code):
    conn = get_conn()
    conn.execute("UPDATE invite_codes SET status = 'disabled' WHERE code = ?", (code,))
    conn.commit()
    conn.close()
    print(f"Disabled invite code: {code}")

def list_users():
    conn = get_conn()
    rows = conn.execute("SELECT id, email, role, banned, created_at FROM users ORDER BY id DESC").fetchall()
    conn.close()
    if not rows:
        print("No users found")
        return
    print(f"{'ID':>4} {'EMAIL':>30} {'ROLE':>6} {'BANNED':>7} {'CREATED':>20}")
    print("-" * 72)
    for r in rows:
        print(f"{r['id']:>4} {r['email']:>30} {r['role']:>6} {r['banned']:>7} {r['created_at']:>20}")

def ban_user(user_id):
    conn = get_conn()
    conn.execute("UPDATE users SET banned = 1 WHERE id = ?", (user_id,))
    conn.commit()
    conn.close()
    print(f"Banned user ID: {user_id}")

def unban_user(user_id):
    conn = get_conn()
    conn.execute("UPDATE users SET banned = 0 WHERE id = ?", (user_id,))
    conn.commit()
    conn.close()
    print(f"Unbanned user ID: {user_id}")

def set_admin(user_id):
    conn = get_conn()
    conn.execute("UPDATE users SET role = 'admin' WHERE id = ?", (user_id,))
    conn.commit()
    conn.close()
    print(f"Set user {user_id} as admin")

def usage():
    print("""Usage: python3 admin_cli.py <command> [args]

Commands:
  gen-invite [count] [max_uses]   Generate invite codes (default: 1 code, 1 use)
  list-invites                    List all invite codes
  disable-invite <CODE>           Disable an invite code
  list-users                      List all users
  ban-user <USER_ID>              Ban a user
  unban-user <USER_ID>            Unban a user
  set-admin <USER_ID>             Set user as admin
""")

if __name__ == "__main__":
    if len(sys.argv) < 2:
        usage()
        sys.exit(1)
    cmd = sys.argv[1]
    if cmd == "gen-invite":
        count = int(sys.argv[2]) if len(sys.argv) > 2 else 1
        max_uses = int(sys.argv[3]) if len(sys.argv) > 3 else 1
        gen_invite(count, max_uses)
    elif cmd == "list-invites":
        list_invites()
    elif cmd == "disable-invite":
        disable_invite(sys.argv[2])
    elif cmd == "list-users":
        list_users()
    elif cmd == "ban-user":
        ban_user(int(sys.argv[2]))
    elif cmd == "unban-user":
        unban_user(int(sys.argv[2]))
    elif cmd == "set-admin":
        set_admin(int(sys.argv[2]))
    else:
        usage()