dlab-cli 0.1.0__py3-none-any.whl

dlab/config.py

@@ -0,0 +1,190 @@
+"""
+Configuration loading and validation for decision-pack config directories.
+"""
+
+from pathlib import Path
+from typing import Any
+
+import yaml
+
+
+REQUIRED_DIRS: list[str] = ["docker", "opencode"]
+REQUIRED_FILES: list[str] = ["config.yaml"]
+CONFIG_KEYS: list[str] = ["name", "description", "docker_image_name", "default_model"]
+
+
+def list_config_issues(config_dir: str) -> list[str]:
+    """
+    Check a decision-pack directory and return a list of issues found.
+
+    Parameters
+    ----------
+    config_dir : str
+        Path to the decision-pack config directory.
+
+    Returns
+    -------
+    list[str]
+        List of issue descriptions. Empty if valid.
+    """
+    issues: list[str] = []
+    config_path: Path = Path(config_dir)
+
+    if not config_path.exists():
+        return [f"Directory does not exist: {config_dir}"]
+    if not config_path.is_dir():
+        return [f"Path is not a directory: {config_dir}"]
+
+    for required_dir in REQUIRED_DIRS:
+        dir_path: Path = config_path / required_dir
+        if not dir_path.exists():
+            issues.append(f"Missing directory: {required_dir}/")
+        elif not dir_path.is_dir():
+            issues.append(f"Expected directory but found file: {required_dir}")
+
+    for required_file in REQUIRED_FILES:
+        file_path: Path = config_path / required_file
+        if not file_path.exists():
+            issues.append(f"Missing file: {required_file}")
+        elif not file_path.is_file():
+            issues.append(f"Expected file but found directory: {required_file}")
+
+    return issues
+
+
+def validate_config_structure(config_dir: str) -> None:
+    """
+    Validate that a decision-pack config directory has the required structure.
+
+    Parameters
+    ----------
+    config_dir : str
+        Path to the decision-pack config directory.
+
+    Raises
+    ------
+    ValueError
+        If the directory structure is invalid.
+    """
+    config_path: Path = Path(config_dir)
+
+    if not config_path.exists():
+        raise ValueError(f"Config directory does not exist: {config_dir}")
+
+    if not config_path.is_dir():
+        raise ValueError(f"Config path is not a directory: {config_dir}")
+
+    for required_dir in REQUIRED_DIRS:
+        dir_path: Path = config_path / required_dir
+        if not dir_path.exists():
+            raise ValueError(f"Missing required directory: {required_dir}")
+        if not dir_path.is_dir():
+            raise ValueError(f"Expected directory but found file: {required_dir}")
+
+    for required_file in REQUIRED_FILES:
+        file_path: Path = config_path / required_file
+        if not file_path.exists():
+            raise ValueError(f"Missing required file: {required_file}")
+        if not file_path.is_file():
+            raise ValueError(f"Expected file but found directory: {required_file}")
+
+
+def load_config_yaml(config_dir: str) -> dict[str, Any]:
+    """
+    Load and validate config.yaml from a decision-pack config directory.
+
+    Parameters
+    ----------
+    config_dir : str
+        Path to the decision-pack config directory.
+
+    Returns
+    -------
+    dict[str, Any]
+        The parsed config.yaml contents.
+
+    Raises
+    ------
+    ValueError
+        If config.yaml is invalid or missing required keys.
+    """
+    config_path: Path = Path(config_dir) / "config.yaml"
+
+    try:
+        with open(config_path, "r") as f:
+            config: dict[str, Any] = yaml.safe_load(f)
+    except yaml.YAMLError as e:
+        raise ValueError(f"Invalid YAML in config.yaml: {e}")
+
+    if not isinstance(config, dict):
+        raise ValueError("config.yaml must contain a YAML mapping")
+
+    missing_keys: list[str] = [key for key in CONFIG_KEYS if key not in config]
+    if missing_keys:
+        raise ValueError(f"config.yaml missing required keys: {missing_keys}")
+
+    return config
+
+
+def load_dpack_config(config_dir: str) -> dict[str, Any]:
+    """
+    Load and validate a complete decision-pack configuration.
+
+    Parameters
+    ----------
+    config_dir : str
+        Path to the decision-pack config directory.
+
+    Returns
+    -------
+    dict[str, Any]
+        Complete decision-pack configuration including:
+        - config_dir: Absolute path to config directory
+        - name: decision-pack name
+        - description: decision-pack description
+        - docker_image_name: Name for the Docker image
+        - default_model: Default LLM model to use
+        - opencode_version: Version of opencode to install (optional, defaults to "latest")
+
+    Raises
+    ------
+    ValueError
+        If the configuration is invalid.
+    """
+    config_path: Path = Path(config_dir).resolve()
+    config_dir_str: str = str(config_path)
+
+    validate_config_structure(config_dir_str)
+    config: dict[str, Any] = load_config_yaml(config_dir_str)
+
+    config["config_dir"] = config_dir_str
+
+    # Autodetect package_manager from docker/ contents if not specified
+    if "package_manager" not in config:
+        docker_dir: Path = config_path / "docker"
+        if (docker_dir / "environment.yml").exists():
+            config["package_manager"] = "conda"
+        elif (docker_dir / "pixi.toml").exists():
+            config["package_manager"] = "pixi"
+        else:
+            config["package_manager"] = "pip"
+
+    # Default opencode_version to "latest" if not specified
+    if "opencode_version" not in config:
+        config["opencode_version"] = "latest"
+
+    # Normalize hooks: string -> list, missing -> empty list
+    hooks: dict[str, Any] = config.get("hooks", {})
+    if not isinstance(hooks, dict):
+        hooks = {}
+    for key in ("pre-run", "post-run"):
+        value: Any = hooks.get(key, [])
+        if isinstance(value, str):
+            hooks[key] = [value]
+        elif isinstance(value, list):
+            hooks[key] = value
+        else:
+            hooks[key] = []
+    config["hooks"] = hooks
+
+    return config