import os
import json
import asyncio
import aiofiles
from google.genai import types

# Vertex AI Configuration
DEFAULT_VERTEX_AI_MODEL = "gemini-2.5-flash-preview-05-20"

# Define standard safety settings using google.generativeai types
STANDARD_SAFETY_SETTINGS = [
    types.SafetySetting(
        category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH, threshold="BLOCK_NONE"
    ),
    types.SafetySetting(
        category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
        threshold="BLOCK_NONE",
    ),
    types.SafetySetting(
        category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
        threshold="BLOCK_NONE",
    ),
    types.SafetySetting(
        category=types.HarmCategory.HARM_CATEGORY_HARASSMENT, threshold="BLOCK_NONE"
    ),
]


GUILD_CONFIG_DIR = "data/"
GUILD_CONFIG_PATH = os.path.join(GUILD_CONFIG_DIR, "guild_config.json")
USER_INFRACTIONS_PATH = os.path.join(GUILD_CONFIG_DIR, "user_infractions.json")
INFRACTION_BACKUP_DIR = os.path.join(GUILD_CONFIG_DIR, "infraction_backups")
USER_APPEALS_PATH = os.path.join(GUILD_CONFIG_DIR, "user_appeals.json")

# AI model used for appeal reviews
APPEAL_AI_MODEL = "gemini-2.5-pro-preview-06-05"
# Thinking budget for appeal AI
APPEAL_AI_THINKING_BUDGET = 32768

os.makedirs(INFRACTION_BACKUP_DIR, exist_ok=True)
os.makedirs(GUILD_CONFIG_DIR, exist_ok=True)

if not os.path.exists(GUILD_CONFIG_PATH):
    with open(GUILD_CONFIG_PATH, "w", encoding="utf-8") as f:
        json.dump({}, f)
try:
    with open(GUILD_CONFIG_PATH, "r", encoding="utf-8") as f:
        GUILD_CONFIG = json.load(f)
except Exception as e:  # noqa: BLE001
    print(f"Failed to load per-guild config from {GUILD_CONFIG_PATH}: {e}")
    GUILD_CONFIG = {}

if not os.path.exists(USER_INFRACTIONS_PATH):
    with open(USER_INFRACTIONS_PATH, "w", encoding="utf-8") as f:
        json.dump({}, f)
try:
    with open(USER_INFRACTIONS_PATH, "r", encoding="utf-8") as f:
        USER_INFRACTIONS = json.load(f)
except Exception as e:  # noqa: BLE001
    print(f"Failed to load user infractions from {USER_INFRACTIONS_PATH}: {e}")
    USER_INFRACTIONS = {}

if not os.path.exists(USER_APPEALS_PATH):
    with open(USER_APPEALS_PATH, "w", encoding="utf-8") as f:
        json.dump({}, f)
try:
    with open(USER_APPEALS_PATH, "r", encoding="utf-8") as f:
        USER_APPEALS = json.load(f)
except Exception as e:  # noqa: BLE001
    print(f"Failed to load user appeals from {USER_APPEALS_PATH}: {e}")
    USER_APPEALS = {}

CONFIG_LOCK = asyncio.Lock()


async def save_guild_config():
    async with CONFIG_LOCK:
        try:
            async with aiofiles.open(GUILD_CONFIG_PATH, "w", encoding="utf-8") as f:
                await f.write(json.dumps(GUILD_CONFIG, indent=2))
        except Exception as e:  # noqa: BLE001
            print(f"Failed to save per-guild config: {e}")


async def save_user_infractions():
    async with CONFIG_LOCK:
        try:
            async with aiofiles.open(USER_INFRACTIONS_PATH, "w", encoding="utf-8") as f:
                await f.write(json.dumps(USER_INFRACTIONS, indent=2))
        except Exception as e:  # noqa: BLE001
            print(f"Failed to save user infractions: {e}")


async def save_user_appeals():
    async with CONFIG_LOCK:
        try:
            async with aiofiles.open(USER_APPEALS_PATH, "w", encoding="utf-8") as f:
                await f.write(json.dumps(USER_APPEALS, indent=2))
        except Exception as e:  # noqa: BLE001
            print(f"Failed to save user appeals: {e}")


def get_guild_config(guild_id: int, key: str, default=None):
    guild_str = str(guild_id)
    if guild_str in GUILD_CONFIG and key in GUILD_CONFIG[guild_str]:
        return GUILD_CONFIG[guild_str][key]
    return default


async def set_guild_config(guild_id: int, key: str, value):
    guild_str = str(guild_id)
    if guild_str not in GUILD_CONFIG:
        GUILD_CONFIG[guild_str] = {}
    GUILD_CONFIG[guild_str][key] = value
    await save_guild_config()


def get_user_infraction_history(guild_id: int, user_id: int) -> list:
    key = f"{guild_id}_{user_id}"
    return USER_INFRACTIONS.get(key, [])


