<?php

declare(strict_types=1);

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

use app\common\service\ClinicalKnowledgeReference;
use app\common\service\NihaixiaClinicalSkill;
use app\common\service\TcmAncientBooksReference;

function ancientExpect(bool $condition, string $message): void
{
    if (!$condition) {
        fwrite(STDERR, "FAIL: {$message}\n");
        exit(1);
    }
}

$indexProperty = new ReflectionProperty(TcmAncientBooksReference::class, 'index');
$indexProperty->setValue(null, [
    'commit' => TcmAncientBooksReference::COMMIT,
    'index_version' => TcmAncientBooksReference::INDEX_VERSION,
    'terms' => [
        '消渴' => [[
            'title' => '测试古籍', 'chapter' => '消渴篇',
            'path' => '123-测试古籍.txt', 'line' => 42,
            'excerpt' => '原文中谈到消渴的历史记载。', 'score' => 15,
        ]],
    ],
    'chapter_terms' => [
        '痛风' => [[
            'title' => '测试古籍二', 'chapter' => '痛风',
            'path' => '124-测试古籍二.txt', 'line' => 88,
            'excerpt' => '痛风相关的原文。', 'score' => 12,
        ]],
    ],
]);

$query = '患者糖尿病、高血糖，请分析病症。';
$sources = TcmAncientBooksReference::sourcesForText($query);
ancientExpect(count($sources) === 1, 'matching patient symptoms select one excerpt');
ancientExpect($sources[0]['title'] === '测试古籍' && $sources[0]['chapter'] === '消渴篇', 'book and chapter retained');
ancientExpect($sources[0]['url'] === '' && $sources[0]['path'] === '123-测试古籍.txt' && $sources[0]['line'] === 42,
    'local citation retains file and line without an external URL');
ancientExpect($sources[0]['excerpt'] === '原文中谈到消渴的历史记载。', 'local citation retains verifiable original text');
ancientExpect(TcmAncientBooksReference::sourcesForText('请分析耳鸣。') === [], 'unmatched symptoms have no invented citation');
ancientExpect(TcmAncientBooksReference::sourcesForText('患者出现痛风症状。')[0]['line'] === 88, 'unlisted diseases can match indexed chapter headings');

$augmented = ClinicalKnowledgeReference::augmentQuery($query);
ancientExpect(str_contains($augmented, TcmAncientBooksReference::COMMIT), 'query includes ancient-book version');
ancientExpect(str_contains($augmented, NihaixiaClinicalSkill::VERSION), 'existing clinical skill remains');
ancientExpect(str_ends_with($augmented, $query), 'original patient question remains intact');
ancientExpect(ClinicalKnowledgeReference::augmentQuery($augmented) === $augmented, 'retry does not duplicate references');

foreach ([$query, '患者出现痛风症状。', '患者有未收录的症状。', '阶段=text 患者糖尿病', '阶段=final 患者糖尿病'] as $case) {
    $selected = ClinicalKnowledgeReference::ancientBookSources($case);
    ancientExpect(
        ClinicalKnowledgeReference::augmentQueryWithAncientBookSources($case, $selected)
            === ClinicalKnowledgeReference::augmentQuery($case),
        'selected sources preserve the exact legacy prompt for each clinical case'
    );
}
ancientExpect(
    ClinicalKnowledgeReference::augmentQueryWithAncientBookSources('患者有未收录的症状。', [])
        === ClinicalKnowledgeReference::augmentQuery('患者有未收录的症状。'),
    'an unrelated request receives no preceding patient citations'
);

$messages = [['role' => 'user', 'content' => $query]];
$wire = ClinicalKnowledgeReference::augmentMessages($messages);
ancientExpect(count($wire) === 2 && str_contains($wire[0]['content'], '测试古籍'), 'chat receives reference in system message');
ancientExpect(ClinicalKnowledgeReference::augmentMessages($wire) === $wire, 'stream retry does not duplicate reference');

$indexProperty->setValue(null, null);
echo "TCM ancient-books reference contract passed.\n";