<?php

declare(strict_types=1);

namespace app\common\service {
    // Exercise chat() offline without loading runtime configuration or opening a connection.
    function config(string $name): array
    {
        return $GLOBALS['upstreamTestConfig'];
    }

    function curl_init(): \stdClass
    {
        return new \stdClass();
    }

    function curl_setopt_array(\stdClass $handle, array $options): bool
    {
        $handle->url = $options[CURLOPT_URL];
        $GLOBALS['upstreamTestRequests'][] = ['url' => $handle->url, 'payload' => json_decode($options[CURLOPT_POSTFIELDS], true)];
        return true;
    }

    function curl_exec(\stdClass $handle): string
    {
        return json_encode(str_ends_with($handle->url, '/chat-messages')
            ? ['answer' => 'offline reply'] : ['choices' => [['message' => ['content' => 'offline reply']]]]);
    }

    function curl_errno(\stdClass $handle): int { return 0; }
    function curl_getinfo(\stdClass $handle, int $option): int { return 200; }
    function curl_close(\stdClass $handle): void {}
}

namespace {

require dirname(__DIR__) . '/vendor/autoload.php';

use app\common\service\DifyChatService;

function expectSame($expected, $actual, string $message): void
{
    if ($expected !== $actual) {
        fwrite(STDERR, "FAIL: {$message}\n");
        exit(1);
    }
}

function callPrivate(string $name, array $arguments)
{
    $method = (new ReflectionClass(DifyChatService::class))->getMethod($name);
    return $method->invoke(null, ...$arguments);
}

$generic = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1',
    'model-name',
    ['prompt_version' => 'test'],
    'clinical prompt',
    'server-user',
]);
expectSame(2, count($generic), 'ambiguous /v1 base should support both protocols');
expectSame('https://ai.example.test/v1/chat-messages', $generic[0]['url'], 'Dify endpoint');
expectSame('blocking', $generic[0]['payload']['response_mode'], 'Dify blocking request');
expectSame('https://ai.example.test/v1/chat/completions', $generic[1]['url'], 'OpenAI endpoint');
expectSame('model-name', $generic[1]['payload']['model'], 'profile model selection');
expectSame('clinical prompt', $generic[1]['payload']['messages'][0]['content'], 'OpenAI prompt');
$serializedSpecs = json_encode($generic, JSON_UNESCAPED_SLASHES) ?: '';
expectSame(false, str_contains($serializedSpecs, 'api_key'), 'credential field is absent from request bodies');
expectSame(false, str_contains($serializedSpecs, 'provider'), 'provider override is absent from request bodies');
expectSame(false, str_contains($serializedSpecs, 'base_url'), 'base URL override is absent from request bodies');

// 附件不得让 OpenAI-compatible 回退失效。历史实现遇到报告/录像等非图片附件时只返回
// Dify 一种协议，网关 404 会直接变成“模型未能处理本次请求”，开处方因此必失败。
$attachments = [
    ['type' => 'document', 'transfer_method' => 'remote_url', 'url' => 'https://cdn.example.test/report.pdf'],
    ['type' => 'video', 'transfer_method' => 'remote_url', 'url' => 'https://cdn.example.test/call.mp4'],
    ['type' => 'image', 'transfer_method' => 'remote_url', 'url' => 'https://cdn.example.test/tongue.jpg'],
];
$withFiles = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1',
    'model-name',
    [],
    'clinical prompt',
    'server-user',
    false,
    $attachments,
]);
expectSame(2, count($withFiles), 'non-image attachments must keep the OpenAI fallback available');
expectSame($attachments, $withFiles[0]['payload']['files'], 'Dify still receives every attachment');
expectSame(
    'clinical prompt',
    $withFiles[0]['payload']['query'],
    'Dify carries attachments in the file channel, not as a manifest'
);
$openAiContent = $withFiles[1]['payload']['messages'][0]['content'];
expectSame(true, is_array($openAiContent), 'OpenAI content becomes multimodal when attachments exist');
expectSame('text', $openAiContent[0]['type'], 'attachment manifest travels in the text part');
expectSame(
    true,
    str_contains($openAiContent[0]['text'], 'https://cdn.example.test/report.pdf'),
    'document attachment is listed instead of being silently dropped'
);
expectSame(
    true,
    str_contains($openAiContent[0]['text'], 'https://cdn.example.test/call.mp4'),
    'recording attachment is listed instead of being silently dropped'
);
expectSame(
    'https://cdn.example.test/tongue.jpg',
    $openAiContent[1]['image_url']['url'],
    'image attachments stay inline for multimodal reading'
);

