zebra-day 0.0.37__py3-none-any.whl → 2.0.0__py3-none-any.whl

zebra_day/__init__.py

@@ -0,0 +1,35 @@
+"""
+zebra_day - A Python library to manage Zebra printer fleets and ZPL print requests.
+"""
+from __future__ import annotations
+
+from zebra_day.exceptions import (
+    ConfigError,
+    ConfigFileNotFoundError,
+    ConfigParseError,
+    LabelTemplateError,
+    LabelTemplateNotFoundError,
+    NetworkScanError,
+    PrinterConnectionError,
+    PrinterNotFoundError,
+    ZebraDayError,
+    ZPLRenderError,
+)
+from zebra_day.logging_config import configure_logging, get_logger
+
+__all__ = [
+    # Logging
+    "configure_logging",
+    "get_logger",
+    # Exceptions
+    "ZebraDayError",
+    "PrinterConnectionError",
+    "PrinterNotFoundError",
+    "ConfigError",
+    "ConfigFileNotFoundError",
+    "ConfigParseError",
+    "LabelTemplateError",
+    "LabelTemplateNotFoundError",
+    "ZPLRenderError",
+    "NetworkScanError",
+]

zebra_day/cli/__init__.py

@@ -0,0 +1,240 @@
+"""zebra_day CLI - Zebra Printer Fleet Management CLI using Typer."""
+
+import os
+import sys
+from pathlib import Path
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+from zebra_day import paths as xdg
+from zebra_day.cli.gui import gui_app
+from zebra_day.cli.printer import printer_app
+from zebra_day.cli.template import template_app
+from zebra_day.cli.cognito import cognito_app
+
+console = Console()
+
+app = typer.Typer(
+    name="zday",
+    help="zebra_day - Zebra Printer Fleet Management CLI",
+    add_completion=True,
+    no_args_is_help=True,
+)
+
+# Register subcommand groups
+app.add_typer(gui_app, name="gui", help="Web UI server management")
+app.add_typer(printer_app, name="printer", help="Printer fleet management")
+app.add_typer(template_app, name="template", help="ZPL template management")
+app.add_typer(cognito_app, name="cognito", help="Cognito authentication management")
+
+
+def _get_version() -> str:
+    """Get zebra_day version."""
+    try:
+        from importlib.metadata import version
+        return version("zebra_day")
+    except Exception:
+        return "dev"
+
+
+@app.command("version")
+def version():
+    """Show zebra_day version."""
+    console.print(f"zebra_day [cyan]{_get_version()}[/cyan]")
+
+
+@app.command("info")
+def info():
+    """Show zebra_day configuration and status."""
+    table = Table(title="zebra_day Info")
+    table.add_column("Property", style="cyan")
+    table.add_column("Value")
+
+    # Version
+    table.add_row("Version", _get_version())
+    table.add_row("Python", sys.version.split()[0])
+
+    # XDG Paths
+    table.add_row("Config Dir", str(xdg.get_config_dir()))
+    table.add_row("Data Dir", str(xdg.get_data_dir()))
+    table.add_row("Logs Dir", str(xdg.get_logs_dir()))
+    table.add_row("Cache Dir", str(xdg.get_cache_dir()))
+
+    # Printer config
+    printer_cfg = xdg.get_printer_config_path()
+    if printer_cfg.exists():
+        table.add_row("Printer Config", f"[green]{printer_cfg}[/green]")
+    else:
+        table.add_row("Printer Config", f"[yellow]not found[/yellow] [dim]({printer_cfg})[/dim]")
+
+    # Check if GUI server is running
+    pid_file = xdg.get_state_dir() / "gui.pid"
+    if pid_file.exists():
+        try:
+            pid = int(pid_file.read_text().strip())
+            os.kill(pid, 0)
+            table.add_row("GUI Server", f"[green]Running[/green] (PID {pid})")
+        except (ValueError, ProcessLookupError, PermissionError):
+            table.add_row("GUI Server", "[dim]Stopped[/dim]")
+    else:
+        table.add_row("GUI Server", "[dim]Stopped[/dim]")
+
+    console.print(table)
+
+
+@app.command("status")
+def status(
+    json_output: bool = typer.Option(False, "--json", "-j", help="Output as JSON"),
+):
+    """Show printer fleet status, network connectivity, and service health."""
+    import json as json_mod
+
+    status_data = {
+        "gui_server": {"running": False, "pid": None, "url": None},
+        "printers": {"configured": 0, "labs": []},
+        "config": {"exists": False, "path": None},
+    }
+
+    # Check GUI server
+    pid_file = xdg.get_state_dir() / "gui.pid"
+    if pid_file.exists():
+        try:
+            pid = int(pid_file.read_text().strip())
+            os.kill(pid, 0)
+            status_data["gui_server"]["running"] = True
+            status_data["gui_server"]["pid"] = pid
+            status_data["gui_server"]["url"] = "http://0.0.0.0:8118"
+        except (ValueError, ProcessLookupError, PermissionError):
+            pass
+
+    # Check printer config
+    printer_cfg = xdg.get_printer_config_path()
+    status_data["config"]["path"] = str(printer_cfg)
+    if printer_cfg.exists():
+        status_data["config"]["exists"] = True
+        try:
+            import zebra_day.print_mgr as zdpm
+            zp = zdpm.zpl()
+            if hasattr(zp, "printers") and "labs" in zp.printers:
+                labs = list(zp.printers["labs"].keys())
+                status_data["printers"]["labs"] = labs
+                total_printers = sum(
+                    len([k for k in zp.printers["labs"][lab].keys()])
+                    for lab in labs
+                )
+                status_data["printers"]["configured"] = total_printers
+        except Exception:
+            pass
+
+    if json_output:
+        console.print(json_mod.dumps(status_data, indent=2))
+        return
+
+    # Human-readable output
+    console.print("\n[bold]Service Status[/bold]")
+    if status_data["gui_server"]["running"]:
+        console.print(f"  [green]●[/green] GUI Server: [green]Running[/green] (PID {status_data['gui_server']['pid']})")
+        console.print(f"    URL: [cyan]{status_data['gui_server']['url']}[/cyan]")
+    else:
+        console.print("  [dim]○[/dim] GUI Server: [dim]Not running[/dim]")
+
+    console.print("\n[bold]Printer Fleet[/bold]")
+    if status_data["config"]["exists"]:
+        console.print(f"  [green]●[/green] Config: [green]Loaded[/green]")
+        console.print(f"    Printers: {status_data['printers']['configured']}")
+        console.print(f"    Labs: {', '.join(status_data['printers']['labs']) or 'none'}")
+    else:
+        console.print("  [yellow]○[/yellow] Config: [yellow]Not found[/yellow]")
+        console.print(f"    Run [cyan]zday bootstrap[/cyan] to initialize")
+    console.print()
+
+
+@app.command("bootstrap")
+def bootstrap(
+    ip_stub: Optional[str] = typer.Option(None, "--ip-stub", "-i", help="IP stub for printer scan (e.g., 192.168.1)"),
+    skip_scan: bool = typer.Option(False, "--skip-scan", "-s", help="Skip printer network scan"),
+    json_output: bool = typer.Option(False, "--json", "-j", help="Output as JSON"),
+):
+    """Initialize configuration, scan for printers, and setup first-time environment.
+
+    This is the recommended first-time setup command. It will:
+    1. Create XDG configuration directories
+    2. Initialize printer configuration
+    3. Scan the network for Zebra printers (unless --skip-scan)
+    """
+    import json as json_mod
+    import time
+    import socket
+
+    result = {
+        "config_dir": str(xdg.get_config_dir()),
+        "data_dir": str(xdg.get_data_dir()),
+        "printers_found": 0,
+        "labs": [],
+    }
+
+    if not json_output:
+        console.print("\n[bold cyan]zebra_day Bootstrap[/bold cyan]\n")
+        console.print("[green]✓[/green] Config directory: " + result["config_dir"])
+        console.print("[green]✓[/green] Data directory: " + result["data_dir"])
+
+    if skip_scan:
+        if not json_output:
+            console.print("[dim]Skipping printer scan (--skip-scan)[/dim]")
+    else:
+        # Determine IP stub
+        if not ip_stub:
+            try:
+                s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+                s.connect(("8.8.8.8", 80))
+                local_ip = s.getsockname()[0]
+                s.close()
+                ip_stub = ".".join(local_ip.split(".")[:-1])
+            except Exception:
+                ip_stub = "192.168.1"
+
+        if not json_output:
+            console.print(f"\n[cyan]→[/cyan] Scanning network for Zebra printers ({ip_stub}.*)...")
+            console.print("[dim]  This may take a few minutes...[/dim]")
+
+        try:
+            import zebra_day.print_mgr as zdpm
+            zp = zdpm.zpl()
+            zp.probe_zebra_printers_add_to_printers_json(ip_stub=ip_stub)
+
+            if hasattr(zp, "printers") and "labs" in zp.printers:
+                for lab in zp.printers["labs"]:
+                    printers_in_lab = len([k for k in zp.printers["labs"][lab].keys()])
+                    result["printers_found"] += printers_in_lab
+                    result["labs"].append(lab)
+
+            if not json_output:
+                console.print(f"[green]✓[/green] Scan complete: {result['printers_found']} printer(s) found")
+                if result["labs"]:
+                    console.print(f"    Labs: {', '.join(result['labs'])}")
+        except Exception as e:
+            if not json_output:
+                console.print(f"[yellow]⚠[/yellow] Scan error: {e}")
+
+    if json_output:
+        console.print(json_mod.dumps(result, indent=2))
+    else:
+        console.print("\n[bold green]✓ Bootstrap complete![/bold green]")
+        console.print("\n[bold]Next steps:[/bold]")
+        console.print("  zday gui start     Start the web UI")
+        console.print("  zday printer list  Show configured printers")
+        console.print("  zday info          Show configuration details")
+        console.print()
+
+
+def main():
+    """Main CLI entry point."""
+    app()
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())
+

