from pathlib import Path
import zipfile, os, shutil, re, textwrap

src = Path("/mnt/data/taiwan_stock_line_landing_page_upgraded.zip")
work = Path("/mnt/data/tw_v2_work")
out = Path("/mnt/data/taiwan_stock_line_landing_page_v2.zip")

if work.exists():
    shutil.rmtree(work)
work.mkdir()

with zipfile.ZipFile(src, "r") as z:
    z.extractall(work)

# Find html entry
html_files = list(work.rglob("*.html"))
if html_files:
    html = html_files[0]
    content = html.read_text(encoding="utf-8", errors="ignore")
    
    # Add v2 enhancement styles/content before closing body
    addon = r"""
<section class="v2-section" style="padding:50px 20px;background:#07111f;color:#fff;text-align:center;">
  <h2>每日掌握台股市場機會</h2>
  <p style="opacity:.85;max-width:720px;margin:15px auto;">
    專業整理市場趨勢、熱門產業與投資觀察，讓您快速掌握市場資訊。
  </p>
  <div style="display:flex;gap:16px;justify-content:center;flex-wrap:wrap;margin-top:30px;">
    <div style="padding:25px;border-radius:16px;background:#12233a;min-width:220px;">
      <h3>市場趨勢</h3><p>追蹤台股熱門方向</p>
    </div>
    <div style="padding:25px;border-radius:16px;background:#12233a;min-width:220px;">
      <h3>產業分析</h3><p>掌握熱門產業變化</p>
    </div>
    <div style="padding:25px;border-radius:16px;background:#12233a;min-width:220px;">
      <h3>即時資訊</h3><p>手機快速查看更新</p>
    </div>
  </div>
</section>
<a href="#" class="line-fixed" style="position:fixed;right:20px;bottom:20px;background:#06c755;color:white;padding:14px 24px;border-radius:30px;text-decoration:none;font-weight:bold;z-index:999;">
免費加入 LINE
</a>
"""
    if "</body>" in content:
        content = content.replace("</body>", addon + "\n</body>")
    else:
        content += addon
    html.write_text(content, encoding="utf-8")

if out.exists():
    out.unlink()
with zipfile.ZipFile(out, "w", zipfile.ZIP_DEFLATED) as z:
    for p in work.rglob("*"):
        if p.is_file():
            z.write(p, p.relative_to(work))

str(out)