nginx-lens 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

commands/cli.py

@@ -10,6 +10,9 @@ from commands.graph import graph
 from commands.logs import logs
 from commands.syntax import syntax
 from commands.resolve import resolve
+from commands.validate import validate
+from commands.metrics import metrics
+from commands.completion import app as completion_app
 
 app = typer.Typer(help="nginx-lens — анализ и диагностика конфигураций Nginx")
 console = Console()
@@ -24,6 +27,9 @@ app.command()(graph)
 app.command()(logs)
 app.command()(syntax)
 app.command()(resolve)
+app.command()(validate)
+app.command()(metrics)
+app.add_typer(completion_app, name="completion", help="Генерация скриптов автодополнения")
 
 if __name__ == "__main__":
     app() 

commands/completion.py

@@ -0,0 +1,174 @@
+"""
+Команда для генерации скриптов автодополнения.
+"""
+import sys
+import typer
+from rich.console import Console
+from rich.panel import Panel
+from rich import box
+
+app = typer.Typer()
+console = Console()
+
+
+def _generate_completion_script(shell: str) -> str:
+    """
+    Генерирует базовый completion скрипт для указанного shell.
+    
+    Args:
+        shell: Тип shell (bash, zsh, fish, powershell)
+        
+    Returns:
+        Строка с completion скриптом
+    """
+    commands = [
+        "health", "analyze", "tree", "diff", "route", "include",
+        "graph", "logs", "syntax", "resolve", "validate", "metrics"
+    ]
+    
+    if shell == "bash":
+        script = f"""# Bash completion for nginx-lens
+_nginx_lens_completion() {{
+    local cur="${{COMP_WORDS[COMP_CWORD]}}"
+    local commands="{' '.join(commands)}"
+    
+    if [[ ${{COMP_CWORD}} -eq 1 ]]; then
+        COMPREPLY=($(compgen -W "$commands" -- "$cur"))
+    else
+        COMPREPLY=($(compgen -f -- "$cur"))
+    fi
+}}
+
+complete -F _nginx_lens_completion nginx-lens
+"""
+    elif shell == "zsh":
+        script = f"""# Zsh completion for nginx-lens
+#compdef nginx-lens
+
+_nginx_lens() {{
+    local -a commands
+    commands=(
+        {' '.join(f'"{cmd}":{cmd}' for cmd in commands)}
+    )
+    
+    _describe 'command' commands
+}}
+
+_nginx_lens "$@"
+"""
+    elif shell == "fish":
+        script = f"""# Fish completion for nginx-lens
+complete -c nginx-lens -f -a "{' '.join(commands)}"
+"""
+    elif shell == "powershell":
+        script = f"""# PowerShell completion for nginx-lens
+Register-ArgumentCompleter -Native -CommandName nginx-lens -ScriptBlock {{
+    param($commandName, $wordToComplete, $cursorPosition)
+    
+    $commands = @({', '.join(f'"{cmd}"' for cmd in commands)})
+    
+    $commands | Where-Object {{ $_ -like "$wordToComplete*" }} | ForEach-Object {{
+        [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
+    }}
+}}
+"""
+    else:
+        script = f"# Completion script for {shell}\n# Use: typer commands.cli app {shell}\n"
+    
+    return script
+
+
+@app.command()
+def install(
+    shell: str = typer.Argument(..., help="Тип shell: bash, zsh, fish, powershell"),
+    output: str = typer.Option(None, "--output", "-o", help="Путь для сохранения скрипта (по умолчанию выводится в stdout)"),
+):
+    """
+    Генерирует скрипт автодополнения для указанного shell.
+    
+    После генерации нужно добавить его в конфигурацию shell:
+    
+    Bash:
+        nginx-lens completion install bash >> ~/.bashrc
+    
+    Zsh:
+        nginx-lens completion install zsh >> ~/.zshrc
+    
+    Fish:
+        nginx-lens completion install fish > ~/.config/fish/completions/nginx-lens.fish
+    
+    PowerShell:
+        nginx-lens completion install powershell > nginx-lens-completion.ps1
+        # Затем выполните: . .\nginx-lens-completion.ps1
+    """
+    import subprocess
+    import sys
+    
+    # Пробуем использовать typer CLI для генерации completion
+    completion_script = None
+    try:
+        result = subprocess.run(
+            [sys.executable, "-m", "typer", "commands.cli", "app", shell],
+            capture_output=True,
+            text=True,
+            check=False,
+            timeout=10
+        )
+        
+        if result.returncode == 0 and result.stdout and len(result.stdout.strip()) > 0:
+            completion_script = result.stdout
+    except (subprocess.TimeoutExpired, FileNotFoundError, Exception):
+        pass
+    
+    # Если typer CLI не сработал, используем базовый скрипт
+    if not completion_script:
+        completion_script = _generate_completion_script(shell)
+    
+    if shell not in ["bash", "zsh", "fish", "powershell"]:
+        console.print(f"[red]Неподдерживаемый shell: {shell}[/red]")
+        console.print("[yellow]Поддерживаемые shell: bash, zsh, fish, powershell[/yellow]")
+        raise typer.Exit(1)
+    
+    if output:
+        try:
+            with open(output, 'w') as f:
+                f.write(completion_script)
+            console.print(f"[green]✓ Скрипт автодополнения сохранен в {output}[/green]")
+            console.print(f"[yellow]Не забудьте добавить его в конфигурацию {shell}[/yellow]")
+        except Exception as e:
+            console.print(f"[red]Ошибка при сохранении файла: {e}[/red]")
+            raise typer.Exit(1)
+        else:
+            typer.echo(completion_script)
+            # Выводим инструкцию в stderr, чтобы не мешать скрипту
+            console.print("\n[yellow]Добавьте этот скрипт в конфигурацию вашего shell[/yellow]", file=sys.stderr)
+
+
+@app.command()
+def show_instructions():
+    """
+    Показывает инструкции по установке автодополнения для всех поддерживаемых shell.
+    """
+    console.print(Panel("[bold blue]Инструкции по установке автодополнения[/bold blue]", box=box.ROUNDED))
+    
+    console.print("\n[bold]Bash:[/bold]")
+    console.print("  nginx-lens completion install bash >> ~/.bashrc")
+    console.print("  source ~/.bashrc")
+    
+    console.print("\n[bold]Zsh:[/bold]")
+    console.print("  nginx-lens completion install zsh >> ~/.zshrc")
+    console.print("  source ~/.zshrc")
+    
+    console.print("\n[bold]Fish:[/bold]")
+    console.print("  mkdir -p ~/.config/fish/completions")
+    console.print("  nginx-lens completion install fish > ~/.config/fish/completions/nginx-lens.fish")
+    
+    console.print("\n[bold]PowerShell:[/bold]")
+    console.print("  nginx-lens completion install powershell > nginx-lens-completion.ps1")
+    console.print("  . .\\nginx-lens-completion.ps1")
+    console.print("  # Или добавьте в профиль PowerShell: $PROFILE")
+
+
+if __name__ == "__main__":
+    app()
+