zebra_day/cli/cognito.py

@@ -0,0 +1,121 @@
+"""Cognito authentication management commands for zebra_day CLI.
+
+This module delegates to daylily-cognito if available, otherwise provides
+basic status and info commands.
+"""
+
+import os
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.table import Table
+
+console = Console()
+
+
+def _is_cognito_available() -> bool:
+    """Check if daylily-cognito is installed."""
+    try:
+        from daylily_cognito.cli import cognito_app as _  # noqa: F401
+        return True
+    except ImportError:
+        return False
+
+
+def _get_cognito_app() -> typer.Typer:
+    """Get the Cognito CLI app, either from daylily-cognito or a fallback."""
+    if _is_cognito_available():
+        # Import and return the full cognito CLI from daylily-cognito
+        from daylily_cognito.cli import cognito_app
+        return cognito_app
+    else:
+        # Return a minimal fallback app
+        return _create_fallback_app()
+
+
+def _create_fallback_app() -> typer.Typer:
+    """Create a fallback cognito app with basic commands."""
+    app = typer.Typer(help="Cognito authentication management (limited - daylily-cognito not installed)")
+
+    @app.command("status")
+    def status():
+        """Show current Cognito authentication configuration."""
+        table = Table(title="Cognito Configuration")
+        table.add_column("Variable", style="cyan")
+        table.add_column("Value")
+        table.add_column("Status")
+
+        pool_id = os.environ.get("COGNITO_USER_POOL_ID")
+        client_id = os.environ.get("COGNITO_APP_CLIENT_ID")
+        region = os.environ.get("COGNITO_REGION", os.environ.get("AWS_DEFAULT_REGION"))
+
+        if pool_id:
+            # Truncate for display
+            display = pool_id[:15] + "..." if len(pool_id) > 15 else pool_id
+            table.add_row("COGNITO_USER_POOL_ID", display, "[green]Set[/green]")
+        else:
+            table.add_row("COGNITO_USER_POOL_ID", "-", "[yellow]Not set[/yellow]")
+
+        if client_id:
+            display = client_id[:15] + "..." if len(client_id) > 15 else client_id
+            table.add_row("COGNITO_APP_CLIENT_ID", display, "[green]Set[/green]")
+        else:
+            table.add_row("COGNITO_APP_CLIENT_ID", "-", "[yellow]Not set[/yellow]")
+
+        if region:
+            table.add_row("COGNITO_REGION", region, "[green]Set[/green]")
+        else:
+            table.add_row("COGNITO_REGION", "-", "[yellow]Not set[/yellow]")
+
+        console.print(table)
+
+        # Summary
+        if pool_id and client_id:
+            console.print("\n[green]✓[/green] Cognito is configured")
+            console.print("  Start server with: [cyan]zday gui start --auth cognito[/cyan]")
+        else:
+            console.print("\n[yellow]⚠[/yellow] Cognito is not fully configured")
+            console.print("  Set environment variables or install daylily-cognito for full management")
+
+    @app.command("info")
+    def info():
+        """Display information about Cognito setup requirements."""
+        console.print("\n[bold]Cognito Authentication Setup[/bold]\n")
+        console.print("To enable Cognito authentication for zebra_day:\n")
+
+        console.print("[bold]1. Install auth dependencies:[/bold]")
+        console.print("   [cyan]pip install -e \".[auth]\"[/cyan]\n")
+
+        console.print("[bold]2. Set environment variables:[/bold]")
+        console.print("   [cyan]export COGNITO_USER_POOL_ID=your-pool-id[/cyan]")
+        console.print("   [cyan]export COGNITO_APP_CLIENT_ID=your-client-id[/cyan]")
+        console.print("   [cyan]export COGNITO_REGION=us-west-2[/cyan]  # optional\n")
+
+        console.print("[bold]3. Start server with authentication:[/bold]")
+        console.print("   [cyan]zday gui start --auth cognito[/cyan]\n")
+
+        if not _is_cognito_available():
+            console.print("[dim]For full Cognito management (create, teardown), install daylily-cognito:[/dim]")
+            console.print("[dim]  pip install daylily-cognito[/dim]")
+
+    @app.command("create")
+    def create():
+        """Create/configure a Cognito user pool (requires daylily-cognito)."""
+        console.print("[yellow]⚠[/yellow] This command requires daylily-cognito")
+        console.print("  Install with: [cyan]pip install -e \".[auth]\"[/cyan]")
+        raise typer.Exit(1)
+
+    @app.command("teardown")
+    def teardown():
+        """Remove Cognito configuration (requires daylily-cognito)."""
+        console.print("[yellow]⚠[/yellow] This command requires daylily-cognito")
+        console.print("  Install with: [cyan]pip install -e \".[auth]\"[/cyan]")
+        raise typer.Exit(1)
+
+    return app
+
+
+# Export the cognito app - either the full version from daylily-cognito or the fallback
+cognito_app = _get_cognito_app()
+