wip: refactor

src/maildir_gtd/screens/CreateTask.py

@@ -0,0 +1,115 @@
+import logging
+from textual.screen import ModalScreen
+from textual.widgets import Input, Label, Button, ListView, ListItem
+from textual.containers import Vertical, Horizontal, Container
+from textual import on, work
+from src.services.taskwarrior import client as taskwarrior_client
+
+
+class CreateTaskScreen(ModalScreen):
+    """Screen for creating a new task."""
+
+    def __init__(self, subject="", from_addr="", **kwargs):
+        super().__init__(**kwargs)
+        self.subject = subject
+        self.from_addr = from_addr
+        self.selected_project = None
+
+    def compose(self):
+        yield Container(
+            Vertical(
+                Label("Create Task", id="create_task_title"),
+                Horizontal(
+                    Label("Subject:"),
+                    Input(
+                        placeholder="Task subject",
+                        value=self.subject,
+                        id="subject_input",
+                    ),
+                ),
+                Horizontal(
+                    Label("Project:"),
+                    Input(placeholder="Project name", id="project_input"),
+                ),
+                Horizontal(
+                    Label("Tags:"),
+                    Input(placeholder="Comma-separated tags", id="tags_input"),
+                ),
+                Horizontal(
+                    Label("Due:"),
+                    Input(
+                        placeholder="Due date (e.g., today, tomorrow, fri)",
+                        id="due_input",
+                    ),
+                ),
+                Horizontal(
+                    Label("Priority:"),
+                    Input(placeholder="Priority (H, M, L)", id="priority_input"),
+                ),
+                Horizontal(
+                    Button("Create", id="create_btn", variant="primary"),
+                    Button("Cancel", id="cancel_btn", variant="error"),
+                ),
+                id="create_task_form",
+            ),
+            id="create_task_container",
+        )
+
+    def on_mount(self):
+        self.styles.align = ("center", "middle")
+
+    @on(Button.Pressed, "#create_btn")
+    def on_create_pressed(self):
+        """Create the task when the Create button is pressed."""
+        # Get input values
+        subject = self.query_one("#subject_input").value
+        project = self.query_one("#project_input").value
+        tags_input = self.query_one("#tags_input").value
+        due = self.query_one("#due_input").value
+        priority = self.query_one("#priority_input").value
+
+        # Process tags (split by commas and trim whitespace)
+        tags = [tag.strip() for tag in tags_input.split(",")] if tags_input else []
+
+        # Add a tag for the sender, if provided
+        if self.from_addr and "@" in self.from_addr:
+            domain = self.from_addr.split("@")[1].split(".")[0]
+            if domain and domain not in ["gmail", "yahoo", "hotmail", "outlook"]:
+                tags.append(domain)
+
+        # Create the task
+        self.create_task_worker(subject, tags, project, due, priority)
+
+    @on(Button.Pressed, "#cancel_btn")
+    def on_cancel_pressed(self):
+        """Dismiss the screen when Cancel is pressed."""
+        self.dismiss()
+
+    @work(exclusive=True)
+    async def create_task_worker(
+        self, subject, tags=None, project=None, due=None, priority=None
+    ):
+        """Worker to create a task using the Taskwarrior API client."""
+        if not subject:
+            self.app.show_status("Task subject cannot be empty.", "error")
+            return
+
+        # Validate priority
+        if priority and priority not in ["H", "M", "L"]:
+            self.app.show_status("Priority must be H, M, or L.", "warning")
+            priority = None
+
+        # Create the task
+        success, result = await taskwarrior_client.create_task(
+            task_description=subject,
+            tags=tags or [],
+            project=project,
+            due=due,
+            priority=priority,
+        )
+
+        if success:
+            self.app.show_status(f"Task created: {subject}", "success")
+            self.dismiss()
+        else:
+            self.app.show_status(f"Failed to create task: {result}", "error")