data-sourcerer 0.1.0__py3-none-any.whl

sourcerer/presentation/screens/main/widgets/storage_list_sidebar.py

@@ -0,0 +1,143 @@
+"""Storage list sidebar widget for the Sourcerer application.
+
+This module provides widgets for displaying and interacting with the list of
+storage providers in the sidebar. It handles storage grouping by provider type
+and selection of storage items.
+"""
+
+from collections import namedtuple
+from itertools import groupby
+from typing import Dict, List
+
+from textual import events
+from textual.app import ComposeResult
+from textual.containers import VerticalScroll, Horizontal, Container
+from textual.reactive import reactive
+from textual.widgets import Label, Rule
+
+from sourcerer.domain.shared.entities import StorageProvider
+from sourcerer.domain.storage_provider.entities import Storage
+from sourcerer.presentation.screens.main.messages.select_storage_item import (
+    SelectStorageItem,
+)
+from sourcerer.presentation.screens.main.widgets.gradient import GradientWidget
+
+STORAGE_ICONS = {
+    StorageProvider.S3: "🟠",
+    StorageProvider.GoogleCloudStorage: "🔵",
+    "azure": "⚪️",
+}
+"""Mapping of storage provider types to their display icons."""
+
+
+class StorageItem(Label):
+    """Widget for displaying and interacting with a single storage item.
+
+    This widget represents a storage instance in the sidebar list, allowing
+    selection and visual feedback on hover.
+    """
+
+    DEFAULT_CSS = """
+    StorageItem {
+        width: 90%;
+        padding-left: 1;
+        height: auto;
+        margin:0;
+        text-overflow: ellipsis;
+        text-wrap: nowrap;
+
+        & > :hover {
+            background: $warning;
+            color: $panel;
+        }
+    }
+    """
+
+    def __init__(self, storage_name, access_credentials_uuid, *args, **kwargs):
+        """Initialize a storage item widget.
+
+        Args:
+            storage_name: The name of the storage instance
+            access_credentials_uuid: UUID of the access credentials being used
+        """
+        self.storage_name = storage_name
+        self.access_credentials_uuid = access_credentials_uuid
+
+        super().__init__(*args, **kwargs)
+
+    def on_click(self, _: events.Click) -> None:
+        """Handle click events to select the storage item."""
+        self.post_message(
+            SelectStorageItem(
+                self.storage_name, access_credentials_uuid=self.access_credentials_uuid
+            )
+        )
+
+
+class StorageListSidebar(VerticalScroll):
+    """Sidebar widget for displaying the list of storage providers.
+
+    This widget manages the display of storage providers grouped by their type,
+    showing provider icons and storage names in a scrollable list.
+
+    Attributes:
+        storages: Dictionary mapping provider types to lists of storage instances
+    """
+
+    storages: reactive[Dict[str, List[Storage]]] = reactive({}, recompose=True)
+
+    DEFAULT_CSS = """
+    StorageListSidebar {
+        padding-right:  0;
+        margin-right: 0;
+        & > .storage-group {
+            height: auto;
+            margin-bottom: 1;
+            #rule-left {
+                width: 1;
+            }
+            
+            Horizontal {
+                height: auto;
+            }
+            Rule.-horizontal {
+                height: 1;
+                margin: 0 0;
+                    
+            }
+            .storage-letter {
+                color: $secondary;
+                padding: 0 1;
+            }
+
+        }
+    }
+    """
+
+    def compose(self) -> ComposeResult:
+        with Container(id="header"):
+            yield GradientWidget("🧙SOURCERER", id="left-middle")
+        StorageData = namedtuple("Storage", ["access_credentials_uuid", "storage"])
+        storages = [
+            StorageData(access_credentials_uuid, storage)
+            for access_credentials_uuid, storages in self.storages.items()
+            for storage in storages
+        ]
+        storages = sorted(storages, key=lambda x: x.storage.storage)
+
+        for letter, storages in groupby(storages, key=lambda x: x.storage.storage[0]):
+            with Container(id=f"group-{letter}", classes="storage-group"):
+                yield Horizontal(
+                    Rule(id="rule-left"),
+                    Label(letter.upper(), classes="storage-letter"),
+                    Rule(),
+                )
+
+                for item in storages:
+                    yield StorageItem(
+                        renderable=STORAGE_ICONS.get(item.storage.provider, "")
+                        + " "
+                        + item.storage.storage,
+                        storage_name=item.storage.storage,
+                        access_credentials_uuid=item.access_credentials_uuid,
+                    )

