data-sourcerer 0.2.2__py3-none-any.whl → 0.3.0__py3-none-any.whl

sourcerer/presentation/screens/preview_content/main.py

@@ -1,5 +1,6 @@
 from pathlib import Path
 
+from dependency_injector.wiring import Provide
 from rich.syntax import Syntax
 from textual import on
 from textual.app import ComposeResult
@@ -11,6 +12,7 @@ from sourcerer.infrastructure.access_credentials.services import CredentialsServ
 from sourcerer.infrastructure.storage_provider.exceptions import (
     ReadStorageItemsError,
 )
+from sourcerer.presentation.di_container import DiContainer
 from sourcerer.presentation.screens.shared.widgets.button import Button
 from sourcerer.presentation.utils import get_provider_service_by_access_uuid
 
@@ -18,12 +20,23 @@ from sourcerer.presentation.utils import get_provider_service_by_access_uuid
 class PreviewContentScreen(ModalScreen):
     CSS_PATH = "styles.tcss"
 
-    def __init__(self, storage_name, key, access_credentials_uuid, *args, **kwargs):
+    def __init__(
+        self,
+        storage_name,
+        key,
+        access_credentials_uuid,
+        *args,
+        credentials_service: CredentialsService = Provide[
+            DiContainer.credentials_repository
+        ],
+        **kwargs
+    ):
         super().__init__(*args, **kwargs)
 
         self.storage_name = storage_name
         self.key = key
         self.access_credentials_uuid = access_credentials_uuid
+        self.credentials_service = credentials_service
 
     def compose(self) -> ComposeResult:
         with Container(id="PreviewContentScreen"):
@@ -35,9 +48,8 @@ class PreviewContentScreen(ModalScreen):
     def on_mount(self) -> None:
         """Called when the DOM is ready."""
 
-        credentials_service = CredentialsService()
         provider_service = get_provider_service_by_access_uuid(
-            self.access_credentials_uuid, credentials_service
+            self.access_credentials_uuid, self.credentials_service
         )
         if not provider_service:
             self.notify("Could not read file :(", severity="error")

sourcerer/presentation/screens/provider_creds_list/main.py

@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 from enum import Enum
 
+from dependency_injector.wiring import Provide
 from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, VerticalScroll
@@ -10,7 +11,9 @@ from textual.screen import ModalScreen
 from textual.widgets import Checkbox, Label
 
 from sourcerer.domain.access_credentials.entities import Credentials
+from sourcerer.domain.access_credentials.repositories import BaseCredentialsRepository
 from sourcerer.infrastructure.access_credentials.services import CredentialsService
+from sourcerer.presentation.di_container import DiContainer
 from sourcerer.presentation.screens.provider_creds_list.messages.reload_credentials_request import (
     ReloadCredentialsRequest,
 )
@@ -32,9 +35,12 @@ class ProviderCredentialsRow(Horizontal):
         uuid: str
         active: bool
 
-    def __init__(self, row: Credentials, *args, **kwargs):
+    def __init__(
+        self, row: Credentials, credentials_service: CredentialsService, *args, **kwargs
+    ):
         super().__init__(*args, **kwargs)
         self.row = row
