Add script to generate session data

mauigpapi/scripts/iglogin.py

@@ -0,0 +1,81 @@
+import asyncio
+import base64
+import datetime
+import getpass
+import logging
+import zlib
+
+from mauigpapi import AndroidAPI, AndroidState
+from mauigpapi.errors import IGChallengeError, IGLoginTwoFactorRequiredError
+from mautrix.util.logging import TraceLogger
+
+
+async def main():
+    logging.basicConfig(level=logging.DEBUG)
+    username = password = twofactor = None
+    while not username:
+        username = input("Username: ").strip()
+    state = AndroidState()
+    state.device.generate(username + str(datetime.date.today()))
+    api_log: TraceLogger = logging.getLogger("api")
+    api = AndroidAPI(state, log=api_log)
+    try:
+        print("Getting mobile config...")
+        await api.get_mobile_config()
+        while not password:
+            password = getpass.getpass("Password: ")
+        try:
+            try:
+                try:
+                    print("Logging in...")
+                    await api.login(username, password)
+                except IGLoginTwoFactorRequiredError as e:
+                    print(e)
+                    print("Enter `r` to re-request SMS")
+                    inf = e.body.two_factor_info
+                    while not twofactor:
+                        twofactor = input("2FA code: ").lower().strip()
+                        if twofactor == "r":
+                            if inf.sms_two_factor_on:
+                                print("Re-requesting SMS code...")
+                                await api.send_two_factor_login_sms(
+                                    username, identifier=inf.two_factor_identifier
+                                )
+                                print("SMS code re-requested")
+                                inf.totp_two_factor_on = False
+                            else:
+                                print("You don't have SMS 2FA on 🤔")
+                            twofactor = None
+                    print("Sending 2FA code...")
+                    await api.two_factor_login(
+                        username,
+                        code=twofactor,
+                        identifier=inf.two_factor_identifier,
+                        is_totp=inf.totp_two_factor_on,
+                    )
+                print("Fetching current user...")
+                user = await api.current_user()
+            except IGChallengeError as e:
+                print(e)
+                print("Resetting challenge...")
+                await api.challenge_auto(reset=True)
+                print("Fetching current user...")
+                user = await api.current_user()
+        except Exception as e:
+            print("💥", e)
+            return
+        if not user or not user.user:
+            print("Login failed?")
+            return
+        print(f"Logged in as @{user.user.username}")
+        print()
+        print(
+            base64.b64encode(zlib.compress(state.json().encode("utf-8"), level=9)).decode("utf-8")
+        )
+        print()
+    finally:
+        await api.http.close()
+
+
+if __name__ == "__main__":
+    asyncio.run(main())