splight-cli 5.4.1__tar.gz → 5.5.0__tar.gz

{splight_cli-5.4.1 → splight_cli-5.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: splight-cli
-Version: 5.4.1
+Version: 5.5.0
 Summary: Splight Command Line Interface
 Author: Splight Dev
 Author-email: dev@splight-ae.com
@@ -43,8 +43,6 @@ Description-Content-Type: text/markdown
   - [Commands](#commands)
     - [Component](#component)
     - [Configure](#configure)
-    - [Hub](#hub)
-      - [Component](#component-1)
     - [Engine](#engine)
     - [Workspace](#workspace)
   - [Developing Components](#developing-components)
@@ -231,11 +229,6 @@ In the same way you can modify one parameter using the `set` command
 splight configure set <parameter> <value>
 ```
 
-### Hub
-
-This command allows you to interact with the Splight HUB, places where you can find all the
-existing components that can be used in the platform.
-
 #### Component
 
 - List component

{splight_cli-5.4.1 → splight_cli-5.5.0}/README.md

@@ -15,8 +15,6 @@
   - [Commands](#commands)
     - [Component](#component)
     - [Configure](#configure)
-    - [Hub](#hub)
-      - [Component](#component-1)
     - [Engine](#engine)
     - [Workspace](#workspace)
   - [Developing Components](#developing-components)
@@ -203,11 +201,6 @@ In the same way you can modify one parameter using the `set` command
 splight configure set <parameter> <value>
 ```
 
-### Hub
-
-This command allows you to interact with the Splight HUB, places where you can find all the
-existing components that can be used in the platform.
-
 #### Component
 
 - List component

{splight_cli-5.4.1 → splight_cli-5.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "splight-cli"
-version = "5.4.1"
+version = "5.5.0"
 description = "Splight Command Line Interface"
 authors = [
     {name = "Splight Dev",email = "dev@splight-ae.com"}

{splight_cli-5.4.1 → splight_cli-5.5.0}/splight_cli/cli.py

@@ -7,7 +7,6 @@ from splight_lib.settings import workspace_settings
 from splight_cli.component import component_app
 from splight_cli.config import config_app
 from splight_cli.engine import engine_app
-from splight_cli.hub import hub_app
 from splight_cli.version import __version__
 from splight_cli.workspace import workspace_app
 
@@ -24,7 +23,6 @@ app = typer.Typer(
 app.add_typer(config_app, name="configure")
 app.add_typer(component_app, name="component")
 app.add_typer(engine_app, name="engine")
-app.add_typer(hub_app, name="hub")
 app.add_typer(workspace_app, name="workspace")
 
 

{splight_cli-5.4.1 → splight_cli-5.5.0}/splight_cli/engine/manager/manager.py

@@ -34,7 +34,6 @@ from splight_cli.engine.manager.exceptions import (
     UpdateParametersError,
     VersionUpdateError,
 )
-from splight_cli.hub.component.exceptions import HubComponentNotFound
 from splight_cli.utils.input import prompt_data_address_value, prompt_param
 
 SplightModel = Type[SplightDatabaseBaseModel]

splight_cli-5.4.1/splight_cli/hub/__init__.py

@@ -1,14 +0,0 @@
-import typer
-
-from splight_cli.hub.component import component_app
-from splight_cli.hub.server import server_app
-
-hub_app = typer.Typer(
-    name="Splight Hub",
-    add_completion=True,
-    rich_markup_mode="rich",
-    no_args_is_help=True,
-)
-
-hub_app.add_typer(component_app, name="component")
-hub_app.add_typer(server_app, name="server")

splight_cli-5.4.1/splight_cli/hub/component/__init__.py

@@ -1,73 +0,0 @@
-import typer
-from rich.console import Console
-from splight_lib.models import Component
-
-from splight_cli.constants import error_style
-from splight_cli.hub.component.hub_manager import HubComponentManager
-
-component_app = typer.Typer(
-    name="Splight Hub Component",
-    add_completion=True,
-    rich_markup_mode="rich",
-    no_args_is_help=True,
-)
-
-console = Console()
-MODEL = Component
-
-
-@component_app.command()
-def push(
-    ctx: typer.Context,
-    path: str = typer.Argument(..., help="Path to component source code"),
-    force: bool = typer.Option(
-        False,
-        "--force",
-        "-f",
-        help="Overwrite existing component in Splight HUB",
-    ),
-):
-    try:
-        manager = HubComponentManager()
-        manager.push(path, force=force)
-    except Exception as exc:
-        console.print(f"Error pushing component: {exc}", style=error_style)
-        raise typer.Exit(code=1)
-
-
-@component_app.command()
-def pull(
-    ctx: typer.Context,
-    name: str = typer.Argument(..., help="The component's name"),
-    version: str = typer.Argument(..., help="The component's version"),
-):
-    try:
-        manager = HubComponentManager()
-        manager.pull(name=name, version=version)
-    except Exception as exc:
-        console.print(f"Error pulling component: {exc}", style=error_style)
-        raise typer.Exit(code=1)
-
-
-@component_app.command()
-def list(ctx: typer.Context):
-    try:
-        manager = HubComponentManager()
-        manager.list_components()
-    except Exception as exc:
-        console.print(f"Error listing components: {exc}", style=error_style)
-        raise typer.Exit(code=1)
-
-
-@component_app.command()
-def versions(
-    ctx: typer.Context, name: str = typer.Argument(..., help="Componet's name")
-):
-    try:
-        manager = HubComponentManager()
-        manager.versions(name=name)
-    except Exception as exc:
-        console.print(
-            f"Error getting component versions: {exc}", style=error_style
-        )
-        raise typer.Exit(code=1)

splight_cli-5.4.1/splight_cli/hub/component/component_builder.py

@@ -1,109 +0,0 @@
-import gzip
-import os
-import re
-import shutil
-import sys
-
-import docker
-
-from splight_cli.hub.component.exceptions import BuildError
-
-
-class ComponentBuilder:
-    _RESOURCES_PATH = "resources"
-    _RESOURCES = [
-        "Dockerfile",
-        "installer.py",
-    ]
-
-    def __init__(self, component_spec: dict, component_path: str):
-        self._base_path = os.path.dirname(__file__)
-        self._component_name = component_spec["name"]
-        self._component_version = component_spec["version"]
-        self._spl_pkg = (
-            "splight-lib"
-            if component_spec["splight_lib_version"]
-            else "splight-cli"
-        )
-        self._spl_pkg_version = (
-            component_spec["splight_lib_version"]
-            if component_spec["splight_lib_version"]
-            else component_spec["splight_cli_version"]
-        )
-        self._component_path = component_path
-        self._docker_client = docker.from_env()
-
-    def build(self) -> str:
-        original_path = os.getcwd()
-        self._copy_resources()
-        os.chdir(self._component_path)
-        image_file = self._build_component()
-        self._clean_up()
-        os.chdir(original_path)
-        return image_file
-
-    def _build_component(self) -> str:
-        full_name = f"{self._component_name}-{self._component_version}"
-        tag = f"component:{full_name}"
-        streamer = self._build_component_image(tag)
-        for chunk in streamer:
-            try:
-                log = self._parse_chunk(chunk)
-            except BuildError as exc:
-                sys.stdout.write(exc)
-                sys.stdout.flush()
-                raise exc
-            if log:
-                sys.stdout.write(f"{log}\n")
-                sys.stdout.flush()
-
-            if "stream" in chunk:
-                match = re.search(
-                    r"(^Successfully built |sha256:)([0-9a-f]+)$",
-                    chunk["stream"],
-                )
-                if match:
-                    image_id = match.group(2)
-        image = self._docker_client.images.get(image_id)
-        file_name = f"{full_name}.tgz"
-        with gzip.open(file_name, "wb") as image_file:
-            for chunk in image.save(named=True):
-                image_file.write(chunk)
-        return file_name
-
-    def _build_component_image(self, tag: str):
-        client = self._docker_client.api
-        streamer = client.build(
-            path=".",
-            tag=tag,
-            dockerfile="Dockerfile",
-            rm=True,
-            pull=True,
-            quiet=False,
-            buildargs={
-                "COMPONENT_NAME": self._component_name,
-                "COMPONENT_VERSION": self._component_version,
-                "SPL_PACKAGE": self._spl_pkg,
-                "SPL_PACKAGE_VERSION": self._spl_pkg_version,
-            },
-            decode=True,
-        )
-        return streamer
-
-    def _copy_resources(self):
-        for resource in self._RESOURCES:
-            resource_path = os.path.join(
-                self._base_path, self._RESOURCES_PATH, resource
-            )
-            shutil.copy(resource_path, self._component_path)
-
-    def _clean_up(self):
-        for resource in self._RESOURCES:
-            os.remove(resource)
-
-    def _parse_chunk(self, chunk: dict) -> str:
-        if "errorDetail" in chunk:
-            message = chunk["error"]
-            raise BuildError(message)
-        log = "".join([str(item) for item in chunk.values()])
-        return log.strip("\n") if log != "\n" else ""

splight_cli-5.4.1/splight_cli/hub/component/exceptions.py

@@ -1,28 +0,0 @@
-class ComponentAlreadyExists(Exception):
-    def __init__(self, name: str, version: str):
-        self._msg = f"Component {name}-{version} already exists in HUB"
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class ComponentDirectoryAlreadyExists(Exception):
-    def __init__(self, directory: str):
-        self._msg = f"Directory with name {directory} already exists in path"
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class HubComponentNotFound(Exception):
-    """Exception raised when a version is invalid."""
-
-    def __init__(self, name: str, version: str):
-        self._msg = f"Hub component {name} version {version} not found."
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class BuildError(Exception):
-    pass

splight_cli-5.4.1/splight_cli/hub/component/hub_manager.py

@@ -1,137 +0,0 @@
-import json
-import os
-import shutil
-from typing import Optional
-
-import pathspec
-import py7zr
-from pydantic import ValidationError
-from rich.console import Console
-from rich.table import Table
-from splight_lib.models import HubComponent
-
-from splight_cli.constants import (
-    COMPRESSION_TYPE,
-    SPEC_FILE,
-    SPLIGHT_IGNORE,
-    success_style,
-)
-from splight_cli.hub.component.component_builder import ComponentBuilder
-from splight_cli.hub.component.exceptions import (
-    ComponentAlreadyExists,
-    ComponentDirectoryAlreadyExists,
-    HubComponentNotFound,
-)
-from splight_cli.hub.exceptions import SpecFormatError, SpecValidationError
-from splight_cli.utils.loader import Loader
-
-console = Console()
-
-
-class HubComponentManager:
-    def push(self, path: str, force: Optional[bool] = False):
-        try:
-            with open(os.path.join(path, SPEC_FILE)) as fid:
-                spec = json.load(fid)
-        except Exception as exc:
-            raise SpecFormatError(exc)
-
-        # Validate spec fields before pushing the model
-        try:
-            component = HubComponent.model_validate(spec)
-        except ValidationError as exc:
-            raise SpecValidationError(exc)
-
-        name = spec["name"]
-        version = spec["version"]
-
-        if not force and self._exists_in_hub(name, version):
-            raise ComponentAlreadyExists(name, version)
-
-        builder = ComponentBuilder(spec, path)
-        image_file = builder.build()
-
-        with Loader("Pushing Component to Splight Hub"):
-            component = HubComponent.upload(path, image_file)
-
-        console.print(
-            f"Component {component.id} pushed succesfully", style=success_style
-        )
-
-    def pull(self, name: str, version: str):
-        with Loader("Pulling component from Splight Hub"):
-            self._pull_component(name, version)
-        console.print(
-            f"Component {name} pulled succesfully", style=success_style
-        )
-
-    def _pull_component(self, name: str, version: str):
-        components = HubComponent.list_mine(name=name, version=version)
-        if not components:
-            raise HubComponentNotFound(name, version)
-
-        component_file_wrapper = components[0].download()
-        component_data = component_file_wrapper.read()
-
-        # TODO: search for a better approach
-        version_modified = version.replace(".", "_")
-        component_path = f"{name}/{version_modified}"
-        versioned_name = f"{name}-{version}"
-        file_name = f"{versioned_name}.{COMPRESSION_TYPE}"
-        if os.path.exists(component_path):
-            raise ComponentDirectoryAlreadyExists(component_path)
-
-        try:
-            with open(file_name, "wb") as fid:
-                fid.write(component_data)
-
-            with py7zr.SevenZipFile(file_name, "r") as z:
-                z.extractall(path=".")
-            shutil.move(f"{versioned_name}", component_path)
-
-        except Exception as exc:
-            raise exc
-        finally:
-            if os.path.exists(file_name):
-                os.remove(file_name)
-
-    def list_components(self):
-        components = HubComponent.list_mine(limit_=10000)
-        names = set([component.name for component in components])
-        table = Table("Name")
-        [table.add_row(name) for name in names]
-        console.print(table)
-
-    def versions(self, name: str):
-        components = HubComponent.list_mine(name=name)
-        table = Table("Name", "Version", "Verification", "Privacy Policy")
-        for item in components:
-            table.add_row(
-                name, item.version, item.verification, item.privacy_policy
-            )
-        console.print(table)
-
-    def _get_ignore_pathspec(self, path):
-        try:
-            with open(
-                os.path.join(path, SPLIGHT_IGNORE), "r"
-            ) as splightignore:
-                return pathspec.PathSpec.from_lines(
-                    "gitwildmatch", splightignore
-                )
-        except FileNotFoundError:
-            return None
-
-    def _get_component(self, name: str, version: str):
-        components = HubComponent.list_all(name=name, version=version)
-        return components
-
-    def _exists_in_hub(self, name: str, version: str) -> bool:
-        components = self._get_component(name, version)
-        return len(components) > 0
-
-    def fetch_component_version(self, name: str, version: str):
-        components = self._get_component(name, version)
-        if not components:
-            raise HubComponentNotFound(name, version)
-        return components[0]

splight_cli-5.4.1/splight_cli/hub/component/resources/Dockerfile

@@ -1,21 +0,0 @@
-# syntax=docker/dockerfile:1
-FROM --platform=linux/amd64 python:3.11.7-slim
-
-WORKDIR /whl
-RUN pip install --upgrade splight-runner
-
-ARG SPL_PACKAGE
-ARG SPL_PACKAGE_VERSION
-RUN pip install ${SPL_PACKAGE}==${SPL_PACKAGE_VERSION}
-
-ARG COMPONENT_NAME
-ARG COMPONENT_VERSION
-
-WORKDIR /app
-COPY . .
-RUN python installer.py --name ${COMPONENT_NAME} --version ${COMPONENT_VERSION}
-
-# Cleaning up files
-RUN rm installer.py Dockerfile
-
-ENTRYPOINT ["splight-runner", "run-component"]

splight_cli-5.4.1/splight_cli/hub/component/resources/installer.py

@@ -1,80 +0,0 @@
-import argparse
-import os
-import subprocess
-from typing import List
-
-
-class HubComponentInstaller:
-    _INIT_FILE = "Initialization"
-
-    def __init__(self, name: str, version: str) -> None:
-        self._name = name
-        self._version = version
-        self._component_path = os.getenv("APP_DIR", "/app")
-
-    def _handle_run(self, command: List[str]) -> None:
-        command: str = " ".join(command)
-        try:
-            subprocess.run(
-                command, check=True, cwd=self._component_path, shell=True
-            )
-        except subprocess.CalledProcessError as e:
-            raise Exception(
-                f"Failed to run command: {e.cmd}. Output: {e.output}"
-            )
-
-    def _validate(self):
-        # VALIDATE INIT FILE EXISTS
-        if not os.path.isfile(
-            os.path.join(self._component_path, self._INIT_FILE)
-        ):
-            raise Exception(
-                f"{self._INIT_FILE} file is missing in {self._component_path}"
-            )
-
-    def _load_commands(self) -> List[List[str]]:
-        file = os.path.join(self._component_path, self._INIT_FILE)
-        lines: List[List[str]] = []
-        lines.append(["RUN", "pip", "cache", "purge"])
-        with open(file) as f:
-            for line in f:
-                line = line.strip()
-                if line.startswith("#") or line == "":
-                    continue
-                lines.append(line.split(" "))
-        return lines
-
-    def install(self):
-        self._validate()
-        commands = self._load_commands()
-        for command in commands:
-            prefix: str = command[0].lower()
-            handler = getattr(self, f"_handle_{prefix}", None)
-            if not handler:
-                raise Exception(f"Invalid command: {prefix}")
-            handler(command[1:])
-
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Install Hub Component")
-    parser.add_argument(
-        "-n",
-        "--name",
-        type=str,
-        nargs=1,
-        help="Hub Component Name",
-        required=True,
-    )
-    parser.add_argument(
-        "-v",
-        "--version",
-        type=str,
-        nargs=1,
-        help="Hub Component Version",
-        required=True,
-    )
-
-    args = parser.parse_args()
-
-    manager = HubComponentInstaller(name=args.name[0], version=args.version[0])
-    manager.install()

splight_cli-5.4.1/splight_cli/hub/component/tests/test_hub_manager.py

@@ -1,135 +0,0 @@
-import os
-from unittest.mock import mock_open, patch
-from uuid import uuid4
-
-import pytest
-from splight_lib.models import HubComponent
-
-from splight_cli.component import ComponentManager
-from splight_cli.hub.component.component_builder import ComponentBuilder
-from splight_cli.hub.component.exceptions import (
-    ComponentAlreadyExists,
-    HubComponentNotFound,
-)
-from splight_cli.hub.component.hub_manager import HubComponentManager
-from splight_cli.hub.exceptions import SpecFormatError, SpecValidationError
-
-BASE_PATH = os.getcwd()
-TEST_COMPONENT_PATH = os.path.join(
-    BASE_PATH, "splight_cli/tests/test_component/"
-)
-with open(os.path.join(TEST_COMPONENT_PATH, "spec.json"), "r") as fid:
-    HUB_COMPONENT = HubComponent.model_validate_json(fid.read())
-HUB_COMPONENT.id = str(uuid4())
-
-
-@patch.object(ComponentBuilder, "build", return_value="image_file")
-@patch.object(ComponentManager, "test", return_value=None)
-@patch.object(HubComponentManager, "_exists_in_hub", return_value=False)
-def test_push(mock_exists, mock_component_manager, mock_builder):
-    manager = HubComponentManager()
-    test_component_path = os.path.join(
-        os.getcwd(), "splight_cli/tests/test_component/"
-    )
-    with patch.object(
-        HubComponent, "upload", return_value=HUB_COMPONENT
-    ) as mock_hub:
-        manager.push(test_component_path)
-        mock_builder.assert_called()
-        mock_hub.assert_called_with(test_component_path, "image_file")
-
-
-@patch.object(ComponentManager, "test", return_value=None)
-@patch(
-    "splight_cli.hub.component.hub_manager.json.load", side_effect=Exception
-)
-def test_push_spec_format_error(mock_exists, mock_json_load):
-    manager = HubComponentManager()
-    test_component_path = os.path.join(
-        os.getcwd(), "splight_cli/tests/test_component/"
-    )
-    with pytest.raises(SpecFormatError):
-        manager.push(test_component_path, force=False)
-
-
-@patch.object(ComponentManager, "test", return_value=None)
-@patch(
-    "splight_cli.hub.component.hub_manager.json.load",
-    return_value={"name": "mycomponent", "splight_lib_version": "1.1.0"},
-)
-def test_push_spec_missing_keys(mock_exists, mock_json_load):
-    manager = HubComponentManager()
-    test_component_path = os.path.join(
-        os.getcwd(), "splight_cli/tests/test_component/"
-    )
-    with pytest.raises(SpecValidationError):
-        manager.push(test_component_path, force=False)
-
-
-@patch.object(ComponentManager, "test", return_value=None)
-@patch(
-    "splight_cli.hub.component.hub_manager.json.load",
-    return_value={
-        "name": "mycomponent",
-        "version": "0.1.0",
-        "splight_lib_version": "3.0.0",
-        "input": "wrong_input_type",
-    },
-)
-def test_push_spec_wrong_attr_type(mock_exists, mock_json_load):
-    manager = HubComponentManager()
-    test_component_path = os.path.join(
-        os.getcwd(), "splight_cli/tests/test_component/"
-    )
-    with pytest.raises(SpecValidationError):
-        manager.push(test_component_path, force=False)
-
-
-@patch.object(ComponentManager, "test", return_value=None)
-@patch.object(HubComponentManager, "_exists_in_hub", return_value=True)
-def test_push_already_exists(mock_exists, mock_component_manager):
-    manager = HubComponentManager()
-    test_component_path = os.path.join(
-        os.getcwd(), "splight_cli/tests/test_component/"
-    )
-    with pytest.raises(ComponentAlreadyExists):
-        manager.push(test_component_path, force=False)
-
-
-@patch.object(ComponentBuilder, "build", return_value="image_file")
-@patch.object(ComponentManager, "test", return_value=None)
-@patch.object(HubComponentManager, "_exists_in_hub", return_value=True)
-def test_push_already_exists_with_force(
-    mock_exists, mock_component_manager, mock_builder
-):
-    manager = HubComponentManager()
-    test_component_path = os.path.join(
-        os.getcwd(), "splight_cli/tests/test_component/"
-    )
-    with patch.object(
-        HubComponent, "upload", return_value=HUB_COMPONENT
-    ) as mock_hub:
-        manager.push(test_component_path, force=True)
-        mock_builder.assert_called()
-        mock_hub.assert_called_with(test_component_path, "image_file")
-
-
-@patch.object(HubComponent, "list_mine", return_value=[HUB_COMPONENT])
-@patch.object(
-    HubComponent,
-    "download",
-    return_value=open(os.path.join(TEST_COMPONENT_PATH, "spec.json"), "r"),
-)
-@patch("builtins.open", new_callable=mock_open())
-@patch("py7zr.SevenZipFile")
-@patch("shutil.move", return_value=None)
-def test_pull(mock_move, mock_zip, mock_open, mock_download, mock_list):
-    manager = HubComponentManager()
-    manager.pull(name="TestComponent", version="1.1.0")
-
-
-@patch.object(HubComponent, "list_mine", return_value=[])
-def test_pull_not_found(mock_list):
-    manager = HubComponentManager()
-    with pytest.raises(HubComponentNotFound):
-        manager._pull_component(name="TestComponent", version="1.1.0")

splight_cli-5.4.1/splight_cli/hub/exceptions.py

@@ -1,21 +0,0 @@
-from json import loads
-
-
-class SpecFormatError(Exception):
-    def __init__(self, error):
-        self._msg = f"Error while parsing 'spec.json': {error}"
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class SpecValidationError(Exception):
-    def __init__(self, error):
-        field = loads(error.json())[0]["loc"][0]
-        message = loads(error.json())[0]["msg"]
-        self._msg = (
-            f"Validation error for field in 'spec.json': '{field}' {message}"
-        )
-
-    def __str__(self) -> str:
-        return self._msg

splight_cli-5.4.1/splight_cli/hub/server/__init__.py

@@ -1,60 +0,0 @@
-import typer
-from rich.console import Console
-from splight_lib.models import Component
-
-from splight_cli.constants import error_style
-from splight_cli.hub.server.server_manager import HubServerManager
-
-server_app = typer.Typer(
-    name="Splight Hub Server",
-    add_completion=True,
-    rich_markup_mode="rich",
-    no_args_is_help=True,
-)
-
-console = Console()
-MODEL = Component
-
-
-@server_app.command()
-def push(
-    ctx: typer.Context,
-    path: str = typer.Argument(..., help="Path to server files"),
-    force: bool = typer.Option(
-        False,
-        "--force",
-        "-f",
-        help="Overwrite existing server in Splight HUB",
-    ),
-):
-    try:
-        manager = HubServerManager()
-        manager.push(path, force=force)
-    except Exception as exc:
-        console.print(f"Error pushing server: {exc}", style=error_style)
-        raise
-        raise typer.Exit(code=1)
-
-
-@server_app.command()
-def pull(
-    ctx: typer.Context,
-    name: str = typer.Argument(..., help="The servers's name"),
-    version: str = typer.Argument(..., help="The servers's version"),
-):
-    try:
-        manager = HubServerManager()
-        manager.pull(name=name, version=version)
-    except Exception as exc:
-        console.print(f"Error pulling server: {exc}", style=error_style)
-        raise typer.Exit(code=1)
-
-
-@server_app.command()
-def list(ctx: typer.Context):
-    try:
-        manager = HubServerManager()
-        manager.list_servers()
-    except Exception as exc:
-        console.print(f"Error listing servers: {exc}", style=error_style)
-        raise typer.Exit(code=1)

splight_cli-5.4.1/splight_cli/hub/server/exceptions.py

@@ -1,28 +0,0 @@
-class ServerAlreadyExists(Exception):
-    def __init__(self, name: str, version: str):
-        self._msg = f"Server {name}-{version} already exists in HUB"
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class ServerDirectoryAlreadyExists(Exception):
-    def __init__(self, directory: str):
-        self._msg = f"Directory with name {directory} already exists in path"
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class HubServerNotFound(Exception):
-    """Exception raised when a version is invalid."""
-
-    def __init__(self, name: str, version: str):
-        self._msg = f"Hub server {name} version {version} not found."
-
-    def __str__(self) -> str:
-        return self._msg
-
-
-class BuildError(Exception):
-    pass

splight_cli-5.4.1/splight_cli/hub/server/server_builder.py

@@ -1,73 +0,0 @@
-import gzip
-import os
-import re
-import sys
-
-import docker
-
-from splight_cli.hub.server.exceptions import BuildError
-
-
-class ServerBuilder:
-    def __init__(self, server_spec: dict, server_path: str):
-        self._base_path = os.path.dirname(__file__)
-        self._server_name = server_spec["name"]
-        self._server_version = server_spec["version"]
-        self._server_path = server_path
-        self._docker_client = docker.from_env()
-
-    def build(self) -> str:
-        original_path = os.getcwd()
-        os.chdir(self._server_path)
-        image_file = self._build_server()
-        os.chdir(original_path)
-        return image_file
-
-    def _build_server(self) -> str:
-        full_name = f"{self._server_name}-{self._server_version}"
-        tag = f"server:{full_name}"
-        streamer = self._build_server_image(tag)
-        for chunk in streamer:
-            try:
-                log = self._parse_chunk(chunk)
-            except BuildError as exc:
-                sys.stdout.write(exc)
-                sys.stdout.flush()
-                raise exc
-            if log:
-                sys.stdout.write(f"{log}\n")
-                sys.stdout.flush()
-
-            if "stream" in chunk:
-                match = re.search(
-                    r"(^Successfully built |sha256:)([0-9a-f]+)$",
-                    chunk["stream"],
-                )
-                if match:
-                    image_id = match.group(2)
-        image = self._docker_client.images.get(image_id)
-        file_name = f"{full_name}.tgz"
-        with gzip.open(file_name, "wb") as image_file:
-            for chunk in image.save(named=True):
-                image_file.write(chunk)
-        return file_name
-
-    def _build_server_image(self, tag: str):
-        client = self._docker_client.api
-        streamer = client.build(
-            path=".",
-            tag=tag,
-            dockerfile="Dockerfile",
-            rm=True,
-            pull=True,
-            quiet=False,
-            decode=True,
-        )
-        return streamer
-
-    def _parse_chunk(self, chunk: dict) -> str:
-        if "errorDetail" in chunk:
-            message = chunk["error"]
-            raise BuildError(message)
-        log = "".join([str(item) for item in chunk.values()])
-        return log.strip("\n") if log != "\n" else ""

splight_cli-5.4.1/splight_cli/hub/server/server_manager.py

@@ -1,116 +0,0 @@
-import json
-import os
-import shutil
-from collections import namedtuple
-from typing import Optional
-
-import py7zr
-from pydantic import ValidationError
-from rich.console import Console
-from rich.table import Table
-from splight_lib.models import HubServer
-
-from splight_cli.constants import COMPRESSION_TYPE, SPEC_FILE, success_style
-from splight_cli.hub.exceptions import SpecFormatError, SpecValidationError
-from splight_cli.hub.server.exceptions import (
-    HubServerNotFound,
-    ServerAlreadyExists,
-    ServerDirectoryAlreadyExists,
-)
-from splight_cli.hub.server.server_builder import ServerBuilder
-from splight_cli.utils.loader import Loader
-
-console = Console()
-
-
-class HubServerManager:
-    def push(self, path: str, force: Optional[bool] = False):
-        try:
-            with open(os.path.join(path, SPEC_FILE)) as fid:
-                spec = json.load(fid)
-        except Exception as exc:
-            raise SpecFormatError(exc)
-
-        # Validate spec fields before pushing the model
-        try:
-            server = HubServer.model_validate(spec)
-        except ValidationError as exc:
-            raise SpecValidationError(exc)
-
-        name = spec["name"]
-        version = spec["version"]
-
-        if not force and self._exists_in_hub(name, version):
-            raise ServerAlreadyExists(name, version)
-
-        builder = ServerBuilder(spec, path)
-        image_file = builder.build()
-
-        with Loader("Pushing Server to Splight Hub"):
-            server = HubServer.upload(path, image_file)
-
-        if image_file:
-            os.remove(image_file)
-
-        console.print(
-            f"Server {server.id} pushed succesfully", style=success_style
-        )
-
-    def pull(self, name: str, version: str):
-        with Loader("Pulling server from Splight Hub"):
-            self._pull_server(name, version)
-        console.print(f"Server {name} pulled succesfully", style=success_style)
-
-    def _pull_server(self, name: str, version: str):
-        server = self._get_hub_server(name, version)
-        name, version = server.name, server.version
-
-        file_wrapper = server.download()
-        file_data = file_wrapper.read()
-
-        version_modified = version.replace(".", "_")
-        server_path = f"{name}/{version_modified}"
-        versioned_name = f"{name}-{version}"
-        file_name = f"{versioned_name}.{COMPRESSION_TYPE}"
-        if os.path.exists(server_path):
-            raise ServerDirectoryAlreadyExists(server_path)
-
-        try:
-            with open(file_name, "wb") as fid:
-                fid.write(file_data)
-
-            with py7zr.SevenZipFile(file_name, "r") as z:
-                z.extractall(path=".")
-            shutil.move(f"{versioned_name}", server_path)
-
-        except Exception as exc:
-            raise exc
-        finally:
-            if os.path.exists(file_name):
-                os.remove(file_name)
-
-    def _get_hub_server(self, name: str, version: str) -> HubServer:
-        servers = HubServer.list(name=name, version=version)
-        if not servers:
-            raise HubServerNotFound(name, version)
-        return servers[0]
-
-    def list_servers(self):
-        servers_tuple = namedtuple("Server", ["name", "version"])
-        all_servers = HubServer.list()
-        servers = set(
-            [
-                servers_tuple(server.name, server.version)
-                for server in all_servers
-            ]
-        )
-        table = Table("Name", "Version", title="Hub Server List")
-        [table.add_row(server.name, server.version) for server in servers]
-        console.print(table)
-
-    def _exists_in_hub(self, name: str, version: str) -> bool:
-        try:
-            self._get_hub_server(name, version)
-            return True
-        except HubServerNotFound:
-            return False