+        self.credentials_service = credentials_service
 
     def compose(self) -> ComposeResult:
         yield Checkbox(
@@ -88,8 +94,7 @@ class ProviderCredentialsRow(Horizontal):
         """
         if not result:
             return
-        credentials_service = CredentialsService()
-        credentials_service.delete(self.row.uuid)
+        self.credentials_service.delete(self.row.uuid)
         self.post_message(ReloadCredentialsRequest())
 
 
@@ -107,9 +112,16 @@ class ProviderCredsListScreen(ModalScreen):
 
     credentials_list = reactive([], recompose=True)
 
-    def __init__(self, *args, **kwargs):
+    def __init__(
+        self,
+        credentials_service: CredentialsService = Provide[
+            DiContainer.credentials_service
+        ],
+        *args,
+        **kwargs,
+    ):
         super().__init__(*args, **kwargs)
-        self.credentials_service = CredentialsService()
+        self.credentials_service = credentials_service
 
     def compose(self) -> ComposeResult:
         with Container(id=self.MAIN_CONTAINER_ID):
@@ -129,7 +141,9 @@ class ProviderCredsListScreen(ModalScreen):
                     yield Label("Auth method", classes="credentials_auth_method")
                     yield Label("Delete", classes="credentials_auth_delete")
                 for row in self.credentials_list:
-                    yield ProviderCredentialsRow(row, classes="credentials_row")
+                    yield ProviderCredentialsRow(
+                        row, self.credentials_service, classes="credentials_row"
+                    )
             with Horizontal(id="controls"):
                 yield Button(ControlsEnum.CANCEL.value, name=ControlsEnum.CANCEL.name)
 
@@ -146,7 +160,11 @@ class ProviderCredsListScreen(ModalScreen):
         self.credentials_list = self.credentials_service.list()
 
     def create_provider_creds_registration(
-        self, credentials_entry: ProviderCredentialsEntry
+        self,
+        credentials_entry: ProviderCredentialsEntry,
+        credentials_repo: BaseCredentialsRepository = Provide[
+            DiContainer.credentials_repository
+        ],
     ):
         """
         Create a new provider credentials registration.
@@ -155,10 +173,13 @@ class ProviderCredsListScreen(ModalScreen):
 
         Args:
             credentials_entry (ProviderCredentialsEntry): The credentials entry to register.
+            credentials_repo (BaseCredentialsRepository): The repository to store the credentials.
         """
         if not credentials_entry:
             return
-        service = credentials_entry.cloud_storage_provider_credentials_service()  # type: ignore
+        service = credentials_entry.cloud_storage_provider_credentials_service(
+            credentials_repo
+        )
         service.store(credentials_entry.name, credentials_entry.fields)
         self.refresh_credentials_list()
 

sourcerer/presentation/screens/provider_creds_registration/main.py

@@ -15,6 +15,9 @@ from sourcerer.domain.access_credentials.services import (
 from sourcerer.infrastructure.access_credentials.exceptions import (
     MissingAuthFieldsError,
 )
+from sourcerer.infrastructure.access_credentials.registry import (
+    AccessCredentialsRegistry,
+)
 from sourcerer.infrastructure.utils import generate_unique_name
 from sourcerer.presentation.di_container import DiContainer
 from sourcerer.presentation.screens.shared.widgets.button import Button
@@ -48,8 +51,8 @@ class ProviderCredsRegistrationScreen(ModalScreen):
     def __init__(
         self,
         *args,
-        credentials_type_registry=Provide[
-            DiContainer.config.access_credential_method_registry
+        credentials_type_registry: AccessCredentialsRegistry = Provide[  # type: ignore
+            DiContainer.config.access_credential_method_registry  # type: ignore
         ],
         **kwargs,
     ):
@@ -123,7 +126,7 @@ class ProviderCredsRegistrationScreen(ModalScreen):
 
         # If only one authentication method exists, set it and mount its fields
         self.auth_method = next(iter(auth_methods.values()))
-        cls: BaseAccessCredentialsService = self.auth_method
+        cls: type[BaseAccessCredentialsService] = self.auth_method
         await self._mount_credentials_fields(cls.auth_fields())
 
     @on(Select.Changed)
@@ -206,7 +209,7 @@ class ProviderCredsRegistrationScreen(ModalScreen):
             ProviderCredentialsEntry: An object containing the authentication name, method, and fields.
         """
         if not self.auth_method:
-            return
+            return None
 
         fields = {
             input_field.get().name: input_field.get().value

sourcerer/presentation/screens/shared/widgets/labeled_input.py

@@ -75,10 +75,10 @@ class LabeledInput(Container):
             Value: A dataclass containing the name and value of the input field.
 
         """
-        input_area = self.query_one(".form_input", expect_type=Input)
+        input_area = self.query_one(".form_input")
         text = (
             input_area.document.text
             if isinstance(input_area, TextArea)
-            else input_area.value
+            else input_area.value  # type: ignore
         )
         return self.Value(name=self.key, value=text)

sourcerer/presentation/screens/shared/widgets/spinner.py

@@ -0,0 +1,57 @@
+from collections.abc import Iterator
+from enum import Enum
+from itertools import cycle
+
+from textual.app import RenderResult
+from textual.widgets import Static
+
+
+class SpinnerType(Enum):
+    """Enumeration of various spinner animations."""
+
+    dots_1 = "⣷⣯⣟⡿⢿⣻⣽⣾"
+    dots_2 = "⠋⠙⠹⠸⠼⠴⠦⠧⠇⠏"
+    dots_3 = "⠋⠙⠚⠞⠖⠦⠴⠲⠳⠓"
+    dots_4 = "⠄⠆⠇⠋⠙⠸⠰⠠⠰⠸⠙⠋⠇⠆"
+    dots_5 = "⠈⠐⠠⢀⡀⠄⠂⠁"
+    dots_6 = "⋯⋱⋮⋰"
+    circles = "◐◓◑◒"
+    angles = "┐┤┘┴└├┌┬"
+    arrows = "←↖↑↗→↘↓↙"
+    moon = "🌑🌒🌓🌔🌕🌖🌗🌘"
+    clock = "🕛🕐🕑🕒🕓🕔🕕🕖🕗🕘🕙🕚"
+    histogram = "▁▃▄▅▆▇█▇▆▅▄▃"
+    shade = "░▒▓█▓▒░"
+    colors = "🟨🟨🟧🟧🟥🟥🟦🟦🟪🟪🟩🟩"
+    triangles = "◢◣◤◥"
+
+
+class Spinner(Static):
+    """A loading spinner widget for Textual apps."""
+
+    DEFAULT_CSS = """
+    Spinner {
+        color: #9E53E0;
+    }
+    """
+
+    def __init__(
+        self, spinner: SpinnerType = SpinnerType.dots_1, interval: float = 0.1, **kwargs
+    ):
+        """
+        Initialize the Loader widget.
+
+        Args:
+            spinner (SpinnerType): The spinner animation type.
+            interval (float): Time in seconds between frames.
+            **kwargs: Additional keyword arguments for Static.
+        """
+        super().__init__(**kwargs)
+        self._frames: Iterator[str] = cycle(spinner.value)
+        self._interval = interval
+
+    def render(self) -> RenderResult:
+        return next(self._frames)
+
+    def on_mount(self) -> None:
+        self.auto_refresh = self._interval

sourcerer/presentation/screens/storage_action_progress/main.py

@@ -185,9 +185,9 @@ class StorageActionProgressScreen(ModalScreen):
         Sets the border title and starts the appropriate worker thread based on the action type
         (download, delete, or upload).
         """
-        self.query_one("#StorageActionProgress").border_title = (
-            f"{self.action.capitalize()} {len(self.keys)} files from {self.storage_name}"
-        )
+        self.query_one(
+            "#StorageActionProgress"
+        ).border_title = f"{self.action.capitalize()} {len(self.keys)} files from {self.storage_name}"
 
         if self.action == "download":
             self.active_worker = self.run_worker(self.download_files, thread=True)
@@ -298,7 +298,7 @@ class StorageActionProgressScreen(ModalScreen):
                 progress_bar.total = file_size
             except Exception as ex:
                 self.notify(
-                    f"Failed to get file size for {key}: {str(ex)}", severity="error"
+                    f"Failed to get file size for {key}: {ex}", severity="error"
                 )
                 self.log.error(f"Error getting file size: {ex}")
                 return
@@ -311,7 +311,7 @@ class StorageActionProgressScreen(ModalScreen):
                     lambda chunk: progress_callback(progress_bar, chunk),
                 )
             except Exception as ex:
-                self.notify(f"Failed to download {key}: {str(ex)}", severity="error")
+                self.notify(f"Failed to download {key}: {ex}", severity="error")
                 self.log.error(f"Error downloading file: {ex}")
                 return
 
@@ -324,9 +324,7 @@ class StorageActionProgressScreen(ModalScreen):
                     # Non-critical error, continue execution
         except Exception as ex:
             # Catch any unexpected exceptions
-            self.notify(
-                f"Unexpected error downloading {key}: {str(ex)}", severity="error"
-            )
+            self.notify(f"Unexpected error downloading {key}: {ex}", severity="error")
             self.log.error(f"Unexpected error: {ex}")
         finally:
             main_progress_bar.advance(1)
@@ -447,9 +445,7 @@ class StorageActionProgressScreen(ModalScreen):
                     self.files_has_been_processed = True
                 self.active_executor = None
                 try:
-                    await self.query_one(
-                        f"#progress_file_details_{key.uuid}", Label
-                    ).remove()
+                    self.query_one(f"#progress_file_details_{key.uuid}", Label).remove()
                 except NoMatches:
                     self.log(f"Failed to remove progress details for {key.uuid}")
             main_progress_bar.advance(1)

sourcerer/presentation/screens/storages_list/main.py

@@ -0,0 +1,180 @@
+import datetime
+import uuid
+from enum import Enum
+
+from dependency_injector.wiring import Provide
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal, VerticalScroll
+from textual.reactive import reactive
+from textual.screen import ModalScreen
+from textual.widgets import Label
+
+from sourcerer.domain.storage.entities import Storage
+from sourcerer.infrastructure.access_credentials.services import CredentialsService
+from sourcerer.infrastructure.storage.services import StoragesService
+from sourcerer.presentation.di_container import DiContainer
+from sourcerer.presentation.screens.question.main import QuestionScreen
+from sourcerer.presentation.screens.shared.widgets.button import Button
+from sourcerer.presentation.screens.storages_list.messages.reload_storages_request import (
+    ReloadStoragesRequest,
+)
+from sourcerer.presentation.screens.storages_registration.main import (
+    StorageEntry,
+    StoragesRegistrationScreen,
+)
+
+
+class ControlsEnum(Enum):
+    CANCEL = "Cancel"
+    ADD_STORAGE = "Add Storage"
+
+
+class StorageRow(Horizontal):
+    def __init__(
+        self, storage: Storage, storages_service: StoragesService, *args, **kwargs
+    ):
+        super().__init__(*args, **kwargs)
+        self.storage = storage
+        self.storages_service = storages_service
+
+    def compose(self):
+        yield Label(self.storage.name, classes="storage_name")
+        yield Label(self.storage.credentials_name or "🚫", classes="credentials_name")
+        yield Button(
+            "❌",
+            name="delete_storage",
+            classes="storage_delete",
+        )
+
+    @on(Button.Click)
+    def on_button_click(self, _: Button.Click):
+        """
+        Handle delete button click events by showing a confirmation dialog for storage deletion.
+        Args:
+            _ (Button.Click): The button click event.
+        """
+        self.app.push_screen(
+            QuestionScreen(
+                f"Are you sure you want to delete {self.storage.credentials_name} {self.storage.name} storage?"
+            ),
+            callback=self.delete_callback,  # type: ignore
+        )
+
+    def delete_callback(self, result: bool):
+        """
+        Callback function to handle the result of the confirmation screen.
+
+        Args:
+            result (bool): True if the user confirmed, False otherwise.
+        """
+        if not result:
+            return
+        self.storages_service.delete(self.storage.uuid)
+        self.post_message(ReloadStoragesRequest())
+
+
+class StoragesListScreen(ModalScreen):
+    CSS_PATH = "styles.tcss"
+
+    MAIN_CONTAINER_ID = "StoragesListScreen"
+    SETTINGS_CONTAINER_ID = "settings"
+
+    storages_list = reactive([], recompose=True)
+
+    def __init__(
+        self,
+        credentials_service: CredentialsService = Provide[
+            DiContainer.credentials_service
+        ],
+        storages_service: StoragesService = Provide[DiContainer.storages_service],
+        *args,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+        self.storage_service = storages_service
+        self.credentials_service = credentials_service
+
+    def compose(self) -> ComposeResult:
+        with Container(id=self.MAIN_CONTAINER_ID):
+            yield Container(
+                Button(
+                    "+Add new storage",
+                    name=ControlsEnum.ADD_STORAGE.name,
+                    classes="add_storage_button",
+                ),
+                id="right-top",
+            )
+            with VerticalScroll(id=self.SETTINGS_CONTAINER_ID):
+                with Horizontal():
+                    yield Label("Storage Name", classes="storage_name")
+                    yield Label("Credentials Name", classes="credentials_name")
+                    yield Label("Delete", classes="storage_delete")
+                for storage in self.storages_list:
+                    yield StorageRow(storage, self.storage_service)
+            with Horizontal(id="controls"):
+                yield Button(ControlsEnum.CANCEL.value, name=ControlsEnum.CANCEL.name)
+
+    def on_compose(self):
+        """
+        Initialize the screen by refreshing the credentials list when the screen is composed.
+        """
+        self.refresh_storages_list()
+
+    def refresh_storages_list(self):
+        """
+        Refresh the storages list by retrieving the latest storages from the storage service.
+        """
+        self.storages_list = self.storage_service.list()
+
+    @on(ReloadStoragesRequest)
+    def on_reload_storages_request(self, _: ReloadStoragesRequest):
+        """
+        Handle the reload storages request by refreshing the storages list.
+
+        Args:
+            _: ReloadStoragesRequest: The reload storages request message.
+        """
+        self.refresh_storages_list()
+
+    @on(Button.Click)
+    def on_control_button_click(self, event: Button.Click):
+        """
+        Handle click events for control buttons.
+
+        Dismisses the screen if the cancel button is clicked, or opens the provider credentials registration screen if
+        the add registration button is clicked.
+
+        Args:
+            event (Button.Click): The button click event.
+        """
+        if event.action == ControlsEnum.CANCEL.name:
+            self.dismiss()
+        if event.action == ControlsEnum.ADD_STORAGE.name:
+            self.app.push_screen(
+                StoragesRegistrationScreen(),
+                callback=self.create_storage_entry,  # type: ignore
+            )
+
+    def create_storage_entry(self, storage: StorageEntry | None):
+        """
+        Create a new storage entry.
+
+        Creates a new storage entry using the provided data and refreshes the storage list.
+        """
+        if storage is None:
+            return
+
+        credentials = self.credentials_service.get(storage.credentials_uuid)
+        if not credentials:
+            self.notify("Credentials not found", severity="error")
+            return
+        self.storage_service.create(
+            Storage(
+                uuid=str(uuid.uuid4()),
+                name=storage.name,
+                credentials_id=credentials.id,
+                date_created=datetime.datetime.now(),
+            )
+        )
+        self.refresh_storages_list()

sourcerer/presentation/screens/storages_list/messages/reload_storages_request.py

@@ -0,0 +1,8 @@
+from dataclasses import dataclass
+
+from textual.message import Message
+
+
+@dataclass
+class ReloadStoragesRequest(Message):
+    pass

sourcerer/presentation/screens/storages_list/styles.tcss

@@ -0,0 +1,55 @@
+
+Container {
+  height: auto;
+}
+
+Label {
+  padding-left: 1;
+}
+
+StoragesListScreen {
+  align: center middle;
+  content-align: center top;
+
+  & > #StoragesListScreen {
+    padding: 1 2 0 2;
+    margin: 0 0;
+    width: 60%;
+    height: 60%;
+    border: solid $secondary-background;
+
+    #right-top {
+      align: right top;
+
+      .add_storage_button {
+        color: $success-darken-2;
+      }
+
+      margin-bottom: 1;
+    }
+
+    ProviderCredentialsRow.active {
+      background: $primary-lighten-2;
+    }
+
+    ProviderCredentialsRow, Horizontal {
+      height: auto;
+    }
+
+    .storage_name {
+      width: 45%;
+    }
+
+    .credentials_name {
+      width: 35%;
+    }
+
+    .storage_delete {
+      width: 20%;
+    }
+
+    & > Select {
+      margin-bottom: 1;
+    }
+  }
+}

