from pathlib import Path
from io import BytesIO
from functools import lru_cache
from xml.etree import ElementTree as ET
import json,base64,html,re,zipfile,hashlib
from PIL import Image,ImageDraw,ImageFont
from reportlab.pdfgen import canvas
from reportlab.lib.utils import ImageReader

ROOT=Path(__file__).resolve().parent
BOOK=json.loads((ROOT/'book.json').read_text())
FONT='/System/Library/AssetsV2/com_apple_MobileAsset_Font7/8dc7805506cc9f233dcc19aabf593196842a47ae.asset/AssetData/Hannotate.ttc'
SCALE=2
@lru_cache(None)
def font(size):return ImageFont.truetype(FONT,size*SCALE)

def wrap(text,size,width,draw):
    tokens=re.findall(r'GPT-6 Astra|Opus 5\.5|Sonnet 5\.5|Gemini 4\.1|Sol 6\.1|3\.1 Pro|3\.8 Flash|\d+(?:\.\d+)?%|第[一二三四五六七八九十百千万0-9]+|[A-Za-z0-9]+(?:[.%-][A-Za-z0-9]+)*|\s+|.',text)
    closing='，。！？；：、）》”’…';opening='（《“‘'
    n=len(tokens);maxw=width*SCALE;face=font(size);dp={n:(0,[])}
    for i in range(n-1,-1,-1):
        if tokens[i].isspace():
            if i+1 in dp:dp[i]=dp[i+1]
            continue
        if tokens[i][0] in closing:continue
        segment='';best=None
        for j in range(i,n):
            segment+=tokens[j];line=segment.strip();measured=draw.textlength(line,font=face)
            if measured>maxw:break
            if not line or line[-1] in opening:continue
            nxt=j+1
            while nxt<n and tokens[nxt].isspace():nxt+=1
            if nxt<n and tokens[nxt][0] in closing:continue
            if nxt not in dp:continue
            cost=1000000+100*((maxw-measured)/maxw)**2+dp[nxt][0]
            candidate=(cost,[line]+dp[nxt][1])
            if best is None or cost<best[0]:best=candidate
        if best is not None:dp[i]=best
    if 0 not in dp:raise ValueError('Cannot wrap: '+text)
    return dp[0][1]

layout=[]
def compose(p):
    n=p['number'];artpath=ROOT/'art'/f'page-{n:02d}.png'
    source=Image.open(artpath).convert('RGB')
    im=source.resize((2048,3072),Image.Resampling.LANCZOS);draw=ImageDraw.Draw(im)
    svg=[f'<svg xmlns="http://www.w3.org/2000/svg" width="1024" height="1536" viewBox="0 0 1024 1536"><image width="1024" height="1536" href="data:image/png;base64,{base64.b64encode(artpath.read_bytes()).decode()}"/>']
    for k,panel in enumerate(p['panels']):
        if not panel['dialogue']:continue
        x,y,w,h=panel['text_box'];fit=None
        for size in [30,29,28,27,26,25,24]:
            step=round(size*1.2);gap=10
            groups=[(speaker,wrap(text,size,w,draw)) for speaker,text in panel['dialogue']]
            used=sum(len(lines)*step for _,lines in groups)+(len(groups)-1)*gap
            if used<=h:fit=(size,step,groups,used);break
        if fit is None:raise ValueError(f'Page {n} panel {k+1}: text needs more space')
        size,step,groups,used=fit
        start=y
        for speaker,lines in groups:
            color=BOOK['characters'][speaker]['color']
            for line in lines:
                draw.text((x*SCALE,y*SCALE),line,font=font(size),fill=color,anchor='lt')
                svg.append(f'<text x="{x}" y="{y}" dominant-baseline="text-before-edge" font-family="Hannotate SC,sans-serif" font-size="{size}" fill="{color}">{html.escape(line)}</text>')
                y+=step
            y+=gap
        layout.append({'page':n,'panel':k+1,'size':size,'lines':sum(len(ls) for _,ls in groups),'text_bottom':start+used,'box_bottom':start+h})
    svg.append('</svg>')
    out=ROOT/'pages'/f'page-{n:02d}.png';im.save(out)
    svgpath=ROOT/'editable'/f'page-{n:02d}.svg';svgpath.write_text('\n'.join(svg))
    texts=[e.text or '' for e in ET.parse(svgpath).findall('{http://www.w3.org/2000/svg}text')]
    expected=''.join(t for panel in p['panels'] for _,t in panel['dialogue'])
    assert re.sub(r'\s+','',expected)==re.sub(r'\s+','',''.join(texts))
    assert not any(t[0] in '，。！？；：、）》”’…' for t in texts)
    return out

if __name__=='__main__':
    import sys
    requested=[int(x) for x in sys.argv[1:]]
    pages=[p for p in BOOK['pages'] if not requested or p['number'] in requested]
    paths=[compose(p) for p in pages]
    (ROOT/'tmp/layout.json').write_text(json.dumps(layout,ensure_ascii=False,indent=2))
    if not requested:
        out=ROOT/'output/pdf';out.mkdir(parents=True,exist_ok=True)
        pdf=out/'Argon讨论-第二组.pdf'
        c=canvas.Canvas(str(pdf),pagesize=(512,768));c.setTitle(BOOK['title']);c.setAuthor('Codex')
        for path in paths:
            b=BytesIO();Image.open(path).convert('RGB').save(b,format='JPEG',quality=96,subsampling=0);b.seek(0)
            c.drawImage(ImageReader(b),0,0,width=512,height=768);c.showPage()
        c.save()
        sheet=Image.new('RGB',(1840,1410),'#e9e6ed')
        for i,p in enumerate(paths):
            im=Image.open(p);im.thumbnail((440,660));sheet.paste(im,((i%4)*465,(i//4)*710))
        sheet.save(ROOT/'overview.jpg',quality=95)
        reader='''<!doctype html><html lang="zh-CN"><meta charset="utf-8"><meta name="viewport" content="width=device-width,initial-scale=1"><title>漫画</title><style>body{margin:0;background:#e9e6ed}main{max-width:1000px;margin:auto}img{width:100%;height:auto;display:block;margin-bottom:24px}@media print{img{margin:0;break-after:page}}</style><main>'''
        reader+=''.join(f'<img src="pages/page-{i:02d}.png" alt="">' for i in range(1,9))+'</main></html>'
        (ROOT/'index.html').write_text(reader)
        (ROOT/'tmp/art-sha256.json').write_text(json.dumps({p.name:hashlib.sha256(p.read_bytes()).hexdigest() for p in (ROOT/'art').glob('*.png')},indent=2))
    print(json.dumps({'rendered':[p['number'] for p in pages],'layout':layout},ensure_ascii=False))