#!/usr/bin/env python3
from pathlib import Path

p = Path("/Users/john/Project/Memind/server.mjs")
text = p.read_text()
if '"/learn"' in text and "public/learn" in text:
    print("ALREADY_PATCHED")
    raise SystemExit(0)

needle = """  app2.use(
    "/assets",
    staticMiddleware(
      pathApi.join(h5Root, "public/assets"),
      { maxAge: "1d" }
    )
  );
  app2.get(
    "/dev/wechat-share-preview","""

insert = """  app2.use(
    "/assets",
    staticMiddleware(
      pathApi.join(h5Root, "public/assets"),
      { maxAge: "1d" }
    )
  );
  app2.use(
    "/learn",
    staticMiddleware(
      pathApi.join(h5Root, "public/learn"),
      { maxAge: "5m", index: "index.html" }
    )
  );
  app2.get(
    "/dev/wechat-share-preview","""

if needle not in text:
    print("NEEDLE_NOT_FOUND")
    raise SystemExit(1)

p.write_text(text.replace(needle, insert, 1))
print("PATCHED_OK")