centra-sdk 0.1.0__tar.gz

centra_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,21 @@
+Metadata-Version: 2.3
+Name: centra-sdk
+Version: 0.1.0
+Summary: 
+License: MIT
+Author: ivasylenko
+Author-email: ivasylen@akamai.com
+Requires-Python: >=3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: fastapi (>=0.116.1,<0.117.0)
+Requires-Dist: uvicorn (>=0.35.0,<0.36.0)
+Project-URL: Homepage, https://github.com/guardicore/contracts
+Project-URL: Repository, https://github.com/guardicore/contracts
+Description-Content-Type: text/markdown
+
+## Contracts SDK
+That is an SDK for Centra Connectors
+

centra_sdk-0.1.0/README.md

@@ -0,0 +1,2 @@
+## Contracts SDK
+That is an SDK for Centra Connectors

centra_sdk-0.1.0/pyproject.toml

@@ -0,0 +1,27 @@
+[project]
+name = "centra-sdk"
+version = "0.1.0"
+description = ""
+authors = [
+    {name = "ivasylenko",email = "ivasylen@akamai.com"}
+]
+license = { text = "MIT" }
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "fastapi (>=0.116.1,<0.117.0)",
+    "uvicorn (>=0.35.0,<0.36.0)"
+]
+
+[tool.poetry]
+packages = [{include = "centra_sdk", from = "src"}]
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[project.urls]
+homepage = "https://github.com/guardicore/contracts"
+repository = "https://github.com/guardicore/contracts"
+

centra_sdk-0.1.0/src/centra_sdk/dependencies.py

@@ -0,0 +1,45 @@
+# generated by fastapi-codegen:
+#   filename:  connector
+#   timestamp: 2025-07-31T14:14:32+00:00
+
+from __future__ import annotations
+
+from typing import Optional, Union
+from uuid import UUID
+
+from pydantic import conint
+
+from .models.connector.v1.common.inventory import InventoryItem
+from .models.connector.v1.consumer.agents import AgentsInfo
+from .models.connector.v1.consumer.enforcement import (
+    Action,
+    EnforcementPolicy,
+    EnforcementPolicyInventory,
+)
+from .models.connector.v1.operations.config import (
+    InternalConfig,
+    InternalConfigMetadata,
+)
+from .models.connector.v1.operations.health import (
+    V1OperationsFlagsGetResponse,
+    V1OperationsHealthGetResponse,
+    V1OperationsMetricsGetResponse,
+)
+from .models.connector.v1.operations.info import EnvUnits, IntegrationEnvInfo
+from .models.connector.v1.operations.log import LogStart, LogStatus
+from .models.connector.v1.operations.onboard import (
+    Onboard,
+    StatusRequest,
+    StatusResponse,
+)
+from .models.connector.v1.provider.inventory import (
+    Inventory,
+    InventoryAssetType,
+    InventoryReporterType,
+    Labels,
+    NetworkTopology,
+    V1ProviderInventoryIdAssetsAssetTypePostResponse,
+    V1ProviderInventoryIdAssetsAssetTypePostResponse1,
+)
+from .models.connector.v1.provider.lookup import LookupRequest
+from .models.connector.version import VersionHandshakeData

centra_sdk-0.1.0/src/centra_sdk/handler_registry.py

