# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
"""AppSync Async IO Gql Client"""
from urllib.parse import urlparse

from gql.client import Client
from gql.transport.aiohttp import AIOHTTPTransport
from gql.transport.appsync_auth import AppSyncIAMAuthentication


class AppsyncAioGqlClient(Client):
    """AppSync Async IO Gql Client"""

    def __init__(
        self,
        url: str,
        **kwargs,
    ):
        host = str(urlparse(url).netloc)
        auth = AppSyncIAMAuthentication(host=host)
        transport = AIOHTTPTransport(url=url, auth=auth)

        super().__init__(transport=transport, **kwargs)