slidge-whatsapp 0.2.0a0__cp311-cp311-manylinux_2_36_x86_64.whl

slidge_whatsapp/go.mod

@@ -0,0 +1,28 @@
+module git.sr.ht/~nicoco/slidge-whatsapp/slidge_whatsapp
+
+go 1.21
+
+toolchain go1.21.5
+
+require (
+	github.com/go-python/gopy v0.4.10
+	github.com/h2non/filetype v1.1.3
+	github.com/mattn/go-sqlite3 v1.14.22
+	go.mau.fi/libsignal v0.1.1-0.20240705162345-47e713a595ab
+	go.mau.fi/whatsmeow v0.0.0-20240710112833-d732338c041f
+	golang.org/x/image v0.15.0
+)
+
+require (
+	filippo.io/edwards25519 v1.1.0 // indirect
+	github.com/google/uuid v1.6.0 // indirect
+	github.com/gorilla/websocket v1.5.1 // indirect
+	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/rs/zerolog v1.33.0 // indirect
+	go.mau.fi/util v0.5.0 // indirect
+	golang.org/x/crypto v0.25.0 // indirect
+	golang.org/x/net v0.27.0 // indirect
+	golang.org/x/sys v0.22.0 // indirect
+	google.golang.org/protobuf v1.34.2 // indirect
+)

slidge_whatsapp/go.sum

@@ -0,0 +1,55 @@
+filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
+filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
+github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/go-python/gopy v0.4.10 h1:Ec3x+NTSzLsw9f6FTdDLwQCQlmlNmJIu4J6nSnyugqE=
+github.com/go-python/gopy v0.4.10/go.mod h1:zMV/gSSYa9u/8Zp0WYR+L/z+kOIqIUtMg/a1/GRy5uw=
+github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
+github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU=
+github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
+github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
+github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
+github.com/gorilla/websocket v1.5.1 h1:gmztn0JnHVt9JZquRuzLw3g4wouNVzKL15iLr/zn/QY=
+github.com/gorilla/websocket v1.5.1/go.mod h1:x3kM2JMyaluk02fnUJpQuwD2dCS5NDG2ZHL0uE0tcaY=
+github.com/h2non/filetype v1.1.3 h1:FKkx9QbD7HR/zjK1Ia5XiBsq9zdLi5Kf3zGyFTAFkGg=
+github.com/h2non/filetype v1.1.3/go.mod h1:319b3zT68BvV+WRj7cwy856M2ehB3HqNOt6sy1HndBY=
+github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
+github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
+github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
+github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
+github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
+github.com/mattn/go-sqlite3 v1.14.22 h1:2gZY6PC6kBnID23Tichd1K+Z0oS6nE/XwU+Vz/5o4kU=
+github.com/mattn/go-sqlite3 v1.14.22/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
+github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/rs/xid v1.5.0/go.mod h1:trrq9SKmegXys3aeAKXMUTdJsYXVwGY3RLcfgqegfbg=
+github.com/rs/zerolog v1.33.0 h1:1cU2KZkvPxNyfgEmhHAz/1A9Bz+llsdYzklWFzgp0r8=
+github.com/rs/zerolog v1.33.0/go.mod h1:/7mN4D5sKwJLZQ2b/znpjC3/GQWY/xaDXUM0kKWRHss=
+github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
+github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+go.mau.fi/libsignal v0.1.1-0.20240705162345-47e713a595ab h1:/tnRxsaaG/xBGjXTb6tzTr+XY4T5fQlrGHE1p9ir/wM=
+go.mau.fi/libsignal v0.1.1-0.20240705162345-47e713a595ab/go.mod h1:R8ovrTezxtUNzCQE5PH30StOQWWeBskBsWE55vMfY9I=
+go.mau.fi/util v0.5.0 h1:8yELAl+1CDRrwGe9NUmREgVclSs26Z68pTWePHVxuDo=
+go.mau.fi/util v0.5.0/go.mod h1:DsJzUrJAG53lCZnnYvq9/mOyLuPScWwYhvETiTrpdP4=
+go.mau.fi/whatsmeow v0.0.0-20240710112833-d732338c041f h1:ni8K5zVngwOWrrZ1HzwEmvAovj0+p0cq464l9g0/dt0=
+go.mau.fi/whatsmeow v0.0.0-20240710112833-d732338c041f/go.mod h1:lMW+LxRTakgyNasZwYNB+2uqjKox75GcEfeUXSJhe8I=
+golang.org/x/crypto v0.25.0 h1:ypSNr+bnYL2YhwoMt2zPxHFmbAN1KZs/njMG3hxUp30=
+golang.org/x/crypto v0.25.0/go.mod h1:T+wALwcMOSE0kXgUAnPAHqTLW+XHgcELELW8VaDgm/M=
+golang.org/x/image v0.15.0 h1:kOELfmgrmJlw4Cdb7g/QGuB3CvDrXbqEIww/pNtNBm8=
+golang.org/x/image v0.15.0/go.mod h1:HUYqC05R2ZcZ3ejNQsIHQDQiwWM4JBqmm6MKANTp4LE=
+golang.org/x/net v0.27.0 h1:5K3Njcw06/l2y9vpGCSdcxWOYHOUk3dVNGDXN+FvAys=
+golang.org/x/net v0.27.0/go.mod h1:dDi0PyhWNoiUOrAS8uXv/vnScO4wnHQO4mj9fn/RytE=
+golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.22.0 h1:RI27ohtqKCnwULzJLqkv897zojh5/DwS/ENaMzUOaWI=
+golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+google.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg=
+google.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
+gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

