titan-cli 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

titan_plugin_jira/steps/search_jql_step.py

@@ -0,0 +1,191 @@
+"""
+Search JIRA issues using custom JQL query
+"""
+
+from titan_cli.engine import WorkflowContext, WorkflowResult, Success, Error
+from titan_cli.ui.tui.widgets import Table
+from ..exceptions import JiraAPIError
+from ..messages import msg
+from ..utils import IssueSorter
+
+
+def search_jql_step(ctx: WorkflowContext) -> WorkflowResult:
+    """
+    Search JIRA issues using a custom JQL query.
+
+    This is a generic search step that accepts raw JQL and allows variable substitution.
+    Use this when you need a specific query that isn't covered by saved queries.
+
+    Inputs (from ctx.data):
+        jql (str): JQL query string (supports variable substitution with ${var_name})
+        max_results (int, optional): Maximum number of results (default: 100)
+
+    Outputs (saved to ctx.data):
+        jira_issues (list): List of JiraTicket objects
+        jira_issue_count (int): Number of issues found
+
+    Returns:
+        Success: Issues found
+        Error: Search failed or JQL not provided
+
+    Variable substitution:
+        You can use ${variable_name} in the JQL and it will be replaced with values from ctx.data.
+        Example: "project = ${project_key} AND fixVersion = ${fix_version}"
+
+    Example usage in workflow:
+        ```yaml
+        - id: search_issues
+          plugin: jira
+          step: search_jql
+          params:
+            jql: "project = MYPROJECT AND status = 'In Progress' ORDER BY created DESC"
+            max_results: 50
+
+        # With variable substitution:
+        - id: search_release
+          plugin: jira
+          step: search_jql
+          params:
+            jql: "project = ${project_key} AND fixVersion = ${fix_version} ORDER BY created DESC"
+        ```
+    """
+    if not ctx.textual:
+        return Error("Textual UI context is not available for this step.")
+
+    # Begin step container
+    ctx.textual.begin_step("Search JIRA Issues")
+
+    if not ctx.jira:
+        ctx.textual.error_text(msg.Plugin.CLIENT_NOT_AVAILABLE_IN_CONTEXT)
+        ctx.textual.end_step("error")
+        return Error(msg.Plugin.CLIENT_NOT_AVAILABLE_IN_CONTEXT)
+
+    # Get JQL query
+    jql = ctx.get("jql")
+    if not jql:
+        error_msg = "JQL query is required but not provided"
+        ctx.textual.error_text(error_msg)
+        ctx.textual.dim_text("Provide 'jql' parameter in workflow step")
+        ctx.textual.end_step("error")
+        return Error(error_msg)
+
+    # Perform variable substitution: replace ${var} with ctx.data values
+    # This allows dynamic queries like "project = ${project_key}"
+    import re
+
+    def replace_var(match):
+        var_name = match.group(1)
+        value = ctx.get(var_name)
+        if value is None:
+            # Variable not found in context, keep original placeholder
+            return match.group(0)
+        return str(value)
+
+    # Replace ${variable} patterns
+    jql = re.sub(r'\$\{([^}]+)\}', replace_var, jql)
+
+    # Show which query is being executed
+    ctx.textual.text("")
+    ctx.textual.bold_text("Executing JQL Query:")
+    ctx.textual.dim_text(f"  {jql}")
+    ctx.textual.text("")
+
+    # Get max results
+    max_results = ctx.get("max_results", 100)
+
+    try:
+        # Execute search with loading indicator
+        # Request ALL fields including custom fields
+        with ctx.textual.loading("Searching JIRA issues..."):
+            issues = ctx.jira.search_tickets(jql=jql, max_results=max_results, fields=["*all"])
+
+        if not issues:
+            ctx.textual.dim_text("No issues found")
+            ctx.textual.end_step("success")
+            return Success(
+                "No issues found",
+                metadata={
+                    "jira_issues": [],
+                    "issues": [],  # Alias for compatibility
+                    "jira_issue_count": 0
+                }
+            )
+
+        # Show results
+        ctx.textual.text("")  # spacing
+        ctx.textual.success_text(f"Found {len(issues)} issues")
+        ctx.textual.text("")
+
+        # Show detailed table
+        ctx.textual.bold_text("Found Issues:")
+        ctx.textual.text("")
+
+        try:
+            # Sort issues intelligently
+            sorter = IssueSorter()
+            sorted_issues = sorter.sort(issues)
+
+            # Prepare table data with row numbers for selection
+            headers = ["#", "Key", "Status", "Summary", "Assignee", "Type", "Priority"]
+            rows = []
+            for i, issue in enumerate(sorted_issues, 1):
+                assignee = issue.assignee or "Unassigned"
+                status = issue.status or "Unknown"
+                priority = issue.priority or "Unknown"
+                issue_type = issue.issue_type or "Unknown"
+                summary = (issue.summary or "No summary")[:60]
+
+                rows.append([
+                    str(i),
+                    issue.key,
+                    status,
+                    summary,
+                    assignee,
+                    issue_type,
+                    priority
+                ])
+
+            # Render table using textual widget
+            ctx.textual.mount(
+                Table(
+                    headers=headers,
+                    rows=rows,
+                    title=f"Issues (sorted by {sorter.get_sort_description()})"
+                )
+            )
+
+            # Use sorted issues for downstream steps
+            issues = sorted_issues
+        except Exception as e:
+            # If table rendering fails, show error but continue with raw issue list
+            ctx.textual.error_text(f"Error rendering table: {e}")
+            ctx.textual.primary_text(f"Found {len(issues)} issues (showing raw data)")
+            for i, issue in enumerate(issues, 1):
+                ctx.textual.text(f"{i}. {issue.key} - {getattr(issue, 'summary', 'N/A')}")
+            ctx.textual.text("")
+
+        ctx.textual.end_step("success")
+        return Success(
+            f"Found {len(issues)} issues",
+            metadata={
+                "jira_issues": issues,
+                "issues": issues,  # Alias for compatibility
+                "jira_issue_count": len(issues)
+            }
+        )
+
+    except JiraAPIError as e:
+        error_msg = f"JIRA search failed: {e}"
+        ctx.textual.error_text(error_msg)
+        ctx.textual.end_step("error")
+        return Error(error_msg)
+    except Exception as e:
+        import traceback
+        error_detail = traceback.format_exc()
+        error_msg = f"Unexpected error: {e}\n\nTraceback:\n{error_detail}"
+        ctx.textual.error_text(error_msg)
+        ctx.textual.end_step("error")
+        return Error(error_msg)
+
+
+__all__ = ["search_jql_step"]

