toDus-API 1.3.3__py3-none-any.whl

todus/stanzas/private.py

@@ -0,0 +1,203 @@
+"""Generadores de stanzas XML para chat privado en ToDus."""
+
+import hashlib
+from .. import util
+
+
+def _generate_msg_id() -> str:
+    """Genera msg_id en formato hex 32 chars como usa ToDus oficial."""
+    return hashlib.md5(util.generate_token(16).encode()).hexdigest()
+
+
+def message(to: str, body: str, msg_id: str = "", msg_type: str = "c") -> str:
+    """Stanza <m> de ToDus para chat privado."""
+    mid = msg_id or _generate_msg_id()
+    body_esc = util.escape_xml(body)
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<b>{body_esc}</b>"
+        f"</m>"
+    )
+
+
+def edit_message(to: str, new_body: str, original_msg_id: str, edit_id: str = "") -> str:
+    """Edita un mensaje existente en chat privado."""
+    eid = edit_id or _generate_msg_id()
+    body_esc = util.escape_xml(new_body)
+    return (
+        f"<m to='{to}' t='c' i='{original_msg_id}' xmlns='jc'>"
+        f"<edited xmlns='edited:n' i='{eid}' mi='{original_msg_id}'/>"
+        f"<k xmlns='x8'/>"
+        f"<b>{body_esc}</b>"
+        f"</m>"
+    )
+
+
+def file_message(to: str, url: str, file_type: int, caption: str = "", msg_id: str = "", msg_type: str = "c", file_name: str = "", file_size: int = 0) -> str:
+    """Stanza con archivo adjunto para chat privado."""
+    mid = msg_id or _generate_msg_id()
+    fid = _generate_msg_id()
+    cap_esc = util.escape_xml(caption)
+    name_esc = util.escape_xml(file_name)
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<b>{cap_esc}</b>"
+        f"<file xmlns='file:n' i='{fid}' mi='{mid}' n='{name_esc}' url='{util.escape_xml(url)}' s='{file_size}' h=''/>"
+        f"</m>"
+    )
+
+
+def image_message(to: str, url: str, file_name: str, file_size: int, width: int = 0, height: int = 0, thumbnail: str = "", caption: str = "", msg_id: str = "", msg_type: str = "c") -> str:
+    """Stanza con imagen adjunta para chat privado."""
+    mid = msg_id or _generate_msg_id()
+    fid = _generate_msg_id()
+    cap_esc = util.escape_xml(caption)
+    name_esc = util.escape_xml(file_name)
+    url_esc = util.escape_xml(url)
+
+    tnail = thumbnail if thumbnail else "U6688O?Hr=xu^-w2sp-;,^VZnm-;_3xHMyt5"
+
+    wh_attrs = ""
+    if width > 0:
+        wh_attrs += f" w='{width}'"
+    if height > 0:
+        wh_attrs += f" he='{height}'"
+
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<image xmlns='image:n' i='{fid}' mi='{mid}' url='{url_esc}' n='{name_esc}' s='{file_size}' h=''{wh_attrs} tnail='{tnail}'/>"
+        f"<b>{cap_esc}</b>"
+        f"</m>"
+    )
+
+
+def image_message_simple(to: str, url: str, file_name: str, file_size: int, msg_id: str = "", msg_type: str = "c") -> str:
+    """Versión simple sin metadata para chat privado."""
+    mid = msg_id or _generate_msg_id()
+    fid = _generate_msg_id()
+    name_esc = util.escape_xml(file_name)
+    url_esc = util.escape_xml(url)
+
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<image xmlns='image:n' i='{fid}' mi='{mid}' url='{url_esc}' n='{name_esc}' s='{file_size}' h=''/>"
+        f"<b/>"
+        f"</m>"
+    )
+
+
+def button_message(to: str, text: str, buttons: list[dict], msg_id: str = "", msg_type: str = "c") -> str:
+    """Stanza con botones interactivos."""
+    mid = msg_id or _generate_msg_id()
+    text_esc = util.escape_xml(text)
+
+    buttons_xml = ""
+    for btn in buttons:
+        btn_text = util.escape_xml(btn.get("text", ""))
+        btn_cmd = btn.get("command", "cmd_type_send")
+        btn_msg = util.escape_xml(btn.get("data", ""))
+        btn_size = btn.get("size", "0.82")
+        buttons_xml += f"<button xmlns='button:n' btn_t='{btn_text}' btn_cmd='{btn_cmd}' btn_msg_c='{btn_msg}' btn_size='{btn_size}'/>"
+
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<b>{text_esc}</b>"
+        f"{buttons_xml}"
+        f"</m>"
+    )
+
+
+def contact_message(to: str, contact_id: str, contact_name: str, contact_phone: str, msg_id: str = "", msg_type: str = "c") -> str:
+    """Stanza con contacto adjunto."""
+    mid = msg_id or _generate_msg_id()
+    name_esc = util.escape_xml(contact_name)
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<contact xmlns='contact:n' i='{contact_id}' mi='{mid}' n='{name_esc}' num='{contact_phone}'/>"
+        f"<b/>"
+        f"</m>"
+    )
+
+
+def sticker_message(to: str, sticker_id: str, sticker_name: str, sticker_pack: str, sticker_hash: str, msg_id: str = "", msg_type: str = "c") -> str:
+    """Stanza con sticker adjunto."""
+    mid = msg_id or _generate_msg_id()
+    name_esc = util.escape_xml(sticker_name)
+    pack_esc = util.escape_xml(sticker_pack)
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<sticker xmlns='sticker:n' i='{sticker_id}' mi='{mid}' n='{name_esc}' f='{pack_esc}' url='' s='0' h='{sticker_hash}' json=''/>"
+        f"<b/>"
+        f"</m>"
+    )
+
+
+def video_message(to: str, url: str, video_id: str, file_name: str, file_size: int, duration: int, width: int, height: int, thumbnail: str, msg_id: str = "", msg_type: str = "c", info_text: str = "") -> str:
+    """Stanza con video adjunto."""
+    mid = msg_id or _generate_msg_id()
+    name_esc = util.escape_xml(file_name)
+    url_esc = util.escape_xml(url)
+
+    body_tag = "<b/>" if not info_text else f"<b>{util.escape_xml(info_text)}</b>"
+
+    return (
+        f"<m xml:lang='en' o='{to}' t='{msg_type}' i='{mid}'>"
+        f"<k xmlns='x8'/>"
+        f"<video xmlns='video:n' i='{video_id}' mi='{mid}' url='{url_esc}' s='{file_size}' h='' d='{duration}' n='{name_esc}' w='{width}' he='{height}' tnail='{thumbnail}'/>"
+        f"{body_tag}"
+        f"</m>"
+    )
+
+
+def delete_message(to: str, message_id: str, msg_id: str = "", msg_type: str = "c", body: str = "", media_xml: str = "") -> str:
+    """Eliminar mensaje."""
+    mid = msg_id or message_id
+    did = _generate_msg_id()
+    body_xml = f"<b>{util.escape_xml(body)}</b>" if body or not media_xml else "<b/>"
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"{media_xml}"
+        f"<deleted xmlns='deleted:n' i='{did}' mi='{message_id}'/>"
+        f"{body_xml}"
+        f"</m>"
+    )
+
+
+def location_message(to: str, lat: float, lon: float, zoom: float = 11.0, text: str = "", msg_id: str = "") -> str:
+    """Stanza con ubicación adjunta para chat privado."""
+    mid = msg_id or _generate_msg_id()
+    lid = _generate_msg_id()
+    text_esc = util.escape_xml(text)
+    return (
+        f"<m to='{to}' t='c' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<location xmlns='location:n' i='{lid}' mi='{mid}' lat='{lat}' lon='{lon}' z='{zoom}' t='{text_esc}'/>"
+        f"<b/>"
+        f"</m>"
+    )
+
+
+def event_message(to: str, event_id: str, title: str, start: int, end: int, all_day: bool, ics_data: str, msg_id: str = "") -> str:
+    """Stanza con evento/calendario adjunto para chat privado."""
+    mid = msg_id or _generate_msg_id()
+    eid = event_id or _generate_msg_id()
+    ad_str = "true" if all_day else "false"
+    title_esc = util.escape_xml(title)
+    ics_esc = util.escape_xml(ics_data)
+    return (
+        f"<m to='{to}' t='c' i='{mid}' xmlns='jc'>"
+        f"<k xmlns='x8'/>"
+        f"<event xmlns='event:n' i='{eid}' mi='{mid}' ti='{title_esc}' s='{start}' e='{end}' ad='{ad_str}'>"
+        f"<ics>{ics_esc}</ics>"
+        f"</event>"
+        f"<b/>"
+        f"</m>"
+    )