sourcerer/presentation/screens/preview_content/main.py

@@ -0,0 +1,59 @@
+from rich.syntax import Syntax
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal
+from textual.screen import ModalScreen
+from textual.widgets import RichLog, LoadingIndicator
+
+from sourcerer.infrastructure.access_credentials.services import CredentialsService
+from sourcerer.infrastructure.storage_provider.exceptions import (
+    ReadStorageItemsException,
+)
+from sourcerer.presentation.screens.shared.widgets.button import Button
+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):
+        super().__init__(*args, **kwargs)
+
+        self.storage_name = storage_name
+        self.key = key
+        self.access_credentials_uuid = access_credentials_uuid
+
+    def compose(self) -> ComposeResult:
+        with Container(id="PreviewContentScreen"):
+            yield LoadingIndicator(id="loading")
+            yield RichLog(highlight=True, markup=True, auto_scroll=False)
+            with Horizontal(id="controls"):
+                yield Button("Close", name="cancel")
+
+    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
+        )
+        if not provider_service:
+            self.notify("Could not read file :(", severity="error")
+            return
+        try:
+            content = provider_service.read_storage_item(self.storage_name, self.key)
+        except ReadStorageItemsException:
+            self.notify("Could not read file :(", severity="error")
+            return
+        self.query_one("#loading").remove()
+
+        text_log = self.query_one(RichLog)
+        lexer = Syntax.guess_lexer(self.key, content)
+        content = Syntax(content, lexer, line_numbers=True, theme="ansi_dark")
+        text_log.write(content)
+
+    @on(Button.Click)
+    def on_button_click(self, event: Button.Click) -> None:
+        """Handle button click events."""
+        if event.action == "cancel":
+            self.dismiss()

sourcerer/presentation/screens/preview_content/styles.tcss

@@ -0,0 +1,26 @@
+
+Container {
+  height: auto;
+}
+
+
+PreviewContentScreen {
+  align: center middle;
+  content-align: center top;
+
+
+  & > #PreviewContentScreen {
+    padding: 1 2 0 2;
+    margin: 0 0;
+    width: 90;
+    height: 40;
+    border: solid $secondary-background;
+    border-title-color: $primary-lighten-2;
+
+    RichLog {
+      background: transparent;
+    }
+
+  }
+}
+

sourcerer/presentation/screens/provider_creds_list/main.py

