mirror of
https://github.com/FuzzingLabs/fuzzforge_ai.git
synced 2026-06-09 07:33:56 +02:00
AFredefon
115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
"""FuzzForge Runner settings configuration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from enum import StrEnum
|
|
from pathlib import Path
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
|
|
class EngineType(StrEnum):
|
|
"""Supported container engine types."""
|
|
|
|
DOCKER = "docker"
|
|
PODMAN = "podman"
|
|
|
|
|
|
class EngineSettings(BaseModel):
|
|
"""Container engine configuration."""
|
|
|
|
#: Type of container engine to use.
|
|
type: EngineType = EngineType.PODMAN
|
|
|
|
#: Path to the container engine socket (only used as fallback).
|
|
socket: str = Field(default="")
|
|
|
|
#: Custom graph root for container storage (isolated from system).
|
|
#: When set, uses CLI mode instead of socket for better portability.
|
|
graphroot: Path = Field(default=Path.home() / ".fuzzforge" / "containers" / "storage")
|
|
|
|
#: Custom run root for container runtime state.
|
|
runroot: Path = Field(default=Path.home() / ".fuzzforge" / "containers" / "run")
|
|
|
|
|
|
class StorageSettings(BaseModel):
|
|
"""Storage configuration for local or S3 storage."""
|
|
|
|
#: Storage backend type.
|
|
type: Literal["local", "s3"] = "local"
|
|
|
|
#: Base path for local storage (used when type is "local").
|
|
path: Path = Field(default=Path.home() / ".fuzzforge" / "storage")
|
|
|
|
#: S3 endpoint URL (used when type is "s3").
|
|
s3_endpoint: str | None = None
|
|
|
|
#: S3 access key (used when type is "s3").
|
|
s3_access_key: str | None = None
|
|
|
|
#: S3 secret key (used when type is "s3").
|
|
s3_secret_key: str | None = None
|
|
|
|
|
|
class ProjectSettings(BaseModel):
|
|
"""Project configuration."""
|
|
|
|
#: Default path for FuzzForge projects.
|
|
default_path: Path = Field(default=Path.home() / ".fuzzforge" / "projects")
|
|
|
|
|
|
class RegistrySettings(BaseModel):
|
|
"""Container registry configuration for module images."""
|
|
|
|
#: Registry URL for pulling module images.
|
|
url: str = Field(default="ghcr.io/fuzzinglabs")
|
|
|
|
#: Default tag to use when pulling images.
|
|
default_tag: str = Field(default="latest")
|
|
|
|
#: Registry username for authentication (optional).
|
|
username: str | None = None
|
|
|
|
#: Registry password/token for authentication (optional).
|
|
password: str | None = None
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
"""FuzzForge Runner settings.
|
|
|
|
Settings can be configured via environment variables with the prefix
|
|
``FUZZFORGE_``. Nested settings use underscore as delimiter.
|
|
|
|
Example:
|
|
``FUZZFORGE_ENGINE_TYPE=docker``
|
|
``FUZZFORGE_STORAGE_PATH=/data/fuzzforge``
|
|
``FUZZFORGE_MODULES_PATH=/path/to/modules``
|
|
|
|
"""
|
|
|
|
model_config = SettingsConfigDict(
|
|
case_sensitive=False,
|
|
env_nested_delimiter="__",
|
|
env_prefix="FUZZFORGE_",
|
|
)
|
|
|
|
#: Container engine settings.
|
|
engine: EngineSettings = Field(default_factory=EngineSettings)
|
|
|
|
#: Storage settings.
|
|
storage: StorageSettings = Field(default_factory=StorageSettings)
|
|
|
|
#: Project settings.
|
|
project: ProjectSettings = Field(default_factory=ProjectSettings)
|
|
|
|
#: Container registry settings.
|
|
registry: RegistrySettings = Field(default_factory=RegistrySettings)
|
|
|
|
#: Path to modules directory (for development/local builds).
|
|
modules_path: Path = Field(default=Path.home() / ".fuzzforge" / "modules")
|
|
|
|
#: Enable debug logging.
|
|
debug: bool = False
|