todus/stanzas/utils.py

@@ -0,0 +1,122 @@
+"""Stanzas XML de utilidad y de protocolo XMPP para ToDus."""
+
+import hashlib
+from .. import util
+
+
+def _generate_msg_id() -> str:
+    """Genera msg_id en formato hex 32 chars como usa ToDus oficial."""
+    return hashlib.md5(util.generate_token(16).encode()).hexdigest()
+
+
+def iq(type_: str, iq_id: str, payload: str = "", to: str = "") -> str:
+    """Stanza IQ genérica."""
+    to_attr = f" to='{to}'" if to else ""
+    return f"<iq i='{iq_id}' t='{type_}'{to_attr}>{payload}</iq>"
+
+
+def ping(ping_id: str) -> str:
+    """XMPP ping (urn:xmpp:ping)."""
+    return f"<iq i='{ping_id}' t='get'><ping xmlns='urn:xmpp:ping'/></iq>"
+
+
+def chat_state(to: str, state: str, msg_id: str = "", msg_type: str = "c") -> str:
+    """Notificación de estado de chat para ToDus."""
+    mid = msg_id or _generate_msg_id()
+    tag = "csp" if state == "composing" else "csc"
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{mid}' xmlns='jc'>"
+        f"<{tag} xmlns='uc1'/>"
+        f"</m>"
+    )
+
+
+def receipt(to: str, msg_id: str, receipt_id: str = "", msg_type: str = "c") -> str:
+    """Delivery receipt para ToDus."""
+    rid = receipt_id or _generate_msg_id()
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{rid}' xmlns='jc'>"
+        f"<dd xmlns='x8' i='{msg_id}'/>"
+        f"</m>"
+    )
+
+
+def read_receipt(to: str, msg_id: str, receipt_id: str = "", msg_type: str = "c") -> str:
+    """Read receipt para ToDus."""
+    rid = receipt_id or _generate_msg_id()
+    return (
+        f"<m to='{to}' t='{msg_type}' i='{rid}' xmlns='jc'>"
+        f"<rd xmlns='x8' i='{msg_id}'/>"
+        f"</m>"
+    )
+
+
+def ack(msg_id: str, to: str = "") -> str:
+    """ACK de mensaje recibido (tdack)."""
+    return f"<tdack xmlns='x8' mi='{msg_id}'/>"
+
+
+def keepalive() -> str:
+    """Keepalive: espacio en blanco."""
+    return " "
+
+
+def stream_open(host: str = "im.todus.cu") -> str:
+    """Stream header inicial."""
+    return f"<stream:stream xmlns='jc' o='{host}' xmlns:stream='x1' v='1.0'>"
+
+
+def stream_restart(host: str = "im.todus.cu") -> str:
+    """Stream header post-auth."""
+    return f"<stream:stream xmlns='jc' o='{host}' xmlns:stream='x1' v='1.0'>"
+
+
+def stream_close() -> str:
+    """Cierre graceful del stream."""
+    return "</stream:stream>"
+
+
+def sasl_auth(authstr: bytes) -> str:
+    """SASL PLAIN auth."""
+    return b"<ah xmlns='ah:ns' e='PLAIN'>" + authstr + b"</ah>"
+
+
+def bind(iq_id: str) -> str:
+    """Resource bind."""
+    return f"<iq i='{iq_id}' t='set'><b1 xmlns='x4'></b1></iq>"
+
+
+def mam_query(query_id: str, since: str = "", before: str = "", limit: int = 50) -> str:
+    """Query de Message Archive Management (MAM)."""
+    filters = ""
+    if since:
+        filters += f"<start>{since}</start>"
+    if before:
+        filters += f"<end>{before}</end>"
+    return (
+        f"<iq i='{query_id}' t='set'>"
+        f"<query xmlns='todus:mam'>{filters}"
+        f"<set xmlns='http://jabber.org/protocol/rsm'><max>{limit}</max></set>"
+        f"</query></iq>"
+    )
+
+
+def upload_query(iq_id: str, size: int, file_type: int, persistent: bool = False, file_name: str = "") -> str:
+    """Reserva URL de subida."""
+    persist = "true" if persistent else "false"
+    n_attr = f" n='{util.escape_xml(file_name)}'" if file_name else ""
+    return (
+        f"<iq i='{iq_id}-3' t='get'>"
+        f"<query xmlns='todus:purl' type='{file_type}' "
+        f"persistent='{persist}' size='{size}' room=''{n_attr}></query>"
+        f"</iq>"
+    )
+
+
+def download_query(iq_id: str, url: str) -> str:
+    """Resuelve URL real de descarga."""
+    return (
+        f"<iq i='{iq_id}-2' t='get'>"
+        f"<query xmlns='todus:gurl' url='{url}'></query>"
+        f"</iq>"
+    )

