feat: implementar caché de traducciones y persistencia de mensajes en MySQL para Discord y Telegram

botdiscord/database.py

@@ -1,5 +1,6 @@
 import sqlite3
 import mysql.connector
+import json
 from mysql.connector import Error as MySQLError
 from botdiscord.config import get_db_config, get_db_type
 
@@ -87,6 +88,26 @@ def init_db():
         cursor.execute('''CREATE TABLE IF NOT EXISTS bot_languages
                      (bot_type VARCHAR(50) NOT NULL, lang_code VARCHAR(10) NOT NULL, 
                       PRIMARY KEY (bot_type, lang_code)) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4''')
+        
+        # Nuevas tablas para caché y persistencia
+        cursor.execute('''CREATE TABLE IF NOT EXISTS messages
+                     (message_id BIGINT PRIMARY KEY, 
+                      guild_id BIGINT, 
+                      author_id BIGINT, 
+                      content TEXT NOT NULL, 
+                      mentions_map JSON, 
+                      bot_type ENUM('discord', 'telegram') NOT NULL,
+                      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4''')
+        
+        cursor.execute('''CREATE TABLE IF NOT EXISTS translations
+                     (id INT AUTO_INCREMENT PRIMARY KEY,
+                      message_id BIGINT NOT NULL,
+                      target_lang VARCHAR(10) NOT NULL,
+                      translated_text TEXT NOT NULL,
+                      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                      FOREIGN KEY (message_id) REFERENCES messages(message_id) ON DELETE CASCADE,
+                      UNIQUE KEY idx_msg_lang (message_id, target_lang)) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4''')
+        
         conn.commit()
         cursor.close()
     else:
@@ -100,6 +121,24 @@ def init_db():
                      (code TEXT PRIMARY KEY, name TEXT, flag TEXT)''')
         c.execute('''CREATE TABLE IF NOT EXISTS bot_languages
                      (bot_type TEXT, lang_code TEXT, PRIMARY KEY (bot_type, lang_code))''')
+        
+        # SQLite equivalents
+        c.execute('''CREATE TABLE IF NOT EXISTS messages
+                     (message_id INTEGER PRIMARY KEY, 
+                      guild_id INTEGER, 
+                      author_id INTEGER, 
+                      content TEXT NOT NULL, 
+                      mentions_map TEXT, 
+                      bot_type TEXT NOT NULL,
+                      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP)''')
+        
+        c.execute('''CREATE TABLE IF NOT EXISTS translations
+                     (id INTEGER PRIMARY KEY AUTOINCREMENT,
+                      message_id INTEGER NOT NULL,
+                      target_lang TEXT NOT NULL,
+                      translated_text TEXT NOT NULL,
+                      created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                      UNIQUE(message_id, target_lang))''')
         conn.commit()
         conn.close()
 
@@ -256,3 +295,87 @@ def set_config_value(key: str, value: str):
         c.execute("INSERT OR REPLACE INTO bot_config (key, value) VALUES (?, ?)", (key, value))
         conn.commit()
         conn.close()
+
+def save_message(message_id: int, guild_id: int, author_id: int, content: str, mentions_map: dict, bot_type: str):
+    db_type = get_db_type()
+    mentions_json = json.dumps(mentions_map)
+    
+    if db_type == "mysql":
+        conn = get_connection()
+        cursor = conn.cursor()
+        query = """INSERT INTO messages (message_id, guild_id, author_id, content, mentions_map, bot_type) 
+                   VALUES (%s, %s, %s, %s, %s, %s) 
+                   ON DUPLICATE KEY UPDATE content = %s, mentions_map = %s"""
+        cursor.execute(query, (message_id, guild_id, author_id, content, mentions_json, bot_type, content, mentions_json))
+        conn.commit()
+        cursor.close()
+    else:
+        conn = get_connection()
+        c = conn.cursor()
+        c.execute("INSERT OR REPLACE INTO messages (message_id, guild_id, author_id, content, mentions_map, bot_type) VALUES (?, ?, ?, ?, ?, ?)", 
+                  (message_id, guild_id, author_id, content, mentions_json, bot_type))
+        conn.commit()
+        conn.close()
+
+def get_message(message_id: int):
+    db_type = get_db_type()
+    
+    if db_type == "mysql":
+        conn = get_connection()
+        cursor = conn.cursor(dictionary=True)
+        cursor.execute("SELECT content, mentions_map, bot_type FROM messages WHERE message_id = %s", (message_id,))
+        row = cursor.fetchone()
+        cursor.close()
+        if row:
+            if row['mentions_map']:
+                row['mentions_map'] = json.loads(row['mentions_map'])
+            return row
+    else:
+        conn = get_connection()
+        c = conn.cursor()
+        c.execute("SELECT content, mentions_map, bot_type FROM messages WHERE message_id = ?", (message_id,))
+        row = c.fetchone()
+        conn.close()
+        if row:
+            return {
+                'content': row[0],
+                'mentions_map': json.loads(row[1]) if row[1] else {},
+                'bot_type': row[2]
+            }
+    return None
+
+def save_translation(message_id: int, target_lang: str, translated_text: str):
+    db_type = get_db_type()
+    
+    if db_type == "mysql":
+        conn = get_connection()
+        cursor = conn.cursor()
+        query = "INSERT INTO translations (message_id, target_lang, translated_text) VALUES (%s, %s, %s) ON DUPLICATE KEY UPDATE translated_text = %s"
+        cursor.execute(query, (message_id, target_lang, translated_text, translated_text))
+        conn.commit()
+        cursor.close()
+    else:
+        conn = get_connection()
+        c = conn.cursor()
+        c.execute("INSERT OR REPLACE INTO translations (message_id, target_lang, translated_text) VALUES (?, ?, ?)", 
+                  (message_id, target_lang, translated_text))
+        conn.commit()
+        conn.close()
+
+def get_cached_translation(message_id: int, target_lang: str) -> str:
+    db_type = get_db_type()
+    
+    if db_type == "mysql":
+        conn = get_connection()
+        cursor = conn.cursor()
+        cursor.execute("SELECT translated_text FROM translations WHERE message_id = %s AND target_lang = %s", (message_id, target_lang))
+        row = cursor.fetchone()
+        cursor.close()
+        return row[0] if row else None
+    else:
+        conn = get_connection()
+        c = conn.cursor()
+        c.execute("SELECT translated_text FROM translations WHERE message_id = ? AND target_lang = ?", (message_id, target_lang))
+        row = c.fetchone()
+        conn.close()
+        return row[0] if row else None