# This file was auto-generated by Fern from our API Definition.

from __future__ import annotations

import typing

import httpx
from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
from .environment import BrowserUseEnvironment

if typing.TYPE_CHECKING:
    from .billing.client import AsyncBillingClient, BillingClient
    from .browsers.client import AsyncBrowsersClient, BrowsersClient
    from .files.client import AsyncFilesClient, FilesClient
    from .profiles.client import AsyncProfilesClient, ProfilesClient
    from .sessions.client import AsyncSessionsClient, SessionsClient
    from .skills.client import AsyncSkillsClient, SkillsClient
    from .skills_marketplace.client import AsyncSkillsMarketplaceClient, SkillsMarketplaceClient
    from .tasks.client import AsyncTasksClient, TasksClient


class BaseClient:
    """
    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.

    Parameters
    ----------
    base_url : typing.Optional[str]
        The base url to use for requests from the client.

    environment : BrowserUseEnvironment
        The environment to use for requests from the client. from .environment import BrowserUseEnvironment



        Defaults to BrowserUseEnvironment.PRODUCTION



    api_key : str
    headers : typing.Optional[typing.Dict[str, str]]
        Additional headers to send with every request.

    timeout : typing.Optional[float]
        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.

    follow_redirects : typing.Optional[bool]
        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.

    httpx_client : typing.Optional[httpx.Client]
        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.

    Examples
    --------
    from browser_use_sdk import BrowserUse

    client = BrowserUse(
        api_key="YOUR_API_KEY",
    )
    """

    def __init__(
        self,
        *,
        base_url: typing.Optional[str] = None,
        environment: BrowserUseEnvironment = BrowserUseEnvironment.PRODUCTION,
        api_key: str,
        headers: typing.Optional[typing.Dict[str, str]] = None,
        timeout: typing.Optional[float] = None,
        follow_redirects: typing.Optional[bool] = True,
        httpx_client: typing.Optional[httpx.Client] = None,
    ):
        _defaulted_timeout = (
            timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
        )
        self._client_wrapper = SyncClientWrapper(
            base_url=_get_base_url(base_url=base_url, environment=environment),
            api_key=api_key,
            headers=headers,
            httpx_client=httpx_client
            if httpx_client is not None
            else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
            if follow_redirects is not None
            else httpx.Client(timeout=_defaulted_timeout),
            timeout=_defaulted_timeout,
        )
        self._billing: typing.Optional[BillingClient] = None
        self._tasks: typing.Optional[TasksClient] = None
        self._sessions: typing.Optional[SessionsClient] = None
        self._files: typing.Optional[FilesClient] = None
        self._profiles: typing.Optional[ProfilesClient] = None
        self._browsers: typing.Optional[BrowsersClient] = None
        self._skills: typing.Optional[SkillsClient] = None
        self._skills_marketplace: typing.Optional[SkillsMarketplaceClient] = None

    @property
    def billing(self):
        if self._billing is None:
            from .billing.client import BillingClient  # noqa: E402

            self._billing = BillingClient(client_wrapper=self._client_wrapper)
        return self._billing

    @property
    def tasks(self):
        if self._tasks is None:
            from .tasks.client import TasksClient  # noqa: E402

            self._tasks = TasksClient(client_wrapper=self._client_wrapper)
        return self._tasks

    @property
    def sessions(self):
        if self._sessions is None:
            from .sessions.client import SessionsClient  # noqa: E402

            self._sessions = SessionsClient(client_wrapper=self._client_wrapper)
        return self._sessions

    @property
    def files(self):
        if self._files is None:
            from .files.client import FilesClient  # noqa: E402

            self._files = FilesClient(client_wrapper=self._client_wrapper)
        return self._files

    @property
    def profiles(self):
        if self._profiles is None:
            from .profiles.client import ProfilesClient  # noqa: E402

            self._profiles = ProfilesClient(client_wrapper=self._client_wrapper)
        return self._profiles

    @property
    def browsers(self):
        if self._browsers is None:
            from .browsers.client import BrowsersClient  # noqa: E402

            self._browsers = BrowsersClient(client_wrapper=self._client_wrapper)
        return self._browsers

    @property
    def skills(self):
        if self._skills is None:
            from .skills.client import SkillsClient  # noqa: E402

            self._skills = SkillsClient(client_wrapper=self._client_wrapper)
        return self._skills

    @property
    def skills_marketplace(self):
        if self._skills_marketplace is None:
            from .skills_marketplace.client import SkillsMarketplaceClient  # noqa: E402

            self._skills_marketplace = SkillsMarketplaceClient(client_wrapper=self._client_wrapper)
        return self._skills_marketplace


