zebra-day 0.0.32__py3-none-any.whl → 1.0.2__py3-none-any.whl

zebra_day/templates/printer_status.html

@@ -0,0 +1,70 @@
+{% extends "base.html" %}
+
+{% block title %}{{ title }}{% endblock %}
+
+{% block content %}
+<h1>Printer Status Summary For {{ lab }}</h1>
+<small><a href="/">BACK HOME</a></small>
+<ul>
+    <hr>
+    <br>
+    <table border="1">
+        <tr>
+            <th>Printer Name</th>
+            <th>Printer IP</th>
+            <th>Label Styles Available</th>
+            <th>Status on Network</th>
+        </tr>
+        {% for printer in printers %}
+        <tr>
+            <td>
+                {{ printer.name }}<br>
+                <small>{{ printer.model }}<br>{{ printer.serial }}</small>
+            </td>
+            <td>
+                {% if printer.ip_address != 'dl_png' %}
+                <a href="http://{{ printer.ip_address }}" target="_blank">{{ printer.ip_address }}</a>
+                {% else %}
+                {{ printer.ip_address }}
+                {% endif %}
+                <br><small>{{ printer.arp_data }}</small>
+            </td>
+            <td>
+                {% for style in printer.label_zpl_styles %}
+                <small>
+                    <a href="/edit_zpl">{{ style }}</a>
+                    (<a href="/_print_label?lab={{ lab }}&printer={{ printer.name }}&label_zpl_style={{ style }}&uid_barcode={{ printer.ip_address }}&alt_a={{ printer.model }}&alt_b={{ printer.serial }}&alt_c={{ lab }}&alt_d={{ lab }}" target="_blank">test</a>)
+                </small><br>
+                {% endfor %}
+            </td>
+            <td>
+                {{ printer.status }}
+                <br><small><a href="/printer_details?printer_name={{ printer.name }}&lab={{ lab }}" target="_blank">...printer details</a></small>
+            </td>
+        </tr>
+        {% endfor %}
+    </table>
+</ul>
+
+<br><hr>
+<h2>Build New Printers Config JSON // Scan Network</h2>
+<ul>
+    <p>This will probe for printers, and with all discovered printers will create a new printers JSON and overwrite the current fleet data for this lab.</p>
+    <form id="myForm" action="/probe_zebra_printers_add_to_printers_json">
+        <label>Enter IP Root To Scan:</label>
+        <input type="text" value="{{ ip_root }}" name="ip_stub">
+        <br>
+        <label>Select (or enter new) lab:</label>
+        <input type="text" name="lab" list="lab_names" value="{{ lab }}">
+        <datalist id="lab_names">
+            {% for lab_name in labs %}
+            <option value="{{ lab_name }}">{{ lab_name }}</option>
+            {% endfor %}
+        </datalist>
+        <br>
+        <input type="submit" value="Scan Network">
+    </form>
+    <small><a href="/list_prior_printer_config_files">The current printers JSON will be backed up and can be found here.</a></small>
+</ul>
+{% endblock %}
+

zebra_day/templates/save_result.html

@@ -0,0 +1,17 @@
+{% extends "base.html" %}
+
+{% block title %}{{ title }}{% endblock %}
+
+{% block content %}
+<h1>Template Saved</h1>
+<small><a href="/">home</a></small>
+<hr>
+
+<p>Changes saved to temp file: <strong>{{ new_filename }}</strong></p>
+
+<ul>
+    <li><a href="/edit_zpl">Go back to the ZPL file list</a></li>
+    <li><a href="/">Go home</a></li>
+</ul>
+{% endblock %}
+

zebra_day/templates/send_print_request.html

