#!/usr/bin/env python3
"""Office to PDF using the preinstalled LibreOffice, isolated per invocation."""
from pathlib import Path
import shutil
import subprocess
import sys
import tempfile

from _pdf_common import SkillArgumentParser, output_pdf, new_temp_pdf, publish_temp_file, run_cli

FORMATS={'.docx','.doc','.odt','.rtf','.pptx','.ppt','.odp','.xlsx','.xls','.ods'}


def convert(args):
    from pypdf import PdfReader
    source=Path(args.input).expanduser().resolve()
    if not source.is_file() or source.suffix.lower() not in FORMATS or not 0 < source.stat().st_size <= 25*1024*1024:
        raise ValueError('输入需为不超过 25 MiB 的本地 Office 文档；不支持把 PDF 直接反向转成可编辑 Office')
    if not 1 <= args.timeout <= 600:
        raise ValueError('timeout 必须在 1–600 秒之间')
    executable=shutil.which('soffice') or shutil.which('libreoffice')
    if not executable:
        raise RuntimeError('基础镜像缺少 LibreOffice，需要更新镜像')
    target=output_pdf(args.output,args.overwrite)
    temporary=new_temp_pdf(target)
    try:
        with tempfile.TemporaryDirectory(prefix='pdf-office-') as folder:
            root=Path(folder)
            incoming=root/'input'; outgoing=root/'output'; profile=root/'profile'
            incoming.mkdir(); outgoing.mkdir(); profile.mkdir()
            local=incoming/('source'+source.suffix.lower()); shutil.copyfile(source,local)
            command=[executable,'-env:UserInstallation='+profile.as_uri(),'--headless','--nologo','--nodefault',
                     '--nofirststartwizard','--convert-to','pdf','--outdir',str(outgoing),str(local)]
            completed=subprocess.run(command,capture_output=True,text=True,timeout=args.timeout,check=False)
            result=outgoing/'source.pdf'
            if completed.returncode or not result.is_file():
                raise RuntimeError('Office 转 PDF 失败：'+(completed.stderr or completed.stdout)[-1500:])
            count=len(PdfReader(result).pages)
            if not count:
                raise ValueError('转换结果没有页面')
            shutil.copyfile(result,temporary)
        publish_temp_file(temporary,target,args.overwrite)
    finally:
        temporary.unlink(missing_ok=True)
    return {'source':str(source),'path':str(target),'page_count':count,'engine':'libreoffice','requires_visual_review':True,
            'note':'转换前需在对应 Office skill 中重算公式并核对字体、图表和打印范围。'}


def main(argv=None):
    parser=SkillArgumentParser(description='Office 文档导出 PDF，源文件保持不变')
    parser.add_argument('--input',required=True); parser.add_argument('--output',required=True)
    parser.add_argument('--timeout',type=int,default=180); parser.add_argument('--overwrite',action='store_true')
    return run_cli(lambda:convert(parser.parse_args(sys.argv[1:] if argv is None else argv)))


if __name__=='__main__':
    raise SystemExit(main())