todus/types.py

@@ -0,0 +1,54 @@
+"""Tipos y enumeraciones de ToDus."""
+
+from enum import IntEnum, StrEnum
+
+
+class FileType(IntEnum):
+    """Tipos de archivo soportados por ToDus."""
+    FILE = 0
+    VOICE = 1
+    AUDIO = 2
+    VIDEO = 3
+    PICTURE = 4
+    PROFILE = 5
+    PROFILE_THUMBNAIL = 6
+
+
+class ChatState(StrEnum):
+    """Estados de chat XEP-0085."""
+    COMPOSING = "composing"
+    PAUSED = "paused"
+    ACTIVE = "active"
+    GONE = "gone"
+    INACTIVE = "inactive"
+
+
+class MessageType(StrEnum):
+    """Tipos de mensaje XMPP."""
+    CHAT = "chat"
+    GROUPCHAT = "groupchat"
+    ERROR = "error"
+    HEADLINE = "headline"
+    NORMAL = "normal"
+
+
+class PresenceShow(StrEnum):
+    """Estados de presencia XMPP."""
+    CHAT = "chat"
+    AWAY = "away"
+    XA = "xa"
+    DND = "dnd"
+
+
+class ButtonSize(StrEnum):
+    """Tamaños de botón interactivo."""
+    FULL = "0.82"
+    HALF = "0.5"
+
+
+class ButtonCommand(StrEnum):
+    """Tipos de comando para botones interactivos."""
+    SEND = "cmd_type_send"
+    URL = "cmd_type_url"
+    COPY = "cmd_type_copy"
+    CALL = "cmd_type_call"