commands/health.py

@@ -1,36 +1,64 @@
 import sys
+from typing import Optional
 import typer
 from rich.console import Console
 from rich.table import Table
 from upstream_checker.checker import check_upstreams, resolve_upstreams
+from upstream_checker.dns_cache import disable_cache, enable_cache
 from parser.nginx_parser import parse_nginx_config
 from exporter.json_yaml import format_health_results, print_export
+from config.config_loader import get_config
+from utils.progress import ProgressManager
 
 app = typer.Typer()
 console = Console()
 
 def health(
     config_path: str = typer.Argument(..., help="Путь к nginx.conf"),
-    timeout: float = typer.Option(2.0, help="Таймаут проверки (сек)"),
-    retries: int = typer.Option(1, help="Количество попыток"),
-    mode: str = typer.Option("tcp", help="Режим проверки: tcp или http", case_sensitive=False),
+    timeout: Optional[float] = typer.Option(None, help="Таймаут проверки (сек)"),
+    retries: Optional[int] = typer.Option(None, help="Количество попыток"),
+    mode: Optional[str] = typer.Option(None, help="Режим проверки: tcp или http", case_sensitive=False),
     resolve: bool = typer.Option(False, "--resolve", "-r", help="Показать резолвленные IP-адреса"),
-    max_workers: int = typer.Option(10, "--max-workers", "-w", help="Максимальное количество потоков для параллельной обработки"),
+    max_workers: Optional[int] = typer.Option(None, "--max-workers", "-w", help="Максимальное количество потоков для параллельной обработки"),
     json: bool = typer.Option(False, "--json", help="Экспортировать результаты в JSON"),
     yaml: bool = typer.Option(False, "--yaml", help="Экспортировать результаты в YAML"),
+    no_cache: bool = typer.Option(False, "--no-cache", help="Отключить кэширование DNS резолвинга"),
+    cache_ttl: Optional[int] = typer.Option(None, "--cache-ttl", help="Время жизни кэша в секундах"),
 ):
     """
     Проверяет доступность upstream-серверов, определённых в nginx.conf. Выводит таблицу.
     Использует параллельную обработку для ускорения проверки множества upstream серверов.
+    Поддерживает кэширование результатов DNS резолвинга для ускорения повторных запусков.
 
     Пример:
         nginx-lens health /etc/nginx/nginx.conf
         nginx-lens health /etc/nginx/nginx.conf --timeout 5 --retries 3 --mode http
         nginx-lens health /etc/nginx/nginx.conf --resolve
         nginx-lens health /etc/nginx/nginx.conf --max-workers 20
+        nginx-lens health /etc/nginx/nginx.conf --resolve --no-cache
+        nginx-lens health /etc/nginx/nginx.conf --resolve --cache-ttl 600
     """
     exit_code = 0
     
