Applying previous commit.

cogs/aimod_config.py

@@ -0,0 +1,176 @@
+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"
+    ),
+]
+
+MOD_LOG_API_SECRET_ENV_VAR = "MOD_LOG_API_SECRET"
+
+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")
+
+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 = {}
+
+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}")
+
+
+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,
+):
+    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,
+    }
+    USER_INFRACTIONS[key].append(infraction_record)
+    USER_INFRACTIONS[key] = USER_INFRACTIONS[key][-10:]
+    await save_user_infractions()
+
+
+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:** Do not post AI-generated pornography.
+
+* **Zero Tolerance for Pedophilia:** Any form of pedophilia, including lolicon and shotacon content, is strictly forbidden and will result in an immediate ban.
+
+* **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`
+"""
+
+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.
+"""