sourcerer/presentation/screens/storages_registration/main.py

@@ -0,0 +1,100 @@
+from dataclasses import dataclass
+from enum import Enum
+
+from dependency_injector.wiring import Provide
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal, VerticalScroll
+from textual.screen import ModalScreen
+from textual.widgets import Label, Select
+
+from sourcerer.infrastructure.access_credentials.services import CredentialsService
+from sourcerer.presentation.di_container import DiContainer
+from sourcerer.presentation.screens.shared.widgets.button import Button
+from sourcerer.presentation.screens.shared.widgets.labeled_input import LabeledInput
+
+
+class ControlsEnum(Enum):
+    CANCEL = "Cancel"
+    CREATE = "Create"
+
+
+@dataclass
+class StorageEntry:
+    name: str
+    credentials_uuid: str
+
+
+class StoragesRegistrationScreen(ModalScreen):
+    CSS_PATH = "styles.tcss"
+
+    MAIN_CONTAINER_ID = "StoragesRegistrationScreen"
+    SETTINGS_CONTAINER_ID = "settings"
+    CREDENTIALS_SELECTOR_ID = "credentials_selector"
+
+    PROVIDERS_NAME = "credentials"
+
+    def __init__(
+        self,
+        credentials_service: CredentialsService = Provide[
+            DiContainer.credentials_repository
+        ],
+        *args,
+        **kwargs,
+    ):
+        super().__init__(*args, **kwargs)
+        self.credentials = credentials_service.list()
+        self.auth_method = None
+
+    def compose(self) -> ComposeResult:
+        with Container(id=self.MAIN_CONTAINER_ID):
+            with VerticalScroll(id=self.SETTINGS_CONTAINER_ID):
+                yield LabeledInput(
+                    "storage_name",
+                    "Name:",
+                    True,
+                    multiline=False,
+                    id="storage_name",
+                )
+                yield Label("* Credentials:", classes="form_label")
+                yield Select(
+                    options=((item.name, item.uuid) for item in self.credentials),
+                    name=self.PROVIDERS_NAME,
+                    id=self.CREDENTIALS_SELECTOR_ID,
+                )
+            with Horizontal(id="controls"):
+                yield Button(ControlsEnum.CANCEL.value, name=ControlsEnum.CANCEL.name)
+                yield Button(ControlsEnum.CREATE.value, name=ControlsEnum.CREATE.name)
+
+    @on(Button.Click)
+    def on_control_button_click(self, event: Button.Click):
+        """
+        Handle click events for control buttons on the registration screen.
+
+        Depending on the action associated with the button click event, either dismiss
+        the screen or gather authentication fields and then dismiss the screen with
+        the collected data.
+
+        Args:
+            event (Button.Click): The click event containing the action to be performed.
+
+        Flow:
+            1. Check if the event.action is ControlsEnum.cancel.name. If true, dismiss the screen.
+            2. If event.action is ControlsEnum.create.name, gather authentication fields. Dismiss the screen with the
+               collected authentication fields.
+        """
+        if event.action == ControlsEnum.CANCEL.name:
+            self.dismiss()
+        elif event.action == ControlsEnum.CREATE.name:
+            storage_name = self.query_one("#storage_name", LabeledInput).get().value
+            if not storage_name:
+                self.notify("Storage name is required", severity="error")
+                return
+            credentials_uuid = self.query_one(
+                f"#{self.CREDENTIALS_SELECTOR_ID}", Select
+            ).value
+            if not credentials_uuid or credentials_uuid == Select.BLANK:
+                self.notify("Credentials are required", severity="error")
+                return
+
+            self.dismiss(StorageEntry(storage_name, credentials_uuid))  # type: ignore