class AsyncBaseClient:
    """
    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.

    Parameters
    ----------
    base_url : typing.Optional[str]
        The base url to use for requests from the client.

    environment : BrowserUseEnvironment
        The environment to use for requests from the client. from .environment import BrowserUseEnvironment



        Defaults to BrowserUseEnvironment.PRODUCTION



    api_key : str
    headers : typing.Optional[typing.Dict[str, str]]
        Additional headers to send with every request.

    timeout : typing.Optional[float]
        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.

    follow_redirects : typing.Optional[bool]
        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.

    httpx_client : typing.Optional[httpx.AsyncClient]
        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.

    Examples
    --------
    from browser_use_sdk import AsyncBrowserUse

    client = AsyncBrowserUse(
        api_key="YOUR_API_KEY",
    )
    """

    def __init__(
        self,
        *,
        base_url: typing.Optional[str] = None,
        environment: BrowserUseEnvironment = BrowserUseEnvironment.PRODUCTION,
        api_key: str,
        headers: typing.Optional[typing.Dict[str, str]] = None,
        timeout: typing.Optional[float] = None,
        follow_redirects: typing.Optional[bool] = True,
        httpx_client: typing.Optional[httpx.AsyncClient] = None,
    ):
        _defaulted_timeout = (
            timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
        )
        self._client_wrapper = AsyncClientWrapper(
            base_url=_get_base_url(base_url=base_url, environment=environment),
            api_key=api_key,
            headers=headers,
            httpx_client=httpx_client
            if httpx_client is not None
            else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
            if follow_redirects is not None
            else httpx.AsyncClient(timeout=_defaulted_timeout),
            timeout=_defaulted_timeout,
        )
        self._billing: typing.Optional[AsyncBillingClient] = None
        self._tasks: typing.Optional[AsyncTasksClient] = None
        self._sessions: typing.Optional[AsyncSessionsClient] = None
        self._files: typing.Optional[AsyncFilesClient] = None
        self._profiles: typing.Optional[AsyncProfilesClient] = None
        self._browsers: typing.Optional[AsyncBrowsersClient] = None
        self._skills: typing.Optional[AsyncSkillsClient] = None
        self._skills_marketplace: typing.Optional[AsyncSkillsMarketplaceClient] = None

    @property
    def billing(self):
        if self._billing is None:
            from .billing.client import AsyncBillingClient  # noqa: E402

            self._billing = AsyncBillingClient(client_wrapper=self._client_wrapper)
        return self._billing

    @property
    def tasks(self):
        if self._tasks is None:
            from .tasks.client import AsyncTasksClient  # noqa: E402

            self._tasks = AsyncTasksClient(client_wrapper=self._client_wrapper)
        return self._tasks

    @property
    def sessions(self):
        if self._sessions is None:
            from .sessions.client import AsyncSessionsClient  # noqa: E402

            self._sessions = AsyncSessionsClient(client_wrapper=self._client_wrapper)
        return self._sessions

    @property
    def files(self):
        if self._files is None:
            from .files.client import AsyncFilesClient  # noqa: E402

            self._files = AsyncFilesClient(client_wrapper=self._client_wrapper)
        return self._files

    @property
    def profiles(self):
        if self._profiles is None:
            from .profiles.client import AsyncProfilesClient  # noqa: E402

            self._profiles = AsyncProfilesClient(client_wrapper=self._client_wrapper)
        return self._profiles

    @property
    def browsers(self):
        if self._browsers is None:
            from .browsers.client import AsyncBrowsersClient  # noqa: E402

            self._browsers = AsyncBrowsersClient(client_wrapper=self._client_wrapper)
        return self._browsers

    @property
    def skills(self):
        if self._skills is None:
            from .skills.client import AsyncSkillsClient  # noqa: E402

            self._skills = AsyncSkillsClient(client_wrapper=self._client_wrapper)
        return self._skills

    @property
    def skills_marketplace(self):
        if self._skills_marketplace is None:
            from .skills_marketplace.client import AsyncSkillsMarketplaceClient  # noqa: E402

            self._skills_marketplace = AsyncSkillsMarketplaceClient(client_wrapper=self._client_wrapper)
        return self._skills_marketplace


def _get_base_url(*, base_url: typing.Optional[str] = None, environment: BrowserUseEnvironment) -> str:
    if base_url is not None:
        return base_url
    elif environment is not None:
        return environment.value
    else:
        raise Exception("Please pass in either base_url or environment to construct the client")
