fluidattacks_cloudflare_sdk 0.1.0__tar.gz

fluidattacks_cloudflare_sdk-0.1.0/.envrc

@@ -0,0 +1,3 @@
+# use https://github.com/nix-community/nix-direnv
+ln -f ../../common/mypy.ini ./mypy.ini
+use_flake ".#python311.devShell"

fluidattacks_cloudflare_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,10 @@
+Metadata-Version: 2.4
+Name: fluidattacks_cloudflare_sdk
+Version: 0.1.0
+Summary: cloudflare SDK
+Author-email: Product Team <development@fluidattacks.com>
+Requires-Python: >=3.11
+Requires-Dist: cloudflare >=4.0.0, <5.0.0
+Requires-Dist: fa-purity >=2.5.0, <3.0.0
+Requires-Dist: fluidattacks-etl-utils >=4.0.0, <5.0.0
+Requires-Dist: fluidattacks-utils-logger >=3.0.0, <4.0.0

fluidattacks_cloudflare_sdk-0.1.0/build/default.nix

@@ -0,0 +1,13 @@
+{
+  nixpkgs,
+  builders,
+  scripts,
+  src,
+}:
+{
+  inherit src;
+  root_path = "observes/sdk/cloudflare";
+  module_name = "fluidattacks_cloudflare_sdk";
+  pypi_token_var = "CLOUDFLARE_SDK_TOKEN";
+  override = b: b;
+}

fluidattacks_cloudflare_sdk-0.1.0/build/filter.nix

@@ -0,0 +1,12 @@
+path_filter: src:
+path_filter {
+  root = src;
+  include = [
+    "fluidattacks_cloudflare_sdk"
+    "tests"
+    "pyproject.toml"
+    "mypy.ini"
+    "ruff.toml"
+    "uv.lock"
+  ];
+}

fluidattacks_cloudflare_sdk-0.1.0/flake.lock

