mautrix-instagram/mauigpapi/state/state.py

# mautrix-instagram - A Matrix-Instagram puppeting bridge.
# Copyright (C) 2021 Tulir Asokan
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
from typing import Optional
from uuid import uuid4
import random
import time

from attr import dataclass

from mautrix.types import SerializableAttrs, field

from ..errors import IGNoChallengeError, IGUserIDNotFoundError
from ..types import ChallengeContext, ChallengeStateResponse
from .application import AndroidApplication
from .cookies import Cookies
from .device import AndroidDevice
from .session import AndroidSession


@dataclass
class AndroidState(SerializableAttrs):
    device: AndroidDevice = field(factory=lambda: AndroidDevice())
    session: AndroidSession = field(factory=lambda: AndroidSession())
    application: AndroidApplication = field(factory=lambda: AndroidApplication())
    challenge: Optional[ChallengeStateResponse] = None
    challenge_context: Optional[ChallengeContext] = None
    _challenge_path: Optional[str] = field(default=None, json="challenge_path")
    cookies: Cookies = field(factory=lambda: Cookies())
    login_2fa_username: Optional[str] = field(default=None, hidden=True)
    _pigeon_session_id: Optional[str] = field(default=None, hidden=True)

    def __attrs_post_init__(self) -> None:
        if self.application.APP_VERSION_CODE != AndroidApplication().APP_VERSION_CODE:
            self.application = AndroidApplication()

    @property
    def pigeon_session_id(self) -> str:
        if not self._pigeon_session_id:
            self._pigeon_session_id = str(uuid4())
        return self._pigeon_session_id

    def reset_pigeon_session_id(self) -> None:
        self._pigeon_session_id = None

    @property
    def user_agent(self) -> str:
        return (
            f"Instagram {self.application.APP_VERSION} Android ({self.device.descriptor}; "
            f"{self.device.language}; {self.application.APP_VERSION_CODE})"
        )

    @property
    def user_id(self) -> str:
        if self.session.ds_user_id:
            return self.session.ds_user_id
        elif self.challenge and self.challenge.user_id:
            return str(self.challenge.user_id)
        else:
            raise IGUserIDNotFoundError()

    @property
    def challenge_path(self) -> str:
        if not self._challenge_path:
            raise IGNoChallengeError()
        return self._challenge_path

    @challenge_path.setter
    def challenge_path(self, val: str) -> None:
        self._challenge_path = val

    @staticmethod
    def gen_client_context() -> str:
        return str((int(time.time() * 1000) << 22) + random.randint(10000, 5000000))