<?php

namespace app\ai\controller;

use app\common\service\AiGatewayService;
use app\common\util\JwtUtil;
use library\ResponseHelper;
use think\Controller;

/**
 * AI Gateway · 各端统一消费入口（§2.5）
 *
 * - GET  /v1/ai/gateway/config       脱敏配置（各端只读）
 * - POST /v1/ai/chat/completions      统一补全（slug / tier? / messages[] / companyId）
 */
class AiGatewayController extends Controller
{
    /** 脱敏配置（不含 Key） */
    public function config()
    {
        return ResponseHelper::success(AiGatewayService::config());
    }

    /** 统一补全 */
    public function chat()
    {
        $slug = (string) $this->request->param('slug', '');
        $tier = (string) $this->request->param('tier', '');
        $messages = $this->request->param('messages', []);
        if (!is_array($messages)) {
            $messages = json_decode((string) $messages, true) ?: [];
        }
        $companyId = (int) $this->request->param('companyId', 0);
        $maxTokens = (int) $this->request->param('maxTokens', 0);

        // companyId 缺省时从 JWT 兜底
        if ($companyId <= 0) {
            $companyId = $this->companyIdFromToken();
        }
        $userId = $this->userIdFromToken();

        if (empty($messages)) {
            return ResponseHelper::error('messages 不能为空', 400);
        }

        $result = AiGatewayService::complete($slug, $messages, $companyId, $userId, $tier, [
            'maxTokens' => $maxTokens,
        ]);

        if (!$result['ok']) {
            return ResponseHelper::error($result['error'] ?? 'AI 调用失败', (int) ($result['code'] ?? 500), [
                'tier'    => $result['tier'] ?? '',
                'balance' => $result['balance'] ?? null,
            ]);
        }
        return ResponseHelper::success([
            'content'    => $result['content'],
            'providerId' => $result['providerId'],
            'model'      => $result['model'],
            'tier'       => $result['tier'],
            'strategy'   => $result['strategy'] ?? '',
            'deducted'   => $result['deducted'],
            'balance'    => $result['balance'],
            'usage'      => $result['usage'] ?? [],
        ]);
    }

    protected function companyIdFromToken(): int
    {
        $payload = $this->jwtPayload();
        if (!$payload) {
            return 0;
        }
        return (int) (is_array($payload) ? ($payload['companyId'] ?? 0) : ($payload->companyId ?? 0));
    }

    protected function userIdFromToken(): int
    {
        $payload = $this->jwtPayload();
        if (!$payload) {
            return 0;
        }
        return (int) (is_array($payload) ? ($payload['id'] ?? 0) : ($payload->id ?? 0));
    }

    protected function jwtPayload()
    {
        try {
            $token = JwtUtil::getRequestToken();
            return $token ? JwtUtil::verifyToken($token) : false;
        } catch (\Throwable $e) {
            return false;
        }
    }
}