import asyncio
from pathlib import Path
import subprocess
import sys
import pytest
from app.contracts import ExportRequest
from app.export import service
from app.export.document import ExportResult
from app.export import browser_pdf
from app.export.browser_pdf import render_snapshot, browser_executable


def test_renderer_command_dispatches_frozen_build_to_sidecar_entry(monkeypatch, tmp_path):
    source, output = tmp_path / 'snapshot.html', tmp_path / 'document.pdf'
    monkeypatch.setattr(sys, 'frozen', True, raising=False)
    assert browser_pdf.renderer_command(source, output, 'A4') == [
        sys.executable, '--opennexus-pdf-render', str(source), str(output), 'A4'
    ]


def test_frozen_render_worker_isolated_from_core_bootstrap(monkeypatch):
    calls = []
    monkeypatch.setattr(sys, 'frozen', True, raising=False)

    def run(command, **kwargs):
        calls.append((command, kwargs))
        Path(command[3]).write_bytes(b'%PDF-frozen')
        return subprocess.CompletedProcess(command, 0, '', '')

    monkeypatch.setattr(browser_pdf.subprocess, 'run', run)
    result = render_snapshot('<h1>Frozen</h1>', 'A4')
    command, kwargs = calls[0]
    assert command[1] == '--opennexus-pdf-render'
    assert kwargs['stdin'] is subprocess.DEVNULL
    assert kwargs['timeout'] == browser_pdf.PDF_RENDER_TIMEOUT_SECONDS
    assert result.content == b'%PDF-frozen'


def test_browser_render_timeout_is_bounded(monkeypatch):
    def run(command, **kwargs):
        raise subprocess.TimeoutExpired(command, kwargs['timeout'])

    monkeypatch.setattr(browser_pdf.subprocess, 'run', run)
    with pytest.raises(RuntimeError, match='timed out'):
        render_snapshot('<h1>Timeout</h1>', 'A4')


def test_browser_render_falls_back_in_process_when_worker_fails(monkeypatch):
    class Failed:
        returncode = 1
        stderr = 'desktop child launch failed'

    monkeypatch.setattr(browser_pdf.subprocess, 'run', lambda *args, **kwargs: Failed())
    called = {}

    def fallback(source, output, page_size):
        called['page_size'] = page_size
        output.write_bytes(b'%PDF-fallback')

    monkeypatch.setattr(browser_pdf, 'print_snapshot', fallback)
    result = render_snapshot('<h1>Fallback</h1>', 'Letter')
    assert result.content == b'%PDF-fallback'
    assert called == {'page_size': 'Letter'}


def test_browser_snapshot_uses_print_pipeline(monkeypatch):
    calls=[]
    def render(html,size):
        calls.append((html,size));return ExportResult(content=b'%PDF-browser',mime_type='application/pdf')
    monkeypatch.setattr('app.export.browser_pdf.render_snapshot',render)
    monkeypatch.setattr(service,'parse_document',lambda _:pytest.fail('Browser snapshots must not be reparsed by ReportLab'))
    async def run():
        request=ExportRequest(format='pdf',source={'type':'markdown','markdown':'snapshot'},print_html='<style>h1::before{content:"tape"}</style><h1>Note</h1>')
        job=await service.create_export(request);done=await service.wait_for_export(job.job_id)
        assert done.status.value=='completed'
        assert calls==[(request.print_html,'A4')]
    asyncio.run(run())


def test_preview_resources_keeps_vault_boundary_and_plot_quota_removed():
    async def run():
        source='![outside](../../private.png)\n\n```function-plot\n'+'\n'.join(f'y=x+{i}' for i in range(17))+'\n```'
        resources=await service.preview_resources(ExportRequest(format='pdf',source={'type':'markdown','markdown':source}))
        assert resources['images'][0]['data'] is None
        assert resources['images'][0]['warnings']
        assert resources['plots'][0]['svg'].startswith('<svg')
    asyncio.run(run())


@pytest.mark.skipif(browser_executable() is None,reason='No installed Chromium browser')
def test_browser_prints_css_without_executing_document_scripts(tmp_path):
    # 若脚本被执行会清空正文；测试同时确认 CSS/字体可用且网络、文件资源保持禁用。
    html='<style>h1{color:#875343;font-size:37px} h1::before{content:"Theme "}</style><h1>Snapshot</h1><script>document.body.innerHTML="EXECUTED"</script><img src="file:///private.png">'
    result=render_snapshot(html,'A4')
    assert result.content.startswith(b'%PDF')
    assert b'/Subtype /Type0' in result.content or b'/Type /Font' in result.content
    import shutil, subprocess
    if shutil.which('pdftotext'):
        pdf=tmp_path/'snapshot.pdf'; pdf.write_bytes(result.content)
        text=subprocess.check_output(['pdftotext',str(pdf),'-']).decode('utf-8')
        assert 'Theme Snapshot' in text
        assert 'EXECUTED' not in text


@pytest.mark.parametrize('source', ['<div><IMG SRC="assets/a&amp;b.png"></div>', 'inline <img src="assets/a&amp;b.png"/> image'])
def test_preview_embeds_html_images(source):
    from app.config import get_settings
    from PIL import Image
    import base64
    folder=get_settings().vault_path/'notes'/'assets'
    folder.mkdir(parents=True)
    Image.new('RGBA',(2,2),(10,20,30,128)).save(folder/'a&b.png')
    resources=asyncio.run(service.preview_resources(ExportRequest(format='pdf',source={'type':'markdown','markdown':source,'file_path':'notes/test.md'})))
    image=resources['images'][0]
    assert image['source']=='assets/a&b.png'
    assert base64.b64decode(image['data'].split(',')[1]).startswith(b'\x89PNG')
    assert image['warnings']==[]


def test_html_images_keep_path_validation_and_code_is_not_an_image():
    source='<img src="../../private.png">\n\ninline <img src="https://example.com/a.png">\n\n`<img src="code.png">`\n\n```html\n<img src="fenced.png">\n```'
    resources=asyncio.run(service.preview_resources(ExportRequest(format='pdf',source={'type':'markdown','markdown':source})))
    assert [image['source'] for image in resources['images']]==['../../private.png','https://example.com/a.png']
    assert all(image['data'] is None and image['warnings'] for image in resources['images'])