slidge_whatsapp/group.py

@@ -0,0 +1,240 @@
+import re
+from datetime import datetime, timezone
+from typing import TYPE_CHECKING, Optional
+
+from slidge.group import LegacyBookmarks, LegacyMUC, LegacyParticipant, MucType
+from slidge.util.archive_msg import HistoryMessage
+from slidge.util.types import Hat, HoleBound, Mention, MucAffiliation
+from slixmpp.exceptions import XMPPError
+
+from .generated import whatsapp
+from .util import get_bytes_temp
+
+if TYPE_CHECKING:
+    from .contact import Contact
+    from .session import Session
+
+
+class Participant(LegacyParticipant):
+    contact: "Contact"
+    muc: "MUC"
+
+
+class MUC(LegacyMUC[str, str, Participant, str]):
+    session: "Session"
+    type = MucType.GROUP
+
+    REACTIONS_SINGLE_EMOJI = True
+    _ALL_INFO_FILLED_ON_STARTUP = True
+
+    HAS_DESCRIPTION = False
+
+    async def update_info(self):
+        try:
+            avatar = self.session.whatsapp.GetAvatar(self.legacy_id, self.avatar or "")
+        except RuntimeError:
+            # no avatar
+            await self.set_avatar(None)
+        else:
+            if avatar.URL:
+                await self.set_avatar(avatar.URL, avatar.ID)
+
+    async def backfill(
+        self,
+        after: HoleBound | None = None,
+        before: HoleBound | None = None,
+    ):
+        """
+        Request history for messages older than the oldest message given by ID and date.
+        """
+
+        if before is None:
+            return
+            # WhatsApp requires a full reference to the last seen message in performing on-demand sync.
+
+        assert isinstance(before.id, str)
+        oldest_message = whatsapp.Message(
+            ID=before.id,
+            IsCarbon=self.session.message_is_carbon(self, before.id),
+            Timestamp=int(before.timestamp.timestamp()),
+        )
+        self.session.whatsapp.RequestMessageHistory(self.legacy_id, oldest_message)
+
+    def get_message_sender(self, legacy_msg_id: str):
+        assert self.pk is not None
+        stored = self.xmpp.store.mam.get_by_legacy_id(self.pk, legacy_msg_id)
+        if stored is None:
+            raise XMPPError("internal-server-error", "Unable to find message sender")
+        msg = HistoryMessage(stored.stanza)
+        occupant_id = msg.stanza["occupant-id"]["id"]
+        if occupant_id == "slidge-user":
+            return self.session.contacts.user_legacy_id
+        if "@" in occupant_id:
+            jid_username = occupant_id.split("@")[0]
+            return jid_username.removeprefix("+") + "@" + whatsapp.DefaultUserServer
+        raise XMPPError("internal-server-error", "Unable to find message sender")
+
+    async def update_whatsapp_info(self, info: whatsapp.Group):
+        """
+        Set MUC information based on WhatsApp group information, which may or may not be partial in
+        case of updates to existing MUCs.
+        """
+        if info.Nickname:
+            self.user_nick = info.Nickname
+        if info.Name:
+            self.name = info.Name
+        if info.Subject.Subject:
+            self.subject = info.Subject.Subject
+            if info.Subject.SetAt:
+                set_at = datetime.fromtimestamp(info.Subject.SetAt, tz=timezone.utc)
+                self.subject_date = set_at
+            if info.Subject.SetByJID:
+                participant = await self.get_participant_by_legacy_id(
+                    info.Subject.SetByJID
+                )
+                if name := participant.nickname:
+                    self.subject_setter = name
+        for data in info.Participants:
+            participant = await self.get_participant_by_legacy_id(data.JID)
+            if data.Action == whatsapp.GroupParticipantActionRemove:
+                self.remove_participant(participant)
+            else:
+                if data.Affiliation == whatsapp.GroupAffiliationAdmin:
+                    # Only owners can change the group name according to
+                    # XEP-0045, so we make all "WA admins" "XMPP owners"
+                    participant.affiliation = "owner"
+                    participant.role = "moderator"
+                elif data.Affiliation == whatsapp.GroupAffiliationOwner:
+                    # The WA owner is in fact the person who created the room
+                    participant.set_hats(
+                        [Hat("https://slidge.im/hats/slidge-whatsapp/owner", "Owner")]
+                    )
+                    participant.affiliation = "owner"
+                    participant.role = "moderator"
+                else:
+                    participant.affiliation = "member"
+                    participant.role = "participant"
+
+    async def replace_mentions(self, t: str):
+        return replace_whatsapp_mentions(
+            t,
+            participants=(
+                {
+                    p.contact.jid_username: p.nickname
+                    async for p in self.get_participants()
+                    if p.contact is not None  # should not happen
+                }
+                | {self.session.user_phone: self.user_nick}
+                if self.session.user_phone  # user_phone *should* be set at this point,
+                else {}  # but better safe than sorry
+            ),
+        )
+
+    async def on_avatar(self, data: Optional[bytes], mime: Optional[str]) -> None:
+        return self.session.whatsapp.SetAvatar(
+            self.legacy_id, await get_bytes_temp(data) if data else ""
+        )
+
+    async def on_set_config(
+        self,
+        name: Optional[str],
+        description: Optional[str],
+    ):
+        # there are no group descriptions in WA, but topics=subjects
+        if self.name != name:
+            self.session.whatsapp.SetGroupName(self.legacy_id, name)
+
+    async def on_set_subject(self, subject: str):
+        if self.subject != subject:
+            self.session.whatsapp.SetGroupTopic(self.legacy_id, subject)
+
+    async def on_set_affiliation(
+        self,
+        contact: "Contact",  # type:ignore
+        affiliation: MucAffiliation,
+        reason: Optional[str],
+        nickname: Optional[str],
+    ):
+        assert contact.contact_pk is not None
+        assert self.pk is not None
+        if affiliation == "member":
+            if (
+                self.xmpp.store.participants.get_by_contact(self.pk, contact.contact_pk)
+                is not None
+            ):
+                change = "demote"
+            else:
+                change = "add"
+        elif affiliation == "admin":
+            change = "promote"
+        elif affiliation == "outcast" or affiliation == "none":
+            change = "remove"
+        else:
+            raise XMPPError(
+                "bad-request",
+                f"You can't make a participant '{affiliation}' in whatsapp",
+            )
+        self.session.whatsapp.SetAffiliation(self.legacy_id, contact.legacy_id, change)
+
+
+class Bookmarks(LegacyBookmarks[str, MUC]):
+    session: "Session"
+
+    def __init__(self, session: "Session"):
+        super().__init__(session)
+        self.__filled = False
+
+    async def fill(self):
+        groups = self.session.whatsapp.GetGroups()
+        for group in groups:
+            await self.add_whatsapp_group(group)
+        self.__filled = True
+
+    async def add_whatsapp_group(self, data: whatsapp.Group):
+        muc = await self.by_legacy_id(data.JID)
+        await muc.update_whatsapp_info(data)
+        await muc.add_to_bookmarks()
+
+    async def legacy_id_to_jid_local_part(self, legacy_id: str):
+        return "#" + legacy_id[: legacy_id.find("@")]
+
+    async def jid_local_part_to_legacy_id(self, local_part: str):
+        if not local_part.startswith("#"):
+            raise XMPPError("bad-request", "Invalid group ID, expected '#' prefix")
+
+        if not self.__filled:
+            raise XMPPError(
+                "recipient-unavailable", "Still fetching group info, please retry later"
+            )
+
+        whatsapp_group_id = (
+            local_part.removeprefix("#") + "@" + whatsapp.DefaultGroupServer
+        )
+
+        if (
+            self.xmpp.store.rooms.get_by_legacy_id(
+                self.session.user_pk, whatsapp_group_id
+            )
+            is None
+        ):
+            raise XMPPError("item-not-found", f"No group found for {whatsapp_group_id}")
+
+        return whatsapp_group_id
+
+
+def replace_xmpp_mentions(text: str, mentions: list[Mention]):
+    offset: int = 0
+    result: str = ""
+    for m in mentions:
+        legacy_id = "@" + m.contact.legacy_id[: m.contact.legacy_id.find("@")]
+        result = result + text[offset : m.start] + legacy_id
+        offset = m.end
+    return result + text[offset:] if offset > 0 else text
+
+
+def replace_whatsapp_mentions(text: str, participants: dict[str, str]):
+    def match(m: re.Match):
+        group = m.group(0)
+        return participants.get(group.replace("@", "+"), group)
+
+    return re.sub(r"@\d+", match, text)