LehrerApp/ai-backend/db.php

<?php
declare(strict_types=1);

require_once __DIR__ . '/providers/AnthropicProvider.php';
require_once __DIR__ . '/providers/FakeProvider.php';

/** Baut eine PDO-Verbindung aus config.php auf. */
function ai_backend_db(array $config): PDO
{
    $db = $config['db'];
    $dsn = "mysql:host={$db['host']};dbname={$db['name']};charset=utf8mb4";
    return new PDO($dsn, $db['user'], $db['pass'], [
        PDO::ATTR_ERRMODE            => PDO::ERRMODE_EXCEPTION,
        PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
        PDO::ATTR_EMULATE_PREPARES   => false,
    ]);
}

/**
 * Liest den Authorization-Header — auf vielen Apache/PHP-FPM-Setups landet er NICHT in
 * $_SERVER['HTTP_AUTHORIZATION'] (Header wird von PHP-FPM standardmäßig verworfen), sondern nur
 * in $_SERVER['REDIRECT_HTTP_AUTHORIZATION'] (nach der .htaccess-Weiterleitung, siehe dort) oder
 * ist nur über getallheaders() erreichbar. Alle drei Quellen abklappern, statt sich auf eine zu
 * verlassen — genau das war die Ursache für "Login klappt, danach sofort Session abgelaufen".
 */
function ai_backend_authorization_header(): string
{
    if (!empty($_SERVER['HTTP_AUTHORIZATION'])) {
        return $_SERVER['HTTP_AUTHORIZATION'];
    }
    if (!empty($_SERVER['REDIRECT_HTTP_AUTHORIZATION'])) {
        return $_SERVER['REDIRECT_HTTP_AUTHORIZATION'];
    }
    if (function_exists('getallheaders')) {
        foreach (getallheaders() as $name => $value) {
            if (strcasecmp($name, 'Authorization') === 0) {
                return $value;
            }
        }
    }
    return '';
}

/**
 * Bearer-Token aus dem Authorization-Header lesen, gegen tokens.token_hash prüfen
 * (SHA-256, der Klartext wird nie gespeichert) und den zugehörigen aktiven User zurückgeben.
 * Sendet bei fehlendem/ungültigem/abgelaufenem Token selbst eine 401-Antwort und beendet das Skript.
 */
function ai_backend_authenticate(PDO $pdo): array
{
    $header = ai_backend_authorization_header();
    if (!preg_match('/^Bearer\s+(.+)$/i', $header, $m)) {
        ai_backend_fail(401, 'Kein gültiges Token übermittelt.');
    }
    $tokenHash = hash('sha256', $m[1]);

    $stmt = $pdo->prepare(
        'SELECT u.* FROM users u
         JOIN tokens t ON t.user_id = u.id
         WHERE t.token_hash = ? AND t.expires_at > NOW() AND u.is_active = 1
         LIMIT 1'
    );
    $stmt->execute([$tokenHash]);
    $user = $stmt->fetch();
    if (!$user) {
        ai_backend_fail(401, 'Ungültiges oder abgelaufenes Token.');
    }
    return $user;
}

/** Einheitliche Fehlerantwort als JSON, beendet danach das Skript. */
function ai_backend_fail(int $httpStatus, string $message): never
{
    http_response_code($httpStatus);
    header('Content-Type: application/json');
    echo json_encode(['error' => $message]);
    exit;
}

/**
 * Ruft das konfigurierte LLM (oder den FakeProvider für lokale Tests, siehe README.md) auf und
 * verrechnet die echten Token-Kosten gegen das Guthaben des Nutzers. Gemeinsame Logik für jeden
 * Endpunkt, der einen LLM-Call abrechnet (plan.php, explain.php) — die SELECT-FOR-UPDATE-
 * Absicherung gegen Race Conditions bei gleichzeitigen Anfragen desselben Nutzers soll nicht an
 * mehreren Stellen gepflegt werden müssen. Gibt bei Erfolg das Provider-Ergebnis unverändert
 * zurück (inkl. "content", das der Aufrufer je nach Endpunkt selbst auswertet).
 */
function ai_backend_call_and_charge(PDO $pdo, array $config, array $user, string $systemPrompt, string $userContent): array
{
    $useFake = getenv('AI_BACKEND_FAKE_PROVIDER') === '1';
    if ($useFake) {
        $provider = new FakeProvider();
        $modelKey = 'fake';
    } else {
        $providerName = $config['llm_provider'];
        if ($providerName !== 'anthropic') {
            ai_backend_fail(500, "Provider '$providerName' ist nicht implementiert.");
        }
        $modelKey = $config['anthropic']['model'];
        $provider = new AnthropicProvider($config['anthropic']['api_key'], $modelKey);
    }

    try {
        $result = $provider->sendMessage($systemPrompt, $userContent, $config['max_output_tokens']);
    } catch (RuntimeException $e) {
        ai_backend_fail(502, $e->getMessage());
    }

    $defaultPricing = ['input' => 0, 'output' => 0, 'cache_write' => 0, 'cache_read' => 0];
    $pricing = $config['pricing'][$modelKey] ?? ($useFake ? $defaultPricing : null);
    if ($pricing === null) {
        ai_backend_fail(500, "Kein Preis für Modell '$modelKey' konfiguriert.");
    }
    $pricing += $defaultPricing; // fehlende cache_write/cache_read in älteren config.php-Einträgen -> 0

    $cacheCreationTokens = $result['cacheCreationInputTokens'] ?? 0;
    $cacheReadTokens = $result['cacheReadInputTokens'] ?? 0;
    $cost = ($result['inputTokens'] / 1_000_000 * $pricing['input'])
          + ($result['outputTokens'] / 1_000_000 * $pricing['output'])
          + ($cacheCreationTokens / 1_000_000 * $pricing['cache_write'])
          + ($cacheReadTokens / 1_000_000 * $pricing['cache_read']);

    // Guthaben abziehen und Transaktion protokollieren — mit Zeilensperre, damit zwei gleichzeitige
    // Anfragen desselben Nutzers das Guthaben nicht versehentlich unter 0 drücken können.
    $pdo->beginTransaction();
    try {
        $stmt = $pdo->prepare('SELECT balance_usd FROM users WHERE id = ? FOR UPDATE');
        $stmt->execute([$user['id']]);
        $currentBalance = (float) $stmt->fetchColumn();

        if ($currentBalance - $cost < 0) {
            $pdo->rollBack();
            ai_backend_fail(402, 'Guthaben würde durch diese Anfrage negativ werden.');
        }

        $newBalance = $currentBalance - $cost;
        $pdo->prepare('UPDATE users SET balance_usd = ? WHERE id = ?')->execute([$newBalance, $user['id']]);
        $pdo->prepare(
            'INSERT INTO transactions
                 (user_id, type, model, input_tokens, output_tokens,
                  cache_creation_input_tokens, cache_read_input_tokens, cost_usd, balance_after)
             VALUES (?, "usage", ?, ?, ?, ?, ?, ?, ?)'
        )->execute([
            $user['id'], $modelKey, $result['inputTokens'], $result['outputTokens'],
            $cacheCreationTokens, $cacheReadTokens, $cost, $newBalance,
        ]);

        $pdo->commit();
    } catch (Throwable $e) {
        $pdo->rollBack();
        throw $e;
    }

    return $result;
}