@@ -0,0 +1,164 @@
+from dataclasses import dataclass
+from enum import Enum
+
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Container, Horizontal, VerticalScroll
+from textual.message import Message
+from textual.reactive import reactive
+from textual.screen import ModalScreen
+from textual.widgets import Label, Checkbox
+
+from sourcerer.domain.access_credentials.entities import Credentials
+from sourcerer.infrastructure.access_credentials.services import CredentialsService
+from sourcerer.presentation.screens.provider_creds_registration.main import (
+    ProviderCredsRegistrationScreen,
+    ProviderCredentialsEntry,
+)
+from sourcerer.presentation.screens.shared.widgets.button import Button
+
+
+class ControlsEnum(Enum):
+    CANCEL = "Cancel"
+
+
+class ProviderCredentialsRow(Horizontal):
+    @dataclass
+    class ChangeActiveStatus(Message):
+        uuid: str
+        active: bool
+
+    def __init__(self, row: Credentials, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.row = row
+
+    def compose(self) -> ComposeResult:
+        yield Checkbox(
+            value=self.row.active, classes="credentials_active", name=self.row.uuid
+        )
+        yield Label(self.row.name, classes="credentials_name")
+        yield Label(self.row.provider, classes="credentials_provider")
+        yield Label(self.row.credentials_type, classes="credentials_auth_method")
+
+    def on_mouse_move(self, _) -> None:
+        """Change background color when hovered."""
+        self.add_class("active")
+
+    def on_leave(self, _) -> None:
+        """Reset background color when mouse leaves."""
+        self.remove_class("active")
+
+    @on(Checkbox.Changed)
+    def on_checkbox_change(self, event: Checkbox.Changed):
+        """
+        Handle checkbox state changes by updating the row's active status and posting a ChangeActiveStatus message.
+
+        Args:
+            event (Checkbox.Changed): The checkbox change event containing the new value.
+        """
+        self.row.active = event.value
+        self.post_message(self.ChangeActiveStatus(self.row.uuid, self.row.active))
+
+
+class ProviderCredsListScreen(ModalScreen):
+    CSS_PATH = "styles.tcss"
+
+    MAIN_CONTAINER_ID = "ProviderCredsListScreen"
+    SETTINGS_CONTAINER_ID = "settings"
+    PROVIDER_SELECTOR_ID = "provider_selector"
+    CREDENTIALS_TYPE_SELECTOR_ID = "credentials_type_select"
+    CREDENTIALS_FIELDS_CONTAINER_ID = "credentials_fields_container"
+
+    PROVIDERS_NAME = "providers"
+    AUTH_METHODS_NAME = "auth_methods"
+
+    credentials_list = reactive([], recompose=True)
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.credentials_service = CredentialsService()
+
+    def compose(self) -> ComposeResult:
+        with Container(id=self.MAIN_CONTAINER_ID):
+            yield Container(
+                Button(
+                    "+Add new registration",
+                    name="add_registration",
+                    classes="add_registration_button",
+                ),
+                id="right-top",
+            )
+            with VerticalScroll(id=self.SETTINGS_CONTAINER_ID):
+                with Horizontal():
+                    yield Label("Active", classes="credentials_active")
+                    yield Label("Name", classes="credentials_name")
+                    yield Label("Provider", classes="credentials_provider")
+                    yield Label("Auth method", classes="credentials_auth_method")
+                for row in self.credentials_list:
+                    yield ProviderCredentialsRow(row, classes="credentials_row")
+            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_credentials_list()
+
+    def refresh_credentials_list(self):
+        """
+        Refresh the credentials list by retrieving the latest credentials from the credentials service.
+        """
+        self.credentials_list = self.credentials_service.list()
+
+    def create_provider_creds_registration(
+        self, credentials_entry: ProviderCredentialsEntry
+    ):
+        """
+        Create a new provider credentials registration.
+
+        Stores the provided credentials entry using its associated service and refreshes the credentials list.
+
+        Args:
+            credentials_entry (ProviderCredentialsEntry): The credentials entry to register.
+        """
+        if not credentials_entry:
+            return
+        service = credentials_entry.cloud_storage_provider_credentials_service()  # type: ignore
+        service.store(credentials_entry.name, credentials_entry.fields)
+        self.refresh_credentials_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 == "add_registration":
+            self.app.push_screen(
+                ProviderCredsRegistrationScreen(),
+                callback=self.create_provider_creds_registration,  # type: ignore
+            )
+
+    @on(ProviderCredentialsRow.ChangeActiveStatus)
+    def on_change_active_status(self, event: ProviderCredentialsRow.ChangeActiveStatus):
+        """
+        Handle changes to the active status of a provider credentials row.
+
+        Activates or deactivates the credentials based on the event, then refreshes the credentials list.
+
+        Args:
+            event (ProviderCredentialsRow.ChangeActiveStatus): Event containing the credentials UUID and new status.
+        """
+        if event.active:
+            self.credentials_service.activate(event.uuid)
+        else:
+            self.credentials_service.deactivate(event.uuid)
+        self.refresh_credentials_list()

sourcerer/presentation/screens/provider_creds_list/styles.tcss

@@ -0,0 +1,65 @@
+
+Container {
+  height: auto;
+}
+
+Label {
+  padding-left: 1;
+}
+
+ProviderCredsListScreen {
+  align: center middle;
+  content-align: center top;
+
+  & > #ProviderCredsListScreen {
+    padding: 1 2 0 2;
+    margin: 0 0;
+    width: 60%;
+    height: 60%;
+    border: solid $secondary-background;
+
+    #right-top {
+      align: right top;
+
+      .add_registration_button {
+        color: $success-darken-2;
+      }
+
+      margin-bottom: 1;
+    }
+
+    ProviderCredentialsRow.active {
+      background: $primary-lighten-2;
+    }
+
+    ProviderCredentialsRow, Horizontal {
+      height: auto;
+    }
+
+    .credentials_active {
+      width: 10%;
+      background: transparent;
+      border: none;
+    }
+
+    .credentials_name {
+      width: 30%;
+    }
+
+    .credentials_provider {
+      width: 30%;
+    }
+
+    .credentials_auth_method {
+      width: 30%;
+    }
+
+
+    & > Select {
+      margin-bottom: 1;
+    }
+  }
+}
+
+
+