+    # Загружаем конфигурацию
+    config = get_config()
+    defaults = config.get_defaults()
+    cache_config = config.get_cache_config()
+    
+    # Применяем значения из конфига, если не указаны через CLI
+    timeout = timeout if timeout is not None else defaults.get("timeout", 2.0)
+    retries = retries if retries is not None else defaults.get("retries", 1)
+    mode = mode if mode is not None else defaults.get("mode", "tcp")
+    max_workers = max_workers if max_workers is not None else defaults.get("max_workers", 10)
+    cache_ttl = cache_ttl if cache_ttl is not None else cache_config.get("ttl", defaults.get("dns_cache_ttl", 300))
+    
+    # Управление кэшем
+    use_cache = not no_cache and cache_config.get("enabled", True)
+    if no_cache:
+        disable_cache()
+    else:
+        enable_cache()
+    
     try:
         tree = parse_nginx_config(config_path)
     except FileNotFoundError:
@@ -41,12 +69,19 @@ def health(
         sys.exit(1)
 
     upstreams = tree.get_upstreams()
-    results = check_upstreams(upstreams, timeout=timeout, retries=retries, mode=mode.lower(), max_workers=max_workers)
+    
+    # Подсчитываем общее количество серверов для прогресс-бара
+    total_servers = sum(len(servers) for servers in upstreams.values())
+    
+    # Проверка upstream с прогресс-баром
+    with ProgressManager(description="Проверка upstream серверов", show_progress=total_servers > 5) as pm:
+        results = check_upstreams(upstreams, timeout=timeout, retries=retries, mode=mode.lower(), max_workers=max_workers, progress_manager=pm)
     
     # Если нужно показать резолвленные IP-адреса
     resolved_info = {}
     if resolve:
-        resolved_info = resolve_upstreams(upstreams, max_workers=max_workers)
+        with ProgressManager(description="Резолвинг DNS", show_progress=total_servers > 5) as pm:
+            resolved_info = resolve_upstreams(upstreams, max_workers=max_workers, use_cache=use_cache, cache_ttl=cache_ttl, progress_manager=pm)
 
     # Экспорт в JSON/YAML
     if json or yaml:

commands/logs.py

@@ -1,4 +1,5 @@
 import sys
+from typing import Optional, List, Dict, Any
 import typer
 from rich.console import Console
 from rich.table import Table
@@ -6,9 +7,9 @@ import re
 import gzip
 from datetime import datetime, timedelta
 from collections import Counter, defaultdict
-from typing import Optional, List, Dict, Any
 from exporter.json_yaml import format_logs_results, print_export
 from exporter.csv import export_logs_to_csv
+from config.config_loader import get_config
 
 app = typer.Typer(help="Анализ access.log/error.log: топ-статусы, пути, IP, User-Agent, ошибки.")
 console = Console()
@@ -23,7 +24,7 @@ log_line_re = re.compile(
 
 def logs(
     log_path: str = typer.Argument(..., help="Путь к access.log или error.log"),
-    top: int = typer.Option(10, help="Сколько топ-значений выводить"),
+    top: Optional[int] = typer.Option(None, help="Сколько топ-значений выводить"),
     json: bool = typer.Option(False, "--json", help="Экспортировать результаты в JSON"),
     yaml: bool = typer.Option(False, "--yaml", help="Экспортировать результаты в YAML"),
     csv: bool = typer.Option(False, "--csv", help="Экспортировать результаты в CSV"),
@@ -49,6 +50,13 @@ def logs(
         nginx-lens logs /var/log/nginx/access.log --since "2024-01-01" --status 404,500
         nginx-lens logs /var/log/nginx/access.log.gz --detect-anomalies --json
     """
+    # Загружаем конфигурацию
+    config = get_config()
+    defaults = config.get_defaults()
+    
+    # Применяем значения из конфига, если не указаны через CLI
+    top = top if top is not None else defaults.get("top", 10)
+    
     # Парсинг фильтров
     status_filter = None
     if status: