intuned-runtime 1.0.0__py3-none-any.whl

intuned_runtime-1.0.0.dist-info/LICENSE

@@ -0,0 +1,42 @@
+Acceptance
+By using the software, you agree to all of the terms and conditions below.
+
+Copyright License
+The licensor grants you a non-exclusive, royalty-free, worldwide, non-sublicensable, non-transferable license to use, copy, distribute, make available, and prepare derivative works of the software, in each case subject to the limitations and conditions below.
+
+Limitations
+You may not provide the software to third parties as a hosted or managed service, where the service provides users with access to any substantial set of the features or functionality of the software.
+
+You may not move, change, disable, or circumvent the license key functionality in the software, and you may not remove or obscure any functionality in the software that is protected by the license key.
+
+You may not alter, remove, or obscure any licensing, copyright, or other notices of the licensor in the software. Any use of the licensor’s trademarks is subject to applicable law.
+
+Patents
+The licensor grants you a license, under any patent claims the licensor can license, or becomes able to license, to make, have made, use, sell, offer for sale, import and have imported the software, in each case subject to the limitations and conditions in this license. This license does not cover any patent claims that you cause to be infringed by modifications or additions to the software. If you or your company make any written claim that the software infringes or contributes to infringement of any patent, your patent license for the software granted under these terms ends immediately. If your company makes such a claim, your patent license ends immediately for work on behalf of your company.
+
+Notices
+You must ensure that anyone who gets a copy of any part of the software from you also gets a copy of these terms.
+
+If you modify the software, you must include in any modified copies of the software prominent notices stating that you have modified the software.
+
+No Other Rights
+These terms do not imply any licenses other than those expressly granted in these terms.
+
+Termination
+If you use the software in violation of these terms, such use is not licensed, and your licenses will automatically terminate. If the licensor provides you with a notice of your violation, and you cease all violation of this license no later than 30 days after you receive that notice, your licenses will be reinstated retroactively. However, if you violate these terms after such reinstatement, any additional violation of these terms will cause your licenses to terminate automatically and permanently.
+
+No Liability
+As far as the law allows, the software comes as is, without any warranty or condition, and the licensor will not be liable to you for any damages arising out of these terms or the use or nature of the software, under any kind of legal claim.
+
+Definitions
+The licensor is the entity offering these terms, and the software is the software the licensor makes available under these terms, including any portion of it.
+
+you refers to the individual or entity agreeing to these terms.
+
+your company is any legal entity, sole proprietorship, or other kind of organization that you work for, plus all organizations that have control over, are under the control of, or are under common control with that organization. control means ownership of substantially all the assets of an entity, or the power to direct its management and policies by vote, contract, or otherwise. Control can be direct or indirect.
+
+your licenses are all the licenses granted to you for the software under these terms.
+
+use means anything you do with the software requiring one of your licenses.
+
+trademark means trademarks, service marks, and similar rights.

intuned_runtime-1.0.0.dist-info/METADATA