$openAi = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat/completions',
    'model-name',
    [],
    'prompt',
    'server-user',
]);
expectSame(1, count($openAi), 'explicit OpenAI endpoint should not probe Dify');
expectSame('openai', $openAi[0]['protocol'], 'explicit OpenAI protocol');

$openAiWithFiles = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat/completions',
    'model-name',
    [],
    'prompt',
    'server-user',
    false,
    $attachments,
]);
expectSame(1, count($openAiWithFiles), 'explicit OpenAI endpoint stays OpenAI even with attachments');
expectSame('openai', $openAiWithFiles[0]['protocol'], 'explicit OpenAI protocol with attachments');

$difyWithFiles = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat-messages',
    'model-name',
    [],
    'prompt',
    'server-user',
    false,
    $attachments,
]);
expectSame(1, count($difyWithFiles), 'explicit Dify endpoint stays Dify with attachments');
expectSame($attachments, $difyWithFiles[0]['payload']['files'], 'explicit Dify keeps the file channel');

$dify = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat-messages',
    'model-name',
    [],
    'prompt',
    'server-user',
]);
expectSame(1, count($dify), 'explicit Dify endpoint should not probe OpenAI');
expectSame('dify', $dify[0]['protocol'], 'explicit Dify protocol');

expectSame('Dify answer', callPrivate('extractContent', [['answer' => ' Dify answer ']]), 'Dify response');
expectSame(
    'OpenAI answer',
    callPrivate('extractContent', [['choices' => [['message' => ['content' => ' OpenAI answer ']]]]]),
    'OpenAI response'
);
expectSame(
    'multipart answer',
    callPrivate('extractContent', [['choices' => [['message' => ['content' => [
        ['type' => 'text', 'text' => 'multipart '],
        ['type' => 'text', 'text' => 'answer'],
    ]]]]]]),
    'OpenAI multipart response'
);

// 附件数量必须按上游应用的 file_upload.number_limits 截断。Dify 超限时返回
// 400 invalid_param 并整单拒绝，历史实现会把患者的全部舌象/报告一次性送上去，
// 导致该患者的每一次 AI 请求都固定失败（UPSTREAM_REJECTED）。
$manyFiles = [];
for ($i = 0; $i < 5; $i++) {
    $manyFiles[] = ['type' => 'image', 'transfer_method' => 'remote_url', 'url' => "https://cdn.example.test/tongue{$i}.jpg"];
}
for ($i = 0; $i < 4; $i++) {
    $manyFiles[] = ['type' => 'document', 'transfer_method' => 'remote_url', 'url' => "https://cdn.example.test/report{$i}.pdf"];
}
$capped = callPrivate('normalizeFiles', [$manyFiles, 3]);
expectSame(3, count($capped['kept']), 'the total attachment count is capped, not each type');
expectSame(6, count($capped['dropped']), 'attachments past the cap are recorded, not discarded');
expectSame(
    'https://cdn.example.test/tongue3.jpg',
    $capped['dropped'][0]['url'],
    'the earliest attachments are the ones kept'
);
expectSame(
    ['kept' => [], 'dropped' => []],
    callPrivate('normalizeFiles', [[['type' => 'image', 'url' => 'ftp://cdn.example.test/x.jpg']], 3]),
    'non-http attachments are still rejected outright'
);

