golf-mcp 0.1.20__py3-none-any.whl → 0.2.1__py3-none-any.whl

golf/commands/init.py

@@ -7,6 +7,13 @@ from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn
 from rich.prompt import Confirm
 
+from golf.cli.branding import (
+    create_success_message,
+    create_info_panel,
+    STATUS_ICONS,
+    GOLF_ORANGE,
+)
+
 from golf.core.telemetry import (
     track_command,
     track_event,
@@ -20,35 +27,21 @@ console = Console()
 def initialize_project(
     project_name: str,
     output_dir: Path,
-    template: str = "basic",
 ) -> None:
-    """Initialize a new GolfMCP project with the specified template.
+    """Initialize a new GolfMCP project.
 
     Args:
         project_name: Name of the project
         output_dir: Directory where the project will be created
-        template: Template to use (basic or api_key)
     """
     try:
-        # Validate template
-        valid_templates = ("basic", "api_key")
-        if template not in valid_templates:
-            console.print(f"[bold red]Error:[/bold red] Unknown template '{template}'")
-            console.print(f"Available templates: {', '.join(valid_templates)}")
-            track_command(
-                "init",
-                success=False,
-                error_type="InvalidTemplate",
-                error_message=f"Unknown template: {template}",
-            )
-            return
+        # Use the basic template by default
+        template = "basic"
 
         # Check if directory exists
         if output_dir.exists():
             if not output_dir.is_dir():
-                console.print(
-                    f"[bold red]Error:[/bold red] '{output_dir}' exists but is not a directory."
-                )
+                console.print(f"[bold red]Error:[/bold red] '{output_dir}' exists but is not a directory.")
                 track_command(
                     "init",
                     success=False,
@@ -78,9 +71,7 @@ def initialize_project(
         template_dir = package_init_file.parent / "examples" / template
 
         if not template_dir.exists():
-            console.print(
-                f"[bold red]Error:[/bold red] Could not find template '{template}'"
-            )
+            console.print(f"[bold red]Error:[/bold red] Could not find template '{template}'")
             track_command(
                 "init",
                 success=False,
@@ -92,7 +83,9 @@ def initialize_project(
         # Copy template files
         with Progress(
             SpinnerColumn(),
-            TextColumn("[bold green]Creating project structure...[/bold green]"),
+            TextColumn(
+                f"[bold {GOLF_ORANGE}]{STATUS_ICONS['building']} Creating project structure...[/bold {GOLF_ORANGE}]"
+            ),
             transient=True,
         ) as progress:
             progress.add_task("copying", total=None)
@@ -103,11 +96,13 @@ def initialize_project(
         # Ask for telemetry consent
         _prompt_for_telemetry_consent()
 
-        # Create virtual environment
-        console.print("[bold green]Project initialized successfully![/bold green]")
-        console.print("\nTo get started, run:")
-        console.print(f"  cd {output_dir.name}")
-        console.print("  golf build dev")
+        # Show success message
+        console.print()
+        create_success_message("Project initialized successfully!", console)
+
+        # Show next steps
+        next_steps = f"cd {output_dir.name}\ngolf build dev\ngolf run"
+        create_info_panel("Next Steps", next_steps, console)
 
         # Track successful initialization
         track_event("cli_init_success", {"success": True, "template": template})
@@ -116,12 +111,8 @@ def initialize_project(
         error_type = type(e).__name__
         error_message = str(e)
 
-        console.print(
-            f"[bold red]Error during initialization:[/bold red] {error_message}"
-        )
-        track_command(
-            "init", success=False, error_type=error_type, error_message=error_message
-        )
+        console.print(f"[bold red]Error during initialization:[/bold red] {error_message}")
+        track_command("init", success=False, error_type=error_type, error_message=error_message)
 
         # Re-raise to maintain existing behavior
         raise
@@ -160,9 +151,7 @@ def _copy_template(source_dir: Path, target_dir: Path, project_name: str) -> Non
 
             # Replace template variables
             content = content.replace("{{project_name}}", project_name)
-            content = content.replace(
-                "{{project_name_lowercase}}", project_name.lower()
-            )
+            content = content.replace("{{project_name_lowercase}}", project_name.lower())
 
             with open(target_path, "w", encoding="utf-8") as f:
                 f.write(content)
@@ -170,13 +159,6 @@ def _copy_template(source_dir: Path, target_dir: Path, project_name: str) -> Non
             # Binary file, just copy
             shutil.copy2(source_path, target_path)
 
-    # Create .env file
-    env_file = target_dir / ".env"
-    with open(env_file, "w", encoding="utf-8") as f:
-        f.write(f"GOLF_NAME={project_name}\n")
-        f.write("GOLF_HOST=127.0.0.1\n")
-        f.write("GOLF_PORT=3000\n")
-
     # Create a .gitignore if it doesn't exist
     gitignore_file = target_dir / ".gitignore"
     if not gitignore_file.exists():
@@ -219,7 +201,8 @@ def _prompt_for_telemetry_consent() -> None:
     """Prompt user for telemetry consent and save their preference."""
     import os
 
-    # Skip prompt in test mode, when telemetry is explicitly disabled, or if preference already exists
+    # Skip prompt in test mode, when telemetry is explicitly disabled, or if
+    # preference already exists
     if os.environ.get("GOLF_TEST_MODE", "").lower() in ("1", "true", "yes", "on"):
         return
 
@@ -235,9 +218,7 @@ def _prompt_for_telemetry_consent() -> None:
     console.print()
     console.rule("[bold blue]Anonymous usage analytics[/bold blue]", style="blue")
     console.print()
-    console.print(
-        "Golf can collect [bold]anonymous usage analytics[/bold] to help improve the tool."
-    )
+    console.print("Golf can collect [bold]anonymous usage analytics[/bold] to help improve the tool.")
     console.print()
     console.print("[dim]What we collect:[/dim]")
     console.print("  • Command usage (init, build, run)")
@@ -251,14 +232,10 @@ def _prompt_for_telemetry_consent() -> None:
     console.print("  • Personal information")
     console.print("  • IP addresses")
     console.print()
-    console.print(
-        "You can change this anytime by setting GOLF_TELEMETRY=0 in your environment."
-    )
+    console.print("You can change this anytime by setting GOLF_TELEMETRY=0 in your environment.")
     console.print()
 
-    enable_telemetry = Confirm.ask(
-        "[bold]Enable anonymous usage analytics?[/bold]", default=False
-    )
+    enable_telemetry = Confirm.ask("[bold]Enable anonymous usage analytics?[/bold]", default=False)
 
     set_telemetry_enabled(enable_telemetry, persist=True)
 

golf/commands/run.py

@@ -6,7 +6,11 @@ import sys
 from pathlib import Path
 
 from rich.console import Console
+from rich.panel import Panel
+from rich.align import Align
+from rich.text import Text
 
+from golf.cli.branding import create_command_header, get_status_text, STATUS_ICONS, GOLF_BLUE, GOLF_GREEN, GOLF_ORANGE
 from golf.core.config import Settings
 
 console = Console()
@@ -24,7 +28,8 @@ def run_server(
     Args:
         project_path: Path to the project root
         settings: Project settings
-        dist_dir: Path to the directory containing the built server (defaults to project_path/dist)
+        dist_dir: Path to the directory containing the built server
+            (defaults to project_path/dist)
         host: Host to bind the server to (overrides settings)
         port: Port to bind the server to (overrides settings)
 
@@ -38,11 +43,41 @@ def run_server(
     # Check if server file exists
     server_path = dist_dir / "server.py"
     if not server_path.exists():
-        console.print(
-            f"[bold red]Error: Server file {server_path} not found.[/bold red]"
-        )
+        console.print(get_status_text("error", f"Server file {server_path} not found"))
         return 1
 
+    # Display server startup header
+    create_command_header("Starting Server", f"{settings.name}", console)
+
+    # Show server info with flashy styling
+    server_host = host or settings.host or "localhost"
+    server_port = port or settings.port or 3000
+
+    server_content = Text()
+    server_content.append("🚀 ", style=f"bold {GOLF_ORANGE}")
+    server_content.append(f"{STATUS_ICONS['server']} Server starting on ", style=f"bold {GOLF_BLUE}")
+    server_content.append(f"http://{server_host}:{server_port}", style=f"bold {GOLF_GREEN}")
+    server_content.append(" 🚀", style=f"bold {GOLF_ORANGE}")
+    server_content.append("\n")
+
+    # Add telemetry status indicator
+    if settings.opentelemetry_enabled:
+        server_content.append("📊 Golf telemetry enabled", style=f"dim {GOLF_BLUE}")
+        server_content.append("\n")
+
+    server_content.append("⚡ Press Ctrl+C to stop ⚡", style=f"dim {GOLF_ORANGE}")
+
+    console.print(
+        Panel(
+            Align.center(server_content),
+            border_style=GOLF_BLUE,
+            padding=(1, 2),
+            title="[bold]🌐 SERVER READY 🌐[/bold]",
+            title_align="center",
+        )
+    )
+    console.print()
+
     # Prepare environment variables
     env = os.environ.copy()
     if host is not None:
@@ -65,31 +100,26 @@ def run_server(
         )
 
         # Provide more context about the exit
+        console.print()
         if process.returncode == 0:
-            console.print("[green]Server stopped successfully[/green]")
+            console.print(get_status_text("success", "Server stopped successfully"))
         elif process.returncode == 130:
-            console.print("[yellow]Server stopped by user interrupt (Ctrl+C)[/yellow]")
+            console.print(get_status_text("info", "Server stopped by user interrupt (Ctrl+C)"))
         elif process.returncode == 143:
-            console.print(
-                "[yellow]Server stopped by SIGTERM (graceful shutdown)[/yellow]"
-            )
+            console.print(get_status_text("info", "Server stopped by SIGTERM (graceful shutdown)"))
         elif process.returncode == 137:
-            console.print(
-                "[yellow]Server stopped by SIGKILL (forced shutdown)[/yellow]"
-            )
+            console.print(get_status_text("warning", "Server stopped by SIGKILL (forced shutdown)"))
         elif process.returncode in [1, 2]:
-            console.print(
-                f"[red]Server exited with error code {process.returncode}[/red]"
-            )
+            console.print(get_status_text("error", f"Server exited with error code {process.returncode}"))
         else:
-            console.print(
-                f"[orange]Server exited with code {process.returncode}[/orange]"
-            )
+            console.print(get_status_text("warning", f"Server exited with code {process.returncode}"))
 
         return process.returncode
     except KeyboardInterrupt:
-        console.print("\n[yellow]Server stopped by user (Ctrl+C)[/yellow]")
+        console.print()
+        console.print(get_status_text("info", "Server stopped by user (Ctrl+C)"))
         return 130  # Standard exit code for SIGINT
     except Exception as e:
-        console.print(f"\n[bold red]Error running server:[/bold red] {e}")
+        console.print()
+        console.print(get_status_text("error", f"Error running server: {e}"))
         return 1