titan_plugin_jira/steps/search_saved_query_step.py

@@ -3,7 +3,7 @@ Search JIRA issues using saved query from utils registry
 """
 
 from titan_cli.engine import WorkflowContext, WorkflowResult, Success, Error
-from titan_cli.ui.tui.widgets import Panel, Table
+from titan_cli.ui.tui.widgets import Table
 from ..exceptions import JiraAPIError
 from ..messages import msg
 from ..utils import SAVED_QUERIES, IssueSorter
@@ -57,14 +57,19 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Search Open Issues")
+
     if not ctx.jira:
-        ctx.textual.mount(Panel(msg.Plugin.CLIENT_NOT_AVAILABLE_IN_CONTEXT, panel_type="error"))
+        ctx.textual.error_text(msg.Plugin.CLIENT_NOT_AVAILABLE_IN_CONTEXT)
+        ctx.textual.end_step("error")
         return Error(msg.Plugin.CLIENT_NOT_AVAILABLE_IN_CONTEXT)
 
     # Get query name
     query_name = ctx.get("query_name")
     if not query_name:
-        ctx.textual.mount(Panel(msg.Steps.Search.QUERY_NAME_REQUIRED, panel_type="error"))
+        ctx.textual.error_text(msg.Steps.Search.QUERY_NAME_REQUIRED)
+        ctx.textual.end_step("error")
         return Error(msg.Steps.Search.QUERY_NAME_REQUIRED)
 
     # Get all predefined queries from utils
@@ -102,7 +107,8 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
             error_msg += "\n\n" + msg.Steps.Search.ADD_CUSTOM_HINT + "\n"
             error_msg += msg.Steps.Search.CUSTOM_QUERY_EXAMPLE
 
-        ctx.textual.mount(Panel(error_msg, panel_type="error"))
+        ctx.textual.error_text(error_msg)
+        ctx.textual.end_step("error")
         return Error(error_msg)
 
     jql = all_queries[query_name]
@@ -119,7 +125,8 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
 
         if not project:
             error_msg = msg.Steps.Search.PROJECT_REQUIRED.format(query_name=query_name, jql=jql)
-            ctx.textual.mount(Panel(error_msg, panel_type="error"))
+            ctx.textual.error_text(error_msg)
+            ctx.textual.end_step("error")
             return Error(error_msg)
 
         jql = jql.format(project=project)
@@ -129,8 +136,8 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
     source_label = "Custom" if is_custom else "Predefined"
 
     ctx.textual.text("")
-    ctx.textual.text(f"Using {source_label} Query: {query_name}", markup="bold")
-    ctx.textual.text(f"  JQL: {jql}", markup="dim")
+    ctx.textual.bold_text(f"Using {source_label} Query: {query_name}")
+    ctx.textual.dim_text(f"  JQL: {jql}")
     ctx.textual.text("")
 
     # Get max results
@@ -142,12 +149,8 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
             issues = ctx.jira.search_tickets(jql=jql, max_results=max_results)
 
         if not issues:
-            ctx.textual.mount(
-                Panel(
-                    text=f"No issues found for query: {query_name}",
-                    panel_type="info"
-                )
-            )
+            ctx.textual.dim_text(f"No issues found for query: {query_name}")
+            ctx.textual.end_step("success")
             return Success(
                 "No issues found",
                 metadata={
@@ -158,16 +161,12 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
             )
 
         # Show results
-        ctx.textual.mount(
-            Panel(
-                text=f"Found {len(issues)} issues",
-                panel_type="success"
-            )
-        )
+        ctx.textual.text("")  # spacing
+        ctx.textual.success_text(f"Found {len(issues)} issues")
         ctx.textual.text("")
 
         # Show detailed table
-        ctx.textual.text("Found Issues:", markup="bold")
+        ctx.textual.bold_text("Found Issues:")
         ctx.textual.text("")
 
         try:
@@ -208,12 +207,13 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
             issues = sorted_issues
         except Exception as e:
             # If table rendering fails, show error but continue with raw issue list
-            ctx.textual.text(f"Error rendering table: {e}", markup="red")
-            ctx.textual.text(f"Found {len(issues)} issues (showing raw data)", markup="cyan")
+            ctx.textual.error_text(f"Error rendering table: {e}")
+            ctx.textual.primary_text(f"Found {len(issues)} issues (showing raw data)")
             for i, issue in enumerate(issues, 1):
                 ctx.textual.text(f"{i}. {issue.key} - {getattr(issue, 'summary', 'N/A')}")
             ctx.textual.text("")
 
+        ctx.textual.end_step("success")
         return Success(
             f"Found {len(issues)} issues using query: {query_name}",
             metadata={
@@ -225,13 +225,15 @@ def search_saved_query_step(ctx: WorkflowContext) -> WorkflowResult:
 
     except JiraAPIError as e:
         error_msg = f"JIRA search failed: {e}"
-        ctx.textual.mount(Panel(error_msg, panel_type="error"))
+        ctx.textual.error_text(error_msg)
+        ctx.textual.end_step("error")
         return Error(error_msg)
     except Exception as e:
         import traceback
         error_detail = traceback.format_exc()
         error_msg = f"Unexpected error: {e}\n\nTraceback:\n{error_detail}"
-        ctx.textual.mount(Panel(error_msg, panel_type="error"))
+        ctx.textual.error_text(error_msg)
+        ctx.textual.end_step("error")
         return Error(error_msg)
 
 

titan_plugin_jira/utils/__init__.py

@@ -9,5 +9,5 @@ __all__ = [
     "SavedQueries",
     "SAVED_QUERIES",
     "IssueSorter",
-    "IssueSortConfig"
+    "IssueSortConfig",
 ]