$duplicateFiles = [
    ['file_id' => 'file:1', 'source_ids' => ['source:1'], 'type' => 'image', 'url' => 'https://cdn.example.test/shared.jpg'],
    ['file_id' => 'file:2', 'source_ids' => ['source:2'], 'type' => 'image', 'url' => 'https://cdn.example.test/shared.jpg'],
    ['file_id' => 'file:3', 'source_ids' => ['source:3'], 'type' => 'image', 'url' => 'https://cdn.example.test/other.jpg'],
];
expectSame(2, count(callPrivate('normalizeFiles', [$duplicateFiles, 3])['kept']), 'default normalization still deduplicates shared URLs');
$upstreamTestConfig = ['enable' => true, 'base_url' => '', 'timeout' => 30, 'max_files' => 3,
    'models' => ['qwen' => ['name' => 'offline-model', 'api_key' => 'offline-fixture']]];
foreach (['dify' => 'chat-messages', 'openai' => 'chat/completions'] as $protocol => $endpoint) {
    $upstreamTestConfig['base_url'] = 'https://ai.example.test/v1/' . $endpoint;
    $upstreamTestRequests = [];
    $strictResult = DifyChatService::chat('qwen', [], 'offline query', 'offline-user', $duplicateFiles, ['strict_files' => true]);
    expectSame(true, $strictResult['ok'], 'strict ' . $protocol . ' accepts distinct logical attachments sharing a URL');
    expectSame(1, count($upstreamTestRequests), 'strict ' . $protocol . ' submits the complete batch once');
    $payload = $upstreamTestRequests[0]['payload'];
    $wireUrls = $protocol === 'dify' ? array_column($payload['files'], 'url')
        : array_column(array_column(array_slice($payload['messages'][0]['content'], 1), 'image_url'), 'url');
    expectSame(array_column($duplicateFiles, 'url'), $wireUrls, 'strict ' . $protocol . ' transmits every attachment in manifest order');
    expectSame(count($wireUrls), $strictResult['transmitted_file_count'], 'strict ' . $protocol . ' acknowledgment matches actual wire attachment count');
    expectSame($protocol, $strictResult['attachment_transport'], 'strict response identifies the actual attachment protocol');

    $upstreamTestRequests = [];
    $ordinaryResult = DifyChatService::chat('qwen', [], 'offline query', 'offline-user', $duplicateFiles);
    expectSame(true, $ordinaryResult['ok'], 'ordinary ' . $protocol . ' chat remains successful');
    $payload = $upstreamTestRequests[0]['payload'];
    $wireUrls = $protocol === 'dify' ? array_column($payload['files'], 'url')
        : array_column(array_column(array_slice($payload['messages'][0]['content'], 1), 'image_url'), 'url');
    expectSame(array_values(array_unique(array_column($duplicateFiles, 'url'))), $wireUrls, 'ordinary ' . $protocol . ' still deduplicates URLs');
}
foreach ([
    ['type' => 'image', 'url' => 'ftp://cdn.example.test/invalid.jpg'],
    ['type' => 'image', 'url' => 'https://user@cdn.example.test/invalid.jpg'],
    ['type' => 'image', 'url' => "https://cdn.example.test/invalid\n.jpg"],
    ['type' => 'unknown', 'url' => 'https://cdn.example.test/invalid.jpg'],
    null,
] as $invalidFile) {
    $upstreamTestRequests = [];
    $invalidFiles = [$duplicateFiles[0], $duplicateFiles[1], $invalidFile];
    $invalidResult = DifyChatService::chat('qwen', [], 'offline query', 'offline-user', $invalidFiles, ['strict_files' => true]);
    expectSame('STRICT_FILES_INVALID_OR_LIMIT', $invalidResult['error_code'] ?? '', 'strict duplicate preservation never bypasses attachment validation');
    expectSame([], $upstreamTestRequests, 'invalid strict batches are rejected before transport');
}
foreach ([2, 0] as $limit) {
    $upstreamTestConfig['max_files'] = $limit;
    $upstreamTestRequests = [];
    $limitedResult = DifyChatService::chat('qwen', [], 'offline query', 'offline-user', $duplicateFiles, ['strict_files' => true]);
    expectSame('STRICT_FILES_INVALID_OR_LIMIT', $limitedResult['error_code'] ?? '', 'strict limits count logical attachments even when URLs repeat');
    expectSame([], $upstreamTestRequests, 'over-limit strict batches are never partially transmitted');
}

