mirror of
https://github.com/JJTech0130/pypush.git
synced 2025-01-22 19:30:56 +00:00
f80acd2e09
- extract idsuser authentication functionality to separate function so that it can be reused - add more typing hints to make lsp happier - access dictionary values more safely with walrus operator - simplify some list comprehension and iteration - print proxy errors in more detail so they're easier to debug - store apnsconnection in proxy so that we can use it to make a user and decrypt payloads if needed
120 lines
3.5 KiB
Python
120 lines
3.5 KiB
Python
import json
|
|
import logging
|
|
from base64 import b64decode, b64encode
|
|
from subprocess import PIPE, Popen
|
|
|
|
from rich.logging import RichHandler
|
|
|
|
import apns
|
|
import ids
|
|
import imessage
|
|
|
|
import trio
|
|
|
|
logging.basicConfig(
|
|
level=logging.NOTSET, format="%(message)s", datefmt="[%X]", handlers=[RichHandler()]
|
|
)
|
|
|
|
# Set sane log levels
|
|
logging.getLogger("urllib3").setLevel(logging.WARNING)
|
|
logging.getLogger("py.warnings").setLevel(logging.ERROR) # Ignore warnings from urllib3
|
|
logging.getLogger("asyncio").setLevel(logging.WARNING)
|
|
logging.getLogger("jelly").setLevel(logging.INFO)
|
|
logging.getLogger("nac").setLevel(logging.INFO)
|
|
logging.getLogger("apns").setLevel(logging.INFO)
|
|
logging.getLogger("albert").setLevel(logging.INFO)
|
|
logging.getLogger("ids").setLevel(logging.DEBUG)
|
|
logging.getLogger("bags").setLevel(logging.INFO)
|
|
logging.getLogger("imessage").setLevel(logging.INFO)
|
|
|
|
logging.captureWarnings(True)
|
|
|
|
process = Popen(["git", "rev-parse", "HEAD"], stdout=PIPE) # type: ignore
|
|
(commit_hash, err) = process.communicate()
|
|
exit_code = process.wait()
|
|
commit_hash = commit_hash.decode().strip()
|
|
|
|
# Try and load config.json
|
|
try:
|
|
with open("config.json", "r") as f:
|
|
CONFIG = json.load(f)
|
|
except FileNotFoundError:
|
|
CONFIG = {}
|
|
|
|
# Re-register if the commit hash has changed
|
|
if CONFIG.get("commit_hash") != commit_hash:
|
|
logging.warning("pypush commit is different, forcing re-registration...")
|
|
CONFIG["commit_hash"] = commit_hash
|
|
if "id" in CONFIG:
|
|
del CONFIG["id"]
|
|
|
|
|
|
def safe_b64decode(s):
|
|
try:
|
|
return b64decode(s)
|
|
except:
|
|
return None
|
|
|
|
async def main():
|
|
token = CONFIG.get("push", {}).get("token")
|
|
if token is not None:
|
|
token = b64decode(token)
|
|
else:
|
|
token = b""
|
|
|
|
push_creds = apns.PushCredentials(
|
|
CONFIG.get("push", {}).get("key", ""), CONFIG.get("push", {}).get("cert", ""), token)
|
|
|
|
async with apns.APNSConnection.start(push_creds) as conn:
|
|
await conn.set_state(1)
|
|
await conn.filter(["com.apple.madrid"])
|
|
|
|
user = ids.IDSUser(conn)
|
|
user.auth_and_set_encryption_from_config(CONFIG)
|
|
|
|
# Write config.json
|
|
CONFIG["encryption"] = {
|
|
"rsa_key": user.encryption_identity.encryption_key,
|
|
"ec_key": user.encryption_identity.signing_key,
|
|
}
|
|
CONFIG["id"] = {
|
|
"key": user._id_keypair.key,
|
|
"cert": user._id_keypair.cert,
|
|
}
|
|
CONFIG["auth"] = {
|
|
"key": user._auth_keypair.key,
|
|
"cert": user._auth_keypair.cert,
|
|
"user_id": user.user_id,
|
|
"handles": user.handles,
|
|
}
|
|
CONFIG["push"] = {
|
|
"token": b64encode(user.push_connection.credentials.token).decode(),
|
|
"key": user.push_connection.credentials.private_key,
|
|
"cert": user.push_connection.credentials.cert,
|
|
}
|
|
|
|
with open("config.json", "w") as f:
|
|
json.dump(CONFIG, f, indent=4)
|
|
|
|
im = imessage.iMessageUser(conn, user)
|
|
|
|
# Send a message to myself
|
|
async with trio.open_nursery() as nursery:
|
|
nursery.start_soon(input_task, im)
|
|
nursery.start_soon(output_task, im)
|
|
|
|
async def input_task(im: imessage.iMessageUser):
|
|
while True:
|
|
cmd = await trio.to_thread.run_sync(input, "> ", cancellable=True)
|
|
if cmd != "":
|
|
await im.send(imessage.iMessage.create(im, cmd, [im.user.current_handle]))
|
|
|
|
async def output_task(im: imessage.iMessageUser):
|
|
while True:
|
|
msg = await im.receive()
|
|
print(str(msg))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
trio.run(main)
|