Sneed-Group/pypush-plus-plus
2
0
Fork 0
mirror of https://github.com/Sneed-Group/pypush-plus-plus synced 2025-01-09 17:33:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

implement more SMS handling stuff

Browse source
This commit is contained in:
JJTech0130 2023-08-14 20:31:03 -04:00
parent 10a5771791
commit 21fdfc97a9
No known key found for this signature in database
GPG key ID: 23C92EBCCF8F93D6
1 changed files with 24 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
imessage.py
View file

@ -140,6 +140,7 @@ class Message:
id: uuid.UUID
_raw: dict
_compressed: bool = True
xml: str | None = None
def from_raw(message: bytes, sender: str | None = None) -> "Message":
"""Create a `Message` from raw message bytes"""
@ -163,6 +164,8 @@ class SMSReflectedMessage(Message):
message = plistlib.loads(message)
logger.debug(f"Decoding SMSReflectedMessage: {message}")
return SMSReflectedMessage(
text=message["mD"]["plain-body"],
sender=sender,
@ -189,7 +192,7 @@ class SMSIncomingMessage(Message):
message = plistlib.loads(message)
logger.debug(f"Decompressed message : {message}")
logger.debug(f"Decoding SMSIncomingMessage: {message}")
return SMSIncomingMessage(
text=message["k"][0]["data"].decode(),
@ -202,6 +205,14 @@ class SMSIncomingMessage(Message):
def __str__(self):
return f"[SMS {self.sender}] '{self.text}'"
@dataclass
class SMSIncomingImage(Message):
def from_raw(message: bytes, sender: str | None = None) -> "SMSIncomingImage":
"""Create a `SMSIncomingImage` from raw message bytes"""
# TODO: Implement this
return "SMSIncomingImage"
@dataclass
class iMessage(Message):
@ -219,11 +230,14 @@ class iMessage(Message):
message = plistlib.loads(message)
logger.debug(f"Decoding iMessage: {message}")
return iMessage(
text=message["t"],
participants=message["p"],
sender=sender,
id=uuid.UUID(message["r"]),
xml=message["x"] if "x" in message else None,
_raw=message,
_compressed=compressed,
effect=message["iid"] if "iid" in message else None,
@ -533,10 +547,18 @@ class iMessageUser:
# Check for SMS messages
body = self._receive_raw(143, "com.apple.private.alloy.sms")
t = SMSReflectedMessage
if body is None:
# Check for SMS reflected images
body = self._receive_raw(144, "com.apple.private.alloy.sms")
t = SMSReflectedMessage
if body is None:
# Check for SMS incoming messages
body = self._receive_raw(140, "com.apple.private.alloy.sms")
t = SMSIncomingMessage
if body is None:
# Incoming images
body = self._receive_raw(141, "com.apple.private.alloy.sms")
t = SMSIncomingImage
if body is None:
return None
@ -677,7 +699,7 @@ class iMessageUser:
return False
resp_body = plistlib.loads(resp_body)
#logger.debug(f"See type {resp_body['c']}")
#logger.info(f"See type {resp_body['c']}")
if isinstance(c, list):
if not resp_body["c"] in c: