swarmauri_auth_idp_cognito 0.1.0.dev24__tar.gz

swarmauri_auth_idp_cognito-0.1.0.dev24/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [2025] [Jacob Stewart @ Swarmauri]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License 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.

swarmauri_auth_idp_cognito-0.1.0.dev24/PKG-INFO

@@ -0,0 +1,127 @@
+Metadata-Version: 2.4
+Name: swarmauri_auth_idp_cognito
+Version: 0.1.0.dev24
+Summary: AWS Cognito OAuth 2.0 / OAuth 2.1 / OIDC 1.0 identity provider implementations for Swarmauri
+License-Expression: Apache-2.0
+License-File: LICENSE
+Keywords: swarmauri,sdk,experimental,identity,oauth2,oauth21,oidc,cognito,aws,idp
+Author: Jacob Stewart
+Author-email: jacob@swarmauri.com
+Requires-Python: >=3.10,<3.13
+Classifier: Development Status :: 1 - Planning
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Internet :: WWW/HTTP :: Session
+Classifier: Topic :: Security :: Cryptography
+Requires-Dist: httpx
+Requires-Dist: pyjwt[crypto] (>=2.8)
+Requires-Dist: swarmauri_base
+Requires-Dist: swarmauri_core
+Description-Content-Type: text/markdown
+
+![Swarmauri Logo](https://github.com/swarmauri/swarmauri-sdk/blob/master/assets/swarmauri.brand.theme.svg)
+
+<p align="center">
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/dm/swarmauri_auth_idp_cognito" alt="PyPI - Downloads"/></a>
+    <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/experimental/swarmauri_auth_idp_cognito/">
+        <img alt="Hits" src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/experimental/swarmauri_auth_idp_cognito.svg"/></a>
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/pyversions/swarmauri_auth_idp_cognito" alt="PyPI - Python Version"/></a>
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/l/swarmauri_auth_idp_cognito" alt="PyPI - License"/></a>
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/v/swarmauri_auth_idp_cognito?label=swarmauri_auth_idp_cognito&color=green" alt="PyPI - swarmauri_auth_idp_cognito"/></a>
+</p>
+
+---
+
+# Swarmauri Auth IDP Cognito
+
+AWS Cognito OAuth 2.0, OAuth 2.1, and OIDC 1.0 login and app-client flows packaged for the Swarmauri ecosystem.
+
+## Features
+
+- PKCE-enabled authorization code logins with signed state payloads to prevent tampering.
+- Automatic discovery of Cognito endpoints with resilient HTTP retry semantics.
+- ID token verification against Cognito JWKS with graceful fallback to the UserInfo endpoint.
+- Machine-to-machine app clients supporting shared secrets or JWT-based client assertions.
+- ComponentBase-registered classes for seamless Swarmauri plugin discovery and configuration.
+
+## Installation
+
+### pip
+
+```bash
+pip install swarmauri_auth_idp_cognito
+```
+
+### uv (project)
+
+```bash
+uv add swarmauri_auth_idp_cognito
+```
+
+### uv (environment)
+
+```bash
+uv pip install swarmauri_auth_idp_cognito
+```
+
+## Usage
+
+Instantiate the login classes with your Cognito issuer, app client credentials, and redirect URI.
+Persist the returned state between `auth_url` and `exchange*` calls to prevent replay attacks.
+
+```python
+from pydantic import SecretBytes, SecretStr
+from swarmauri_auth_idp_cognito import CognitoOAuth21Login
+
+login = CognitoOAuth21Login(
+    issuer="https://example-domain.auth.us-east-1.amazoncognito.com",
+    client_id="example-client-id",
+    client_secret=SecretStr("example-secret"),
+    redirect_uri="https://example.com/callback",
+    state_secret=SecretBytes(b"super-secret-state-key"),
+)
+
+# Within an async context:
+# auth_payload = await login.auth_url()
+# identity = await login.exchange_and_identity(code, auth_payload["state"])
+print(login.client_id)
+```
+
+### Expected Workflow
+
+1. Call `auth_url()` and redirect the user agent to the returned authorization URL.
+2. Persist the state value and later validate it when Cognito posts back to your callback.
+3. Call `exchange_and_identity()` (or `exchange()` for the OIDC login) to normalize identity claims.
+4. Use the normalized payload to provision sessions, issue downstream tokens, or audit login activity.
+
+App client classes expose the same `access_token` coroutine to support background services
+and machine-to-machine integrations.
+
+## Entry Points
+
+The distribution registers the following entry points:
+
+- `swarmauri.auth_idp:CognitoOAuth20Login`
+- `swarmauri.auth_idp:CognitoOAuth21Login`
+- `swarmauri.auth_idp:CognitoOIDC10Login`
+- `swarmauri.auth_idp:CognitoOAuth20AppClient`
+- `swarmauri.auth_idp:CognitoOAuth21AppClient`
+- `swarmauri.auth_idp:CognitoOIDC10AppClient`
+
+## Contributing
+
+To contribute to swarmauri-sdk, review the
+[guidelines for contributing](https://github.com/swarmauri/swarmauri-sdk/blob/master/CONTRIBUTING.md),
+including development workflow, testing, and coding standards.
+

swarmauri_auth_idp_cognito-0.1.0.dev24/README.md

@@ -0,0 +1,98 @@
+![Swarmauri Logo](https://github.com/swarmauri/swarmauri-sdk/blob/master/assets/swarmauri.brand.theme.svg)
+
+<p align="center">
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/dm/swarmauri_auth_idp_cognito" alt="PyPI - Downloads"/></a>
+    <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/experimental/swarmauri_auth_idp_cognito/">
+        <img alt="Hits" src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/experimental/swarmauri_auth_idp_cognito.svg"/></a>
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/pyversions/swarmauri_auth_idp_cognito" alt="PyPI - Python Version"/></a>
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/l/swarmauri_auth_idp_cognito" alt="PyPI - License"/></a>
+    <a href="https://pypi.org/project/swarmauri_auth_idp_cognito/">
+        <img src="https://img.shields.io/pypi/v/swarmauri_auth_idp_cognito?label=swarmauri_auth_idp_cognito&color=green" alt="PyPI - swarmauri_auth_idp_cognito"/></a>
+</p>
+
+---
+
+# Swarmauri Auth IDP Cognito
+
+AWS Cognito OAuth 2.0, OAuth 2.1, and OIDC 1.0 login and app-client flows packaged for the Swarmauri ecosystem.
+
+## Features
+
+- PKCE-enabled authorization code logins with signed state payloads to prevent tampering.
+- Automatic discovery of Cognito endpoints with resilient HTTP retry semantics.
+- ID token verification against Cognito JWKS with graceful fallback to the UserInfo endpoint.
+- Machine-to-machine app clients supporting shared secrets or JWT-based client assertions.
+- ComponentBase-registered classes for seamless Swarmauri plugin discovery and configuration.
+
+## Installation
+
+### pip
+
+```bash
+pip install swarmauri_auth_idp_cognito
+```
+
+### uv (project)
+
+```bash
+uv add swarmauri_auth_idp_cognito
+```
+
+### uv (environment)
+
+```bash
+uv pip install swarmauri_auth_idp_cognito
+```
+
+## Usage
+
+Instantiate the login classes with your Cognito issuer, app client credentials, and redirect URI.
+Persist the returned state between `auth_url` and `exchange*` calls to prevent replay attacks.
+
+```python
+from pydantic import SecretBytes, SecretStr
+from swarmauri_auth_idp_cognito import CognitoOAuth21Login
+
+login = CognitoOAuth21Login(
+    issuer="https://example-domain.auth.us-east-1.amazoncognito.com",
+    client_id="example-client-id",
+    client_secret=SecretStr("example-secret"),
+    redirect_uri="https://example.com/callback",
+    state_secret=SecretBytes(b"super-secret-state-key"),
+)
+
+# Within an async context:
+# auth_payload = await login.auth_url()
+# identity = await login.exchange_and_identity(code, auth_payload["state"])
+print(login.client_id)
+```
+
+### Expected Workflow
+
+1. Call `auth_url()` and redirect the user agent to the returned authorization URL.
+2. Persist the state value and later validate it when Cognito posts back to your callback.
+3. Call `exchange_and_identity()` (or `exchange()` for the OIDC login) to normalize identity claims.
+4. Use the normalized payload to provision sessions, issue downstream tokens, or audit login activity.
+
+App client classes expose the same `access_token` coroutine to support background services
+and machine-to-machine integrations.
+
+## Entry Points
+
+The distribution registers the following entry points:
+
+- `swarmauri.auth_idp:CognitoOAuth20Login`
+- `swarmauri.auth_idp:CognitoOAuth21Login`
+- `swarmauri.auth_idp:CognitoOIDC10Login`
+- `swarmauri.auth_idp:CognitoOAuth20AppClient`
+- `swarmauri.auth_idp:CognitoOAuth21AppClient`
+- `swarmauri.auth_idp:CognitoOIDC10AppClient`
+
+## Contributing
+
+To contribute to swarmauri-sdk, review the
+[guidelines for contributing](https://github.com/swarmauri/swarmauri-sdk/blob/master/CONTRIBUTING.md),
+including development workflow, testing, and coding standards.

swarmauri_auth_idp_cognito-0.1.0.dev24/pyproject.toml

@@ -0,0 +1,88 @@
+[project]
+name = "swarmauri_auth_idp_cognito"
+version = "0.1.0.dev24"
+description = "AWS Cognito OAuth 2.0 / OAuth 2.1 / OIDC 1.0 identity provider implementations for Swarmauri"
+license = "Apache-2.0"
+readme = "README.md"
+requires-python = ">=3.10,<3.13"
+repository = "https://github.com/swarmauri/swarmauri-sdk"
+authors = [{ name = "Jacob Stewart", email = "jacob@swarmauri.com" }]
+classifiers = [
+    "Development Status :: 1 - Planning",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: Apache Software License",
+    "Natural Language :: English",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Internet :: WWW/HTTP :: Session",
+    "Topic :: Security :: Cryptography",
+]
+dependencies = [
+    "swarmauri_core",
+    "swarmauri_base",
+    "httpx",
+    "pyjwt[crypto]>=2.8",
+]
+keywords = [
+    "swarmauri",
+    "sdk",
+    "experimental",
+    "identity",
+    "oauth2",
+    "oauth21",
+    "oidc",
+    "cognito",
+    "aws",
+    "idp",
+]
+
+[tool.uv.sources]
+swarmauri_tests_pylicense = { workspace = true }
+swarmauri_core = { workspace = true }
+swarmauri_base = { workspace = true }
+
+[tool.pytest.ini_options]
+markers = [
+    "test: standard test",
+    "unit: Unit tests",
+    "i9n: Integration tests",
+    "r8n: Regression tests",
+    "acceptance: Acceptance tests",
+    "perf: Performance tests",
+    "example: Documentation examples",
+]
+timeout = 300
+log_cli = true
+log_cli_level = "INFO"
+log_cli_format = "%(asctime)s [%(levelname)s] %(message)s"
+log_cli_date_format = "%Y-%m-%d %H:%M:%S"
+asyncio_default_fixture_loop_scope = "function"
+addopts = "--pylicense-accept-deps=sniffio"
+
+[dependency-groups]
+dev = [
+    "pytest>=8.0",
+    "pytest-asyncio>=0.24.0",
+    "pytest-xdist>=3.6.1",
+    "pytest-json-report>=1.5.0",
+    "pytest-timeout>=2.3.1",
+    "pytest-benchmark>=4.0.0",
+    "ruff>=0.9.9",
+    "swarmauri_tests_pylicense",
+]
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[project.entry-points.'swarmauri.auth_idp']
+CognitoOAuth20Login = "swarmauri_auth_idp_cognito:CognitoOAuth20Login"
+CognitoOAuth21Login = "swarmauri_auth_idp_cognito:CognitoOAuth21Login"
+CognitoOIDC10Login = "swarmauri_auth_idp_cognito:CognitoOIDC10Login"
+CognitoOAuth20AppClient = "swarmauri_auth_idp_cognito:CognitoOAuth20AppClient"
+CognitoOAuth21AppClient = "swarmauri_auth_idp_cognito:CognitoOAuth21AppClient"
+CognitoOIDC10AppClient = "swarmauri_auth_idp_cognito:CognitoOIDC10AppClient"

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoLoginMixin.py

@@ -0,0 +1,105 @@
+"""Shared helpers for Cognito login implementations."""
+
+from __future__ import annotations
+
+import json
+from typing import Any, Callable, Mapping, Optional
+
+import jwt
+from pydantic import ConfigDict, PrivateAttr
+
+from swarmauri_base.auth_idp.http import RetryingAsyncClient
+from swarmauri_base.auth_idp.utils import verify_state
+
+
+def _load_signing_key(jwks: Mapping[str, Any], kid: str) -> Any:
+    """Return the cryptographic key referenced by ``kid``."""
+
+    for entry in jwks.get("keys", []):
+        if entry.get("kid") == kid:
+            kty = entry.get("kty", "RSA").upper()
+            serialized = json.dumps(entry)
+            if kty == "EC":
+                return jwt.algorithms.ECAlgorithm.from_jwk(serialized)
+            return jwt.algorithms.RSAAlgorithm.from_jwk(serialized)
+    raise ValueError("signing key not found")
+
+
+class CognitoLoginMixin:
+    """Provide shared discovery, HTTP, and state helpers for Cognito logins."""
+
+    model_config = ConfigDict(extra="forbid")
+
+    _http_client_factory: Callable[[], RetryingAsyncClient] = PrivateAttr(
+        default=RetryingAsyncClient
+    )
+    _discovery: Optional[Mapping[str, Any]] = PrivateAttr(default=None)
+
+    def _state_secret_value(self) -> bytes:
+        return self.state_secret.get_secret_value()
+
+    def _client_secret_value(self) -> str:
+        return self.client_secret.get_secret_value()
+
+    def _state_payload(self, state: str) -> Mapping[str, Any]:
+        return verify_state(self._state_secret_value(), state)
+
+    async def _metadata(self) -> Mapping[str, Any]:
+        if self._discovery is None:
+            url = f"{self.issuer.rstrip('/')}/.well-known/openid-configuration"
+            self._discovery = await self._http_get(url)
+        return self._discovery
+
+    async def _http_get(
+        self, url: str, *, headers: Optional[Mapping[str, str]] = None
+    ) -> Mapping[str, Any]:
+        request_headers = {"Accept": "application/json"}
+        if headers:
+            request_headers.update(headers)
+        async with self._http_client_factory() as client:
+            response = await client.get_retry(url, headers=request_headers)
+            response.raise_for_status()
+            return response.json()
+
+    async def _http_post(
+        self,
+        url: str,
+        *,
+        data: Mapping[str, Any],
+        headers: Optional[Mapping[str, str]] = None,
+    ) -> Mapping[str, Any]:
+        request_headers = {"Accept": "application/json"}
+        if headers:
+            request_headers.update(headers)
+        async with self._http_client_factory() as client:
+            response = await client.post_retry(url, data=data, headers=request_headers)
+            response.raise_for_status()
+            return response.json()
+
+    async def _decode_id_token(
+        self,
+        id_token: Optional[str],
+        *,
+        jwks_uri: str,
+        expected_audience: str,
+        issuer: str,
+    ) -> Mapping[str, Any]:
+        if not id_token:
+            raise ValueError("no id_token in response")
+        header = jwt.get_unverified_header(id_token)
+        kid = header.get("kid")
+        if not kid:
+            raise ValueError("missing kid in id_token header")
+        jwks = await self._http_get(jwks_uri)
+        signer = _load_signing_key(jwks, kid)
+        algorithm = header.get("alg", "RS256")
+        return jwt.decode(
+            id_token,
+            signer,
+            algorithms=[algorithm],
+            audience=expected_audience,
+            issuer=issuer,
+        )
+
+
+__all__ = ["CognitoLoginMixin"]

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoOAuth20AppClient.py

@@ -0,0 +1,59 @@
+"""Cognito OAuth 2.0 client credentials app client implementation."""
+
+from __future__ import annotations
+
+import time
+from typing import Callable, Optional, Tuple
+
+from pydantic import ConfigDict, Field, PrivateAttr, SecretStr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.auth_idp import OAuth20AppClientBase
+from swarmauri_base.auth_idp.http import RetryingAsyncClient
+
+
+@ComponentBase.register_type(OAuth20AppClientBase, "CognitoOAuth20AppClient")
+class CognitoOAuth20AppClient(OAuth20AppClientBase):
+    """Request machine-to-machine access tokens from Cognito's OAuth 2.0 endpoint."""
+
+    model_config = ConfigDict(extra="forbid")
+
+    base_url: str
+    client_id: str
+    client_secret: SecretStr
+    cache_skew_seconds: int = Field(default=30, ge=0)
+    _http_client_factory: Callable[[], RetryingAsyncClient] = PrivateAttr(
+        default=RetryingAsyncClient
+    )
+    _cached_token: Optional[Tuple[str, float]] = PrivateAttr(default=None)
+
+    def _token_endpoint(self) -> str:
+        return f"{self.base_url.rstrip('/')}/oauth2/token"
+
+    def _client_secret_value(self) -> str:
+        return self.client_secret.get_secret_value()
+
+    async def _fetch_token(self, scope: Optional[str]) -> Tuple[str, float]:
+        form = {"grant_type": "client_credentials"}
+        if scope:
+            form["scope"] = scope
+        auth = (self.client_id, self._client_secret_value())
+        async with self._http_client_factory() as client:
+            response = await client.post_retry(
+                self._token_endpoint(),
+                data=form,
+                auth=auth,
+                headers={"Accept": "application/json"},
+            )
+            response.raise_for_status()
+            payload = response.json()
+        expires_at = time.time() + int(payload.get("expires_in", 3600))
+        return payload["access_token"], expires_at
+
+    async def access_token(self, scope: Optional[str] = None) -> str:
+        cached = self._cached_token
+        now = time.time()
+        if cached and now < cached[1] - self.cache_skew_seconds:
+            return cached[0]
+        token, expires_at = await self._fetch_token(scope)
+        self._cached_token = (token, expires_at)
+        return token

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoOAuth20Login.py

@@ -0,0 +1,68 @@
+"""Cognito OAuth 2.0 Authorization Code login implementation."""
+
+from __future__ import annotations
+
+from typing import Any, Mapping
+
+from pydantic import Field, SecretBytes, SecretStr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.auth_idp import OAuth20LoginBase, make_pkce_pair, sign_state
+
+from .CognitoLoginMixin import CognitoLoginMixin
+
+
+@ComponentBase.register_type(OAuth20LoginBase, "CognitoOAuth20Login")
+class CognitoOAuth20Login(CognitoLoginMixin, OAuth20LoginBase):
+    """AWS Cognito OAuth 2.0 Authorization Code flow with PKCE."""
+
+    issuer: str
+    client_id: str
+    client_secret: SecretStr
+    redirect_uri: str
+    state_secret: SecretBytes
+    scope: str = Field(default="openid email profile")
+
+    async def auth_url(self) -> Mapping[str, str]:
+        metadata = await self._metadata()
+        verifier, challenge = make_pkce_pair()
+        state = sign_state(self._state_secret_value(), {"verifier": verifier})
+        endpoint = metadata["authorization_endpoint"]
+        url = (
+            f"{endpoint}?response_type=code"
+            f"&client_id={self.client_id}&redirect_uri={self.redirect_uri}"
+            f"&scope={self.scope}&state={state}"
+            f"&code_challenge={challenge}&code_challenge_method=S256"
+        )
+        return {"url": url, "state": state}
+
+    async def exchange_and_identity(self, code: str, state: str) -> Mapping[str, Any]:
+        payload = self._state_payload(state)
+        metadata = await self._metadata()
+        form = {
+            "grant_type": "authorization_code",
+            "code": code,
+            "redirect_uri": self.redirect_uri,
+            "client_id": self.client_id,
+            "client_secret": self._client_secret_value(),
+            "code_verifier": payload["verifier"],
+        }
+        token_json = await self._http_post(metadata["token_endpoint"], data=form)
+        access_token = token_json["access_token"]
+        userinfo_endpoint = metadata.get("userinfo_endpoint") or (
+            f"{self.issuer.rstrip('/')}/oauth2/userInfo"
+        )
+        userinfo_headers = {
+            "Authorization": f"Bearer {access_token}",
+            "Accept": "application/json",
+        }
+        userinfo_json = await self._http_get(
+            userinfo_endpoint, headers=userinfo_headers
+        )
+        name = userinfo_json.get("name") or userinfo_json.get("given_name") or "Unknown"
+        return {
+            "issuer": "cognito-oauth2",
+            "sub": userinfo_json.get("sub"),
+            "email": userinfo_json.get("email"),
+            "name": name,
+            "raw": {"tokens": token_json, "userinfo": userinfo_json},
+        }

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoOAuth21AppClient.py

@@ -0,0 +1,103 @@
+"""Cognito OAuth 2.1 client credentials app client implementation."""
+
+from __future__ import annotations
+
+import json
+import time
+from typing import Any, Callable, Dict, Mapping, Optional, Tuple
+
+import jwt
+from pydantic import ConfigDict, Field, PrivateAttr, SecretStr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.auth_idp import OAuth21AppClientBase
+from swarmauri_base.auth_idp.http import RetryingAsyncClient
+
+
+@ComponentBase.register_type(OAuth21AppClientBase, "CognitoOAuth21AppClient")
+class CognitoOAuth21AppClient(OAuth21AppClientBase):
+    """Request OAuth 2.1 access tokens using either client secrets or JWT assertions."""
+
+    model_config = ConfigDict(extra="forbid")
+
+    base_url: str
+    client_id: str
+    client_secret: Optional[SecretStr] = None
+    private_key_jwk: Optional[Dict[str, Any]] = None
+    cache_skew_seconds: int = Field(default=30, ge=0)
+    _http_client_factory: Callable[[], RetryingAsyncClient] = PrivateAttr(
+        default=RetryingAsyncClient
+    )
+    _cached_token: Optional[Tuple[str, float]] = PrivateAttr(default=None)
+
+    def _token_endpoint(self) -> str:
+        return f"{self.base_url.rstrip('/')}/oauth2/token"
+
+    def _client_secret_value(self) -> Optional[str]:
+        return self.client_secret.get_secret_value() if self.client_secret else None
+
+    @staticmethod
+    def _load_private_key(jwk_payload: Mapping[str, Any], algorithm: str) -> Any:
+        serialized = json.dumps(jwk_payload)
+        if algorithm.upper().startswith("ES"):
+            return jwt.algorithms.ECAlgorithm.from_jwk(serialized)
+        return jwt.algorithms.RSAAlgorithm.from_jwk(serialized)
+
+    def _client_assertion_body(self) -> Dict[str, str]:
+        if not self.private_key_jwk:
+            return {}
+        algorithm = self.private_key_jwk.get("alg", "RS256")
+        key = self._load_private_key(self.private_key_jwk, algorithm)
+        now = int(time.time())
+        payload = {
+            "iss": self.client_id,
+            "sub": self.client_id,
+            "aud": self._token_endpoint(),
+            "iat": now,
+            "exp": now + 300,
+            "jti": str(now),
+        }
+        headers: Dict[str, str] = {}
+        if kid := self.private_key_jwk.get("kid"):
+            headers["kid"] = kid
+        assertion = jwt.encode(
+            payload, key=key, algorithm=algorithm, headers=headers or None
+        )
+        return {
+            "client_assertion_type": "urn:ietf:params:oauth:client-assertion-type:jwt-bearer",
+            "client_assertion": assertion,
+        }
+
+    async def _fetch_token(self, scope: Optional[str]) -> Tuple[str, float]:
+        form = {"grant_type": "client_credentials"}
+        if scope:
+            form["scope"] = scope
+        body = self._client_assertion_body()
+        auth = None
+        if body:
+            form["client_id"] = self.client_id
+        else:
+            secret = self._client_secret_value()
+            if not secret:
+                raise ValueError("client_secret or private_key_jwk must be provided")
+            auth = (self.client_id, secret)
+        payload = {**form, **body}
+        async with self._http_client_factory() as client:
+            response = await client.post_retry(
+                self._token_endpoint(),
+                data=payload,
+                auth=auth,
+                headers={"Accept": "application/json"},
+            )
+            response.raise_for_status()
+            data = response.json()
+        expires_at = time.time() + int(data.get("expires_in", 3600))
+        return data["access_token"], expires_at
+
+    async def access_token(self, scope: Optional[str] = None) -> str:
+        cached = self._cached_token
+        now = time.time()
+        if cached and now < cached[1] - self.cache_skew_seconds:
+            return cached[0]
+        token, expires_at = await self._fetch_token(scope)
+        self._cached_token = (token, expires_at)
+        return token

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoOAuth21Login.py

@@ -0,0 +1,63 @@
+"""Cognito OAuth 2.1 Authorization Code login implementation."""
+
+from __future__ import annotations
+
+from typing import Any, Mapping
+
+from pydantic import Field, SecretBytes, SecretStr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.auth_idp import OAuth21LoginBase, make_pkce_pair, sign_state
+
+from .CognitoLoginMixin import CognitoLoginMixin
+
+
+@ComponentBase.register_type(OAuth21LoginBase, "CognitoOAuth21Login")
+class CognitoOAuth21Login(CognitoLoginMixin, OAuth21LoginBase):
+    """AWS Cognito OAuth 2.1 Authorization Code flow leveraging OIDC discovery."""
+
+    issuer: str
+    client_id: str
+    client_secret: SecretStr
+    redirect_uri: str
+    state_secret: SecretBytes
+    scope: str = Field(default="openid email profile")
+
+    async def auth_url(self) -> Mapping[str, str]:
+        metadata = await self._metadata()
+        verifier, challenge = make_pkce_pair()
+        state = sign_state(self._state_secret_value(), {"verifier": verifier})
+        endpoint = metadata["authorization_endpoint"]
+        url = (
+            f"{endpoint}?response_type=code"
+            f"&client_id={self.client_id}&redirect_uri={self.redirect_uri}"
+            f"&scope={self.scope}&state={state}"
+            f"&code_challenge={challenge}&code_challenge_method=S256"
+        )
+        return {"url": url, "state": state}
+
+    async def exchange_and_identity(self, code: str, state: str) -> Mapping[str, Any]:
+        payload = self._state_payload(state)
+        metadata = await self._metadata()
+        form = {
+            "grant_type": "authorization_code",
+            "code": code,
+            "redirect_uri": self.redirect_uri,
+            "client_id": self.client_id,
+            "client_secret": self._client_secret_value(),
+            "code_verifier": payload["verifier"],
+        }
+        token_json = await self._http_post(metadata["token_endpoint"], data=form)
+        claims = await self._decode_id_token(
+            token_json.get("id_token"),
+            jwks_uri=metadata["jwks_uri"],
+            expected_audience=self.client_id,
+            issuer=metadata.get("issuer", self.issuer),
+        )
+        name = claims.get("name") or claims.get("given_name") or "Unknown"
+        return {
+            "issuer": "cognito-oauth21",
+            "sub": claims["sub"],
+            "email": claims.get("email"),
+            "name": name,
+            "raw": {"tokens": token_json},
+        }

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoOIDC10AppClient.py

@@ -0,0 +1,69 @@
+"""Cognito OIDC 1.0 client credentials app client implementation."""
+
+from __future__ import annotations
+
+import time
+from typing import Any, Callable, Mapping, Optional, Tuple
+
+from pydantic import ConfigDict, Field, PrivateAttr, SecretStr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.auth_idp import OIDC10AppClientBase
+from swarmauri_base.auth_idp.http import RetryingAsyncClient
+
+
+@ComponentBase.register_type(OIDC10AppClientBase, "CognitoOIDC10AppClient")
+class CognitoOIDC10AppClient(OIDC10AppClientBase):
+    """Request machine-to-machine tokens using Cognito's OIDC discovery metadata."""
+
+    model_config = ConfigDict(extra="forbid")
+
+    issuer: str
+    client_id: str
+    client_secret: SecretStr
+    cache_skew_seconds: int = Field(default=30, ge=0)
+    _http_client_factory: Callable[[], RetryingAsyncClient] = PrivateAttr(
+        default=RetryingAsyncClient
+    )
+    _cached_token: Optional[Tuple[str, float]] = PrivateAttr(default=None)
+    _discovery: Optional[Mapping[str, Any]] = PrivateAttr(default=None)
+
+    def _client_secret_value(self) -> str:
+        return self.client_secret.get_secret_value()
+
+    async def _metadata(self) -> Mapping[str, Any]:
+        if self._discovery is None:
+            url = f"{self.issuer.rstrip('/')}/.well-known/openid-configuration"
+            async with self._http_client_factory() as client:
+                response = await client.get_retry(
+                    url, headers={"Accept": "application/json"}
+                )
+                response.raise_for_status()
+                self._discovery = response.json()
+        return self._discovery
+
+    async def _fetch_token(self, scope: Optional[str]) -> Tuple[str, float]:
+        metadata = await self._metadata()
+        form = {"grant_type": "client_credentials"}
+        if scope:
+            form["scope"] = scope
+        auth = (self.client_id, self._client_secret_value())
+        async with self._http_client_factory() as client:
+            response = await client.post_retry(
+                metadata["token_endpoint"],
+                data=form,
+                auth=auth,
+                headers={"Accept": "application/json"},
+            )
+            response.raise_for_status()
+            payload = response.json()
+        expires_at = time.time() + int(payload.get("expires_in", 3600))
+        return payload["access_token"], expires_at
+
+    async def access_token(self, scope: Optional[str] = None) -> str:
+        cached = self._cached_token
+        now = time.time()
+        if cached and now < cached[1] - self.cache_skew_seconds:
+            return cached[0]
+        token, expires_at = await self._fetch_token(scope)
+        self._cached_token = (token, expires_at)
+        return token

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/CognitoOIDC10Login.py

@@ -0,0 +1,74 @@
+"""Cognito OpenID Connect 1.0 Authorization Code login implementation."""
+
+from __future__ import annotations
+
+from typing import Any, Mapping
+
+from pydantic import Field, SecretBytes, SecretStr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.auth_idp import OIDC10LoginBase, make_pkce_pair, sign_state
+
+from .CognitoLoginMixin import CognitoLoginMixin
+
+
+@ComponentBase.register_type(OIDC10LoginBase, "CognitoOIDC10Login")
+class CognitoOIDC10Login(CognitoLoginMixin, OIDC10LoginBase):
+    """AWS Cognito OIDC 1.0 Authorization Code flow with ID token validation."""
+
+    issuer: str
+    client_id: str
+    client_secret: SecretStr
+    redirect_uri: str
+    state_secret: SecretBytes
+    scope: str = Field(default="openid email profile")
+
+    async def auth_url(self) -> Mapping[str, str]:
+        metadata = await self._metadata()
+        verifier, challenge = make_pkce_pair()
+        state = sign_state(self._state_secret_value(), {"verifier": verifier})
+        endpoint = metadata["authorization_endpoint"]
+        url = (
+            f"{endpoint}?response_type=code"
+            f"&client_id={self.client_id}&redirect_uri={self.redirect_uri}"
+            f"&scope={self.scope}&state={state}"
+            f"&code_challenge={challenge}&code_challenge_method=S256"
+        )
+        return {"url": url, "state": state}
+
+    async def exchange(self, code: str, state: str) -> Mapping[str, Any]:
+        payload = self._state_payload(state)
+        metadata = await self._metadata()
+        form = {
+            "grant_type": "authorization_code",
+            "code": code,
+            "redirect_uri": self.redirect_uri,
+            "client_id": self.client_id,
+            "client_secret": self._client_secret_value(),
+            "code_verifier": payload["verifier"],
+        }
+        token_json = await self._http_post(metadata["token_endpoint"], data=form)
+        claims = await self._decode_id_token(
+            token_json.get("id_token"),
+            jwks_uri=metadata["jwks_uri"],
+            expected_audience=self.client_id,
+            issuer=metadata.get("issuer", self.issuer),
+        )
+        email = claims.get("email")
+        name = claims.get("name") or claims.get("given_name") or "Unknown"
+        if (not email or name == "Unknown") and metadata.get("userinfo_endpoint"):
+            headers = {
+                "Authorization": f"Bearer {token_json['access_token']}",
+                "Accept": "application/json",
+            }
+            userinfo_json = await self._http_get(
+                metadata["userinfo_endpoint"], headers=headers
+            )
+            email = userinfo_json.get("email") or email
+            name = userinfo_json.get("name") or name
+        return {
+            "issuer": "cognito-oidc",
+            "sub": claims["sub"],
+            "email": email,
+            "name": name,
+            "raw": {"tokens": token_json},
+        }

swarmauri_auth_idp_cognito-0.1.0.dev24/swarmauri_auth_idp_cognito/__init__.py

@@ -0,0 +1,17 @@
+"""AWS Cognito identity provider implementations for Swarmauri."""
+
+from .CognitoOAuth20AppClient import CognitoOAuth20AppClient
+from .CognitoOAuth20Login import CognitoOAuth20Login
+from .CognitoOAuth21AppClient import CognitoOAuth21AppClient
+from .CognitoOAuth21Login import CognitoOAuth21Login
+from .CognitoOIDC10AppClient import CognitoOIDC10AppClient
+from .CognitoOIDC10Login import CognitoOIDC10Login
+
+__all__ = [
+    "CognitoOAuth20AppClient",
+    "CognitoOAuth20Login",
+    "CognitoOAuth21AppClient",
+    "CognitoOAuth21Login",
+    "CognitoOIDC10AppClient",
+    "CognitoOIDC10Login",
+]