todus/util.py

@@ -0,0 +1,177 @@
+"""Utilidades para ToDus."""
+
+import json
+import re
+import secrets
+import string
+from base64 import b64decode
+from datetime import datetime
+
+
+def generate_token(length: int = 8) -> str:
+    """Genera un token alfanumérico aleatorio criptográficamente seguro."""
+    chars = string.ascii_letters + string.digits
+    return "".join(secrets.choice(chars) for _ in range(length))
+
+
+def normalize_phone(phone_number: str) -> str:
+    """Normaliza número cubano a formato 53XXXXXXXX."""
+    phone_number = "".join(phone_number.lstrip("+").split())
+    match = re.match(r"(53)?(\d{8})", phone_number)
+    if not match:
+        raise ValueError(f"Número inválido: {phone_number}")
+    return "53" + match.group(2)
+
+
+def build_jid(phone_number: str) -> str:
+    """Construye JID ToDus desde número de teléfono."""
+    return normalize_phone(phone_number) + "@im.todus.cu"
+
+
+def parse_jid(jid: str) -> tuple[str, str]:
+    """Extrae (phone, resource) de un JID."""
+    parts = jid.split("/", 1)
+    phone = parts[0].split("@")[0]
+    resource = parts[1] if len(parts) > 1 else ""
+    return phone, resource
+
+
+def escape_xml(text: str) -> str:
+    """Escapa caracteres XML especiales, incluyendo apóstrofos."""
+    return (
+        text.replace("&", "&")
+        .replace("<", "&lt;")
+        .replace(">", "&gt;")
+        .replace("'", "&apos;")
+    )
+
+
+def unescape_xml(text: str) -> str:
+    """Revierte escape XML."""
+    return (
+        text.replace("&lt;", "<")
+        .replace("&gt;", ">")
+        .replace("&amp;", "&")
+        .replace("&apos;", "'")
+    )
+
+
+def jwt_decode_payload(token: str) -> dict:
+    """Decodifica payload de JWT sin verificar firma."""
+    parts = token.split(".")
+    if len(parts) < 2:
+        return {}
+    payload = parts[1]
+    padding = 4 - len(payload) % 4
+    if padding != 4:
+        payload += "=" * padding
+    try:
+        decoded = b64decode(payload).decode("utf-8", errors="ignore")
+        return json.loads(decoded)
+    except Exception:
+        return {}
+
+
+def timestamp_ms() -> int:
+    """Timestamp actual en milisegundos."""
+    return int(datetime.now().timestamp() * 1000)
+
+
+def format_size(size_bytes: int) -> str:
+    """Formatea tamaño en bytes a human readable."""
+    for unit in ["B", "KB", "MB", "GB"]:
+        if size_bytes < 1024:
+            return f"{size_bytes:.1f} {unit}"
+        size_bytes /= 1024
+    return f"{size_bytes:.1f} TB"
+
+
+def get_image_dimensions(data: bytes) -> tuple[int, int]:
+    """Extrae dimensiones de imagen JPEG/PNG sin decodificar completamente."""
+    width = height = 0
+
+    # PNG
+    if data.startswith(b'\x89PNG\r\n\x1a\n'):
+        for i in range(0, len(data) - 8, 8):
+            if data[i:i+4] == b'IHDR':
+                width = int.from_bytes(data[i+4:i+8], 'big')
+                height = int.from_bytes(data[i+8:i+12], 'big')
+                break
+
+    # JPEG
+    elif data.startswith(b'\xff\xd8'):
+        i = 2
+        while i < len(data) - 8:
+            if data[i] != 0xff:
+                i += 1
+                continue
+            marker = data[i+1]
+            if marker == 0xc0 or marker == 0xc2:
+                height = int.from_bytes(data[i+5:i+7], 'big')
+                width = int.from_bytes(data[i+7:i+9], 'big')
+                break
+            i += 2 + int.from_bytes(data[i+2:i+4], 'big')
+
+    return width, height
+
+
+def generate_blurhash(width: int, height: int) -> str:
+    """Genera un BlurHash simple basado en dimensiones."""
+    # Por ahora devolvemos un hash genérico
+    # En producción usar librería blurhash
+    return "LFE_@w00ay00ay00ay00ay00ay00ay"
+
+
+def sanitize_filename(filename: str, file_type: int = 0) -> str:
+    """Limpia caracteres problemáticos en el nombre del archivo para URLs, asegurando extensión según el tipo."""
+    import os
+    from .types import FileType
+    
+    # Mapeo de extensiones por defecto por tipo
+    default_exts = {
+        FileType.PICTURE: ".jpg",
+        FileType.VIDEO: ".mp4",
+        FileType.AUDIO: ".mp3",
+        FileType.VOICE: ".opus",
+        FileType.FILE: ".bin",
+        FileType.PROFILE: ".jpg",
+        FileType.PROFILE_THUMBNAIL: ".jpg",
+    }
+    
+    default_ext = default_exts.get(file_type, ".bin")
+    
+    if not filename:
+        # Generar nombre por defecto por tipo
+        default_names = {
+            FileType.PICTURE: "photo",
+            FileType.VIDEO: "video",
+            FileType.AUDIO: "audio",
+            FileType.VOICE: "voice",
+            FileType.FILE: "file",
+            FileType.PROFILE: "profile",
+            FileType.PROFILE_THUMBNAIL: "thumbnail",
+        }
+        stem = default_names.get(file_type, "file")
+        ext = default_ext
+    else:
+        # Solo el nombre del archivo si es una ruta
+        filename = os.path.basename(filename)
+        
+        # Separar nombre y extensión
+        parts = filename.rsplit(".", 1)
+        stem = parts[0]
+        ext = "." + parts[1] if len(parts) > 1 else ""
+        
+        # Si no tiene extensión, usar la correspondiente al tipo
+        if not ext:
+            ext = default_ext
+            
+    # Reemplazar caracteres no permitidos en nombres de archivos o problemáticos en URLs
+    stem_clean = re.sub(r'[\\/*?:"<>|\s]', "_", stem)
+    
+    # Limitar longitud para evitar URLs excesivamente largas
+    if len(stem_clean) > 50:
+        stem_clean = stem_clean[:47] + "..."
+        
+    return f"{stem_clean}{ext}"
+