contextbase-plugin-imessage-local 0.2.9__py3-none-any.whl

plugin_imessage_local/models/ingress.py

@@ -0,0 +1,321 @@
+from __future__ import annotations
+
+from datetime import datetime
+
+from sqlalchemy import LargeBinary, String
+from sqlalchemy.orm import Mapped, mapped_column
+from shared_plugins.sqlalchemy_types import AppleMessageTimestamp
+
+from .base import Base
+
+
+class IMessageHandle(Base):
+    """Row from handle in chat.db."""
+
+    __tablename__ = "handle"
+
+    rowid: Mapped[int] = mapped_column("ROWID", primary_key=True)
+    id: Mapped[str] = mapped_column("id", String)
+    country: Mapped[str | None] = mapped_column("country", String)
+    service: Mapped[str] = mapped_column("service", String)
+    uncanonicalized_id: Mapped[str | None] = mapped_column("uncanonicalized_id", String)
+    person_centric_id: Mapped[str | None] = mapped_column("person_centric_id", String)
+
+
+class IMessageChat(Base):
+    """Row from chat in chat.db."""
+
+    __tablename__ = "chat"
+
+    rowid: Mapped[int] = mapped_column("ROWID", primary_key=True)
+    guid: Mapped[str] = mapped_column("guid", String)
+    style: Mapped[int | None] = mapped_column("style")
+    state: Mapped[int | None] = mapped_column("state")
+    account_id: Mapped[str | None] = mapped_column("account_id", String)
+    properties: Mapped[bytes | None] = mapped_column("properties", LargeBinary)
+    chat_identifier: Mapped[str | None] = mapped_column("chat_identifier", String)
+    service_name: Mapped[str | None] = mapped_column("service_name", String)
+    room_name: Mapped[str | None] = mapped_column("room_name", String)
+    account_login: Mapped[str | None] = mapped_column("account_login", String)
+    is_archived: Mapped[int | None] = mapped_column("is_archived")
+    last_addressed_handle: Mapped[str | None] = mapped_column(
+        "last_addressed_handle", String
+    )
+    display_name: Mapped[str | None] = mapped_column("display_name", String)
+    group_id: Mapped[str | None] = mapped_column("group_id", String)
+    is_filtered: Mapped[int | None] = mapped_column("is_filtered")
+    successful_query: Mapped[int | None] = mapped_column("successful_query")
+    engram_id: Mapped[str | None] = mapped_column("engram_id", String)
+    server_change_token: Mapped[str | None] = mapped_column(
+        "server_change_token", String
+    )
+    ck_sync_state: Mapped[int | None] = mapped_column("ck_sync_state")
+    original_group_id: Mapped[str | None] = mapped_column("original_group_id", String)
+    last_read_message_timestamp: Mapped[datetime | None] = mapped_column(
+        "last_read_message_timestamp",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    sr_server_change_token: Mapped[str | None] = mapped_column(
+        "sr_server_change_token", String
+    )
+    sr_ck_sync_state: Mapped[int | None] = mapped_column("sr_ck_sync_state")
+    cloudkit_record_id: Mapped[str | None] = mapped_column("cloudkit_record_id", String)
+    sr_cloudkit_record_id: Mapped[str | None] = mapped_column(
+        "sr_cloudkit_record_id", String
+    )
+    last_addressed_sim_id: Mapped[str | None] = mapped_column(
+        "last_addressed_sim_id", String
+    )
+    is_blackholed: Mapped[int | None] = mapped_column("is_blackholed")
+    syndication_date: Mapped[datetime | None] = mapped_column(
+        "syndication_date",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    syndication_type: Mapped[int | None] = mapped_column("syndication_type")
+    is_recovered: Mapped[int | None] = mapped_column("is_recovered")
+    is_deleting_incoming_messages: Mapped[int | None] = mapped_column(
+        "is_deleting_incoming_messages"
+    )
+
+
+class IMessageAttachment(Base):
+    """Row from attachment in chat.db."""
+
+    __tablename__ = "attachment"
+
+    rowid: Mapped[int] = mapped_column("ROWID", primary_key=True)
+    guid: Mapped[str] = mapped_column("guid", String)
+    created_date: Mapped[datetime | None] = mapped_column(
+        "created_date",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    start_date: Mapped[datetime | None] = mapped_column(
+        "start_date",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    filename: Mapped[str | None] = mapped_column("filename", String)
+    uti: Mapped[str | None] = mapped_column("uti", String)
+    mime_type: Mapped[str | None] = mapped_column("mime_type", String)
+    transfer_state: Mapped[int | None] = mapped_column("transfer_state")
+    is_outgoing: Mapped[int | None] = mapped_column("is_outgoing")
+    user_info: Mapped[bytes | None] = mapped_column("user_info", LargeBinary)
+    transfer_name: Mapped[str | None] = mapped_column("transfer_name", String)
+    total_bytes: Mapped[int | None] = mapped_column("total_bytes")
+    is_sticker: Mapped[int | None] = mapped_column("is_sticker")
+    sticker_user_info: Mapped[bytes | None] = mapped_column(
+        "sticker_user_info", LargeBinary
+    )
+    attribution_info: Mapped[bytes | None] = mapped_column(
+        "attribution_info", LargeBinary
+    )
+    hide_attachment: Mapped[int | None] = mapped_column("hide_attachment")
+    ck_sync_state: Mapped[int | None] = mapped_column("ck_sync_state")
+    ck_server_change_token_blob: Mapped[bytes | None] = mapped_column(
+        "ck_server_change_token_blob", LargeBinary
+    )
+    ck_record_id: Mapped[str | None] = mapped_column("ck_record_id", String)
+    original_guid: Mapped[str] = mapped_column("original_guid", String)
+    sr_ck_sync_state: Mapped[int | None] = mapped_column("sr_ck_sync_state")
+    sr_ck_server_change_token_blob: Mapped[bytes | None] = mapped_column(
+        "sr_ck_server_change_token_blob", LargeBinary
+    )
+    sr_ck_record_id: Mapped[str | None] = mapped_column("sr_ck_record_id", String)
+    is_commsafety_sensitive: Mapped[int | None] = mapped_column(
+        "is_commsafety_sensitive"
+    )
+    emoji_image_content_identifier: Mapped[str | None] = mapped_column(
+        "emoji_image_content_identifier", String
+    )
+    emoji_image_short_description: Mapped[str | None] = mapped_column(
+        "emoji_image_short_description", String
+    )
+    preview_generation_state: Mapped[int | None] = mapped_column(
+        "preview_generation_state"
+    )
+
+
+class IMessageMessage(Base):
+    """Row from message in chat.db."""
+
+    __tablename__ = "message"
+
+    rowid: Mapped[int] = mapped_column("ROWID", primary_key=True)
+    guid: Mapped[str] = mapped_column("guid", String)
+    text: Mapped[str | None] = mapped_column("text", String)
+    replace: Mapped[int | None] = mapped_column("replace")
+    service_center: Mapped[str | None] = mapped_column("service_center", String)
+    handle_id: Mapped[int | None] = mapped_column("handle_id")
+    subject: Mapped[str | None] = mapped_column("subject", String)
+    country: Mapped[str | None] = mapped_column("country", String)
+    attributed_body: Mapped[bytes | None] = mapped_column("attributedBody", LargeBinary)
+    version: Mapped[int | None] = mapped_column("version")
+    type: Mapped[int | None] = mapped_column("type")
+    service: Mapped[str | None] = mapped_column("service", String)
+    account: Mapped[str | None] = mapped_column("account", String)
+    account_guid: Mapped[str | None] = mapped_column("account_guid", String)
+    error: Mapped[int | None] = mapped_column("error")
+    date: Mapped[datetime | None] = mapped_column(
+        "date",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    date_read: Mapped[datetime | None] = mapped_column(
+        "date_read",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    date_delivered: Mapped[datetime | None] = mapped_column(
+        "date_delivered",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    is_delivered: Mapped[int | None] = mapped_column("is_delivered")
+    is_finished: Mapped[int | None] = mapped_column("is_finished")
+    is_emote: Mapped[int | None] = mapped_column("is_emote")
+    is_from_me: Mapped[int | None] = mapped_column("is_from_me")
+    is_empty: Mapped[int | None] = mapped_column("is_empty")
+    is_delayed: Mapped[int | None] = mapped_column("is_delayed")
+    is_auto_reply: Mapped[int | None] = mapped_column("is_auto_reply")
+    is_prepared: Mapped[int | None] = mapped_column("is_prepared")
+    is_read: Mapped[int | None] = mapped_column("is_read")
+    is_system_message: Mapped[int | None] = mapped_column("is_system_message")
+    is_sent: Mapped[int | None] = mapped_column("is_sent")
+    has_dd_results: Mapped[int | None] = mapped_column("has_dd_results")
+    is_service_message: Mapped[int | None] = mapped_column("is_service_message")
+    is_forward: Mapped[int | None] = mapped_column("is_forward")
+    was_downgraded: Mapped[int | None] = mapped_column("was_downgraded")
+    is_archive: Mapped[int | None] = mapped_column("is_archive")
+    cache_has_attachments: Mapped[int | None] = mapped_column("cache_has_attachments")
+    cache_roomnames: Mapped[str | None] = mapped_column("cache_roomnames", String)
+    was_data_detected: Mapped[int | None] = mapped_column("was_data_detected")
+    was_deduplicated: Mapped[int | None] = mapped_column("was_deduplicated")
+    is_audio_message: Mapped[int | None] = mapped_column("is_audio_message")
+    is_played: Mapped[int | None] = mapped_column("is_played")
+    date_played: Mapped[datetime | None] = mapped_column(
+        "date_played",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    item_type: Mapped[int | None] = mapped_column("item_type")
+    other_handle: Mapped[int | None] = mapped_column("other_handle")
+    group_title: Mapped[str | None] = mapped_column("group_title", String)
+    group_action_type: Mapped[int | None] = mapped_column("group_action_type")
+    share_status: Mapped[int | None] = mapped_column("share_status")
+    share_direction: Mapped[int | None] = mapped_column("share_direction")
+    is_expirable: Mapped[int | None] = mapped_column("is_expirable")
+    expire_state: Mapped[int | None] = mapped_column("expire_state")
+    message_action_type: Mapped[int | None] = mapped_column("message_action_type")
+    message_source: Mapped[int | None] = mapped_column("message_source")
+    associated_message_guid: Mapped[str | None] = mapped_column(
+        "associated_message_guid", String
+    )
+    associated_message_type: Mapped[int | None] = mapped_column(
+        "associated_message_type"
+    )
+    balloon_bundle_id: Mapped[str | None] = mapped_column("balloon_bundle_id", String)
+    payload_data: Mapped[bytes | None] = mapped_column("payload_data", LargeBinary)
+    expressive_send_style_id: Mapped[str | None] = mapped_column(
+        "expressive_send_style_id", String
+    )
+    associated_message_range_location: Mapped[int | None] = mapped_column(
+        "associated_message_range_location"
+    )
+    associated_message_range_length: Mapped[int | None] = mapped_column(
+        "associated_message_range_length"
+    )
+    time_expressive_send_played: Mapped[datetime | None] = mapped_column(
+        "time_expressive_send_played",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    message_summary_info: Mapped[bytes | None] = mapped_column(
+        "message_summary_info", LargeBinary
+    )
+    ck_sync_state: Mapped[int | None] = mapped_column("ck_sync_state")
+    ck_record_id: Mapped[str | None] = mapped_column("ck_record_id", String)
+    ck_record_change_tag: Mapped[str | None] = mapped_column(
+        "ck_record_change_tag", String
+    )
+    destination_caller_id: Mapped[str | None] = mapped_column(
+        "destination_caller_id", String
+    )
+    sr_ck_sync_state: Mapped[int | None] = mapped_column("sr_ck_sync_state")
+    sr_ck_record_id: Mapped[str | None] = mapped_column("sr_ck_record_id", String)
+    sr_ck_record_change_tag: Mapped[str | None] = mapped_column(
+        "sr_ck_record_change_tag", String
+    )
+    is_corrupt: Mapped[int | None] = mapped_column("is_corrupt")
+    reply_to_guid: Mapped[str | None] = mapped_column("reply_to_guid", String)
+    sort_id: Mapped[int | None] = mapped_column("sort_id")
+    is_spam: Mapped[int | None] = mapped_column("is_spam")
+    has_unseen_mention: Mapped[int | None] = mapped_column("has_unseen_mention")
+    thread_originator_guid: Mapped[str | None] = mapped_column(
+        "thread_originator_guid", String
+    )
+    thread_originator_part: Mapped[str | None] = mapped_column(
+        "thread_originator_part", String
+    )
+    syndication_ranges: Mapped[str | None] = mapped_column("syndication_ranges", String)
+    was_delivered_quietly: Mapped[int | None] = mapped_column("was_delivered_quietly")
+    did_notify_recipient: Mapped[int | None] = mapped_column("did_notify_recipient")
+    synced_syndication_ranges: Mapped[str | None] = mapped_column(
+        "synced_syndication_ranges", String
+    )
+    date_retracted: Mapped[datetime | None] = mapped_column(
+        "date_retracted",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    date_edited: Mapped[datetime | None] = mapped_column(
+        "date_edited",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+    was_detonated: Mapped[int | None] = mapped_column("was_detonated")
+    part_count: Mapped[int | None] = mapped_column("part_count")
+    is_stewie: Mapped[int | None] = mapped_column("is_stewie")
+    is_kt_verified: Mapped[int | None] = mapped_column("is_kt_verified")
+    is_sos: Mapped[int | None] = mapped_column("is_sos")
+    is_critical: Mapped[int | None] = mapped_column("is_critical")
+    bia_reference_id: Mapped[str | None] = mapped_column("bia_reference_id", String)
+    fallback_hash: Mapped[str | None] = mapped_column("fallback_hash", String)
+    associated_message_emoji: Mapped[str | None] = mapped_column(
+        "associated_message_emoji", String
+    )
+    is_pending_satellite_send: Mapped[int | None] = mapped_column(
+        "is_pending_satellite_send"
+    )
+    needs_relay: Mapped[int | None] = mapped_column("needs_relay")
+    schedule_type: Mapped[int | None] = mapped_column("schedule_type")
+    schedule_state: Mapped[int | None] = mapped_column("schedule_state")
+    sent_or_received_off_grid: Mapped[int | None] = mapped_column(
+        "sent_or_received_off_grid"
+    )
+    date_recovered: Mapped[datetime | None] = mapped_column(
+        "date_recovered",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+
+
+class IMessageChatHandleJoin(Base):
+    """Row from chat_handle_join in chat.db."""
+
+    __tablename__ = "chat_handle_join"
+
+    chat_id: Mapped[int] = mapped_column("chat_id", primary_key=True)
+    handle_id: Mapped[int] = mapped_column("handle_id", primary_key=True)
+
+
+class IMessageChatMessageJoin(Base):
+    """Row from chat_message_join in chat.db."""
+
+    __tablename__ = "chat_message_join"
+
+    chat_id: Mapped[int] = mapped_column("chat_id", primary_key=True)
+    message_id: Mapped[int] = mapped_column("message_id", primary_key=True)
+    message_date: Mapped[datetime | None] = mapped_column(
+        "message_date",
+        AppleMessageTimestamp(null_sentinels=(0,)),
+    )
+
+
+class IMessageMessageAttachmentJoin(Base):
+    """Row from message_attachment_join in chat.db."""
+
+    __tablename__ = "message_attachment_join"
+
+    message_id: Mapped[int] = mapped_column("message_id", primary_key=True)
+    attachment_id: Mapped[int] = mapped_column("attachment_id", primary_key=True)