@@ -0,0 +1,34 @@
+{% extends "base.html" %}
+
+{% block title %}{{ title }}{% endblock %}
+
+{% block content %}
+<h1>Send Print Request</h1>
+<small><a href="/">home</a></small>
+<hr>
+
+{% for lab_name, printers in labs.items() %}
+<h2>Lab: {{ lab_name }}</h2>
+<table border="1" width="90%">
+    <tr>
+        <th>Printer</th>
+        <th>Available Label Styles</th>
+    </tr>
+    {% for printer_name, printer_info in printers.items() %}
+    <tr>
+        <td>
+            <strong>{{ printer_name }}</strong><br>
+            <small>{{ printer_info.ip_address }}</small>
+        </td>
+        <td>
+            {% for style in printer_info.label_zpl_styles %}
+            <a href="/build_print_request?lab={{ lab_name }}&printer={{ printer_name }}&printer_ip={{ printer_info.ip_address }}&label_zpl_style={{ style }}">{{ style }}</a><br>
+            {% endfor %}
+        </td>
+    </tr>
+    {% endfor %}
+</table>
+<br>
+{% endfor %}
+{% endblock %}
+

zebra_day/templates/simple_print.html

@@ -0,0 +1,94 @@
+{% extends "base.html" %}
+
+{% block title %}{{ title }}{% endblock %}
+
+{% block content %}
+<h1>Send Label Print Request</h1>
+<ul>
+    <small><a href="/">home</a></small>
+    <hr>
+    
+    <form id="myForm" action="/_print_label" method="get">
+        <ul>
+            <li>
+                <label>UID for Barcode:</label>
+                <input type="text" name="uid_barcode">
+            </li>
+            <li>
+                <label>ALT-A:</label>
+                <input type="text" name="alt_a">
+            </li>
+            <li>
+                <label>ALT-B:</label>
+                <input type="text" name="alt_b">
+            </li>
+            <li>
+                <label>ALT-C:</label>
+                <input type="text" name="alt_c">
+            </li>
+            <li>
+                <label>ALT-D:</label>
+                <input type="text" name="alt_d">
+            </li>
+            <li>
+                <label>ALT-E:</label>
+                <input type="text" name="alt_e">
+            </li>
+            <li>
+                <label>ALT-F:</label>
+                <input type="text" name="alt_f">
+            </li>
+            <li>
+                <label>Select Lab:</label>
+                <select id="labSelect" name="lab" onchange="updatePrinters()">
+                    <option value="">Select a Lab</option>
+                    {% for lab in labs %}
+                    <option value="{{ lab }}">{{ lab }}</option>
+                    {% endfor %}
+                </select>
+            </li>
+            <li>
+                <label>Select Printer:</label>
+                <select id="printerSelect" name="printer">
+                    <option value="">Select a Printer</option>
+                </select>
+            </li>
+            <li>
+                <label>Label ZPL Style:</label>
+                <select name="label_zpl_style">
+                    {% for template in templates %}
+                    <option value="{{ template }}">{{ template }}</option>
+                    {% endfor %}
+                </select>
+            </li>
+        </ul>
+        <br>
+        <input type="submit" value="Print Label">
+    </form>
+</ul>
+{% endblock %}
+
+{% block extra_scripts %}
+<script>
+    var labsAndPrinters = {{ labs_and_printers | safe }};
+
+    function updatePrinters() {
+        var labSelect = document.getElementById('labSelect');
+        var printerSelect = document.getElementById('printerSelect');
+        var selectedLab = labSelect.value;
+
+        // Clear current printer options
+        printerSelect.innerHTML = '<option value="">Select a Printer</option>';
+
+        if (selectedLab in labsAndPrinters) {
+            labsAndPrinters[selectedLab].forEach(function(printer) {
+                var option = document.createElement('option');
+                option.value = printer;
+                option.text = printer;
+                printerSelect.appendChild(option);
+            });
+        }
+    }
+</script>
+{% endblock %}
+

zebra_day/templates/view_pstation_json.html

@@ -0,0 +1,29 @@
+{% extends "base.html" %}
+
+{% block title %}{{ title }}{% endblock %}
+
+{% block content %}
+<h1>View and Edit Printer Config JSON</h1>
+<small><a href="/">home</a></small>
+<hr>
+
+{% if error_msg %}
+<p style="color: red; font-weight: bold;">{{ error_msg }}</p>
+{% endif %}
+
+<form action="/save_pstation_json" method="post">
+    <textarea name="json_data" rows="45" cols="80">{{ config_data }}</textarea>
+    <br>
+    <input type="submit" value="Save New Printers Config JSON & Make Active">
+</form>
+
+<hr>
+<ul>
+    <li><a href="/list_prior_printer_config_files">View backed up printers JSON config files</a></li>
+    <hr>
+    <li><a href="/reset_pstation_json">Restore Printer Settings From Default JSON (THIS WILL DELETE YOUR CURRENT FILE!)</a></li>
+    <hr>
+    <li><a href="/clear_printers_json">CLEAR contents of current printers.json file (This Cannot Be Undone)</a></li>
+</ul>
+{% endblock %}
+