// 被截断的附件必须出现在提示词清单里，否则模型会把“没看到”当成“没有”。
$cappedSpecs = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat-messages',
    'model-name',
    [],
    'clinical prompt',
    'server-user',
    false,
    $capped['kept'],
    $capped['dropped'],
]);
$cappedQuery = $cappedSpecs[0]['payload']['query'];
expectSame(true, str_contains($cappedQuery, '<ATTACHMENTS_NOT_INLINE>'), 'dropped attachments are declared to the model');
expectSame(true, str_contains($cappedQuery, 'https://cdn.example.test/tongue3.jpg'), 'dropped attachment URL is listed');
expectSame(false, str_contains($cappedQuery, 'https://cdn.example.test/tongue0.jpg'), 'delivered attachments are not duplicated in the manifest');

// 附件被整体拒绝时必须降级为纯文本重试，而不是让整次问诊失败。
$plan = callPrivate('buildAttemptPlan', [$capped['kept'], $capped['dropped']]);
expectSame(2, count($plan), 'a request with attachments gets a text-only fallback attempt');
expectSame([], $plan[1]['files'], 'the fallback attempt sends no attachments');
expectSame(9, count($plan[1]['omitted']), 'the fallback attempt declares every attachment');
expectSame(1, count(callPrivate('buildAttemptPlan', [[], []])), 'a request without attachments is attempted once');

$inputPlan = callPrivate('buildInputAttemptPlan', [['prompt_version' => 'v2']]);
expectSame(2, count($inputPlan), 'structured Dify inputs get one compatibility fallback');
expectSame([], $inputPlan[1], 'the compatibility fallback uses an empty inputs object');
expectSame([[]], callPrivate('buildInputAttemptPlan', [[]]), 'empty inputs are not retried twice');

$difyInputSpec = ['protocol' => 'dify'];
$openAiInputSpec = ['protocol' => 'openai'];
expectSame(
    true,
    callPrivate('isInputRejection', [
        ['errno' => 0, 'http_code' => 400, 'body' => '{"code":"invalid_param"}'],
        $difyInputSpec,
        ['prompt_version' => 'v2'],
    ]),
    'Dify invalid_param retries with query-only input'
);
expectSame(
    false,
    callPrivate('isInputRejection', [
        ['errno' => 0, 'http_code' => 400, 'body' => '{"code":"invalid_param"}'],
        $difyInputSpec,
        [],
    ]),
    'an already empty inputs object is never retried'
);
expectSame(
    false,
    callPrivate('isInputRejection', [
        ['errno' => 0, 'http_code' => 400, 'body' => '{"code":"invalid_param"}'],
        $openAiInputSpec,
        ['prompt_version' => 'v2'],
    ]),
    'OpenAI protocol does not use the Dify input fallback'
);
expectSame(
    false,
    callPrivate('isInputRejection', [
        ['errno' => 0, 'http_code' => 400, 'body' => '{"code":"provider_quota_exceeded"}'],
        $difyInputSpec,
        ['prompt_version' => 'v2'],
    ]),
    'quota and provider failures are not submitted twice'
);
expectSame(
    true,
    callPrivate('isInputRejection', [
        [
            'errno' => 0,
            'http_code' => 200,
            'upstream_error' => true,
            'upstream_code' => 'invalid_param',
            'emitted' => false,
        ],
        $difyInputSpec,
        ['prompt_version' => 'v2'],
    ]),
    'a streaming invalid_param before any delta also retries without inputs'
);
expectSame(
    false,
    callPrivate('isInputRejection', [
        [
            'errno' => 0,
            'http_code' => 200,
            'upstream_error' => true,
            'upstream_code' => 'invalid_param',
            'emitted' => true,
        ],
        $difyInputSpec,
        ['prompt_version' => 'v2'],
    ]),
    'a stream that already emitted content is never replayed'
);

