smithy-aws-core 0.0.1__tar.gz

smithy_aws_core-0.0.1/.gitignore

@@ -0,0 +1,36 @@
+# Eclipse
+.classpath
+.project
+.settings/
+
+# Intellij
+.idea/
+*.iml
+*.iws
+
+# VS Code
+.vscode/
+
+# Mac
+.DS_Store
+
+# Maven
+target/
+**/dependency-reduced-pom.xml
+
+# Gradle
+**/.gradle
+build/
+*/out/
+**/*/out/
+
+# Python
+__pycache__
+*.swp
+.mypy_cache
+.coverage
+coverage.xml
+htmlcov
+*.egg-info
+dist
+.venv

smithy_aws_core-0.0.1/CHANGES.md

@@ -0,0 +1,13 @@
+# Changelog
+
+## Unreleased
+
+* <Add new items here>
+
+## v0.0.1
+
+### Feature
+* Added support for Instance Metadata Service (IMDS) credential resolution.
+* Added basic endpoint support.
+* Added basic User Agent support.
+* Added basic AWS specific protocol support for RestJson1 and HTTP bindings.

smithy_aws_core-0.0.1/NOTICE

@@ -0,0 +1 @@
+Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.

smithy_aws_core-0.0.1/PKG-INFO

@@ -0,0 +1,15 @@
+Metadata-Version: 2.4
+Name: smithy-aws-core
+Version: 0.0.1
+Summary: Core Smithy components for AWS services and protocols.
+License-File: NOTICE
+Requires-Python: >=3.12
+Requires-Dist: aws-sdk-signers
+Requires-Dist: smithy-core
+Requires-Dist: smithy-http
+Description-Content-Type: text/markdown
+
+# smithy-aws-core
+
+This is the core package that provides AWS specific interfaces for both client and server
+tooling generated by Smithy.

smithy_aws_core-0.0.1/README.md

@@ -0,0 +1,4 @@
+# smithy-aws-core
+
+This is the core package that provides AWS specific interfaces for both client and server
+tooling generated by Smithy.

smithy_aws_core-0.0.1/pyproject.toml

@@ -0,0 +1,23 @@
+[project]
+name = "smithy-aws-core"
+version = "0.0.1"
+description = "Core Smithy components for AWS services and protocols."
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "smithy-core",
+    "smithy-http",
+    "aws-sdk-signers"
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build]
+exclude = [
+  "tests",
+]
+
+[tool.ruff]
+src = ["src"]

smithy_aws_core-0.0.1/src/smithy_aws_core/__init__.py

@@ -0,0 +1,6 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+
+import importlib.metadata
+
+__version__: str = importlib.metadata.version("smithy-aws-core")

smithy_aws_core-0.0.1/src/smithy_aws_core/aio/__init__.py

@@ -0,0 +1,2 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0

smithy_aws_core-0.0.1/src/smithy_aws_core/aio/protocols.py

@@ -0,0 +1,28 @@
+from typing import Final
+
+from smithy_core.codecs import Codec
+from smithy_core.shapes import ShapeID
+from smithy_http.aio.protocols import HttpBindingClientProtocol
+from smithy_json import JSONCodec
+
+from ..traits import RestJson1Trait
+
+
+class RestJsonClientProtocol(HttpBindingClientProtocol):
+    """An implementation of the aws.protocols#restJson1 protocol."""
+
+    _id: ShapeID = RestJson1Trait.id
+    _codec: JSONCodec = JSONCodec()
+    _contentType: Final = "application/json"
+
+    @property
+    def id(self) -> ShapeID:
+        return self._id
+
+    @property
+    def payload_codec(self) -> Codec:
+        return self._codec
+
+    @property
+    def content_type(self) -> str:
+        return self._contentType

smithy_aws_core-0.0.1/src/smithy_aws_core/auth/__init__.py

@@ -0,0 +1,6 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+
+from .sigv4 import SigV4AuthScheme
+
+__all__ = ("SigV4AuthScheme",)

smithy_aws_core-0.0.1/src/smithy_aws_core/auth/sigv4.py

@@ -0,0 +1,54 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+from dataclasses import dataclass
+from typing import Protocol
+
+from aws_sdk_signers import AsyncSigV4Signer, SigV4SigningProperties
+from smithy_core.aio.interfaces.identity import IdentityResolver
+from smithy_core.exceptions import SmithyIdentityException
+from smithy_core.interfaces.identity import IdentityProperties
+from smithy_http.aio.interfaces.auth import HTTPAuthScheme, HTTPSigner
+
+from ..identity import AWSCredentialsIdentity
+
+
+class SigV4Config(Protocol):
+    aws_credentials_identity_resolver: (
+        IdentityResolver[AWSCredentialsIdentity, IdentityProperties] | None
+    )
+
+
+@dataclass(init=False)
+class SigV4AuthScheme(
+    HTTPAuthScheme[
+        AWSCredentialsIdentity, SigV4Config, IdentityProperties, SigV4SigningProperties
+    ]
+):
+    """SigV4 AuthScheme."""
+
+    scheme_id: str = "aws.auth#sigv4"
+    signer: HTTPSigner[AWSCredentialsIdentity, SigV4SigningProperties]
+
+    def __init__(
+        self,
+        *,
+        signer: HTTPSigner[AWSCredentialsIdentity, SigV4SigningProperties]
+        | None = None,
+    ) -> None:
+        """Constructor.
+
+        :param identity_resolver: The identity resolver to extract the api key identity.
+        :param signer: The signer used to sign the request.
+        """
+        # TODO: There are type mismatches in the signature of the "sign" method.
+        self.signer = signer or AsyncSigV4Signer()  # type: ignore
+
+    def identity_resolver(
+        self, *, config: SigV4Config
+    ) -> IdentityResolver[AWSCredentialsIdentity, IdentityProperties]:
+        if not config.aws_credentials_identity_resolver:
+            raise SmithyIdentityException(
+                "Attempted to use SigV4 auth, but aws_credentials_identity_resolver was not "
+                "set on the config."
+            )
+        return config.aws_credentials_identity_resolver

smithy_aws_core-0.0.1/src/smithy_aws_core/credentials_resolvers/__init__.py

@@ -0,0 +1,11 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+from .environment import EnvironmentCredentialsResolver
+from .imds import IMDSCredentialsResolver
+from .static import StaticCredentialsResolver
+
+__all__ = (
+    "EnvironmentCredentialsResolver",
+    "IMDSCredentialsResolver",
+    "StaticCredentialsResolver",
+)

smithy_aws_core-0.0.1/src/smithy_aws_core/credentials_resolvers/environment.py

@@ -0,0 +1,43 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+import os
+
+from smithy_core.aio.interfaces.identity import IdentityResolver
+from smithy_core.exceptions import SmithyIdentityException
+from smithy_core.interfaces.identity import IdentityProperties
+
+from ..identity import AWSCredentialsIdentity
+
+
+class EnvironmentCredentialsResolver(
+    IdentityResolver[AWSCredentialsIdentity, IdentityProperties]
+):
+    """Resolves AWS Credentials from system environment variables."""
+
+    def __init__(self):
+        self._credentials = None
+
+    async def get_identity(
+        self, *, identity_properties: IdentityProperties
+    ) -> AWSCredentialsIdentity:
+        if self._credentials is not None:
+            return self._credentials
+
+        access_key_id = os.getenv("AWS_ACCESS_KEY_ID")
+        secret_access_key = os.getenv("AWS_SECRET_ACCESS_KEY")
+        session_token = os.getenv("AWS_SESSION_TOKEN")
+        account_id = os.getenv("AWS_ACCOUNT_ID")
+
+        if access_key_id is None or secret_access_key is None:
+            raise SmithyIdentityException(
+                "AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY are required"
+            )
+
+        self._credentials = AWSCredentialsIdentity(
+            access_key_id=access_key_id,
+            secret_access_key=secret_access_key,
+            session_token=session_token,
+            account_id=account_id,
+        )
+
+        return self._credentials

smithy_aws_core-0.0.1/src/smithy_aws_core/credentials_resolvers/imds.py

@@ -0,0 +1,237 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+import asyncio
+import json
+from dataclasses import dataclass
+from datetime import UTC, datetime, timedelta
+from types import MappingProxyType
+from typing import Literal
+
+from smithy_core import URI
+from smithy_core.aio.interfaces.identity import IdentityResolver
+from smithy_core.exceptions import SmithyIdentityException
+from smithy_core.interfaces.identity import IdentityProperties
+from smithy_core.interfaces.retries import RetryStrategy
+from smithy_core.retries import SimpleRetryStrategy
+from smithy_http import Field, Fields
+from smithy_http.aio import HTTPRequest
+from smithy_http.aio.interfaces import HTTPClient
+
+from .. import __version__
+from ..identity import AWSCredentialsIdentity
+
+_USER_AGENT_FIELD = Field(
+    name="User-Agent",
+    values=[f"aws-sdk-python-imds-client/{__version__}"],
+)
+
+
+@dataclass(init=False)
+class Config:
+    """Configuration for EC2Metadata."""
+
+    _HOST_MAPPING = MappingProxyType(
+        {"IPv4": "169.254.169.254", "IPv6": "[fd00:ec2::254]"}
+    )
+    _MIN_TTL = 5
+    _MAX_TTL = 21600
+
+    retry_strategy: RetryStrategy
+    endpoint_uri: URI
+    endpoint_mode: Literal["IPv4", "IPv6"]
+    token_ttl: int
+
+    def __init__(
+        self,
+        *,
+        retry_strategy: RetryStrategy | None = None,
+        endpoint_uri: URI | None = None,
+        endpoint_mode: Literal["IPv4", "IPv6"] = "IPv4",
+        token_ttl: int = _MAX_TTL,
+        ec2_instance_profile_name: str | None = None,
+    ):
+        #  TODO: Implement retries.
+        self.retry_strategy = retry_strategy or SimpleRetryStrategy(max_attempts=3)
+        self.endpoint_mode = endpoint_mode
+        self.endpoint_uri = self._resolve_endpoint(endpoint_uri, endpoint_mode)
+        self.token_ttl = self._validate_token_ttl(token_ttl)
+        self.ec2_instance_profile_name = ec2_instance_profile_name
+
+    def _validate_token_ttl(self, ttl: int) -> int:
+        if not self._MIN_TTL <= ttl <= self._MAX_TTL:
+            raise ValueError(
+                f"Token TTL must be between {self._MIN_TTL} and {self._MAX_TTL} seconds."
+            )
+        return ttl
+
+    def _resolve_endpoint(
+        self, endpoint_uri: URI | None, endpoint_mode: Literal["IPv4", "IPv6"]
+    ) -> URI:
+        if endpoint_uri is not None:
+            return endpoint_uri
+
+        return URI(
+            scheme="http",
+            host=self._HOST_MAPPING.get(endpoint_mode, self._HOST_MAPPING["IPv4"]),
+            port=80,
+        )
+
+
+class Token:
+    """Represents an IMDSv2 session token with a value and method for checking
+    expiration."""
+
+    def __init__(self, value: str, ttl: int):
+        self._value = value
+        self._ttl = ttl
+        self._created_time = datetime.now()
+
+    def is_expired(self) -> bool:
+        return datetime.now() - self._created_time >= timedelta(seconds=self._ttl)
+
+    @property
+    def value(self) -> str:
+        return self._value
+
+
+class TokenCache:
+    """Holds the token needed to fetch instance metadata.
+
+    In addition, it knows how to refresh itself.
+    """
+
+    _TOKEN_PATH = "/latest/api/token"  # noqa: S105
+
+    def __init__(self, http_client: HTTPClient, config: Config):
+        self._http_client = http_client
+        self._config = config
+        self._base_uri = config.endpoint_uri
+        self._refresh_lock = asyncio.Lock()
+        self._token = None
+
+    def _should_refresh(self) -> bool:
+        return self._token is None or self._token.is_expired()
+
+    async def _refresh(self) -> None:
+        async with self._refresh_lock:
+            if not self._should_refresh():
+                return
+            headers = Fields(
+                [
+                    _USER_AGENT_FIELD,
+                    Field(
+                        name="x-aws-ec2-metadata-token-ttl-seconds",
+                        values=[str(self._config.token_ttl)],
+                    ),
+                ]
+            )
+            request = HTTPRequest(
+                method="PUT",
+                destination=URI(
+                    scheme=self._base_uri.scheme,
+                    host=self._base_uri.host,
+                    port=self._base_uri.port,
+                    path=self._TOKEN_PATH,
+                ),
+                fields=headers,
+            )
+            response = await self._http_client.send(request)
+            token_value = await response.consume_body_async()
+            self._token = Token(token_value.decode("utf-8"), self._config.token_ttl)
+
+    async def get_token(self) -> Token:
+        if self._should_refresh():
+            await self._refresh()
+        assert self._token is not None  # noqa: S101
+        return self._token
+
+
+class EC2Metadata:
+    def __init__(self, http_client: HTTPClient, config: Config | None = None):
+        self._http_client = http_client
+        self._config = config or Config()
+        self._token_cache = TokenCache(
+            http_client=self._http_client, config=self._config
+        )
+
+    async def get(self, *, path: str) -> str:
+        token = await self._token_cache.get_token()
+        headers = Fields(
+            [
+                _USER_AGENT_FIELD,
+                Field(
+                    name="x-aws-ec2-metadata-token",
+                    values=[token.value],
+                ),
+            ]
+        )
+        request = HTTPRequest(
+            method="GET",
+            destination=URI(
+                scheme=self._config.endpoint_uri.scheme,
+                host=self._config.endpoint_uri.host,
+                port=self._config.endpoint_uri.port,
+                path=path,
+            ),
+            fields=headers,
+        )
+        response = await self._http_client.send(request=request)
+        body = await response.consume_body_async()
+        return body.decode("utf-8")
+
+
+class IMDSCredentialsResolver(
+    IdentityResolver[AWSCredentialsIdentity, IdentityProperties]
+):
+    """Resolves AWS Credentials from an EC2 Instance Metadata Service (IMDS) client."""
+
+    _METADATA_PATH_BASE = "/latest/meta-data/iam/security-credentials"
+
+    def __init__(self, http_client: HTTPClient, config: Config | None = None):
+        # TODO: Respect IMDS specific config values from aws shared config file and environment.
+        self._http_client = http_client
+        self._ec2_metadata_client = EC2Metadata(http_client=http_client, config=config)
+        self._config = config or Config()
+        self._credentials = None
+        self._profile_name = self._config.ec2_instance_profile_name
+
+    async def get_identity(
+        self, *, identity_properties: IdentityProperties
+    ) -> AWSCredentialsIdentity:
+        if (
+            self._credentials is not None
+            and self._credentials.expiration
+            and datetime.now(UTC) < self._credentials.expiration
+        ):
+            return self._credentials
+
+        profile = self._profile_name
+        if profile is None:
+            profile = await self._ec2_metadata_client.get(path=self._METADATA_PATH_BASE)
+
+        creds_str = await self._ec2_metadata_client.get(
+            path=f"{self._METADATA_PATH_BASE}/{profile}"
+        )
+        creds = json.loads(creds_str)
+
+        access_key_id = creds.get("AccessKeyId")
+        secret_access_key = creds.get("SecretAccessKey")
+        session_token = creds.get("Token")
+        account_id = creds.get("AccountId")
+        expiration = creds.get("Expiration")
+        if expiration is not None:
+            expiration = datetime.fromisoformat(expiration).replace(tzinfo=UTC)
+
+        if access_key_id is None or secret_access_key is None:
+            raise SmithyIdentityException(
+                "AccessKeyId and SecretAccessKey are required"
+            )
+
+        self._credentials = AWSCredentialsIdentity(
+            access_key_id=access_key_id,
+            secret_access_key=secret_access_key,
+            session_token=session_token,
+            expiration=expiration,
+            account_id=account_id,
+        )
+        return self._credentials

smithy_aws_core-0.0.1/src/smithy_aws_core/credentials_resolvers/static.py

@@ -0,0 +1,20 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+from smithy_core.aio.interfaces.identity import IdentityResolver
+from smithy_core.interfaces.identity import IdentityProperties
+
+from smithy_aws_core.identity import AWSCredentialsIdentity
+
+
+class StaticCredentialsResolver(
+    IdentityResolver[AWSCredentialsIdentity, IdentityProperties]
+):
+    """Resolve Static AWS Credentials."""
+
+    def __init__(self, *, credentials: AWSCredentialsIdentity) -> None:
+        self._credentials = credentials
+
+    async def get_identity(
+        self, *, identity_properties: IdentityProperties
+    ) -> AWSCredentialsIdentity:
+        return self._credentials

smithy_aws_core-0.0.1/src/smithy_aws_core/endpoints/__init__.py

@@ -0,0 +1,15 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+from smithy_core.endpoints import StaticEndpointConfig
+from smithy_core.types import PropertyKey
+
+
+class RegionalEndpointConfig(StaticEndpointConfig):
+    """Endpoint config for services with standard regional endpoints."""
+
+    region: str | None
+    """The AWS region to address the request to."""
+
+
+REGIONAL_ENDPOINT_CONFIG = PropertyKey(key="config", value_type=RegionalEndpointConfig)
+"""Endpoint config for services with standard regional endpoints."""

smithy_aws_core-0.0.1/src/smithy_aws_core/endpoints/standard_regional.py

@@ -0,0 +1,33 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+from typing import Any
+
+from smithy_core import URI
+from smithy_core.aio.interfaces import EndpointResolver
+from smithy_core.endpoints import Endpoint, EndpointResolverParams, resolve_static_uri
+from smithy_core.exceptions import EndpointResolutionError
+
+from . import REGIONAL_ENDPOINT_CONFIG
+
+
+class StandardRegionalEndpointsResolver(EndpointResolver):
+    """Resolves endpoints for services with standard regional endpoints."""
+
+    def __init__(self, endpoint_prefix: str = "bedrock-runtime"):
+        self._endpoint_prefix = endpoint_prefix
+
+    async def resolve_endpoint(self, params: EndpointResolverParams[Any]) -> Endpoint:
+        if (static_uri := resolve_static_uri(params)) is not None:
+            return Endpoint(uri=static_uri)
+
+        region_config = params.context.get(REGIONAL_ENDPOINT_CONFIG)
+        if region_config is not None and region_config.region is not None:
+            # TODO: use dns suffix determined from partition metadata
+            dns_suffix = "amazonaws.com"
+            hostname = f"{self._endpoint_prefix}.{region_config.region}.{dns_suffix}"
+
+            return Endpoint(uri=URI(host=hostname))
+
+        raise EndpointResolutionError(
+            "Unable to resolve endpoint - either endpoint_url or region are required."
+        )

smithy_aws_core-0.0.1/src/smithy_aws_core/identity.py

@@ -0,0 +1,68 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+# http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+from datetime import datetime
+
+from smithy_core.aio.interfaces.identity import IdentityResolver
+from smithy_core.identity import Identity
+from smithy_core.interfaces.identity import IdentityProperties
+
+
+class AWSCredentialsIdentity(Identity):
+    """Container for AWS authentication credentials."""
+
+    def __init__(
+        self,
+        *,
+        access_key_id: str,
+        secret_access_key: str,
+        session_token: str | None = None,
+        expiration: datetime | None = None,
+        account_id: str | None = None,
+    ) -> None:
+        """Initialize the AWSCredentialIdentity.
+
+        :param access_key_id: A unique identifier for an AWS user or role.
+        :param secret_access_key: A secret key used in conjunction with the access key
+            ID to authenticate programmatic access to AWS services.
+        :param session_token: A temporary token used to specify the current session for
+            the supplied credentials.
+        :param expiration: The expiration time of the identity. If time zone is
+            provided, it is updated to UTC. The value must always be in UTC.
+        :param account_id: The AWS account's ID.
+        """
+        super().__init__(expiration=expiration)
+        self._access_key_id: str = access_key_id
+        self._secret_access_key: str = secret_access_key
+        self._session_token: str | None = session_token
+        self._account_id: str | None = account_id
+
+    @property
+    def access_key_id(self) -> str:
+        return self._access_key_id
+
+    @property
+    def secret_access_key(self) -> str:
+        return self._secret_access_key
+
+    @property
+    def session_token(self) -> str | None:
+        return self._session_token
+
+    @property
+    def account_id(self) -> str | None:
+        return self._account_id
+
+
+type AWSCredentialsResolver = IdentityResolver[
+    AWSCredentialsIdentity, IdentityProperties
+]

smithy_aws_core-0.0.1/src/smithy_aws_core/interceptors/__init__.py

@@ -0,0 +1,2 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0

smithy_aws_core-0.0.1/src/smithy_aws_core/interceptors/user_agent.py

@@ -0,0 +1,72 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+# pyright: reportMissingTypeStubs=false
+from typing import Any
+
+import smithy_core
+from smithy_core.interceptors import Interceptor, RequestContext
+from smithy_http.interceptors.user_agent import USER_AGENT
+from smithy_http.user_agent import RawStringUserAgentComponent, UserAgentComponent
+
+from .. import __version__
+
+_USERAGENT_SDK_NAME = "aws-sdk-python"
+
+
+class UserAgentInterceptor(Interceptor[Any, Any, Any, Any]):
+    """Adds AWS fields to the UserAgent."""
+
+    def __init__(
+        self,
+        *,
+        ua_suffix: str | None,
+        ua_app_id: str | None,
+        sdk_version: str,
+        service_id: str,
+    ) -> None:
+        """Initialize the UserAgentInterceptor.
+
+        :param ua_suffix: Additional suffix to be added to the UserAgent header.
+        :param ua_app_id: User defined and opaque application ID to be added to the
+            UserAgent header.
+        :param sdk_version: SDK version to be added to the UserAgent header.
+        :param service_id: ServiceId to be added to the UserAgent header.
+        """
+        super().__init__()
+        self._ua_suffix = ua_suffix
+        self._ua_app_id = ua_app_id
+        self._sdk_version = sdk_version
+        self._service_id = service_id
+
+    def read_after_serialization(self, context: RequestContext[Any, Any]) -> None:
+        if USER_AGENT in context.properties:
+            user_agent = context.properties[USER_AGENT]
+            user_agent.sdk_metadata = self._build_sdk_metadata()
+            user_agent.api_metadata.append(
+                UserAgentComponent("api", self._service_id, self._sdk_version)
+            )
+
+            if self._ua_app_id is not None:
+                user_agent.additional_metadata.append(
+                    UserAgentComponent("app", self._ua_app_id)
+                )
+
+            if self._ua_suffix is not None:
+                user_agent.additional_metadata.append(
+                    RawStringUserAgentComponent(self._ua_suffix)
+                )
+
+    def _build_sdk_metadata(self) -> list[UserAgentComponent]:
+        return [
+            UserAgentComponent(_USERAGENT_SDK_NAME, __version__),
+            UserAgentComponent("md", "smithy-core", smithy_core.__version__),
+            *self._crt_version(),
+        ]
+
+    def _crt_version(self) -> list[UserAgentComponent]:
+        try:
+            import awscrt
+
+            return [UserAgentComponent("md", "awscrt", awscrt.__version__)]
+        except (ImportError, AttributeError):
+            return []

smithy_aws_core-0.0.1/src/smithy_aws_core/py.typed

@@ -0,0 +1 @@
+Marker

smithy_aws_core-0.0.1/src/smithy_aws_core/traits.py

@@ -0,0 +1,44 @@
+#  Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#  SPDX-License-Identifier: Apache-2.0
+
+# This ruff check warns against using the assert statement, which can be stripped out
+# when running Python with certain (common) optimization settings. Assert is used here
+# for trait values. Since these are always generated, we can be fairly confident that
+# they're correct regardless, so it's okay if the checks are stripped out.
+# ruff: noqa: S101
+
+from collections.abc import Mapping, Sequence
+from dataclasses import dataclass, field
+
+from smithy_core.shapes import ShapeID
+from smithy_core.traits import DocumentValue, DynamicTrait, Trait
+
+
+@dataclass(init=False, frozen=True)
+class RestJson1Trait(Trait, id=ShapeID("aws.protocols#restJson1")):
+    http: Sequence[str] = field(
+        repr=False, hash=False, compare=False, default_factory=tuple
+    )
+    event_stream_http: Sequence[str] = field(
+        repr=False, hash=False, compare=False, default_factory=tuple
+    )
+
+    def __init__(self, value: DocumentValue | DynamicTrait = None):
+        super().__init__(value)
+        assert isinstance(self.document_value, Mapping)
+
+        http_versions = self.document_value["http"]
+        assert isinstance(http_versions, Sequence)
+        for val in http_versions:
+            assert isinstance(val, str)
+        object.__setattr__(self, "http", tuple(http_versions))
+        event_stream_http_versions = self.document_value.get("eventStreamHttp")
+        if not event_stream_http_versions:
+            object.__setattr__(self, "event_stream_http", self.http)
+        else:
+            assert isinstance(event_stream_http_versions, Sequence)
+            for val in event_stream_http_versions:
+                assert isinstance(val, str)
+            object.__setattr__(
+                self, "event_stream_http", tuple(event_stream_http_versions)
+            )