Merge pull request #28 from baselinrhq/feat/ui-cli

feat: Added ui cli command

Author: bwaite15104

README.md

@@ -27,7 +27,7 @@
 - **Dagster Integration**: Built-in orchestration support with Dagster assets and schedules
 - **Configuration-Driven**: Simple YAML/JSON configuration for defining profiling targets
 - **Historical Tracking**: Store profiling results over time for trend analysis
-- **CLI Interface**: Comprehensive command-line interface for profiling, drift detection, querying, and schema management
+- **CLI Interface**: Comprehensive command-line interface for profiling, drift detection, querying, schema management, and dashboard UI
 
 ## 📋 Requirements
 
@@ -190,7 +190,24 @@ baselinr status --config config.yml --watch
 baselinr status --config config.yml --json
 ```
 
-### 7. Manage Schema Migrations
+### 7. Start Dashboard UI
+
+Launch the web dashboard to view profiling runs, drift alerts, and metrics:
+
+```bash
+# Start dashboard (foreground mode)
+baselinr ui --config config.yml
+
+# Custom ports
+baselinr ui --config config.yml --port-backend 8080 --port-frontend 3001
+
+# Localhost only
+baselinr ui --config config.yml --host 127.0.0.1
+```
+
+Press `Ctrl+C` to stop the dashboard. See [docs/schemas/UI_COMMAND.md](docs/schemas/UI_COMMAND.md) for more details.
+
+### 8. Manage Schema Migrations
 
 Check and apply schema migrations:
 
@@ -447,6 +464,9 @@ baselinr status --config examples/config.yml --watch
 # Status with JSON output
 baselinr status --config examples/config.yml --json
 
+# Start dashboard UI
+baselinr ui --config examples/config.yml
+
 # Check schema migration status
 baselinr migrate status --config examples/config.yml
 

baselinr/cli.py

@@ -584,6 +584,36 @@ def query_command(args):
         return 1
 
 
+def ui_command(args):
+    """Execute UI command."""
+    try:
+        # Load configuration
+        config = ConfigLoader.load_from_file(args.config)
+
+        # Import UI startup function
+        from .ui import start_dashboard_foreground
+        from .ui.dependencies import check_all_dependencies
+
+        # Check dependencies first
+        check_all_dependencies(config, args.port_backend, args.port_frontend, args.host)
+
+        # Start dashboard in foreground
+        start_dashboard_foreground(
+            config,
+            backend_port=args.port_backend,
+            frontend_port=args.port_frontend,
+            backend_host=args.host,
+        )
+        return 0
+    except KeyboardInterrupt:
+        logger.info("UI command interrupted by user")
+        return 0
+    except Exception as e:
+        logger.error(f"UI command failed: {e}", exc_info=True)
+        print(f"\nError: {e}")
+        return 1
+
+
 def status_command(args):
     """Execute status command."""
     try:
@@ -1220,6 +1250,30 @@ def main():
         help="Auto-refresh every N seconds (default: 5). Use --watch 0 to disable.",
     )
 
+    # UI command
+    ui_parser = subparsers.add_parser("ui", help="Start local dashboard")
+    ui_parser.add_argument(
+        "--config", "-c", required=True, help="Path to configuration file (YAML or JSON)"
+    )
+    ui_parser.add_argument(
+        "--port-backend",
+        type=int,
+        default=8000,
+        help="Backend API port (default: 8000)",
+    )
+    ui_parser.add_argument(
+        "--port-frontend",
+        type=int,
+        default=3000,
+        help="Frontend UI port (default: 3000)",
+    )
+    ui_parser.add_argument(
+        "--host",
+        type=str,
+        default="0.0.0.0",
+        help="Backend host (default: 0.0.0.0)",
+    )
+
     # Parse arguments
     args = parser.parse_args()
 
@@ -1246,6 +1300,8 @@ def main():
         return query_command(args)
     elif args.command == "status":
         return status_command(args)
+    elif args.command == "ui":
+        return ui_command(args)
     else:
         parser.print_help()
         return 1

baselinr/ui/__init__.py

@@ -0,0 +1,21 @@
+"""UI command utilities for Baselinr dashboard."""
+
+from .config_builder import build_connection_string
+from .dependencies import (
+    check_all_dependencies,
+    check_database_connection,
+    check_nodejs,
+    check_ports,
+    check_python_packages,
+)
+from .startup import start_dashboard_foreground
+
+__all__ = [
+    "build_connection_string",
+    "check_nodejs",
+    "check_python_packages",
+    "check_ports",
+    "check_database_connection",
+    "check_all_dependencies",
+    "start_dashboard_foreground",
+]

baselinr/ui/config_builder.py

@@ -0,0 +1,49 @@
+"""Configuration builder for dashboard database connection."""
+
+import logging
+
+from ..config.schema import BaselinrConfig, DatabaseType
+
+logger = logging.getLogger(__name__)
+
+
+def build_connection_string(config: BaselinrConfig) -> str:
+    """
+    Build PostgreSQL connection string from BaselinrConfig.
+
+    The dashboard currently only supports PostgreSQL and SQLite, so we convert
+    the storage connection config to a connection string.
+
+    Args:
+        config: Baselinr configuration
+
+    Returns:
+        PostgreSQL or SQLite connection string
+
+    Raises:
+        ValueError: If the configured database type is not PostgreSQL or SQLite.
+    """
+    conn_config = config.storage.connection
+    conn_type = conn_config.type
+
+    if conn_type == DatabaseType.POSTGRES:
+        host = conn_config.host or "localhost"
+        port = conn_config.port or 5432
+        username = conn_config.username or "baselinr"
+        password = conn_config.password or ""
+        database = conn_config.database
+
+        if password:
+            return f"postgresql://{username}:{password}@{host}:{port}/{database}"
+        else:
+            return f"postgresql://{username}@{host}:{port}/{database}"
+    elif conn_type == DatabaseType.SQLITE:
+        filepath = conn_config.filepath
+        if not filepath:
+            raise ValueError("SQLite connection requires 'filepath' to be specified.")
+        return f"sqlite:///{filepath}"
+    else:
+        raise ValueError(
+            f"Dashboard currently only supports PostgreSQL or SQLite. "
+            f"Configured type: {conn_type.value}"
+        )