@@ -0,0 +1,138 @@
+# generated by fastapi-codegen:
+#   filename:  connector
+#   timestamp: 2025-07-31T14:14:32+00:00
+
+# generated by fastapi-codegen:
+#   filename:  connector
+#   timestamp: 2025-07-30T11:25:18+00:00
+
+import logging
+from typing import Any, Callable, Dict, Generic, Optional, TypeVar
+
+from fastapi import HTTPException
+from fastapi.encoders import jsonable_encoder
+from fastapi.responses import JSONResponse
+
+T = TypeVar('T', bound=Callable)
+
+
+logger = logging.getLogger(__name__)
+
+
+class HandlerRegistry(Generic[T]):
+    """A more sophisticated handler registry with decorator support and logging."""
+
+    def __init__(self, name: str = "default"):
+        self.name = name
+        self._handlers: Dict[str, T] = {}
+
+    def register(self, tag: str) -> Callable[[T], T]:
+        """Decorator to register a handler.
+
+        Usage:
+            registry = HandlerRegistry()
+
+            @registry.register("my_handler")
+            def my_handler():
+                pass
+        """
+
+        def decorator(handler: T) -> T:
+            self.set_handler(tag, handler)
+            return handler
+
+        return decorator
+
+    def set_handler(self, tag: str, handler: T) -> None:
+        """Register a handler for the given tag."""
+        if not tag:
+            raise ValueError("Tag cannot be empty")
+        if not callable(handler):
+            raise TypeError("Handler must be callable")
+
+        if tag in self._handlers:
+            logger.warning(
+                f"Overriding existing handler for tag '{tag}' in registry '{self.name}'"
+            )
+
+        self._handlers[tag] = handler()
+        logger.debug(
+            f"Registered handler '{handler.__name__}' for tag '{tag}' in registry '{self.name}'"
+        )
+
+    def get_handler(self, tag: str) -> Optional[T]:
+        """Retrieve a handler for the given tag."""
+        handler = self._handlers.get(tag)
+        if handler is None:
+            logger.warning(
+                f"No handler found for tag '{tag}' in registry '{self.name}'"
+            )
+        return handler
+
+    def has_handler(self, tag: str) -> bool:
+        """Check if a handler exists for the given tag."""
+        return tag in self._handlers
+
+    def remove_handler(self, tag: str) -> bool:
+        """Remove a handler for the given tag."""
+        if tag in self._handlers:
+            del self._handlers[tag]
+            logger.debug(f"Removed handler for tag '{tag}' from registry '{self.name}'")
+            return True
+        return False
+
+    def clear_handlers(self) -> None:
+        """Remove all registered handlers."""
+        count = len(self._handlers)
+        self._handlers.clear()
+        logger.debug(f"Cleared {count} handlers from registry '{self.name}'")
+
+    def get_all_tags(self) -> list[str]:
+        """Get all registered tags."""
+        return list(self._handlers.keys())
+
+    def __len__(self) -> int:
+        """Return the number of registered handlers."""
+        return len(self._handlers)
+
+    def __contains__(self, tag: str) -> bool:
+        """Check if a tag is registered (supports 'in' operator)."""
+        return tag in self._handlers
+
+    def __repr__(self) -> str:
+        return f"HandlerRegistry(name='{self.name}', handlers={len(self._handlers)})"
+
+
+# Global instance for backward compatibility
+default_registry = HandlerRegistry("global")
+
+
+# Convenience functions that delegate to the global registry
+def set_handler(tag: str, handler: Callable) -> None:
+    """Register a handler globally."""
+    default_registry.set_handler(tag, handler)
+
+
+def get_handler(tag: str) -> Optional[Callable]:
+    """Get a handler from the global registry."""
+    return default_registry.get_handler(tag)
+
+
+def register(tag: str) -> Callable:
+    """Decorator to register a handler globally."""
+    return default_registry.register(tag)
+
+
+def call_handler(tag: str, func_name: str, *args, **kwargs) -> Any:
+    """Decorator to call a handler."""
+
+    handler = default_registry.get_handler(tag)
+    if handler:
+        handler_func = getattr(handler, func_name, None)
+        if not handler_func:
+            logger.warning(f"No handler found for '{tag}#{func_name}'")
+        else:
+            content = handler_func(*args, **kwargs)
+            if content:
+                return JSONResponse(content=jsonable_encoder(content), status_code=200)
+    raise HTTPException(status_code=404, detail=f"failed to handle {tag}#{func_name}")

centra_sdk-0.1.0/src/centra_sdk/main.py

@@ -0,0 +1,45 @@
+# generated by fastapi-codegen:
+#   filename:  connector
+#   timestamp: 2025-07-31T14:14:32+00:00
+
+from __future__ import annotations
+
+from fastapi import FastAPI
+
+from .routers import (
+    agents,
+    control,
+    enforcement,
+    health,
+    info,
+    inventory,
+    logging,
+    onboarding,
+    operations,
+)
+
+app = FastAPI(
+    title='Version Handshake API',
+    description='Version API , Centra Integration Controller rely on to integrate with 3rd party platforms.\n\nSome useful links:\n- [Contracts OpenApi spec](https://github.com/guardicore/contracts)',
+    license={
+        'name': 'Apache 2.0',
+        'url': 'https://www.apache.org/licenses/LICENSE-2.0.html',
+    },
+    version='1.10.0',
+    servers=[{'url': 'http://127.0.0.1:8000'}],
+)
+
+app.include_router(agents.router)
+app.include_router(control.router)
+app.include_router(enforcement.router)
+app.include_router(health.router)
+app.include_router(info.router)
+app.include_router(inventory.router)
+app.include_router(logging.router)
+app.include_router(onboarding.router)
+app.include_router(operations.router)
+
+
+@app.get("/")
+async def root():
+    return {"message": "Gateway of the App"}

centra_sdk-0.1.0/src/centra_sdk/models/connector/__init__.py

@@ -0,0 +1,3 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00