@@ -0,0 +1,353 @@
+{
+  "nodes": {
+    "flake-parts": {
+      "inputs": {
+        "nixpkgs-lib": "nixpkgs-lib"
+      },
+      "locked": {
+        "lastModified": 1762810396,
+        "narHash": "sha256-dxFVgQPG+R72dkhXTtqUm7KpxElw3u6E+YlQ2WaDgt8=",
+        "owner": "hercules-ci",
+        "repo": "flake-parts",
+        "rev": "0bdadb1b265fb4143a75bd1ec7d8c915898a9923",
+        "type": "github"
+      },
+      "original": {
+        "owner": "hercules-ci",
+        "repo": "flake-parts",
+        "type": "github"
+      }
+    },
+    "nix_filter": {
+      "locked": {
+        "lastModified": 1757882181,
+        "narHash": "sha256-+cCxYIh2UNalTz364p+QYmWHs0P+6wDhiWR4jDIKQIU=",
+        "owner": "numtide",
+        "repo": "nix-filter",
+        "rev": "59c44d1909c72441144b93cf0f054be7fe764de5",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "nix-filter",
+        "type": "github"
+      }
+    },
+    "nix_filter_2": {
+      "locked": {
+        "lastModified": 1731533336,
+        "narHash": "sha256-oRam5PS1vcrr5UPgALW0eo1m/5/pls27Z/pabHNy2Ms=",
+        "owner": "numtide",
+        "repo": "nix-filter",
+        "rev": "f7653272fd234696ae94229839a99b73c9ab7de0",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "nix-filter",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1736441877,
+        "narHash": "sha256-m3+PhBFkDwqo9lBplG4AyMW8P4/KcioJRS1UG8N7okM=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "ce3899414dab3297cf025bfa356dc2da426feefd",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "nixpkgs-lib": {
+      "locked": {
+        "lastModified": 1761765539,
+        "narHash": "sha256-b0yj6kfvO8ApcSE+QmA6mUfu8IYG6/uU28OFn4PaC8M=",
+        "owner": "nix-community",
+        "repo": "nixpkgs.lib",
+        "rev": "719359f4562934ae99f5443f20aa06c2ffff91fc",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-community",
+        "repo": "nixpkgs.lib",
+        "type": "github"
+      }
+    },
+    "nixpkgs_2": {
+      "locked": {
+        "lastModified": 1758690382,
+        "narHash": "sha256-NY3kSorgqE5LMm1LqNwGne3ZLMF2/ILgLpFr1fS4X3o=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "e643668fd71b949c53f8626614b21ff71a07379d",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "nixpkgs_3": {
+      "locked": {
+        "lastModified": 1758690382,
+        "narHash": "sha256-NY3kSorgqE5LMm1LqNwGne3ZLMF2/ILgLpFr1fS4X3o=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "e643668fd71b949c53f8626614b21ff71a07379d",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "nixpkgs_4": {
+      "locked": {
+        "lastModified": 1761373498,
+        "narHash": "sha256-Q/uhWNvd7V7k1H1ZPMy/vkx3F8C13ZcdrKjO7Jv7v0c=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "6a08e6bb4e46ff7fcbb53d409b253f6bad8a28ce",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "ref": "nixos-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "nixpkgs_flake": {
+      "locked": {
+        "lastModified": 1762812764,
+        "narHash": "sha256-MNqrovkO8AEQeYQK7e3X2GDx2H8vfDIxV4K/VEaIjjo=",
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "rev": "400156fe527ab6b54cf6f70cca2c8385548a29c4",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nixos",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "observes_flake_builder": {
+      "inputs": {
+        "nix_filter": "nix_filter",
+        "nixpkgs_flake": "nixpkgs_flake",
+        "pynix_flake": "pynix_flake",
+        "pyproject-build-systems": "pyproject-build-systems",
+        "pyproject-nix": "pyproject-nix_2",
+        "shell-helpers": "shell-helpers",
+        "uv2nix": "uv2nix_2"
+      },
+      "locked": {
+        "dir": "observes/common/std_flake_2",
+        "lastModified": 1764767408,
+        "narHash": "sha256-CYGyCtQuboXbHQ0OUXNNdXXIcPOM1ZhOM94LUNjEQig=",
+        "ref": "refs/heads/trunk",
+        "rev": "28bda83eeda56c9c4016f43c86aaf93a366446c3",
+        "shallow": true,
+        "type": "git",
+        "url": "ssh://git@gitlab.com/fluidattacks/universe"
+      },
+      "original": {
+        "dir": "observes/common/std_flake_2",
+        "rev": "28bda83eeda56c9c4016f43c86aaf93a366446c3",
+        "shallow": true,
+        "type": "git",
+        "url": "ssh://git@gitlab.com/fluidattacks/universe"
+      }
+    },
+    "pynix_flake": {
+      "inputs": {
+        "nix_filter": "nix_filter_2",
+        "nixpkgs": "nixpkgs"
+      },
+      "locked": {
+        "lastModified": 1758642502,
+        "narHash": "sha256-PD/bQMz2dqZSkydHyvRh+jS/0qoV8SdcIVs6sassteQ=",
+        "owner": "dmurciaatfluid",
+        "repo": "python_nix_builder",
+        "rev": "809aafe2e1995e72c15378d099bf251b78f04a20",
+        "type": "gitlab"
+      },
+      "original": {
+        "owner": "dmurciaatfluid",
+        "repo": "python_nix_builder",
+        "type": "gitlab"
+      }
+    },
+    "pyproject-build-systems": {
+      "inputs": {
+        "nixpkgs": "nixpkgs_2",
+        "pyproject-nix": "pyproject-nix",
+        "uv2nix": "uv2nix"
+      },
+      "locked": {
+        "lastModified": 1761781027,
+        "narHash": "sha256-YDvxPAm2WnxrznRqWwHLjryBGG5Ey1ATEJXrON+TWt8=",
+        "owner": "pyproject-nix",
+        "repo": "build-system-pkgs",
+        "rev": "795a980d25301e5133eca37adae37283ec3c8e66",
+        "type": "github"
+      },
+      "original": {
+        "owner": "pyproject-nix",
+        "repo": "build-system-pkgs",
+        "type": "github"
+      }
+    },
+    "pyproject-nix": {
+      "inputs": {
+        "nixpkgs": [
+          "observes_flake_builder",
+          "pyproject-build-systems",
+          "nixpkgs"
+        ]
+      },
+      "locked": {
+        "lastModified": 1758265079,
+        "narHash": "sha256-amLaLNwKSZPShQHzfgmc/9o76dU8xzN0743dWgvYlr8=",
+        "owner": "nix-community",
+        "repo": "pyproject.nix",
+        "rev": "02e9418fd4af638447dca4b17b1280da95527fc9",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-community",
+        "repo": "pyproject.nix",
+        "type": "github"
+      }
+    },
+    "pyproject-nix_2": {
+      "inputs": {
+        "nixpkgs": "nixpkgs_3"
+      },
+      "locked": {
+        "lastModified": 1762427963,
+        "narHash": "sha256-CkPlAbIQ87wmjy5qHibfzk4DmMGBNqFer+lLfXjpP5M=",
+        "owner": "pyproject-nix",
+        "repo": "pyproject.nix",
+        "rev": "4540ea004e04fcd12dd2738d51383d10f956f7b9",
+        "type": "github"
+      },
+      "original": {
+        "owner": "pyproject-nix",
+        "repo": "pyproject.nix",
+        "type": "github"
+      }
+    },
+    "pyproject-nix_3": {
+      "inputs": {
+        "nixpkgs": [
+          "observes_flake_builder",
+          "uv2nix",
+          "nixpkgs"
+        ]
+      },
+      "locked": {
+        "lastModified": 1760402624,
+        "narHash": "sha256-jF6UKLs2uGc2rtved8Vrt58oTWjTQoAssuYs/0578Z4=",
+        "owner": "pyproject-nix",
+        "repo": "pyproject.nix",
+        "rev": "84c4ea102127c77058ea1ed7be7300261fafc7d2",
+        "type": "github"
+      },
+      "original": {
+        "owner": "pyproject-nix",
+        "repo": "pyproject.nix",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "observes_flake_builder": "observes_flake_builder"
+      }
+    },
+    "shell-helpers": {
+      "inputs": {
+        "flake-parts": "flake-parts",
+        "nixpkgs": [
+          "observes_flake_builder",
+          "nixpkgs_flake"
+        ]
+      },
+      "locked": {
+        "dir": "common/utils/shell-helpers",
+        "lastModified": 1762806066,
+        "narHash": "sha256-Wz99Fl7SpPU1NkTHJ36XOo6DjOEAMXCw3ThRABOS2Hs=",
+        "ref": "refs/heads/trunk",
+        "rev": "d73cfdc7ae9e7f547e22a4481d4c2ccce38b26a8",
+        "shallow": true,
+        "type": "git",
+        "url": "ssh://git@gitlab.com/fluidattacks/universe"
+      },
+      "original": {
+        "dir": "common/utils/shell-helpers",
+        "rev": "d73cfdc7ae9e7f547e22a4481d4c2ccce38b26a8",
+        "shallow": true,
+        "type": "git",
+        "url": "ssh://git@gitlab.com/fluidattacks/universe"
+      }
+    },
+    "uv2nix": {
+      "inputs": {
+        "nixpkgs": [
+          "observes_flake_builder",
+          "pyproject-build-systems",
+          "nixpkgs"
+        ],
+        "pyproject-nix": [
+          "observes_flake_builder",
+          "pyproject-build-systems",
+          "pyproject-nix"
+        ]
+      },
+      "locked": {
+        "lastModified": 1758933732,
+        "narHash": "sha256-HAmm1GBS1myZCFuog0DC2ZLaynvZtiUI2Crmo+cdQI0=",
+        "owner": "pyproject-nix",
+        "repo": "uv2nix",
+        "rev": "273ce18f913d8559e0d04f820d724308966d7c4d",
+        "type": "github"
+      },
+      "original": {
+        "owner": "pyproject-nix",
+        "repo": "uv2nix",
+        "type": "github"
+      }
+    },
+    "uv2nix_2": {
+      "inputs": {
+        "nixpkgs": "nixpkgs_4",
+        "pyproject-nix": "pyproject-nix_3"
+      },
+      "locked": {
+        "lastModified": 1761872265,
+        "narHash": "sha256-i25GRgp2vUOebY70L3NTAgkd+Pr1hnn5xM3qHxH0ONU=",
+        "owner": "pyproject-nix",
+        "repo": "uv2nix",
+        "rev": "74dfb62871be152ad3673b143b0cc56105a4f3c5",
+        "type": "github"
+      },
+      "original": {
+        "owner": "pyproject-nix",
+        "repo": "uv2nix",
+        "type": "github"
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}

fluidattacks_cloudflare_sdk-0.1.0/flake.nix

@@ -0,0 +1,29 @@
+{
+  description = "Cloudflare API SDK";
+
+  inputs = {
+    observes_flake_builder = {
+      url = "git+ssh://git@gitlab.com/fluidattacks/universe?shallow=1&rev=28bda83eeda56c9c4016f43c86aaf93a366446c3&dir=observes/common/std_flake_2";
+    };
+  };
+
+  outputs =
+    { self, ... }@inputs:
+    let
+      build_args =
+        {
+          system,
+          python_version,
+          nixpkgs,
+          builders,
+          scripts,
+        }:
+        import ./build {
+          inherit nixpkgs builders scripts;
+          src = import ./build/filter.nix nixpkgs.nix-filter self;
+        };
+    in
+    {
+      packages = inputs.observes_flake_builder.outputs.build build_args;
+    };
+}

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/__init__.py

@@ -0,0 +1,17 @@
+"""Cloudflare SDK for fluidattacks observes."""
+
+from fa_purity import Unsafe
+
+from fluidattacks_cloudflare_sdk._client import Token
+from fluidattacks_cloudflare_sdk._logger import set_logger
+from fluidattacks_cloudflare_sdk.core import AccountId, GatewayId
+
+__version__ = "0.1.0"
+
+Unsafe.compute(set_logger(__name__, __version__))
+
+__all__ = [
+    "AccountId",
+    "GatewayId",
+    "Token",
+]

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/_client/__init__.py

@@ -0,0 +1,19 @@
+"""Private connection package for the Cloudflare SDK."""
+
+from dataclasses import dataclass
+
+from fluidattacks_cloudflare_sdk._client._client import CloudflareClient
+from fluidattacks_cloudflare_sdk._client._core import Client, Token
+
+
+@dataclass(frozen=True)
+class CloudflareClientFactory:
+    """Factory for a Cloudflare Client."""
+
+    @staticmethod
+    def new(token: Token) -> Client:
+        """Build a Client from a Token."""
+        return CloudflareClient.new(token).client
+
+
+__all__ = ["Client", "CloudflareClientFactory", "Token"]

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/_client/_client.py

@@ -0,0 +1,30 @@
+"""Private Cloudflare client; builds the public Client."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
+
+from cloudflare import Cloudflare
+
+from fluidattacks_cloudflare_sdk._client._core import Client
+
+if TYPE_CHECKING:
+    from fluidattacks_cloudflare_sdk._client._core import Token
+
+
+@dataclass(frozen=True)
+class CloudflareClient:
+    """Private class holding the raw Cloudflare connection."""
+
+    _cloudflare: Cloudflare
+
+    @staticmethod
+    def new(token: Token) -> CloudflareClient:
+        """Build a CloudflareClient from a Token."""
+        return CloudflareClient(Cloudflare(api_token=token.raw))
+
+    @property
+    def client(self) -> Client:
+        """Return the opaque Client."""
+        return Client(self._cloudflare)

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/_client/_core.py

@@ -0,0 +1,23 @@
+"""Core types for the private Cloudflare connection package."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from cloudflare import Cloudflare
+
+
+@dataclass(frozen=True)
+class Token:
+    """Cloudflare API token."""
+
+    raw: str
+
+
+@dataclass(frozen=True)
+class Client:
+    """Opaque Cloudflare connection; raw field is SDK-internal."""
+
+    client: Cloudflare

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/_logger.py

@@ -0,0 +1,30 @@
+"""Logger configuration for the Cloudflare SDK."""
+
+from fa_purity import (
+    Cmd,
+)
+from fluidattacks_utils_logger import (
+    set_main_log,
+)
+from fluidattacks_utils_logger.env import (
+    current_app_env,
+    observes_debug,
+)
+from fluidattacks_utils_logger.handlers import (
+    LoggingConf,
+)
+
+
+def set_logger(root_name: str, version: str) -> Cmd[None]:
+    """Configure structured logging for the SDK."""
+    app_env = current_app_env()
+    debug = observes_debug()
+    conf = app_env.map(
+        lambda env: LoggingConf(
+            app_name="observes_cloudflare",
+            app_type="sdk",
+            app_version=version,
+            release_stage=env,
+        ),
+    )
+    return debug.bind(lambda d: conf.bind(lambda c: set_main_log(root_name, c, d)))

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/ai_gateway/__init__.py

@@ -0,0 +1,56 @@
+"""AI Gateway sub-package for the Cloudflare SDK."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
+
+from fluidattacks_cloudflare_sdk._client import CloudflareClientFactory
+from fluidattacks_cloudflare_sdk.ai_gateway._client import list_logs
+from fluidattacks_cloudflare_sdk.ai_gateway.core import (
+    AiGatewayClient,
+    EndDate,
+    LogRecord,
+    LogsOrderBy,
+    LogsQuery,
+    StartDate,
+)
+
+if TYPE_CHECKING:
+    from fluidattacks_cloudflare_sdk._client import Client, Token
+    from fluidattacks_cloudflare_sdk.core import AccountId, GatewayId
+
+
+def _from_client(
+    cloudflare: Client,
+    account_id: AccountId,
+    gateway_id: GatewayId,
+) -> AiGatewayClient:
+    return AiGatewayClient(
+        list_logs=lambda q: list_logs(cloudflare, account_id, gateway_id, q),
+    )
+
+
+@dataclass(frozen=True)
+class AiGatewayClientFactory:
+    """Factory for AiGatewayClient."""
+
+    @staticmethod
+    def new(
+        token: Token,
+        account_id: AccountId,
+        gateway_id: GatewayId,
+    ) -> AiGatewayClient:
+        """Build an AiGatewayClient from a token and identifiers."""
+        return _from_client(CloudflareClientFactory.new(token), account_id, gateway_id)
+
+
+__all__ = [
+    "AiGatewayClient",
+    "AiGatewayClientFactory",
+    "EndDate",
+    "LogRecord",
+    "LogsOrderBy",
+    "LogsQuery",
+    "StartDate",
+]

fluidattacks_cloudflare_sdk-0.1.0/fluidattacks_cloudflare_sdk/ai_gateway/_client.py

@@ -0,0 +1,69 @@
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING
+
+from cloudflare import NOT_GIVEN
+from fa_purity import Cmd, FrozenList, Result, ResultE, Unsafe, cast_exception
+
+from fluidattacks_cloudflare_sdk.ai_gateway._decode import decode_log
+from fluidattacks_cloudflare_sdk.ai_gateway.core import (
+    LogRecord,
+    LogsOrderBy,
+    LogsQuery,
+)
+
+if TYPE_CHECKING:
+    from fluidattacks_cloudflare_sdk._client import Client
+    from fluidattacks_cloudflare_sdk.core import AccountId, GatewayId
+
+LOG = logging.getLogger(__name__)
+
+
+def list_logs(
+    cloudflare: Client,
+    account_id: AccountId,
+    gateway_id: GatewayId,
+    query: LogsQuery,
+) -> Cmd[ResultE[FrozenList[LogRecord]]]:
+    _order = query.order_by.value_or(LogsOrderBy.default())
+    _start = query.start.map(lambda s: s.raw.date_time.strftime("%Y-%m-%dT%H:%M:%SZ")).value_or(
+        NOT_GIVEN
+    )
+    _end = query.end.map(lambda e: e.raw.date_time.strftime("%Y-%m-%dT%H:%M:%SZ")).value_or(
+        NOT_GIVEN
+    )
+
+    def _action() -> ResultE[FrozenList[LogRecord]]:
+        try:
+            LOG.info("fetching ai gateway logs for gateway %s", gateway_id.raw)
+            first_page = cloudflare.client.ai_gateway.logs.list(
+                gateway_id.raw,
+                account_id=account_id.raw,
+                start_date=_start,
+                end_date=_end,
+                order_by=_order.field,
+                order_by_direction=_order.direction,
+            )
+            items: list[LogRecord] = []
+            for page in first_page.iter_pages():
+                page_records = page.result
+                items.extend(
+                    decode_log(raw).alt(Unsafe.raise_exception).to_union() for raw in page_records
+                )
+                LOG.info(
+                    "page fetched: %d records (%d total so far)",
+                    len(page_records),
+                    len(items),
+                )
+            LOG.info(
+                "finished: %d log records from gateway %s",
+                len(items),
+                gateway_id.raw,
+            )
+            return Result.success(tuple(items))
+        except Exception as err:
+            LOG.exception("failed to fetch ai gateway logs")
+            return Result.failure(err, FrozenList[LogRecord]).alt(cast_exception)
+
+    return Cmd.wrap_impure(_action)