expectSame(true, callPrivate('shouldRetryWithoutFiles', [400, $capped['kept']]), 'invalid_param retries without attachments');
expectSame(true, callPrivate('shouldRetryWithoutFiles', [413, $capped['kept']]), 'oversized attachments retry without attachments');
expectSame(false, callPrivate('shouldRetryWithoutFiles', [400, []]), 'a text-only rejection is not retried');
expectSame(false, callPrivate('shouldRetryWithoutFiles', [401, $capped['kept']]), 'a credential failure is not retried');
expectSame(false, callPrivate('shouldRetryWithoutFiles', [500, $capped['kept']]), 'an upstream outage is not retried here');

// 协议回退会把最初的“附件被拒”换成另一协议的状态码（Dify 400 -> OpenAI 404），
// 降级判断必须按每次响应累计，否则去掉附件的重试永远不会发生。
expectSame(
    true,
    callPrivate('isFileRejection', [['errno' => 0, 'http_code' => 400], $capped['kept']]),
    'an attachment rejection is recognised on the response that carried it'
);
expectSame(
    false,
    callPrivate('isFileRejection', [['errno' => 0, 'http_code' => 404], $capped['kept']]),
    'the fallback protocol 404 is not itself an attachment rejection'
);
expectSame(
    true,
    callPrivate('isFileRejection', [
        ['errno' => 0, 'http_code' => 200, 'upstream_error' => true],
        $capped['kept'],
    ]),
    'a 200 stream carrying event:error counts as an attachment rejection'
);
expectSame(
    false,
    callPrivate('isFileRejection', [
        ['errno' => 0, 'http_code' => 200, 'upstream_error' => true],
        [],
    ]),
    'a text-only request never degrades further'
);
expectSame(
    false,
    callPrivate('isFileRejection', [['errno' => 28, 'http_code' => 0], $capped['kept']]),
    'a transport failure is not mistaken for an attachment rejection'
);

// Dify 的 inputs 必须是 JSON 对象。PHP 空数组会被编码成 []，上游以
// invalid_param 拒绝整单——空 inputs 的调用方会 100% 失败。
$emptyInputs = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat-messages', 'model-name', [], 'prompt', 'server-user',
]);
expectSame(
    true,
    str_contains((string) json_encode($emptyInputs[0]['payload']), '"inputs":{}'),
    'empty inputs are encoded as a JSON object, never as an array'
);
$filledInputs = callPrivate('buildRequestSpecs', [
    'https://ai.example.test/v1/chat-messages', 'model-name', ['prompt_version' => 'v2'], 'prompt', 'server-user',
]);
expectSame(
    true,
    str_contains((string) json_encode($filledInputs[0]['payload']), '"inputs":{"prompt_version":"v2"}'),
    'populated inputs keep their keys'
);

expectSame('invalid_param', callPrivate('cleanUpstreamCode', ['invalid_param']), 'enumerable upstream codes are kept for logs');
expectSame('', callPrivate('cleanUpstreamCode', ["Run failed: 404 for https://cdn.example.test/a.pdf"]), 'upstream prose never reaches the log');

expectSame(true, callPrivate('isValidBaseUrl', ['https://ai.example.test/v1']), 'https URL');
expectSame(true, callPrivate('isValidBaseUrl', ['http://127.0.0.1:8080/v1']), 'internal http URL');
expectSame(false, callPrivate('isValidBaseUrl', ['file:///tmp/socket']), 'non-http URL');
expectSame(false, callPrivate('isValidBaseUrl', ['https://user@example.test/v1']), 'userinfo URL');
expectSame(false, callPrivate('isValidBaseUrl', ['https://ai.example.test/v1?unsafe=query']), 'query URL');
expectSame(true, callPrivate('isValidTimeout', [90]), 'normal timeout');
expectSame(false, callPrivate('isValidTimeout', [0]), 'zero timeout');
expectSame(false, callPrivate('isValidTimeout', [301]), 'excessive timeout');

echo "Prescription AI upstream contract: OK\n";

}