@@ -0,0 +1,113 @@
+Metadata-Version: 2.3
+Name: intuned-runtime
+Version: 1.0.0
+Summary: Runtime commands for Intuned platform Python scrapers
+License: Elastic-2.0
+Keywords: runtime,intuned
+Author: Intuned Developers
+Author-email: engineering@intunedhq.com
+Requires-Python: >=3.9,<4.0
+Classifier: License :: Other/Proprietary License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: aiofiles (>=24.1.0,<25.0.0)
+Requires-Dist: arguably (>=1.3.0,<2.0.0)
+Requires-Dist: gitpython (>=3.1.43,<4.0.0)
+Requires-Dist: httpx (>=0.23.0,<1)
+Requires-Dist: more-termcolor (>=1.1.3,<2.0.0)
+Requires-Dist: pathspec (>=0.12.1,<0.13.0)
+Requires-Dist: pydantic (>=2.10.6,<3.0.0)
+Requires-Dist: pyright (>=1.1.387,<2.0.0)
+Requires-Dist: python-dotenv (==1.0.1)
+Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: ruff (>=0.7.2,<0.8.0)
+Requires-Dist: semver (>=3.0.4,<4.0.0)
+Requires-Dist: tenacity (>=8.5.0,<9.0.0)
+Requires-Dist: toml (>=0.10.2,<0.11.0)
+Requires-Dist: waitress (>=3.0.1,<4.0.0)
+Description-Content-Type: text/markdown
+
+# Intuned Python Runtime
+
+Runtime commands for Intuned platform Python automations.
+
+## Dependencies
+
+- Requires Python 3.12 or higher.
+- Install poetry: `pip install poetry`
+- Install dependencies: `poetry install`
+- Activate virtual environment: `poetry shell`
+- Now you have access to `intuned` cli from within project.
+
+## Install globally
+
+- This project can be installed globally on the system to use `intuned` cli anywhere
+- Make sure you are not in a virtual environment. `which python` should point to system python.
+  - If you are, `deactivate` to exit virtual environment. Or open in an external terminal (from outside vscode if it doesn't work)
+- Run `pip install -e .` from the root of the project.
+
+## Commands
+
+All commands have `-h` flag to show help.
+
+### `intuned init`
+
+- Initializes a project. Creates `pyproject.toml`, `Intuned.json` and `README.md` files.
+- Prompts for confirmation for each file that already exists.
+- Options:
+  - `--yes/-y` flag to overwrite all files.
+  - `--no/-n` flag to not overwrite any files.
+
+### `intuned publish-packages`
+
+- Publishes packages to `python-packages` repository.
+- Options:
+  - `--sdk` flag to publish SDK package. Creates `sdk-<version>` and `sdk-latest` tags for the published version.
+  - `--runtime` flag to publish runtime package. Creates `runtime-<version>` and `runtime-latest` tags for the published version.
+  - `--overwrite` flag to overwrite the existing version if it exists.
+  - `--show-diff` flag to show the diff of the package before publishing. You need to configure a diff tool to be used for `git difftool` command in your git config. [How to configure VS Code as a diff tool](https://www.roboleary.net/vscode/2020/09/15/vscode-git.html#tldr).
+  - `--no-latest` flag to not release `latest` tag for the published version.
+- Uses the version specified in `pyproject.toml` of each package respectively.
+- Uses WebApp directory specified in `WEBAPP_REPO` environment variable or tries to resolve it (only works if installed globally with `-e` flag).
+- Uses `python-packages` directory to be sister to WebApp directory `<webapp path>/../python-packages`.
+- These packages are used on deployed apps.
+
+### `intuned project run`
+
+- Runs the project.
+  - `--mode` to specify the mode to run. Default is `sample`.
+    - `--mode sample` extends a sample of payloads to run.
+    - `--mode full` runs all extended payloads.
+    - `--mode single` runs the initial API only.
+  - `--api-name <name>` to specify the initial API to run. Defaults to `default`
+  - `--params <params json>` to specify the parameters to the initial API.
+  - `--sample-config-str '{<api name>: <sample size>, ...}` to specify the sample config. Only used with `--mode sample`.
+  - `--no-headless` to disable headless mode.
+
+### `intuned project deploy`
+
+- Deploys a project and starts a default job.
+- Options:
+  - `--workspace-info '{"environment_url": <>, "workspace_id": <>, "api_key": <>}'` to specify the workspace info.
+  - `--workspace-info-path` to specify the path to a JSON file containing workspace info.
+  - `-y/--yes` to skip confirmation.
+  - `--project-name` to specify the project name. Resolves the name if not provided.
+- Resolves `.gitignore` from current/parent directories to decide what to deploy.
+- Resolves `.env` from current/parent directories to get environment variables to deploy.
+- Resolves project name from the current/parent directory name if not provided.
+
+### `intuned project serve`
+
+- Serves the project as an HTTP server.
+- Options:
+  - `--env development/production` to specify the environment to run the server.
+    - Development runs using Flask's development server.
+    - Production runs using Waitress.
+  - `--debug` to run the development server in debug mode. Not supported in production.
+- This is used on deployed apps.
+

intuned_runtime-1.0.0.dist-info/RECORD

@@ -0,0 +1,58 @@
+cli/__init__.py,sha256=QvTM82L3VLVOFGpu_bziWupgwjsAgv07u6o_gUwO3z0,1314
+cli/commands/__init__.py,sha256=QF-zy-iOZXY-9m-1O-JjWnugjulgJtsODtnkhvlNKq8,944
+cli/commands/ai_source/__init__.py,sha256=lg7owgcK8owNn2a4VBUP9RKxzFsLruhtnnQV0F_z6pc,149
+cli/commands/ai_source/ai_source.py,sha256=2woQtCmhxKvLfEz832eUoCT9gMsuSvEE6rMnHSYXC7w,138
+cli/commands/ai_source/deploy.py,sha256=XVN_alilwIaoVniuSpE8RkAveuaB3MaUtsRjIcpQTXo,2440
+cli/commands/browser/__init__.py,sha256=AuVbvh7aSBTFKYvewXZyPoIvfBTe2uHiPcnaAkzapas,95
+cli/commands/browser/save_state.py,sha256=eHKfvBfeFR_U9VQbsjOnIZjWepyDjNn9NL3naDYWP2s,841
+cli/commands/init.py,sha256=8rWBenWZfwNtLxOBqhEMbOATyQNEnmDUmrFJ1xBGyxI,4384
+cli/commands/project/__init__.py,sha256=t97wvhSenerYRdbSeCKXqHASA6EWA3lc1hnRhF9baOE,734
+cli/commands/project/auth_session/__init__.py,sha256=gt7mlaW6xmqAc_4-pfF_FiecsR51C6fqCaq_NFbcbwA,300
+cli/commands/project/auth_session/check.py,sha256=AFILp7m34nAO_RD3IfRpuJm5Zh0wnCRtBXqIrerdbwo,4565
+cli/commands/project/auth_session/create.py,sha256=r-eYu3uLUo2mzF836CbVgu4oBzcIIDekzzFwwekxmg0,3374
+cli/commands/project/auth_session/load.py,sha256=UUvg9Vyj15xiR44XlJzojLoFm5esv-o4E3qA3JqnBsE,1201
+cli/commands/project/project.py,sha256=_MSh6Xor2Cbh-ItifwgOPq_BP8UDuKB7S6w796FULKQ,137
+cli/commands/project/run.py,sha256=FDYYkU24aURYbljyYLFo8wLF-nvb86EVr9gMEjAfeE0,12274
+cli/commands/project/run_interface.py,sha256=4RyR8WZriIF7Va4z1wt-q6zZDQOI31n62Ho2dyimzUY,8717
+cli/commands/project/type_check.py,sha256=3BKMqTzJ73_J-RYrjba7Yv2mM_VzHR0gnXQEg_Us7JA,3384
+cli/commands/project/upgrade.py,sha256=XmkZLflM4O-mwvhwcswlZpazRotwi3xesLgE0Zz8fTI,3061
+cli/commands/publish_packages.py,sha256=sijkaG7_s0I1EWgLekGy1qm8Aqi_gYY8poXbMX0B6Yw,10505
+cli/logger.py,sha256=bZK3q-KUdGxk_qzDb6pn-n0LOhKJvi6a9p8oSwZtq3s,594
+cli/utils/ai_source_project.py,sha256=xUCM6p3i1XN4bJbuQz8LCzeI4BwqAdSvCl_vwDAEi0k,831
+cli/utils/code_tree.py,sha256=1wfxZoQ5kRCfqs2SEPAicbAIPTiD6P1LxSuwYu_eeaI,2790
+cli/utils/run_apis.py,sha256=Zee4zkgt9R8XY1XCGzj2Nc4zJ3jlRz1xnO493wotuWw,4690
+cli/utils/unix_socket.py,sha256=UISmkJMHrir5iBLUm6vxC3uzTGIFyOk_wa0C9LUw4Cc,1889
+runtime/__init__.py,sha256=wgS7k3f10LBbjSYJdjKeKUptizNw6OrBgNx-ECT5Ox8,124
+runtime/backend_functions/__init__.py,sha256=j2EaK4FK8bmdFtqc5FxtFwx1KhIn_7qKPChrrAhJI3s,119
+runtime/backend_functions/_call_backend_function.py,sha256=sER0op3vE9EVEB5-ZNwpGraxrqkFh5ZgOIV2B5fgmg4,3028
+runtime/backend_functions/get_auth_session_parameters.py,sha256=pOvB7XiWpphEuBpazdKALw9EWgBU1PeY3gkzBfVLpkc,869
+runtime/browser/__init__.py,sha256=9Nqf-80l1F7KGjy2pYBC5S5SeEXJ4YJ4Dl71hi_GWdI,132
+runtime/browser/launch_chromium.py,sha256=sML_XpjmlMf5QpuqMGeaS42D43pAXWKswm2q-3a2jAM,7459
+runtime/browser/storage_state.py,sha256=AmiHMfWbFVr7XUEYXoaWWZH4pXGdPv4wTNPXunm19rA,3591
+runtime/context/__init__.py,sha256=hg8ejm4bJy4tNkwmZ9lKgYJx6bU7OgOdBS684Uv5XGg,73
+runtime/context/context.py,sha256=pl_0x77_d5CiAznz1qGSk6o9cW-msNvlCt-2eFoMKlA,1739
+runtime/env.py,sha256=h4BJI-XVSZKgtTxjkzj2HsyN3DlY5Ml9GZqeH4CKDE8,238
+runtime/errors/__init__.py,sha256=oqiBSvT_yFLQ3hG0AbCUA3WYFaxkTDVkDMSy59xvBCo,688
+runtime/errors/auth_session_errors.py,sha256=6b4XTI8UCDHDPX4jEA8_HyrNUp4VZ1TrEA8DRh6Z3rM,228
+runtime/errors/run_api_errors.py,sha256=etqS18ZxDzr30qUezG6D8gcH0Yc1XNtQ8GMkUZx3Rmw,3400
+runtime/errors/trace_errors.py,sha256=Lzfo0sH3zGaWz1kn5DHcAXQMn3aR2y2bnauj6xP1LYE,110
+runtime/helpers/__init__.py,sha256=jozYPKHgZJ7Na5U1Wjt83egzjPATMZ_OMInEI6swSbY,234
+runtime/helpers/extend_payload.py,sha256=towZF08WTpTTDBL4AV1bUU3XpKAQHEB66kGUfTICDe0,246
+runtime/helpers/extend_timeout.py,sha256=KjfSLEUrqoz7v00rhnPAKq2OmUzEzcv-eQ3M8c2U46s,348
+runtime/helpers/get_auth_session_parameters.py,sha256=7bopGhJ7vjKAn_UxnHSAah-k2rVOPbq0zi9FQOOCFds,472
+runtime/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+runtime/run/__init__.py,sha256=zxMYVb7hn147YTrhMLsrcX6-KTd71HLrYHstJOWeWXQ,52
+runtime/run/intuned_settings.py,sha256=vy2-ktEzUfUp5Z90dp3l7jPKHNjgB-8GSMDgAY-rYaU,1074
+runtime/run/playwright_constructs.py,sha256=kOa95d4m4DTcIDKq_QajBs-9FkPz5yZFcsafLtx5288,519
+runtime/run/run_api.py,sha256=88u91YS5-m_aGwQnNfeN7Mszm1MRfOmciCpzuAXjdoI,8853
+runtime/run/traces.py,sha256=fKzh11LqV47ujgq_9I2tdp-dgld566wffWaHwU_4gis,1123
+runtime/types/__init__.py,sha256=IJkDfqsau8F8R_j8TO6j-JwW4ElQr6aU6LNaWRehg5U,401
+runtime/types/payload.py,sha256=sty8HgDEn3nJbZrwEOMCXyuG7_ICGDwlBIIWSON5ABY,124
+runtime/types/run_types.py,sha256=_ZoW0rv4VnjqVwVxJBWsdh0S5toppNL3AfL456rn5yM,4076
+runtime_helpers/__init__.py,sha256=XBrEiE9yNC8Lgn8NgIkqNXbI6e4ap237E83Zj_nlhCQ,249
+runtime_helpers/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+intuned_runtime-1.0.0.dist-info/LICENSE,sha256=9LIjQdgyU_ptzNIfItNCR7VmEHqYnrY1f1XwOreKFI0,3714
+intuned_runtime-1.0.0.dist-info/METADATA,sha256=i6-dbJpG78FttN_tw7HgA3qd_fE99cHW7bkASh2z16o,5134
+intuned_runtime-1.0.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+intuned_runtime-1.0.0.dist-info/entry_points.txt,sha256=4VXmTtVfaolAgt2owC9QF1bQpzCVv6Qh_cqW6ZGpftw,35
+intuned_runtime-1.0.0.dist-info/RECORD,,

intuned_runtime-1.0.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.1.3
+Root-Is-Purelib: true
+Tag: py3-none-any

intuned_runtime-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+intuned=cli:run
+

runtime/__init__.py

@@ -0,0 +1,3 @@
+from .browser import launch_chromium, dangerous_launch_chromium
+
+__all__ = ["launch_chromium", "dangerous_launch_chromium"]

runtime/backend_functions/__init__.py

@@ -0,0 +1,5 @@
+from .get_auth_session_parameters import get_auth_session_parameters
+
+__all__ = [
+    "get_auth_session_parameters",
+]

runtime/backend_functions/_call_backend_function.py

@@ -0,0 +1,86 @@
+import json
+from typing import Any
+from typing import Literal
+
+from httpx import AsyncClient
+from pydantic import BaseModel
+
+from runtime.context.context import IntunedContext
+from runtime.env import get_functions_domain
+from runtime.env import get_project_id
+from runtime.env import get_workspace_id
+
+
+async def call_backend_function[T: BaseModel](
+    name: str,
+    validation_model: type[T],
+    *,
+    method: Literal["GET", "POST"] = "GET",
+    params: BaseModel | None = None,
+) -> T:
+    """
+    Get the auth session parameters from the IntunedContext.
+    """
+    functions_domain, workspace_id, project_id = get_functions_domain(), get_workspace_id(), get_project_id()
+
+    if functions_domain is None or workspace_id is None or project_id is None:
+        # todo
+        raise Exception("No workspace ID or project ID found.")
+
+    context = IntunedContext.current()
+    if context.run_context is None:
+        # todo
+        raise Exception("No run context found.")
+
+    auth_session_id = context.run_context.auth_session_id
+    if auth_session_id is None:
+        # todo
+        raise Exception("No auth session ID found.")
+
+    async with AsyncClient() as client:
+        if context.functions_token:
+            client.headers["Authorization"] = f"Bearer {context.functions_token}"
+        if params:
+            client.headers["Content-Type"] = "application/json"
+        path = f"{functions_domain}/api/{workspace_id}/functions/{project_id}/{name}"
+        body = params.model_dump() if params else None
+        print("Calling backend function", method, path, context.functions_token, json.dumps(body, indent=2))
+        response = await client.request(
+            method,
+            path,
+            json=body,
+        )
+        try:
+            response_json = response.json()
+        except json.JSONDecodeError as e:
+            raise CallBackendException(
+                response.status_code,
+                f"Expected JSON response, but got: {response.text}",
+            ) from e
+        if not isinstance(response_json, dict):
+            raise CallBackendException(
+                response.status_code,
+                f"Expected JSON object, but got: {response_json}",
+            )
+        if 200 <= response.status_code < 300:
+            return validation_model.model_validate(response_json)
+        raise CallBackendException(
+            response.status_code,
+            f"Calling backend function errored with status {response.status_code}: {response_json}",
+        )
+
+
+class CallBackendException(Exception):
+    def __init__(self, status_code: int, body: str | dict[str, Any]):
+        message = "Unknown error"
+        if isinstance(body, str):
+            message = body
+        else:
+            body_message = body.get("message") or body.get("error")
+            if body_message:
+                message = str(body_message)
+            else:
+                message = json.dumps(body)
+        super().__init__(message)
+        self.status_code = status_code
+        self.body = body

runtime/backend_functions/get_auth_session_parameters.py

@@ -0,0 +1,30 @@
+from typing import Any
+
+from pydantic import BaseModel
+
+from runtime.context.context import IntunedContext
+
+from ._call_backend_function import call_backend_function
+
+
+class AuthSessionParameters(BaseModel):
+    parameters: dict[str, Any]
+
+
+async def get_auth_session_parameters() -> dict[str, Any]:
+    """
+    Get the auth session parameters from backend.
+    """
+
+    context = IntunedContext.current()
+    if context.run_context is None:
+        raise Exception("get_auth_session_parameters failed due to an internal error (context was not found).")
+    if context.run_context.auth_session_id is None:
+        raise Exception("Auth sessions are not enabled")
+
+    result = await call_backend_function(
+        name=f"auth-session/{context.run_context.auth_session_id}/parameters",
+        validation_model=AuthSessionParameters,
+    )
+
+    return result.parameters

runtime/browser/__init__.py

@@ -0,0 +1,3 @@
+from .launch_chromium import launch_chromium, dangerous_launch_chromium
+
+__all__ = ["launch_chromium", "dangerous_launch_chromium"]

runtime/browser/launch_chromium.py

@@ -0,0 +1,212 @@
+import json
+import os
+import tempfile
+from contextlib import asynccontextmanager
+from os.path import join
+from typing import Any
+from typing import Optional
+from typing import Literal
+import logging
+import aiofiles
+from playwright.async_api import async_playwright
+from playwright.async_api import Browser
+from playwright.async_api import ProxySettings
+
+logger = logging.getLogger(__name__)
+
+
+def get_proxy_env() -> Optional[ProxySettings]:
+    server = os.getenv("PROXY_SERVER")
+    username = os.getenv("PROXY_USERNAME")
+    password = os.getenv("PROXY_PASSWORD")
+    if server is None or username is None or password is None:
+        return None
+    return {
+        "server": server,
+        "username": username,
+        "password": password,
+    }
+
+
+chromium_launch_args_to_ignore = [
+    "--disable-field-trial-config",
+    "--disable-background-networking",
+    "--enable-features=NetworkService,NetworkServiceInProcess",
+    "--disable-background-timer-throttling",
+    "--disable-backgrounding-occluded-windows",
+    "--disable-back-forward-cache",
+    "--disable-breakpad",
+    "--disable-client-side-phishing-detection",
+    "--disable-component-extensions-with-background-pages",
+    "--disable-component-update",
+    "--no-default-browser-check",
+    "--disable-default-apps",
+    "--disable-dev-shm-usage",
+    "--disable-extensions",
+    "--disable-features=ImprovedCookieControls,LazyFrameLoading,GlobalMediaControls,DestroyProfileOnBrowserClose,MediaRouter,DialMediaRouteProvider,AcceptCHFrame,AutoExpandDetailsElement,CertificateTransparencyComponentUpdater,AvoidUnnecessaryBeforeUnloadCheckSync,Translate,TranslateUI",
+    "--allow-pre-commit-input",
+    "--disable-hang-monitor",
+    "--disable-ipc-flooding-protection",
+    "--disable-prompt-on-repost",
+    "--disable-renderer-backgrounding",
+    "--force-color-profile=srgb",
+    "--metrics-recording-only",
+    "--no-first-run",
+    "--enable-automation",
+    "--password-store=basic",
+    "--use-mock-keychain",
+    "--no-service-autorun",
+    "--export-tagged-pdf",
+    "--enable-use-zoom-for-dsf=false",
+    "--disable-popup-blocking",
+]
+
+
+async def create_user_dir_with_preferences():
+    # Create a temporary directory
+    playwright_temp_dir = tempfile.mkdtemp(prefix="pw-")
+    user_dir = join(playwright_temp_dir, "userdir")
+    default_dir = join(user_dir, "Default")
+
+    # Create the default directory recursively
+    os.makedirs(default_dir, exist_ok=True)
+
+    # Preferences data
+    preferences = {
+        "plugins": {
+            "always_open_pdf_externally": True,
+        }
+    }
+
+    # Write preferences to file
+    async with aiofiles.open(join(default_dir, "Preferences"), mode="w") as f:
+        await f.write(json.dumps(preferences))
+
+    return os.path.abspath(user_dir)
+
+
+extra_args = [
+    "--no-first-run",
+    "--disable-sync",
+    "--disable-translate",
+    "--disable-features=TranslateUI",
+    "--disable-features=NetworkService",
+    "--lang=en",
+    "--disable-blink-features=AutomationControlled",
+]
+
+default_user_agent = "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36"
+
+
+@asynccontextmanager
+async def launch_chromium(
+    headless: bool = True,
+    timeout: int = 10,
+    cdp_address: str | None = None,
+    **kwargs: Any,
+):
+    async with async_playwright() as playwright:
+        if cdp_address is not None:
+            browser: Browser = await playwright.chromium.connect_over_cdp(cdp_address)
+            context = browser.contexts[0]
+            dir = None
+        else:
+            dir = await create_user_dir_with_preferences()
+            if kwargs.get("proxy") is None:
+                proxy_env = get_proxy_env()
+            else:
+                proxy_env = kwargs.get("proxy")
+            # Remove proxy from kwargs if it exists
+            kwargs.pop("proxy", None)
+            viewport = kwargs.get("viewport", {"width": 1280, "height": 800})
+            kwargs.pop("viewport", None)
+
+            if headless:
+                chromium_launch_args_to_ignore.append("--headless")
+                extra_args.append("--headless=new")
+
+            context = await playwright.chromium.launch_persistent_context(
+                dir,
+                headless=headless,
+                viewport=viewport,
+                proxy=proxy_env,
+                # ignore_default_args=chromium_launch_args_to_ignore,
+                user_agent=os.environ.get("USER_AGENT", default_user_agent),
+                # args=extra_args,
+                **kwargs,
+            )
+        context.set_default_timeout(timeout * 1000)
+
+        async def remove_dir_after_close(*_: Any, **__: Any) -> None:
+            if not dir:
+                return
+            os.system(f"rm -rf {os.path.realpath(dir)}")
+
+        context.once("close", remove_dir_after_close)
+        yield context, context.pages[0]
+
+
+async def dangerous_launch_chromium(
+    headless: bool = True,
+    timeout: int = 10,
+    web_socket: str | None = None,
+    cdp_url: str | None = None,
+    connection_method: Literal["ws", "cdp"] | None = None,
+    **kwargs: Any,
+):
+    playwright = await async_playwright().start()
+    if web_socket is not None and connection_method == "ws":
+        logging.info(f"Connecting to ws: {web_socket}")
+        browser: Browser = await playwright.chromium.connect(web_socket)
+        browser.on("disconnected", lambda: logging.info("Browser Session disconnected"))
+        await browser.new_context(
+            viewport={"width": 1280, "height": 800}, user_agent=default_user_agent
+        )
+        context = browser.contexts[0]
+        dir = None
+    elif cdp_url is not None and connection_method == "cdp":
+        logging.info(f"Connecting to cdp: {cdp_url}")
+        browser: Browser = await playwright.chromium.connect_over_cdp(cdp_url)
+        browser.on("disconnected", lambda: logging.info("Browser Session disconnected"))
+        context = browser.contexts[0]
+        dir = None
+    elif web_socket is None and cdp_url is None and connection_method is None:
+        logging.info("Launching local browser")
+        dir = await create_user_dir_with_preferences()
+        logging.info(f"Using user data directory: {dir}")
+        if kwargs.get("proxy") is None:
+            proxy_env = get_proxy_env()
+        else:
+            proxy_env = kwargs.get("proxy")
+        # Remove proxy from kwargs if it exists
+        kwargs.pop("proxy", None)
+        viewport = kwargs.get("viewport", {"width": 1280, "height": 800})
+        kwargs.pop("viewport", None)
+
+        if headless:
+            chromium_launch_args_to_ignore.append("--headless")
+            extra_args.append("--headless=new")
+
+        context = await playwright.chromium.launch_persistent_context(
+            dir,
+            headless=headless,
+            viewport=viewport,
+            proxy=proxy_env,
+            ignore_default_args=chromium_launch_args_to_ignore,
+            user_agent=os.environ.get("USER_AGENT", default_user_agent),
+            args=extra_args,
+            **kwargs,
+        )
+    else:
+        raise ValueError(
+            "You have to provide method if you are launching a remote browser with ws or cdp"
+        )
+    context.set_default_timeout(timeout * 1000)
+
+    async def remove_dir_after_close(*_: Any, **__: Any) -> None:
+        if not dir:
+            return
+        os.system(f"rm -rf {os.path.realpath(dir)}")
+
+    context.once("close", remove_dir_after_close)
+    return playwright, context