cursorflow 2.6.2__py3-none-any.whl → 2.7.1__py3-none-any.whl

cursorflow/cli.py

@@ -15,8 +15,12 @@ from typing import Dict
 from rich.console import Console
 from rich.table import Table
 from rich.progress import Progress, SpinnerColumn, TextColumn
+from rich.markup import escape
 
 from .core.agent import TestAgent
+from .core.output_manager import OutputManager
+from .core.data_presenter import DataPresenter
+from .core.query_engine import QueryEngine
 from . import __version__
 
 console = Console()
@@ -185,10 +189,10 @@ def test(base_url, path, actions, output, logs, config, verbose, headless, timeo
                 test_actions = json.loads(actions)
                 console.print(f"📋 Using inline actions")
         except json.JSONDecodeError as e:
-            console.print(f"[red]❌ Invalid JSON in actions: {e}[/red]")
+            console.print(f"[red]❌ Invalid JSON in actions: {escape(str(e))}[/red]")
             return
         except Exception as e:
-            console.print(f"[red]❌ Failed to load actions: {e}[/red]")
+            console.print(f"[red]❌ Failed to load actions: {escape(str(e))}[/red]")
             return
     elif path:
         # Simple path navigation with optional wait conditions
@@ -236,7 +240,7 @@ def test(base_url, path, actions, output, logs, config, verbose, headless, timeo
             **agent_config
         )
     except Exception as e:
-        console.print(f"[red]Error initializing CursorFlow: {e}[/red]")
+        console.print(f"[red]Error initializing CursorFlow: {escape(str(e))}[/red]")
         return
     
     # Execute test actions
@@ -292,20 +296,35 @@ def test(base_url, path, actions, output, logs, config, verbose, headless, timeo
             if timeline:
                 console.print(f"⏰ Timeline events: {len(timeline)}")
         
-        # Save results to file for Cursor analysis
-        if not output:
-            # Auto-generate meaningful filename in .cursorflow/artifacts/
-            session_id = results.get('session_id', 'unknown')
-            path_part = path.replace('/', '_') if path else 'root'
-            
-            # Ensure .cursorflow/artifacts directory exists
-            artifacts_dir = Path('.cursorflow/artifacts')
-            artifacts_dir.mkdir(parents=True, exist_ok=True)
-            
-            output = artifacts_dir / f"cursorflow_{path_part}_{session_id}.json"
+        # Save results in structured multi-file format
+        session_id = results.get('session_id', 'unknown')
+        test_desc = path if path else 'test'
         
-        with open(output, 'w') as f:
-            json.dump(results, f, indent=2, default=str)
+        # Use output manager to save structured results
+        output_mgr = OutputManager()
+        file_paths = output_mgr.save_structured_results(
+            results, 
+            session_id, 
+            test_desc
+        )
+        
+        # Generate AI-optimized data digest
+        session_dir = output_mgr.get_session_path(session_id)
+        data_pres = DataPresenter()
+        digest_content = data_pres.generate_data_digest(session_dir, results)
+        
+        digest_path = session_dir / "data_digest.md"
+        with open(digest_path, 'w', encoding='utf-8') as f:
+            f.write(digest_content)
+        file_paths['data_digest'] = str(digest_path)
+        
+        if not quiet:
+            console.print(f"\n📁 [bold green]Results saved to:[/bold green] [cyan]{session_dir}[/cyan]")
+            console.print(f"📄 [bold]AI Summary:[/bold] [cyan]{digest_path}[/cyan]")
+            console.print(f"\n💡 [dim]Quick commands:[/dim]")
+            console.print(f"   cursorflow query {session_id} --errors")
+            console.print(f"   cursorflow query {session_id} --network")
+            console.print(f"   cat {digest_path}")
         
         # Save command for rerun (Phase 3.3)
         last_test_data = {
@@ -339,12 +358,326 @@ def test(base_url, path, actions, output, logs, config, verbose, headless, timeo
                 console.print(f"💡 View manually: playwright show-trace {trace_path}")
         
     except Exception as e:
-        console.print(f"[red]❌ Test failed: {e}[/red]")
+        console.print(f"[red]❌ Test failed: {escape(str(e))}[/red]")
         if verbose:
             import traceback
             console.print(traceback.format_exc())
         raise
 
+
+@main.command()
+@click.argument('session_id', required=False)
+@click.option('--errors', is_flag=True, help='Query error data')
+@click.option('--network', is_flag=True, help='Query network requests')
+@click.option('--console', 'console_opt', is_flag=True, help='Query console messages')
+@click.option('--performance', is_flag=True, help='Query performance metrics')
+@click.option('--summary', is_flag=True, help='Query summary data')
+@click.option('--dom', is_flag=True, help='Query DOM analysis')
+@click.option('--server-logs', is_flag=True, help='Query server logs')
+@click.option('--screenshots', is_flag=True, help='Query screenshot metadata')
+@click.option('--mockup', is_flag=True, help='Query mockup comparison results')
+@click.option('--responsive', is_flag=True, help='Query responsive testing results')
+@click.option('--css-iterations', is_flag=True, help='Query CSS iteration results')
+@click.option('--timeline', is_flag=True, help='Query timeline events')
+@click.option('--severity', type=str, help='Filter errors by severity (critical)')
+@click.option('--level', type=str, help='Filter server logs by level (error,warning,info)')
+@click.option('--status', type=str, help='Filter network by status codes (404,500 or 4xx,5xx)')
+@click.option('--failed', is_flag=True, help='Show only failed network requests')
+@click.option('--type', type=str, help='Filter console by type (error,warning,log,info)')
+@click.option('--selector', type=str, help='Filter DOM by CSS selector')
+@click.option('--source', type=str, help='Filter server logs by source (ssh,local,docker,systemd)')
+@click.option('--file', type=str, help='Filter server logs by file path')
+@click.option('--pattern', type=str, help='Filter by content pattern')
+@click.option('--contains', type=str, help='Filter by content substring')
+@click.option('--matches', type=str, help='Filter by regex pattern')
+@click.option('--from-file', type=str, help='Filter errors by source file')
+@click.option('--from-pattern', type=str, help='Filter errors by file pattern (*.js, *.ts)')
+@click.option('--url-contains', type=str, help='Filter network by URL substring')
+@click.option('--url-matches', type=str, help='Filter network by URL regex')
+@click.option('--over', type=str, help='Filter network requests over timing threshold (500ms)')
+@click.option('--method', type=str, help='Filter network by HTTP method (GET,POST)')
+@click.option('--visible', is_flag=True, help='Filter DOM to visible elements only')
+@click.option('--interactive', is_flag=True, help='Filter DOM to interactive elements only')
+@click.option('--role', type=str, help='Filter DOM by ARIA role')
+@click.option('--with-attr', type=str, help='Filter DOM by attribute name')
+@click.option('--with-network', is_flag=True, help='Include related network requests (cross-ref)')
+@click.option('--with-console', is_flag=True, help='Include related console messages (cross-ref)')
+@click.option('--with-server-logs', is_flag=True, help='Include related server logs (cross-ref)')
+@click.option('--context-for-error', type=int, help='Get full context for error by index')
+@click.option('--group-by-url', type=str, help='Group all data by URL pattern')
+@click.option('--group-by-selector', type=str, help='Group all data by DOM selector')
+@click.option('--viewport', type=str, help='Filter responsive results by viewport (mobile,tablet,desktop)')
+@click.option('--iteration', type=int, help='Filter by specific iteration number')
+@click.option('--with-errors', is_flag=True, help='Filter screenshots/iterations with errors only')
+@click.option('--around', type=float, help='Query timeline events around timestamp')
+@click.option('--window', type=float, default=5.0, help='Time window in seconds (default: 5)')
+@click.option('--export', type=click.Choice(['json', 'markdown', 'csv']), 
+              default='json', help='Export format')
+@click.option('--compare-with', type=str, help='Compare with another session')
+@click.option('--list', 'list_sessions', is_flag=True, help='List recent sessions')
+@click.option('--verbose', '-v', is_flag=True, help='Verbose output')
+def query(session_id, errors, network, console_opt, performance, summary, dom,
+         server_logs, screenshots, mockup, responsive, css_iterations, timeline,
+         severity, level, status, failed, type, selector, source, file, pattern, 
+         contains, matches, from_file, from_pattern, url_contains, url_matches, over, method,
+         visible, interactive, role, with_attr, with_network, with_console, with_server_logs,
+         context_for_error, group_by_url, group_by_selector,
+         viewport, iteration, with_errors, around, window,
+         export, compare_with, list_sessions, verbose):
+    """
+    Query and filter CursorFlow test results
+    
+    Examples:
+    
+        # List recent sessions
+        cursorflow query --list
+        
+        # Query errors from a session
+        cursorflow query session_123 --errors
+        
+        # Query server logs
+        cursorflow query session_123 --server-logs --severity error
+        
+        # Query network failures
+        cursorflow query session_123 --network --failed
+        
+        # Query responsive results
+        cursorflow query session_123 --responsive --viewport mobile
+        
+        # Export in different formats
+        cursorflow query session_123 --errors --export markdown
+        
+        # Compare two sessions
+        cursorflow query session_123 --compare-with session_456
+    """
+    
+    engine = QueryEngine()
+    
+    # List sessions mode
+    if list_sessions:
+        _list_sessions_func(engine)
+        return
+    
+    if not session_id:
+        console.print("[yellow]⚠️  Provide a session_id or use --list to see available sessions[/yellow]")
+        console.print("Example: [cyan]cursorflow query session_123 --errors[/cyan]")
+        return
+    
+    # Comparison mode
+    if compare_with:
+        _compare_sessions_func(engine, session_id, compare_with, errors, network, performance)
+        return
+    
+    # Determine query type
+    query_type = None
+    if errors:
+        query_type = 'errors'
+    elif network:
+        query_type = 'network'
+    elif console_opt:
+        query_type = 'console'
+    elif performance:
+        query_type = 'performance'
+    elif summary:
+        query_type = 'summary'
+    elif dom:
+        query_type = 'dom'
+    elif server_logs:
+        query_type = 'server_logs'
+    elif screenshots:
+        query_type = 'screenshots'
+    elif mockup:
+        query_type = 'mockup'
+    elif responsive:
+        query_type = 'responsive'
+    elif css_iterations:
+        query_type = 'css_iterations'
+    elif timeline:
+        query_type = 'timeline'
+    
+    # Build filters
+    filters = {}
+    if severity:
+        filters['severity'] = severity
+    if level:
+        filters['level'] = level
+    if status:
+        filters['status'] = status
+    if failed:
+        filters['failed'] = True
+    if type:
+        filters['type'] = type
+    if selector:
+        filters['selector'] = selector
+    if source:
+        filters['source'] = source
+    if file:
+        filters['file'] = file
+    if pattern:
+        filters['pattern'] = pattern
+    if contains:
+        filters['contains'] = contains
+    if matches:
+        filters['matches'] = matches
+    if from_file:
+        filters['from_file'] = from_file
+    if from_pattern:
+        filters['from_pattern'] = from_pattern
+    if url_contains:
+        filters['url_contains'] = url_contains
+    if url_matches:
+        filters['url_matches'] = url_matches
+    if over:
+        filters['over'] = over
+    if method:
+        filters['method'] = method
+    if visible:
+        filters['visible'] = True
+    if interactive:
+        filters['interactive'] = True
+    if role:
+        filters['role'] = role
+    if with_attr:
+        filters['with_attr'] = with_attr
+    if with_network:
+        filters['with_network'] = True
+    if with_console:
+        filters['with_console'] = True
+    if with_server_logs:
+        filters['with_server_logs'] = True
+    if context_for_error is not None:
+        filters['context_for_error'] = context_for_error
+    if group_by_url:
+        filters['group_by_url'] = group_by_url
+    if group_by_selector:
+        filters['group_by_selector'] = group_by_selector
+    if viewport:
+        filters['viewport'] = viewport
+    if iteration:
+        filters['iteration'] = iteration
+    if with_errors:
+        filters['with_errors'] = True
+    if around:
+        filters['around'] = around
+        filters['window'] = window
+    
+    try:
+        # Execute query
+        result = engine.query_session(session_id, query_type, filters, export)
+        
+        # Display results
+        if export == 'json':
+            if verbose:
+                console.print(result)
+            else:
+                # Pretty print JSON
+                data = json.loads(result)
+                console.print_json(data=data)
+        else:
+            console.print(result)
+    
+    except ValueError as e:
+        console.print(f"[red]Error:[/red] {e}")
+        console.print(f"\n💡 [dim]Tip:[/dim] List available sessions with: [cyan]cursorflow query --list[/cyan]")
+    except Exception as e:
+        console.print(f"[red]Error querying session:[/red] {e}")
+        if verbose:
+            import traceback
+            console.print(traceback.format_exc())
+
+
+def _list_sessions_func(engine: QueryEngine):
+    """List recent sessions"""
+    sessions = engine.list_sessions(limit=10)
+    
+    if not sessions:
+        console.print("[yellow]No sessions found[/yellow]")
+        console.print("Run a test first: [cyan]cursorflow test --base-url http://localhost:3000 --path /[/cyan]")
+        return
+    
+    table = Table(title="Recent Test Sessions", box=box.ROUNDED)
+    table.add_column("Session ID", style="cyan")
+    table.add_column("Timestamp", style="white")
+    table.add_column("Status", style="green")
+    table.add_column("Errors", style="red")
+    table.add_column("Network Failures", style="yellow")
+    
+    for session in sessions:
+        status_emoji = "✅" if session['success'] else "⚠️"
+        table.add_row(
+            session['session_id'],
+            session['timestamp'],
+            status_emoji,
+            str(session['errors']),
+            str(session['network_failures'])
+        )
+    
+    console.print(table)
+    console.print(f"\n💡 [dim]Query a session:[/dim] [cyan]cursorflow query <session_id> --errors[/cyan]")
+
+
+def _compare_sessions_func(engine: QueryEngine, session_a: str, session_b: str, 
+                           errors: bool, network: bool, performance: bool):
+    """Compare two sessions"""
+    try:
+        query_type = None
+        if errors:
+            query_type = 'errors'
+        elif network:
+            query_type = 'network'
+        elif performance:
+            query_type = 'performance'
+        
+        comparison = engine.compare_sessions(session_a, session_b, query_type)
+        
+        console.print(f"\n[bold]Comparing Sessions:[/bold]")
+        console.print(f"  Session A: [cyan]{session_a}[/cyan]")
+        console.print(f"  Session B: [cyan]{session_b}[/cyan]")
+        console.print()
+        
+        # Display summary comparison
+        summary_diff = comparison.get('summary_diff', {})
+        
+        table = Table(title="Summary Comparison", box=box.ROUNDED)
+        table.add_column("Metric", style="white")
+        table.add_column("Session A", style="cyan")
+        table.add_column("Session B", style="cyan")
+        table.add_column("Difference", style="yellow")
+        
+        for metric, values in summary_diff.items():
+            diff = values.get('difference', 0)
+            diff_str = f"+{diff}" if diff > 0 else str(diff)
+            table.add_row(
+                metric.replace('_', ' ').title(),
+                str(values.get('session_a', 0)),
+                str(values.get('session_b', 0)),
+                diff_str
+            )
+        
+        console.print(table)
+        
+        # Display specific comparison if requested
+        if errors and 'errors_diff' in comparison:
+            console.print(f"\n[bold]Errors Comparison:[/bold]")
+            errors_diff = comparison['errors_diff']
+            console.print(f"  New errors: [red]{errors_diff.get('new_errors', 0)}[/red]")
+        
+        if network and 'network_diff' in comparison:
+            console.print(f"\n[bold]Network Comparison:[/bold]")
+            network_diff = comparison['network_diff']
+            console.print(f"  Success rate A: {network_diff.get('success_rate_a', 0):.1f}%")
+            console.print(f"  Success rate B: {network_diff.get('success_rate_b', 0):.1f}%")
+        
+        if performance and 'performance_diff' in comparison:
+            console.print(f"\n[bold]Performance Comparison:[/bold]")
+            perf_diff = comparison['performance_diff']
+            exec_diff = perf_diff.get('execution_time', {}).get('difference', 0)
+            console.print(f"  Execution time difference: {exec_diff:.2f}s")
+    
+    except ValueError as e:
+        console.print(f"[red]Error:[/red] {e}")
+
+
 @main.command()
 @click.argument('mockup_url', required=True)
 @click.option('--base-url', '-u', default='http://localhost:3000',
@@ -398,7 +731,7 @@ def compare_mockup(mockup_url, base_url, mockup_actions, implementation_actions,
             comparison_config["viewports"] = viewports_parsed
         
     except Exception as e:
-        console.print(f"[red]Error parsing input parameters: {e}[/red]")
+        console.print(f"[red]Error parsing input parameters: {escape(str(e))}[/red]")
         return
     
     # Initialize CursorFlow
@@ -411,7 +744,7 @@ def compare_mockup(mockup_url, base_url, mockup_actions, implementation_actions,
             browser_config={'headless': True}
         )
     except Exception as e:
-        console.print(f"[red]Error initializing CursorFlow: {e}[/red]")
+        console.print(f"[red]Error initializing CursorFlow: {escape(str(e))}[/red]")
         return
     
     # Execute mockup comparison
@@ -425,7 +758,7 @@ def compare_mockup(mockup_url, base_url, mockup_actions, implementation_actions,
         ))
         
         if "error" in results:
-            console.print(f"[red]❌ Comparison failed: {results['error']}[/red]")
+            console.print(f"[red]❌ Comparison failed: {escape(str(results['error']))}[/red]")
             return
         
         # Display results summary (pure metrics only)
@@ -453,7 +786,7 @@ def compare_mockup(mockup_url, base_url, mockup_actions, implementation_actions,
         console.print(f"📁 Visual diffs stored in: [cyan].cursorflow/artifacts/[/cyan]")
         
     except Exception as e:
-        console.print(f"[red]❌ Comparison failed: {e}[/red]")
+        console.print(f"[red]❌ Comparison failed: {escape(str(e))}[/red]")
         if verbose:
             import traceback
             console.print(traceback.format_exc())
@@ -500,7 +833,7 @@ def iterate_mockup(mockup_url, base_url, css_improvements, base_actions, diff_th
         comparison_config = {"diff_threshold": diff_threshold}
         
     except Exception as e:
-        console.print(f"[red]Error parsing input parameters: {e}[/red]")
+        console.print(f"[red]Error parsing input parameters: {escape(str(e))}[/red]")
         return
     
     # Initialize CursorFlow
@@ -512,7 +845,7 @@ def iterate_mockup(mockup_url, base_url, css_improvements, base_actions, diff_th
             browser_config={'headless': True}
         )
     except Exception as e:
-        console.print(f"[red]Error initializing CursorFlow: {e}[/red]")
+        console.print(f"[red]Error initializing CursorFlow: {escape(str(e))}[/red]")
         return
     
     # Execute iterative mockup matching
@@ -526,7 +859,7 @@ def iterate_mockup(mockup_url, base_url, css_improvements, base_actions, diff_th
         ))
         
         if "error" in results:
-            console.print(f"[red]❌ Iteration failed: {results['error']}[/red]")
+            console.print(f"[red]❌ Iteration failed: {escape(str(results['error']))}[/red]")
             return
         
         # Display results summary
@@ -563,7 +896,7 @@ def iterate_mockup(mockup_url, base_url, css_improvements, base_actions, diff_th
         console.print(f"📁 Iteration progress stored in: [cyan].cursorflow/artifacts/[/cyan]")
         
     except Exception as e:
-        console.print(f"[red]❌ Iteration failed: {e}[/red]")
+        console.print(f"[red]❌ Iteration failed: {escape(str(e))}[/red]")
         if verbose:
             import traceback
             console.print(traceback.format_exc())
@@ -620,7 +953,7 @@ async def _run_auto_tests(framework: str, base_url: str, config: Dict):
         display_smoke_test_summary(results)
         
     except Exception as e:
-        console.print(f"[red]Auto-test failed: {e}[/red]")
+        console.print(f"[red]Auto-test failed: {escape(str(e))}[/red]")
 
 @main.command()
 @click.argument('project_path', default='.')
@@ -648,7 +981,7 @@ def install_rules(project_path, framework, force, yes):
             console.print("[red]❌ Installation failed[/red]")
             
     except Exception as e:
-        console.print(f"[red]Installation error: {e}[/red]")
+        console.print(f"[red]Installation error: {escape(str(e))}[/red]")
 
 @main.command()
 @click.option('--force', is_flag=True, help='Force update even if no updates available')
@@ -670,7 +1003,7 @@ def update(force, project_dir):
             console.print("[red]❌ Update failed[/red]")
             
     except Exception as e:
-        console.print(f"[red]Update error: {e}[/red]")
+        console.print(f"[red]Update error: {escape(str(e))}[/red]")
 
 @main.command()
 @click.option('--project-dir', default='.', help='Project directory')
@@ -684,7 +1017,7 @@ def check_updates(project_dir):
         result = asyncio.run(check_updates(project_dir))
         
         if "error" in result:
-            console.print(f"[red]Error checking updates: {result['error']}[/red]")
+            console.print(f"[red]Error checking updates: {escape(str(result['error']))}[/red]")
             return
         
         # Display update information
@@ -723,7 +1056,7 @@ def check_updates(project_dir):
             console.print("\n💡 Run [bold]cursorflow update[/bold] to install updates")
         
     except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
+        console.print(f"[red]Error: {escape(str(e))}[/red]")
 
 @main.command()
 @click.option('--project-dir', default='.', help='Project directory')
@@ -744,7 +1077,7 @@ def install_deps(project_dir):
             console.print("[red]❌ Dependency installation failed[/red]")
             
     except Exception as e:
-        console.print(f"[red]Error: {e}[/red]")
+        console.print(f"[red]Error: {escape(str(e))}[/red]")
 
 @main.command()
 @click.argument('subcommand', required=False)
@@ -959,7 +1292,7 @@ def inspect(base_url, path, selector, verbose):
             console.print(f"\n📸 Screenshot saved: [cyan]{screenshot_path}[/cyan]")
         
     except Exception as e:
-        console.print(f"[red]❌ Inspection failed: {e}[/red]")
+        console.print(f"[red]❌ Inspection failed: {escape(str(e))}[/red]")
         import traceback
         console.print(traceback.format_exc())
 
@@ -1086,7 +1419,7 @@ def measure(base_url, path, selector, verbose):
         console.print("✅ Measurement complete")
         
     except Exception as e:
-        console.print(f"[red]❌ Measurement failed: {e}[/red]")
+        console.print(f"[red]❌ Measurement failed: {escape(str(e))}[/red]")
         import traceback
         console.print(traceback.format_exc())
 
@@ -1137,7 +1470,7 @@ def count(base_url, path, selector):
             console.print(f"💡 Total elements on page: {len(elements)}")
         
     except Exception as e:
-        console.print(f"[red]❌ Count failed: {e}[/red]")
+        console.print(f"[red]❌ Count failed: {escape(str(e))}[/red]")
         import traceback
         console.print(traceback.format_exc())
 
@@ -1174,7 +1507,7 @@ def rerun(click, hover):
         console.print("✅ Rerun completed")
         
     except Exception as e:
-        console.print(f"[red]❌ Rerun failed: {e}[/red]")
+        console.print(f"[red]❌ Rerun failed: {escape(str(e))}[/red]")
 
 @main.command()
 @click.option('--session', '-s', required=True, help='Session ID to view timeline for')
@@ -1227,7 +1560,7 @@ def timeline(session):
             console.print(f"\n... and {len(timeline) - 50} more events")
         
     except Exception as e:
-        console.print(f"[red]❌ Failed to load timeline: {e}[/red]")
+        console.print(f"[red]❌ Failed to load timeline: {escape(str(e))}[/red]")
 
 @main.command()
 @click.option('--artifacts', is_flag=True, help='Clean all artifacts (screenshots, traces)')
@@ -1337,7 +1670,7 @@ def cleanup(artifacts, sessions, old_only, clean_all, dry_run, yes):
                 item_path.unlink()
             deleted_count += 1
         except Exception as e:
-            console.print(f"[red]⚠️  Failed to delete {item_path}: {e}[/red]")
+            console.print(f"[red]⚠️  Failed to delete {item_path}: {escape(str(e))}[/red]")
     
     console.print(f"\n✅ Cleanup complete!")
     console.print(f"   • Deleted {deleted_count}/{len(items_to_delete)} items")
@@ -1418,7 +1751,7 @@ def _display_test_results(results: Dict, test_description: str, show_console: bo
         if errors:
             console.print(f"\n[red]❌ Console Errors ({len(errors)}):[/red]")
             for error in errors[:5]:  # Show first 5
-                console.print(f"   [red]{error.get('text', 'Unknown error')}[/red]")
+                console.print(f"   [red]{escape(str(error.get('text', 'Unknown error')))}[/red]")
         
         if warnings:
             console.print(f"\n[yellow]⚠️  Console Warnings ({len(warnings)}):[/yellow]")

cursorflow/core/cursorflow.py

@@ -21,6 +21,8 @@ from .css_iterator import CSSIterator
 from .cursor_integration import CursorIntegration
 from .persistent_session import PersistentSession, get_session_manager
 from .mockup_comparator import MockupComparator
+from .output_manager import OutputManager
+from .data_presenter import DataPresenter
 
 
 class CursorFlow:
@@ -74,6 +76,10 @@ class CursorFlow:
         self.cursor_integration = CursorIntegration()
         self.mockup_comparator = MockupComparator()
         
+        # Initialize output manager and data presenter
+        self.output_manager = OutputManager()
+        self.data_presenter = DataPresenter()
+        
         # Session tracking
         self.session_id = None
         self.timeline = []
@@ -166,7 +172,8 @@ class CursorFlow:
                 "server_logs": server_logs,       # Raw server logs
                 "summary": summary,               # Basic counts
                 "artifacts": self.artifacts,
-                "comprehensive_data": comprehensive_data  # Complete page intelligence
+                "comprehensive_data": comprehensive_data,  # Complete page intelligence
+                "test_description": session_options.get('test_description', 'test') if session_options else 'test'
             }
             
             self.logger.info(f"Test execution completed: {success}, timeline events: {len(organized_timeline)}")