zebra_day/web/__init__.py

@@ -0,0 +1,9 @@
+"""
+zebra_day web application module.
+
+FastAPI-based web interface for managing Zebra printers and label templates.
+"""
+from zebra_day.web.app import create_app
+
+__all__ = ["create_app"]
+

zebra_day/web/app.py

@@ -0,0 +1,171 @@
+"""
+FastAPI application factory for zebra_day.
+
+This module provides the main FastAPI application for the zebra_day web interface.
+"""
+from __future__ import annotations
+
+import os
+import subprocess
+from pathlib import Path
+from typing import Literal, Optional
+
+from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
+from importlib.resources import files
+
+from zebra_day.logging_config import get_logger
+from zebra_day import paths as xdg
+from zebra_day.web.middleware import RequestLoggingMiddleware, print_rate_limiter
+
+_log = get_logger(__name__)
+
+# Package paths
+_PKG_PATH = Path(str(files("zebra_day")))
+_STATIC_PATH = _PKG_PATH / "static"
+_TEMPLATES_PATH = _PKG_PATH / "templates"
+
+
+def get_local_ip() -> str:
+    """Get the local IP address of this machine."""
+    ipcmd = r"""(ip addr show | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1' || ifconfig | grep -Eo 'inet (addr:)?([0-9]*\.){3}[0-9]*' | grep -Eo '([0-9]*\.){3}[0-9]*' | grep -v '127.0.0.1') 2>/dev/null"""
+    result = subprocess.run(ipcmd, shell=True, capture_output=True, text=True)
+    lines = result.stdout.strip().split("\n")
+    return lines[0] if lines and lines[0] else "127.0.0.1"
+
+
+def create_app(
+    *,
+    debug: bool = False,
+    css_theme: str = "lsmc.css",
+    auth: Optional[Literal["none", "cognito"]] = None,
+) -> FastAPI:
+    """
+    Create and configure the FastAPI application.
+
+    Args:
+        debug: Enable debug mode
+        css_theme: Default CSS theme file name
+        auth: Authentication mode - "none" (public) or "cognito" (AWS Cognito).
+              If None, reads from ZEBRA_DAY_AUTH_MODE env var (defaults to "none").
+
+    Returns:
+        Configured FastAPI application
+    """
+    # Get auth mode from parameter or environment variable
+    if auth is None:
+        auth = os.environ.get("ZEBRA_DAY_AUTH_MODE", "none")  # type: ignore[assignment]
+
+    # Validate auth parameter
+    if auth not in ("none", "cognito"):
+        raise ValueError(f"Invalid auth mode: {auth!r}. Must be 'none' or 'cognito'.")
+
+    app = FastAPI(
+        title="Zebra Day",
+        description="Zebra printer fleet management and label printing",
+        version="0.5.0",
+        debug=debug,
+    )
+
+    # Add request logging middleware
+    app.add_middleware(RequestLoggingMiddleware)
+
+    # Configure authentication if enabled
+    if auth == "cognito":
+        from zebra_day.web.auth import CognitoAuthMiddleware, setup_cognito_auth
+
+        cognito_auth = setup_cognito_auth(app)
+        app.add_middleware(CognitoAuthMiddleware, cognito_auth=cognito_auth)
+        app.state.cognito_auth = cognito_auth
+        app.state.auth_mode = "cognito"
+        _log.info("Cognito authentication middleware enabled")
+    else:
+        app.state.auth_mode = "none"
+        _log.info("Authentication disabled (auth=none)")
+
+    # Store rate limiter in app state for use in endpoints
+    app.state.print_rate_limiter = print_rate_limiter
+
+    # Store app state
+    app.state.css_theme = css_theme
+    app.state.local_ip = get_local_ip()
+    app.state.pkg_path = _PKG_PATH
+
+    # Mount static files
+    app.mount("/static", StaticFiles(directory=str(_STATIC_PATH)), name="static")
+
+    # Also mount package directories that need to be served
+    files_dir = _PKG_PATH / "files"
+    if files_dir.exists():
+        app.mount("/files", StaticFiles(directory=str(files_dir)), name="files")
+
+    etc_dir = _PKG_PATH / "etc"
+    if etc_dir.exists():
+        app.mount("/etc", StaticFiles(directory=str(etc_dir)), name="etc")
+
+    # Setup Jinja2 templates
+    templates = Jinja2Templates(directory=str(_TEMPLATES_PATH))
+    app.state.templates = templates
+
+    # Register routers
+    from zebra_day.web.routers import ui, api
+
+    app.include_router(ui.router)
+    app.include_router(api.router, prefix="/api/v1", tags=["api"])
+
+    @app.on_event("startup")
+    async def startup_event():
+        """Initialize application state on startup."""
+        import zebra_day.print_mgr as zdpm
+
+        app.state.zp = zdpm.zpl()
+        _log.info(
+            "zebra_day web server starting at %s:8118",
+            app.state.local_ip,
+        )
+
+    @app.get("/healthz")
+    async def healthz():
+        """Health check endpoint."""
+        return {"status": "healthy"}
+
+    @app.get("/readyz")
+    async def readyz():
+        """Readiness check endpoint."""
+        # Check if printer manager is initialized
+        if hasattr(app.state, "zp") and app.state.zp is not None:
+            return {"status": "ready"}
+        return {"status": "not_ready"}, 503
+
+    return app
+
+
+def run_server(
+    host: str = "0.0.0.0",
+    port: int = 8118,
+    reload: bool = False,
+    auth: Literal["none", "cognito"] = "none",
+):
+    """
+    Run the FastAPI server using uvicorn.
+
+    Args:
+        host: Host to bind to
+        port: Port to listen on
+        reload: Enable auto-reload for development
+        auth: Authentication mode - "none" (public) or "cognito" (AWS Cognito)
+    """
+    import uvicorn
+
+    # Store auth mode in environment for factory function
+    os.environ["ZEBRA_DAY_AUTH_MODE"] = auth
+
+    uvicorn.run(
+        "zebra_day.web.app:create_app",
+        host=host,
+        port=port,
+        reload=reload,
+        factory=True,
+    )
+