centra_sdk-0.1.0/src/centra_sdk/models/connector/v1/__init__.py

@@ -0,0 +1,3 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00

centra_sdk-0.1.0/src/centra_sdk/models/connector/v1/common/__init__.py

@@ -0,0 +1,3 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00

centra_sdk-0.1.0/src/centra_sdk/models/connector/v1/common/common.py

@@ -0,0 +1,15 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+
+
+class Connector(Enum):
+    CLOUD_AZURE = 'cloud_azure'
+    CLOUD_AWS = 'cloud_aws'
+    PACKETFENCE = 'packetfence'
+    ALGOSEC = 'algosec'
+    PENSANDO = 'pensando'

centra_sdk-0.1.0/src/centra_sdk/models/connector/v1/common/inventory.py

@@ -0,0 +1,250 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import AwareDatetime, BaseModel, ConfigDict, Field
+
+from ..k8s.k8s_inventory import K8SData
+
+
+class ItemType(Enum):
+    ASSET = 'asset'
+
+
+class Label(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    key: str
+    value: str
+
+
+class Type(Enum):
+    VM = 'VM'
+
+
+class AzureNetworkTopology(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    subscription: Optional[str] = None
+    resource_group: Optional[str] = Field(None, alias='resource-group')
+    region: Optional[str] = None
+    resource_id: Optional[str] = Field(
+        None,
+        alias='resource-id',
+        examples=[
+            '/subscriptions/6974b4cf-6bf8-4454-8d22-a8f33084d699/resourceGroups/DEMO_SETUP/providers/Microsoft.Compute/virtualMachines/spoke1-vm'
+        ],
+    )
+    instance_count: Optional[int] = Field(None, alias='instance-count')
+
+
+class AWSNetworkTopology(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    account_id: Optional[str] = Field(None, alias='account-id')
+    resource_group: Optional[str] = Field(None, alias='resource-group')
+    region: Optional[str] = None
+    availability_zones: Optional[List[str]] = Field(
+        None, alias='availability-zones', examples=[['us-east-1a', 'us-east-1b']]
+    )
+    resource_arn: Optional[str] = Field(
+        None,
+        alias='resource-arn',
+        examples=[
+            'arn:aws:lambda:ap-south-1:905418309376:function:GcappOnboardingServicevnjto4fk7zaw'
+        ],
+    )
+
+
+class AzureNetworkAccess(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    public_access_enabled: Optional[bool] = Field(None, alias='public-access-enabled')
+    private_access_enabled: Optional[bool] = Field(None, alias='private-access-enabled')
+    service_endpoint_enabled: Optional[bool] = Field(
+        None, alias='service-endpoint-enabled'
+    )
+
+
+class AWSNetworkAccess(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    public_access_enabled: Optional[bool] = Field(None, alias='public-access-enabled')
+    private_access_enabled: Optional[bool] = Field(None, alias='private-access-enabled')
+
+
+class NetworkInterfaceData(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    id: Optional[str] = Field(
+        None, description='nic id, managed as resource id in Azure'
+    )
+    mac_address: str = Field(..., alias='mac-address')
+    network: str
+    subnet_id: str = Field(..., alias='subnet-id')
+    private_ip_addresses: Optional[List[str]] = Field(
+        None, alias='private-ip-addresses'
+    )
+    public_ip_addresses: Optional[List[str]] = Field(None, alias='public-ip-addresses')
+    vlan: Optional[int] = None
+    vrf: Optional[str] = None
+
+
+class OnlineStatus(Enum):
+    ONLINE = 'online'
+    OFFLINE = 'offline'
+
+
+class PowerState(Enum):
+    RUNNING = 'running'
+    STOPPED = 'stopped'
+    RESTARTING = 'restarting'
+
+
+class RegistrationStatus(Enum):
+    REGISTERED = 'registered'
+    UNREGISTERED = 'unregistered'
+
+
+class Type1(Enum):
+    MS = 'MS'
+
+
+class ManagedServiceData(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    type: Type1
+    nics: Optional[List[NetworkInterfaceData]] = None
+    network_topology: Optional[
+        Union[AzureNetworkTopology, AWSNetworkTopology, Dict[str, Any]]
+    ] = Field(None, alias='network-topology')
+    network_access: Optional[
+        Union[AzureNetworkAccess, AWSNetworkAccess, Dict[str, Any]]
+    ] = Field(None, alias='network-access')
+
+
+class APISec(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    datatypes_tags: Optional[List[str]] = Field([], alias='datatypes-tags')
+    hosts: Optional[List[str]] = []
+    internet_facing: Optional[List[str]] = Field([], alias='internet-facing')
+    authentication: Optional[List[str]] = []
+    max_risk_score: Optional[float] = Field(0.0, alias='max-risk-score')
+    infrastructure_tags: Optional[Dict[str, Any]] = Field(
+        {}, alias='infrastructure-tags'
+    )
+    api_tag: Optional[List[str]] = Field([], alias='api-tag')
+    api_groups: Optional[List[str]] = Field([], alias='api-groups')
+    nn_ur: Optional[str] = Field('', alias='nn-ur')
+
+
+class Type2(Enum):
+    IOT = 'IOT'
+
+
+class IOTRegistrationStatus(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    reg_status: Optional[RegistrationStatus] = Field(None, alias='reg-status')
+    reg_date: Optional[AwareDatetime] = Field(None, alias='reg-date')
+    unreg_date: Optional[AwareDatetime] = Field(None, alias='unreg-date')
+
+
+class HardwareInfo(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    hw_class: Optional[str] = Field('unknown', alias='hw-class')
+    hw_manufacturer: Optional[str] = Field('unknown', alias='hw-manufacturer')
+    hw_type: Optional[str] = Field('unknown', alias='hw-type')
+    bios_uuid: Optional[str] = Field('unknown', alias='bios-uuid')
+
+
+class VMData(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    type: Type
+    os_details: Optional[Dict[str, Any]] = Field(None, alias='os-details')
+    hw_details: Optional[HardwareInfo] = Field(None, alias='hw-details')
+    nics: Optional[List[NetworkInterfaceData]] = None
+    network_topology: Optional[
+        Union[AzureNetworkTopology, AWSNetworkTopology, Dict[str, Any]]
+    ] = Field(None, alias='network-topology')
+    power_state: Optional[PowerState] = Field(None, alias='power-state')
+    api_sec: Optional[APISec] = Field(None, alias='api-sec')
+
+
+class IOT(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    type: Type2
+    os_details: Optional[Dict[str, Any]] = Field(None, alias='os-details')
+    hw_details: HardwareInfo = Field(..., alias='hw-details')
+    nics: Optional[List[NetworkInterfaceData]] = None
+    power_state: Optional[PowerState] = Field(None, alias='power-state')
+    online_status: Optional[OnlineStatus] = Field(None, alias='online-status')
+    registration_status: Optional[IOTRegistrationStatus] = Field(
+        None, alias='registration-status'
+    )
+    device_score: Optional[str] = Field(None, alias='device-score')
+
+
+class InventoryItem(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    item_type: Optional[ItemType] = Field(
+        'asset',
+        alias='item-type',
+        description='inventory item type - currently asset only',
+    )
+    item_id: Optional[str] = Field(
+        None, alias='item-id', description='inventory item id assigned by Centra'
+    )
+    external_ids: Optional[List[str]] = Field(
+        None,
+        alias='external-ids',
+        description='list of external IDs that are managed by external asset management systems such as Azure resource management',
+    )
+    entity_type: Optional[str] = Field(
+        None,
+        alias='entity-type',
+        description='asset type such as Virtual Machine or Azure SQL Server',
+    )
+    entity_category: Optional[str] = Field(
+        None,
+        alias='entity-category',
+        description='asset category such as Compute or Database',
+    )
+    entity_name: Optional[str] = Field(
+        None, alias='entity-name', description='asset name'
+    )
+    entity_data: Optional[
+        Union[VMData, IOT, ManagedServiceData, K8SData, Dict[str, Any]]
+    ] = Field(
+        None,
+        alias='entity-data',
+        description='additional entity data depending on its type',
+        union_mode='left_to_right',
+    )
+    labels: Optional[List[Label]] = None
+    metadata_attributes: Optional[Dict[str, Any]] = Field(
+        None, alias='metadata-attributes'
+    )

centra_sdk-0.1.0/src/centra_sdk/models/connector/v1/consumer/__init__.py

@@ -0,0 +1,3 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00

centra_sdk-0.1.0/src/centra_sdk/models/connector/v1/consumer/agents.py

@@ -0,0 +1,21 @@
+# generated by fastapi-codegen:
+#   filename:  /home/ivasylen/sdk_project/contracts/guardicore/connector
+#   timestamp: 2025-07-31T14:14:32+00:00
+
+from __future__ import annotations
+
+from typing import List, Optional
+
+from pydantic import BaseModel, ConfigDict, Field
+
+
+class AgentsInfo(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    mac_addresses: Optional[List[str]] = Field(
+        None,
+        alias='mac-addresses',
+        description='Known agents MAC addresses without separator',
+        examples=[['000000000000', '112233445566']],
+    )