<?php

namespace App\Services;

use App\Models\Server;
use App\Support\Os\FirewallTool;
use App\Support\Os\OsDetector;
use App\Support\Os\OsProfile;
use App\Support\Os\PackageManager;
use InvalidArgumentException;
use RuntimeException;
use Throwable;

/**
 * Applies the dashboard server-hardening toggles over SSH (as root), OS-aware
 * across apt/dnf/zypper + ufw/firewalld (resolved per host via OsDetector). Every
 * item is BIDIRECTIONAL — apply($server, $action, $enable) toggles the underlying
 * feature; package items (fail2ban, auto-updates, firewall) install on demand.
 *
 * apply() runs the exact command built by commandFor() (the single source of truth);
 * title()/description() supply the German confirm-modal copy for each direction.
 * Each state() row carries `supported`/`reason`, so a feature unavailable on the
 * host's OS degrades with a German explanation instead of failing opaquely.
 * Disabling SSH password auth is GUARDED so the operator can never lock themselves out.
 */
class HardeningService
{
    // 00- so it sorts + wins before distro drop-ins like 50-cloud-init.conf (sshd uses the first value).
    private const SSHD_DROPIN = '/etc/ssh/sshd_config.d/00-clusev.conf';

    // 99zz- so Clusev's apt periodic setting wins (apt uses the LAST value across apt.conf.d/*).
    private const APT_DROPIN = '/etc/apt/apt.conf.d/99zz-clusev';

    public function __construct(
        private FleetService $fleet,
        private FirewallService $firewall,
        private OsDetector $detector,
    ) {}

    /**
     * Live state of every hardening item, read PRIVILEGED in one shot. SSH uses
     * `sshd -T` (the EFFECTIVE config — honours Include order + Match blocks); the
     * package/firewall/auto-update probes are built per-OS (dpkg vs rpm, ufw vs
     * firewalld, apt periodic vs dnf-automatic).
     *
     * @return array<int, array{key:string, label:string, detail:string, featureOn:bool, secure:bool, supported:bool, reason:?string}>
     */
    public function state(Server $server): array
    {
        $os = $this->detector->detect($server);
        $pm = $os->managesPackages() ? PackageManager::for($os) : null;
        $fw = FirewallTool::for($os);

        // Shell CONDITIONS (exit 0 = true); `false` where the host lacks the tool.
        $f2bInst = $pm ? $pm->isInstalledTest('fail2ban') : 'false';
        $auInst = $pm ? $pm->isInstalledTest($os->packageManager === 'dnf' ? 'dnf-automatic' : 'unattended-upgrades') : 'false';
        $auActive = match ($os->packageManager) {
            'apt' => 'apt-config dump APT::Periodic::Unattended-Upgrade 2>/dev/null | grep -q \'"1"\'',
            'dnf' => 'systemctl is-enabled --quiet dnf-automatic.timer',
            default => 'false',
        };

        $cmd = 'sshd -T 2>/dev/null | grep -iE "^(permitrootlogin|passwordauthentication) "; '
            .'echo "fail2ban $('.$f2bInst.' && echo 1 || echo 0) $(systemctl is-active --quiet fail2ban && echo active || echo inactive)"; '
            .'echo "firewall $('.$fw->installedTest().' && echo 1 || echo 0) $('.$fw->activeTest().' && echo active || echo inactive)"; '
            .'echo "autoupdate $('.$auInst.' && echo 1 || echo 0) $('.$auActive.' && echo active || echo inactive)"';

        $res = $this->fleet->runPrivileged($server, $cmd);
        if (! $res['ok']) {
            throw new RuntimeException('Härtungs-Status konnte nicht gelesen werden.');
        }

        return $this->parseState($res['output'], $os);
    }

    /** @return array<int, array{key:string, label:string, detail:string, featureOn:bool, secure:bool, supported:bool, reason:?string}> */
    private function parseState(string $out, OsProfile $os): array
    {
        $root = 'default';
        $pwauth = 'default';
        $pkg = [];

        foreach (preg_split('/\R/', $out) ?: [] as $line) {
            $line = trim($line);
            if (preg_match('/^permitrootlogin\s+(\S+)/i', $line, $m)) {
                $root = strtolower($m[1]);
            } elseif (preg_match('/^passwordauthentication\s+(\S+)/i', $line, $m)) {
                $pwauth = strtolower($m[1]);
            } else {
                $p = preg_split('/\s+/', $line);
                if (count($p) === 3 && in_array($p[0], ['fail2ban', 'firewall', 'autoupdate'], true)) {
                    $pkg[$p[0]] = ['installed' => $p[1] === '1', 'active' => $p[2] === 'active'];
                }
            }
        }

        $f2b = $pkg['fail2ban'] ?? ['installed' => false, 'active' => false];
        $fw = $pkg['firewall'] ?? ['installed' => false, 'active' => false];
        $upg = $pkg['autoupdate'] ?? ['installed' => false, 'active' => false];

        $detail = fn (string $name, array $st): string => $name.' · '
            .(! $st['installed'] ? 'nicht installiert' : ($st['active'] ? 'aktiv' : 'inaktiv'));
        // A package feature is only effectively ON when installed AND active —
        // stale config (e.g. apt periodic = 1 after removal) must not read as secure.
        $on = fn (array $st): bool => $st['installed'] && $st['active'];

        $row = fn (string $key, string $label, string $detail, bool $featureOn, bool $secure, ?string $reason): array => [
            'key' => $key, 'label' => $label, 'detail' => $detail,
            'featureOn' => $featureOn, 'secure' => $secure,
            'supported' => $reason === null, 'reason' => $reason,
        ];

        $fwName = $os->firewallTool === 'firewalld' ? 'firewalld' : 'ufw';
        $auName = $os->packageManager === 'dnf' ? 'dnf-automatic' : 'unattended-upgrades';

        // `featureOn` drives the button (Aktivieren/Deaktivieren); `secure` drives the OK/Offen pill.
        return [
            $row('ssh_root', 'SSH-Root-Login', 'PermitRootLogin '.$root, $root !== 'no', $root === 'no', $os->supports('ssh')),
            $row('ssh_password', 'SSH-Passwort-Login', 'PasswordAuthentication '.$pwauth, $pwauth !== 'no', $pwauth === 'no', $os->supports('ssh')),
            $row('fail2ban', 'fail2ban', $detail('fail2ban.service', $f2b), $on($f2b), $on($f2b), $os->supports('fail2ban')),
            $row('firewall', $os->firewallLabel(), $detail($fwName, $fw), $on($fw), $on($fw), $os->supports('firewall')),
            $row('unattended', 'Automatische Updates', $detail($auName, $upg), $on($upg), $on($upg), $os->supports('auto_updates')),
        ];
    }

    /** Modal title, reflecting the toggle direction. */
    public function title(string $action, bool $enable): string
    {
        return match ($action) {
            'ssh_root' => $enable ? 'SSH-Root-Login erlauben' : 'SSH-Root-Login deaktivieren',
            'ssh_password' => $enable ? 'SSH-Passwort-Login erlauben' : 'SSH-Passwort-Login deaktivieren',
            'fail2ban' => $enable ? 'fail2ban aktivieren' : 'fail2ban deaktivieren',
            'firewall' => $enable ? 'Firewall aktivieren' : 'Firewall deaktivieren',
            'unattended' => $enable ? 'Automatische Updates aktivieren' : 'Automatische Updates deaktivieren',
            default => throw new InvalidArgumentException("Unbekannte Härtung: {$action}"),
        };
    }

    public function description(string $action, bool $enable): string
    {
        return match ($action) {
            'ssh_root' => $enable
                ? 'Erlaubt direkten Root-Login über SSH wieder (PermitRootLogin yes) und lädt den SSH-Dienst neu.'
                : 'Unterbindet direkten Root-Login über SSH (PermitRootLogin no) und lädt den SSH-Dienst neu.',
            'ssh_password' => $enable
                ? 'Erlaubt die Anmeldung per Passwort wieder (PasswordAuthentication yes).'
                : 'Anmeldung nur noch per SSH-Key (PasswordAuthentication no). Nur möglich, wenn ein Key hinterlegt ist.',
            'fail2ban' => $enable
                ? 'Installiert fail2ban (falls nötig) und startet den Dienst. Wiederholte Fehl-Logins werden gesperrt.'
                : 'Stoppt fail2ban und deaktiviert den Autostart.',
            'firewall' => $enable
                ? 'Installiert die Firewall (falls nötig), gibt vorher SSH + 80/443 frei und aktiviert sie.'
                : 'Deaktiviert die Firewall des Servers.',
            'unattended' => $enable
                ? 'Installiert (falls nötig) und aktiviert die automatischen Sicherheitsupdates des Systems.'
                : 'Schaltet die automatischen Updates ab.',
            default => throw new InvalidArgumentException("Unbekannte Härtung: {$action}"),
        };
    }

    /** @return array{ok: bool, output: string} */
    public function apply(Server $server, string $action, bool $enable): array
    {
        $os = $this->detector->detect($server);

        // Gracefully refuse a feature the host's OS does not support.
        $feature = match ($action) {
            'ssh_root', 'ssh_password' => 'ssh',
            'fail2ban' => 'fail2ban',
            'firewall' => 'firewall',
            'unattended' => 'auto_updates',
            default => throw new InvalidArgumentException("Unbekannte Härtung: {$action}"),
        };
        if ($reason = $os->supports($feature)) {
            return ['ok' => false, 'output' => $reason];
        }

        if ($action === 'firewall') {
            $res = $enable ? $this->firewall->enable($server) : $this->firewall->disable($server);

            return ['ok' => $res['ok'], 'output' => $res['output']];
        }

        // Lock-out guard: never disable password auth without a usable key path.
        if ($action === 'ssh_password' && ! $enable) {
            if ($server->credential?->auth_type === 'password') {
                return ['ok' => false, 'output' => 'Clusev verbindet sich selbst per Passwort — Passwort-Login kann nicht deaktiviert werden, sonst verliert Clusev den Zugang. Hinterlege zuerst einen SSH-Key-Zugang.'];
            }
            if (! $this->hasAuthorizedKey($server)) {
                return ['ok' => false, 'output' => 'Kein SSH-Key hinterlegt — Passwort-Login kann nicht deaktiviert werden (Aussperrgefahr).'];
            }
        }

        // package operations need a long timeout; the 12s default would abort the install.
        $timeout = $enable && in_array($action, ['fail2ban', 'unattended'], true) ? 600 : 60;
        $res = $this->fleet->runPrivileged($server, $this->commandFor($os, $action, $enable), $timeout);

        return ['ok' => $res['ok'], 'output' => $res['output']];
    }

    /** Single source of truth for the shell command of a (non-firewall) toggle. */
    private function commandFor(OsProfile $os, string $action, bool $enable): string
    {
        $pm = PackageManager::for($os);

        return match ($action) {
            'ssh_root' => $this->sshDropInScript('PermitRootLogin '.($enable ? 'yes' : 'no')),
            'ssh_password' => $this->sshDropInScript('PasswordAuthentication '.($enable ? 'yes' : 'no')),
            'fail2ban' => $enable
                ? $pm->ensureInstalledScript('fail2ban').' && systemctl enable --now fail2ban'
                : 'systemctl disable --now fail2ban',
            'unattended' => $this->autoUpdateScript($os, $pm, $enable),
            default => throw new InvalidArgumentException("Unbekannte Härtung: {$action}"),
        };
    }

    /** Auto-update mechanism per family: apt periodic + timers, or dnf-automatic.timer. */
    private function autoUpdateScript(OsProfile $os, PackageManager $pm, bool $enable): string
    {
        if ($os->packageManager === 'dnf') {
            return $enable
                ? $pm->ensureInstalledScript('dnf-automatic')
                    .' && sed -i \'s/^apply_updates.*/apply_updates = yes/\' /etc/dnf/automatic.conf'
                    .' && systemctl enable --now dnf-automatic.timer'
                : 'systemctl disable --now dnf-automatic.timer';
        }

        // apt (default): the periodic drop-in + service/timers.
        return $enable
            ? $pm->ensureInstalledScript('unattended-upgrades')
                .' && printf \'%s\n%s\n\' \'APT::Periodic::Update-Package-Lists "1";\' \'APT::Periodic::Unattended-Upgrade "1";\' > '.self::APT_DROPIN
                .' && systemctl enable --now unattended-upgrades apt-daily-upgrade.timer apt-daily.timer'
            : 'printf \'%s\n%s\n\' \'APT::Periodic::Update-Package-Lists "0";\' \'APT::Periodic::Unattended-Upgrade "0";\' > '.self::APT_DROPIN
                .' && (systemctl disable --now unattended-upgrades 2>/dev/null || true)';
    }

    /** True if the SSH user has at least one authorized key. */
    private function hasAuthorizedKey(Server $server): bool
    {
        try {
            if (count($this->fleet->sshKeys($server)) > 0) {
                return true;
            }
        } catch (Throwable) {
            // fall through to the raw check
        }

        $res = $this->fleet->runPlain($server, 'grep -cE "^(ssh-|ecdsa-|sk-)" ~/.ssh/authorized_keys 2>/dev/null || echo 0');

        return $res['ok'] && (int) trim($res['output']) > 0;
    }

    /**
     * Install one sshd directive into the Clusev drop-in (replacing any prior value
     * of the same key), then reload whichever ssh unit exists.
     */
    private function sshDropInScript(string $directive): string
    {
        $key = strtok($directive, ' ');

        return 'mkdir -p /etc/ssh/sshd_config.d'
            .' && touch '.self::SSHD_DROPIN
            .' && grep -viE "^[[:space:]]*'.$key.'[[:space:]]" '.self::SSHD_DROPIN.' > '.self::SSHD_DROPIN.'.tmp 2>/dev/null; '
            .'mv '.self::SSHD_DROPIN.'.tmp '.self::SSHD_DROPIN.' 2>/dev/null; '
            .'echo "'.$directive.'" >> '.self::SSHD_DROPIN
            .' && (systemctl reload ssh || systemctl reload sshd)';
    }
}