zebra_day/web/auth.py

@@ -0,0 +1,172 @@
+"""Authentication integration for zebra_day web server.
+
+Provides optional Cognito authentication support via the daylily-cognito library.
+"""
+
+from __future__ import annotations
+
+import os
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional
+
+from fastapi import Depends, HTTPException, Request, status
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.responses import Response
+
+from zebra_day.logging_config import get_logger
+
+if TYPE_CHECKING:
+    from fastapi import FastAPI
+
+_log = get_logger(__name__)
+
+# Endpoints that should never require authentication
+PUBLIC_PATHS: List[str] = [
+    "/healthz",
+    "/readyz",
+    "/docs",
+    "/openapi.json",
+    "/redoc",
+]
+
+# Try to import daylily-cognito components
+_COGNITO_AVAILABLE = False
+_COGNITO_IMPORT_ERROR: Optional[str] = None
+
+try:
+    from daylily_cognito import CognitoAuth, CognitoConfig, create_auth_dependency
+
+    _COGNITO_AVAILABLE = True
+except ImportError as e:
+    _COGNITO_IMPORT_ERROR = str(e)
+    CognitoAuth = None  # type: ignore[misc, assignment]
+    CognitoConfig = None  # type: ignore[misc, assignment]
+    create_auth_dependency = None  # type: ignore[misc, assignment]
+
+
+def is_cognito_available() -> bool:
+    """Check if daylily-cognito library is installed."""
+    return _COGNITO_AVAILABLE
+
+
+def get_cognito_import_error() -> Optional[str]:
+    """Get the import error message if daylily-cognito is not available."""
+    return _COGNITO_IMPORT_ERROR
+
+
+class CognitoAuthMiddleware(BaseHTTPMiddleware):
+    """Middleware that enforces Cognito authentication on protected endpoints.
+
+    Exempts health check endpoints and other public paths.
+    """
+
+    def __init__(self, app: "FastAPI", cognito_auth: Any) -> None:
+        super().__init__(app)
+        self.cognito_auth = cognito_auth
+        self.get_current_user = create_auth_dependency(cognito_auth, optional=False)
+
+    async def dispatch(self, request: Request, call_next: Callable) -> Response:
+        """Check authentication for protected endpoints."""
+        path = request.url.path
+
+        # Allow public endpoints without authentication
+        if any(path.startswith(public) for public in PUBLIC_PATHS):
+            return await call_next(request)
+
+        # Allow static files without authentication
+        if path.startswith("/static") or path.startswith("/files") or path.startswith("/etc"):
+            return await call_next(request)
+
+        # Check for Authorization header
+        auth_header = request.headers.get("Authorization")
+        if not auth_header:
+            return Response(
+                content='{"detail":"Authentication required"}',
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                media_type="application/json",
+                headers={"WWW-Authenticate": "Bearer"},
+            )
+
+        # Validate Bearer token
+        if not auth_header.startswith("Bearer "):
+            return Response(
+                content='{"detail":"Invalid authorization header format"}',
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                media_type="application/json",
+                headers={"WWW-Authenticate": "Bearer"},
+            )
+
+        token = auth_header[7:]  # Remove "Bearer " prefix
+        try:
+            # Verify token and attach user to request state
+            user_claims = self.cognito_auth.verify_token(token)
+            request.state.user = user_claims
+        except HTTPException:
+            return Response(
+                content='{"detail":"Invalid or expired token"}',
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                media_type="application/json",
+                headers={"WWW-Authenticate": "Bearer"},
+            )
+        except Exception as e:
+            _log.error("Authentication error: %s", str(e))
+            return Response(
+                content='{"detail":"Authentication failed"}',
+                status_code=status.HTTP_401_UNAUTHORIZED,
+                media_type="application/json",
+                headers={"WWW-Authenticate": "Bearer"},
+            )
+
+        return await call_next(request)
+
+
+def setup_cognito_auth(app: "FastAPI") -> Any:
+    """Configure Cognito authentication for the FastAPI app.
+
+    Reads configuration from environment variables:
+        - COGNITO_USER_POOL_ID: Cognito User Pool ID (required)
+        - COGNITO_APP_CLIENT_ID or COGNITO_CLIENT_ID: App Client ID (required)
+        - COGNITO_REGION or AWS_REGION: AWS region (defaults to us-west-2)
+        - AWS_PROFILE: Optional AWS profile name
+
+    Returns:
+        CognitoAuth instance
+
+    Raises:
+        ValueError: If required environment variables are missing
+        ImportError: If daylily-cognito is not installed
+    """
+    if not _COGNITO_AVAILABLE:
+        raise ImportError(
+            f"daylily-cognito library is required for Cognito authentication. "
+            f"Install with: pip install -e '.[auth]'\n"
+            f"Import error: {_COGNITO_IMPORT_ERROR}"
+        )
+
+    # Load config from environment
+    try:
+        config = CognitoConfig.from_legacy_env()
+    except ValueError as e:
+        raise ValueError(
+            f"Missing required Cognito configuration. {e}\n"
+            "Set the following environment variables:\n"
+            "  COGNITO_USER_POOL_ID=your-pool-id\n"
+            "  COGNITO_APP_CLIENT_ID=your-client-id\n"
+            "  COGNITO_REGION=us-west-2  (optional, defaults to us-west-2)"
+        ) from e
+
+    # Create CognitoAuth instance
+    cognito_auth = CognitoAuth(
+        region=config.region,
+        user_pool_id=config.user_pool_id,
+        app_client_id=config.app_client_id,
+        profile=config.aws_profile,
+    )
+
+    _log.info(
+        "Cognito authentication enabled (region=%s, pool=%s)",
+        config.region,
+        config.user_pool_id,
+    )
+
+    return cognito_auth
+