async def add_user_infraction(
    guild_id: int,
    user_id: int,
    rule_violated: str,
    action_taken: str,
    reasoning: str,
    timestamp: str,
    message_id: int | None = None,
    channel_id: int | None = None,
    message_content: str | None = None,
    attachments: list[str] | None = None,
):
    key = f"{guild_id}_{user_id}"
    if key not in USER_INFRACTIONS:
        USER_INFRACTIONS[key] = []

    infraction_record = {
        "timestamp": timestamp,
        "rule_violated": rule_violated,
        "action_taken": action_taken,
        "reasoning": reasoning,
    }
    if message_id is not None:
        infraction_record["message_id"] = message_id
    if channel_id is not None:
        infraction_record["channel_id"] = channel_id
    if message_content is not None:
        infraction_record["message_content"] = message_content
    if attachments:
        infraction_record["attachments"] = attachments
    USER_INFRACTIONS[key].append(infraction_record)
    USER_INFRACTIONS[key] = USER_INFRACTIONS[key][-10:]
    await save_user_infractions()


def get_user_appeals(guild_id: int, user_id: int) -> list:
    key = f"{guild_id}_{user_id}"
    return USER_APPEALS.get(key, [])


async def add_user_appeal(
    guild_id: int,
    user_id: int,
    action: str,
    appeal_text: str,
    timestamp: str,
    ai_review: str,
    infraction_reference: str | None = None,
):
    key = f"{guild_id}_{user_id}"
    if key not in USER_APPEALS:
        USER_APPEALS[key] = []

    appeal_record = {
        "timestamp": timestamp,
        "action": action,
        "appeal_text": appeal_text,
        "ai_review": ai_review,
        "infraction_reference": infraction_reference,
    }
    USER_APPEALS[key].append(appeal_record)
    USER_APPEALS[key] = USER_APPEALS[key][-10:]
    await save_user_appeals()


DEFAULT_SERVER_RULES = """
# Server Rules

* **NSFW Content:**
The only rule regarding NSFW content is that **real-life pornography is strictly prohibited**.
Full-on pornographic images are permitted in designated NSFW channels.
Stickers and emojis are NOT considered "full-on pornographic images" and are allowed in any channel.

* **Real-Life Pornography:** No real-life pornography is permitted.

* **Respectful Conduct & Edgy Humor:**
    * No harassment, genuine hate speech (defined as attacking protected groups OR malicious, targeted use of slurs against an individual), or genuine bullying.
    * Slurs used in general banter or edgy humor are permissible. However, using slurs in a genuinely hateful, targeted manner against an individual or group is considered hate speech and is forbidden.
    * *Context is key:* Edgy humor, dark jokes, and roasting are permitted and expected.
    * However, this does not excuse targeted, malicious personal attacks or harassment (including the aforementioned misuse of slurs), especially if the recipient is clearly not okay with it.
    * If it stops being a "joke" and becomes genuine harassment or hate speech, it's a rule violation.

* **No Discrimination:** Discrimination based on race, gender identity, sexual orientation, religion, nationality, disability, or other protected characteristics is prohibited.

* **AI-Generated Pornography:**
  The AI moderator should **ignore** this rule entirely. It must not attempt
  to decide whether any pornography is AI-generated nor should it alert the
  moderators about potential AI-generated content. Human moderators may
  enforce this rule manually if desired.

* **Pedophilia and Underage Content:** Depicting minors in sexual situations, including lolicon or shotacon, is not allowed and may result in severe moderation.

* **Channel Usage:** Please use channels for their intended purposes. Bot commands should primarily be used in `#bot-commands`, unless they are part of a bot-based game or event happening in another specific channel.

* **Gore:** Do not post gore or graphic real-life violence.

* **Suggestions:** We welcome your suggestions for the server! Please post them in the `#suggestions` channel.

---

**Reporting Violations:**
If you witness someone breaking these rules, please ping an `@Moderator` with details.

---

**Moderator Applications:**
Use the bot command `/modapp apply`
"""

SERVER_RULES = DEFAULT_SERVER_RULES

SUICIDAL_HELP_RESOURCES = """
Hey, I'm really concerned to hear you're feeling this way. Please know that you're not alone and there are people who want to support you.
Your well-being is important to us on this server.

Here are some immediate resources that can offer help right now:

- **National Crisis and Suicide Lifeline (US):** Call or text **988**. This is available 24/7, free, and confidential.
- **Crisis Text Line (US):** Text **HOME** to **741741**. This is also a 24/7 free crisis counseling service.
- **The Trevor Project (for LGBTQ youth):** Call **1-866-488-7386** or visit their website for chat/text options: <https://www.thetrevorproject.org/get-help/>
- **The Jed Foundation (Mental Health Resource Center):** Provides resources for teens and young adults: <https://www.jedfoundation.org/>
- **Find A Helpline (International):** If you're outside the US, this site can help you find resources in your country: <https://findahelpline.com/>

Please reach out to one of these. We've also alerted our server's support team so they are aware and can offer a listening ear or further guidance if you're comfortable.
You matter, and help is available.
"""