Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
File size: 27,663 Bytes
0b4ac11 6d0999a 0b4ac11 6d0999a 0b4ac11 a812aba 0b4ac11 9df2461 0b4ac11 9cea1a9 0b4ac11 6d0999a 0b4ac11 9cea1a9 6d0999a 0b4ac11 a812aba 0b4ac11 9cea1a9 0b4ac11 9cea1a9 5406824 9cea1a9 9df2461 9cea1a9 9df2461 9cea1a9 9df2461 9cea1a9 9df2461 9cea1a9 9df2461 9cea1a9 0b4ac11 9cea1a9 0b4ac11 9cea1a9 0b4ac11 6d0999a 0b4ac11 9cea1a9 0b4ac11 9cea1a9 0b4ac11 a812aba 0b4ac11 9cea1a9 0b4ac11 9cea1a9 0b4ac11 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 | import argparse
import hashlib
import json
import shutil
import subprocess
import sys
from datetime import UTC, datetime
from pathlib import Path
SCRIPT_DIR = Path(__file__).resolve().parent
REPO_ROOT = SCRIPT_DIR.parent
FRONTEND_DIR = REPO_ROOT / "frontend"
BACKEND_DIR = REPO_ROOT / "backend"
if str(BACKEND_DIR) not in sys.path:
sys.path.insert(0, str(BACKEND_DIR))
from app.services import r2_storage
from app.services.crafted_template_service import (
SUMMARY_OBJECT_NAME,
build_summary_payload,
write_summary_to_r2,
)
from app.config import settings
def crafted_r2_namespace_prefix() -> str:
return str(getattr(settings, "CRAFTED_TEMPLATE_R2_PREFIX", "") or "").strip().strip("/")
def apply_crafted_r2_namespace(prefix: str) -> str:
normalized = str(prefix or "").strip().strip("/")
ns = crafted_r2_namespace_prefix()
if not ns:
return normalized
if not normalized:
return ns
if normalized == ns or normalized.startswith(f"{ns}/"):
return normalized
return f"{ns}/{normalized}"
def sha256_file(path: Path) -> str:
h = hashlib.sha256()
with path.open("rb") as f:
for chunk in iter(lambda: f.read(1024 * 1024), b""):
h.update(chunk)
return h.hexdigest()
def copy_tree(src_dir: Path, dst_dir: Path) -> None:
if not src_dir.exists():
return
for src in src_dir.rglob("*"):
if not src.is_file():
continue
rel = src.relative_to(src_dir)
dst = dst_dir / rel
dst.parent.mkdir(parents=True, exist_ok=True)
shutil.copy2(src, dst)
def first_existing(*paths: Path) -> Path | None:
for p in paths:
if p.exists():
return p
return None
def infer_frontend_entry(frontend_dir: Path, composition_id: str | None = None) -> str:
# Prefer the preview composition matching the render's composition_id
# (`<composition_id>Composition.tsx`) so preview and render stay in sync.
# Without this, sorted(glob) picks the alphabetically-first match, which can
# be a stale/simpler sibling (e.g. FjBriefCustomVideoComposition.tsx beats
# the render-matching FjBriefVideoComposition.tsx).
if composition_id:
preferred = frontend_dir / f"{composition_id}Composition.tsx"
if preferred.exists():
return f"frontend/{preferred.name}"
candidates = sorted(frontend_dir.glob("*VideoComposition.tsx"))
if candidates:
return f"frontend/{candidates[0].name}"
fallback = first_existing(frontend_dir / "TemplateVideoComposition.tsx", frontend_dir / "index.tsx")
if fallback:
return f"frontend/{fallback.name}"
return "frontend/TemplateVideoComposition.tsx"
def template_public_dirs(template_id: str, assets_root: str) -> list[Path]:
"""
All on-disk public folders that belong exclusively to a single crafted
template. Order matches what the bundler looks for (scoped first, then
top-level), and both `frontend/public` and `remotion-video/public`
roots are included so the prune step matches what the bundler reads
and what the fetch script writes.
"""
assets_root = (assets_root or "templates").strip().strip("/").replace("\\", "/")
out: list[Path] = []
for src_root in (REPO_ROOT / "frontend" / "public", REPO_ROOT / "remotion-video" / "public"):
scoped = src_root / Path(assets_root) / template_id
top_level = src_root / template_id
for candidate in (scoped, top_level):
if candidate.exists() and candidate not in out:
out.append(candidate)
return out
def prune_template_sources(
template_id: str,
*,
backend_src: Path,
frontend_src: Path,
remotion_src: Path,
public_assets_root: str,
) -> list[str]:
"""
Remove all local on-disk sources for a crafted template so the bundle
folder + R2 are the only remaining source of truth. Mirrors what
`fetch_crafted_template_from_r2.py` re-creates on restore: the three
source roots plus any template-scoped/top-level public folders.
"""
pruned: list[str] = []
for path in (backend_src, frontend_src, remotion_src):
if path.exists():
shutil.rmtree(path)
pruned.append(path.relative_to(REPO_ROOT).as_posix())
for path in template_public_dirs(template_id, public_assets_root):
if path.exists():
shutil.rmtree(path)
pruned.append(path.relative_to(REPO_ROOT).as_posix())
return pruned
def infer_frontend_preview(frontend_dir: Path) -> str | None:
"""
Locate the lightweight `<TemplateName>Preview.tsx` shipped with each crafted
template. Convention: a single default-exported React component used to
render the marquee thumbnail without loading the full layout package.
"""
if not frontend_dir.exists():
return None
direct = sorted(frontend_dir.glob("*Preview.tsx"))
if direct:
return f"frontend/{direct[0].name}"
nested = sorted(frontend_dir.glob("preview/*Preview.tsx"))
if nested:
rel = nested[0].relative_to(frontend_dir).as_posix()
return f"frontend/{rel}"
return None
def infer_frontend_layout_fields(frontend_dir: Path) -> str | None:
"""
Locate the per-template SceneEditModal field definitions file. Convention:
`frontend/layoutFields.ts` (or `.json`) at the bundle root, exporting a
`LAYOUT_FIELDS` Record keyed by layout id. Fetched with the list summary
so the modal can render template-specific controls without a code change.
"""
if not frontend_dir.exists():
return None
for name in ("layoutFields.ts", "layoutFields.tsx", "layoutFields.json"):
candidate = frontend_dir / name
if candidate.exists():
return f"frontend/{name}"
return None
def infer_remotion_entry(remotion_dir: Path, composition_id: str | None) -> str:
if composition_id:
preferred = remotion_dir / f"{composition_id}.tsx"
if preferred.exists():
return f"remotion-video/{preferred.name}"
candidates = sorted(remotion_dir.glob("*Video.tsx"))
if candidates:
return f"remotion-video/{candidates[0].name}"
fallback = first_existing(remotion_dir / "TemplateVideo.tsx", remotion_dir / "index.tsx")
if fallback:
return f"remotion-video/{fallback.name}"
return "remotion-video/TemplateVideo.tsx"
def gather_files_map(bundle_dir: Path) -> dict[str, dict[str, str | int]]:
out: dict[str, dict[str, str | int]] = {}
for p in sorted(bundle_dir.rglob("*")):
if not p.is_file():
continue
rel = p.relative_to(bundle_dir).as_posix()
out[rel] = {"size": p.stat().st_size, "sha256": sha256_file(p)}
return out
def capture_preview_image(bundle_dir: Path) -> bool:
"""Render the bundle's preview_file to `assets/preview.jpg` (+ manifest entry).
Shells out to the frontend's puppeteer capture
(`npm run thumbs:crafted -- <bundle-dir>`), which renders the bundle's real
marquee preview in headless Chrome and writes the image into the bundle folder
before it is uploaded. Non-fatal: a failure logs a warning and returns False
so bundling still proceeds (the tile falls back to the themed placeholder).
"""
if not FRONTEND_DIR.is_dir():
print(f"[preview] frontend dir not found ({FRONTEND_DIR}); skipping image capture")
return False
cmd = ["npm", "run", "thumbs:crafted", "--", str(bundle_dir)]
print(f"[preview] capturing preview image: {' '.join(cmd)}")
try:
subprocess.run(cmd, cwd=str(FRONTEND_DIR), check=True, timeout=240)
return (bundle_dir / "assets" / "preview.jpg").is_file()
except subprocess.CalledProcessError as e:
print(f"[preview] capture failed (rc={e.returncode}); continuing without image")
return False
except (subprocess.TimeoutExpired, FileNotFoundError, OSError) as e:
print(f"[preview] capture unavailable ({e}); continuing without image")
return False
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Bundle a built-in template into crafted package and upload to R2.")
parser.add_argument("--template-id", required=True, help="Built-in template id (for example: nightfall, default).")
parser.add_argument("--template-key", help="Crafted template_key. Default: <template-id>_bundle")
parser.add_argument("--public-template-id", help="Crafted public id. Default: crafted_<template-id>_bundle")
parser.add_argument("--supported-video-style", default="explainer", help="explainer|promotional|storytelling")
parser.add_argument("--upload", action="store_true", help="Upload bundle files to R2. If omitted, only creates local bundle.")
parser.add_argument("--r2-prefix", help="Optional explicit R2 prefix.")
parser.add_argument(
"--replace-existing",
action="store_true",
help=(
"Replace an existing crafted bundle in-place on R2 using the same prefix "
"(stable URLs/manifest path). Requires --upload and --r2-prefix."
),
)
parser.add_argument(
"--prune-source-after-bundle",
action="store_true",
help=(
"After bundle (and upload if enabled), remove local source folders for this template "
"from backend/frontend/remotion codebase."
),
)
parser.add_argument(
"--upload-existing-bundle",
action="store_true",
help=(
"Skip re-bundling and upload files from an existing local crafted bundle folder "
"(requires manifest.json in bundle root)."
),
)
parser.add_argument(
"--bundle-dir",
help=(
"Optional local crafted bundle folder path for --upload-existing-bundle. "
"Default: crafted-templates/<template-id>-crafted-template"
),
)
parser.add_argument(
"--public-assets-root",
default="templates",
help=(
"Copy only template-scoped public assets from "
"frontend/public/<root>/<template-id> and remotion-video/public/<root>/<template-id> "
"(default root: templates)."
),
)
parser.add_argument(
"--no-capture-preview",
action="store_true",
help=(
"Skip auto-generating assets/preview.jpg from the bundle's preview file "
"(default: capture it via the frontend puppeteer script before upload)."
),
)
return parser.parse_args()
def _read_text_if_exists(path: Path) -> str | None:
if not path.is_file():
return None
try:
return path.read_text(encoding="utf-8")
except Exception as e:
print(f"[summary] failed to read {path}: {e}")
return None
def upload_summary_for_bundle(prefix: str, bundle_dir: Path, manifest: dict) -> bool:
"""
Compose summary.json from the local bundle and upload it to R2 at
`{prefix}/summary.json`. Called after bundle files are uploaded so the
dashboard list endpoint sees a fresh summary without requiring a
follow-up call to /admin/publish.
"""
backend = manifest.get("backend") if isinstance(manifest.get("backend"), dict) else {}
meta_rel = backend.get("meta") if isinstance(backend.get("meta"), str) else "backend/meta.json"
meta_path = bundle_dir / meta_rel
meta_raw = _read_text_if_exists(meta_path)
if meta_raw is None:
print(f"[summary] skipped β meta.json missing at {meta_path}")
return False
try:
meta = json.loads(meta_raw)
except Exception as e:
print(f"[summary] skipped β meta.json invalid: {e}")
return False
if not isinstance(meta, dict):
print("[summary] skipped β meta.json is not a JSON object")
return False
frontend = manifest.get("frontend") if isinstance(manifest.get("frontend"), dict) else {}
preview_file_rel = frontend.get("preview") if isinstance(frontend.get("preview"), str) else None
layout_fields_rel = frontend.get("layout_fields") if isinstance(frontend.get("layout_fields"), str) else None
preview_file = _read_text_if_exists(bundle_dir / preview_file_rel) if preview_file_rel else None
layout_fields = _read_text_if_exists(bundle_dir / layout_fields_rel) if layout_fields_rel else None
preview_image_rel = manifest.get("preview_image", "assets/preview.jpg")
preview_image_url = r2_storage.public_url(f"{prefix.strip('/')}/{preview_image_rel}") if preview_image_rel else None
summary = build_summary_payload(
meta=meta,
preview_image_url=preview_image_url,
preview_file=preview_file,
preview_file_rel=preview_file_rel if preview_file is not None else None,
layout_fields=layout_fields,
layout_fields_rel=layout_fields_rel if layout_fields is not None else None,
)
return write_summary_to_r2(prefix, summary)
def _list_r2_keys(prefix: str) -> list[str]:
bucket = settings.R2_BUCKET_NAME
if not bucket:
return []
client = r2_storage._get_client()
keys: list[str] = []
token: str | None = None
while True:
kwargs = {
"Bucket": bucket,
"Prefix": prefix.strip().strip("/") + "/",
"MaxKeys": 1000,
}
if token:
kwargs["ContinuationToken"] = token
resp = client.list_objects_v2(**kwargs)
for obj in resp.get("Contents", []) or []:
key = obj.get("Key")
if isinstance(key, str):
keys.append(key)
if not resp.get("IsTruncated"):
break
token = resp.get("NextContinuationToken")
if not isinstance(token, str):
break
return keys
def main() -> None:
args = parse_args()
template_id = str(args.template_id).strip().lower()
if not template_id:
raise ValueError("template-id is required")
if args.replace_existing and not args.upload:
raise ValueError("--replace-existing requires --upload")
if args.replace_existing and not (isinstance(args.r2_prefix, str) and args.r2_prefix.strip()):
raise ValueError("--replace-existing requires --r2-prefix")
if args.upload_existing_bundle and args.prune_source_after_bundle:
raise ValueError("--prune-source-after-bundle is only valid when creating a fresh bundle")
existing_manifest = None
existing_prefix_raw = args.r2_prefix.strip().strip("/") if isinstance(args.r2_prefix, str) and args.r2_prefix.strip() else ""
existing_prefix = apply_crafted_r2_namespace(existing_prefix_raw)
if args.replace_existing:
existing_manifest = r2_storage.download_json(f"{existing_prefix}/manifest.json")
if not existing_manifest:
raise FileNotFoundError(f"Cannot replace existing bundle: missing manifest at {existing_prefix}/manifest.json")
backend_src = REPO_ROOT / "backend" / "templates" / template_id
frontend_src = REPO_ROOT / "frontend" / "src" / "components" / "remotion" / template_id
remotion_src = REPO_ROOT / "remotion-video" / "src" / "templates" / template_id
out_root_default = REPO_ROOT / "crafted-templates" / f"{template_id}-crafted-template"
out_root = Path(args.bundle_dir).expanduser().resolve() if args.bundle_dir else out_root_default
manifest_path = out_root / "manifest.json"
if args.upload_existing_bundle:
if not out_root.exists():
raise FileNotFoundError(f"Bundle folder not found: {out_root}")
if not manifest_path.exists():
raise FileNotFoundError(f"manifest.json not found in bundle folder: {manifest_path}")
manifest = json.loads(manifest_path.read_text(encoding="utf-8"))
else:
if not backend_src.exists():
raise FileNotFoundError(f"Backend template folder not found: {backend_src}")
if not remotion_src.exists():
raise FileNotFoundError(f"Remotion template folder not found: {remotion_src}")
if out_root.exists():
shutil.rmtree(out_root)
(out_root / "backend").mkdir(parents=True, exist_ok=True)
(out_root / "frontend").mkdir(parents=True, exist_ok=True)
(out_root / "remotion-video").mkdir(parents=True, exist_ok=True)
for name in ("meta.json", "prompt.md", "layout_prompt.md"):
src = backend_src / name
if not src.exists():
raise FileNotFoundError(f"Required backend file missing: {src}")
shutil.copy2(src, out_root / "backend" / name)
copy_tree(frontend_src, out_root / "frontend")
copy_tree(remotion_src, out_root / "remotion-video")
# Optional shared font modules. Template entries commonly import
# ../../../fonts/<template-id>-defaults from their source location.
# After bundling, these live under <section>/fonts/ so the runtime
# mount can place render fonts back at workspace src/fonts/.
for section, src_dir in (
("frontend", REPO_ROOT / "frontend" / "src" / "fonts"),
("remotion-video", REPO_ROOT / "remotion-video" / "src" / "fonts"),
):
if src_dir.is_dir():
dst_dir = out_root / section / "fonts"
for name in (f"{template_id}-defaults.ts", f"{template_id}-defaults.tsx"):
src = src_dir / name
if src.is_file():
dst_dir.mkdir(parents=True, exist_ok=True)
shutil.copy2(src, dst_dir / name)
# Static assets: copy template-scoped public assets into bundle `public/`.
#
# Two layouts are supported (whichever exists wins; both are merged):
# 1) frontend/public/<assets_root>/<template_id>/** (scoped)
# remotion-video/public/<assets_root>/<template_id>/**
# β bundled under: public/<assets_root>/<template_id>/**
#
# 2) frontend/public/<template_id>/** (top-level)
# remotion-video/public/<template_id>/**
# β bundled under: public/<template_id>/**
#
# Both layouts preserve the relative path so that `staticFile("β¦")`
# keys keep working when the package is mounted into a render
# workspace OR when previewed at runtime.
(out_root / "public").mkdir(parents=True, exist_ok=True)
assets_root = str(args.public_assets_root or "templates").strip().strip("/").replace("\\", "/")
scoped_rel = Path(assets_root) / template_id
for src_root in (REPO_ROOT / "frontend" / "public", REPO_ROOT / "remotion-video" / "public"):
scoped_src = src_root / scoped_rel
if scoped_src.exists():
copy_tree(scoped_src, out_root / "public" / scoped_rel)
top_level_src = src_root / template_id
if top_level_src.exists() and top_level_src != scoped_src:
copy_tree(top_level_src, out_root / "public" / template_id)
# Include shared frontend SocialIcons used by ending CTA layouts.
# Built-in-authored layouts often import "../../SocialIcons"; after bundling
# into `frontend/layouts/*`, that should resolve to `frontend/SocialIcons.tsx`.
# Only copy the shared module when the template does NOT ship its own
# SocialIcons.tsx β otherwise this would clobber a template-local copy
# (e.g. fj_market_brief exports `fjNormalizeSocials`) with the generic one.
shared_frontend_social_icons = (
REPO_ROOT / "frontend" / "src" / "components" / "remotion" / "SocialIcons.tsx"
)
if (
shared_frontend_social_icons.exists()
and not (frontend_src / "SocialIcons.tsx").exists()
):
shutil.copy2(shared_frontend_social_icons, out_root / "frontend" / "SocialIcons.tsx")
# Include shared remotion SocialIcons used by ending CTA layouts during render.
# Built-in-authored remotion layouts often import "../../SocialIcons"; after bundling
# into `remotion-video/layouts/*`, that should resolve to `remotion-video/SocialIcons.tsx`.
# As above, skip when the template ships its own SocialIcons.tsx.
shared_remotion_social_icons = REPO_ROOT / "remotion-video" / "src" / "templates" / "SocialIcons.tsx"
if (
shared_remotion_social_icons.exists()
and not (remotion_src / "SocialIcons.tsx").exists()
):
shutil.copy2(shared_remotion_social_icons, out_root / "remotion-video" / "SocialIcons.tsx")
template_key = (
args.template_key
or (existing_manifest.get("template_key") if isinstance(existing_manifest, dict) else None)
or f"{template_id}_bundle"
).strip()
public_template_id = (
args.public_template_id
or (existing_manifest.get("template_id") if isinstance(existing_manifest, dict) else None)
or f"crafted_{template_id}_bundle"
).strip()
style = str(
args.supported_video_style
or (existing_manifest.get("supported_video_style") if isinstance(existing_manifest, dict) else None)
or "explainer"
).strip().lower()
meta_raw = json.loads((out_root / "backend" / "meta.json").read_text(encoding="utf-8"))
composition_id = meta_raw.get("composition_id") if isinstance(meta_raw, dict) else None
frontend_files = [
p.relative_to(out_root).as_posix()
for p in sorted((out_root / "frontend").rglob("*"))
if p.is_file() and p.suffix in {".ts", ".tsx", ".json"}
]
remotion_files = [
p.relative_to(out_root).as_posix()
for p in sorted((out_root / "remotion-video").rglob("*"))
if p.is_file() and p.suffix in {".ts", ".tsx"}
]
files_map = gather_files_map(out_root)
preview_rel = infer_frontend_preview(out_root / "frontend")
layout_fields_rel = infer_frontend_layout_fields(out_root / "frontend")
manifest = {
"template_id": public_template_id,
"template_key": template_key,
"supported_video_style": style,
"backend": {
"meta": "backend/meta.json",
"prompt": "backend/prompt.md",
"layout_prompt": "backend/layout_prompt.md",
},
"frontend": {
"mount_id": template_key,
"entry": infer_frontend_entry(
out_root / "frontend",
composition_id if isinstance(composition_id, str) else None,
),
"layout_index": "frontend/layouts/index.ts",
# Marquee preview (lightweight, fetched alongside list summary).
# Optional β falls back to preview_image when missing.
"preview": preview_rel,
# Per-template SceneEditModal field definitions.
# Optional β falls back to meta.json layout_prop_schema.fields.
"layout_fields": layout_fields_rel,
"files": frontend_files,
},
"remotion": {
"mount_id": template_key,
"entry": infer_remotion_entry(out_root / "remotion-video", composition_id if isinstance(composition_id, str) else None),
"layout_index": "remotion-video/layouts/index.ts",
"files": remotion_files,
},
"preview_image": "assets/preview.jpg",
"files": files_map,
}
manifest_path.write_text(json.dumps(manifest, ensure_ascii=False, indent=2), encoding="utf-8")
files_map["manifest.json"] = {"size": manifest_path.stat().st_size, "sha256": sha256_file(manifest_path)}
manifest["files"] = files_map
manifest_path.write_text(json.dumps(manifest, ensure_ascii=False, indent=2), encoding="utf-8")
template_key = (
args.template_key
or (manifest.get("template_key") if isinstance(manifest, dict) else None)
or (existing_manifest.get("template_key") if isinstance(existing_manifest, dict) else None)
or f"{template_id}_bundle"
).strip()
public_template_id = (
args.public_template_id
or (manifest.get("template_id") if isinstance(manifest, dict) else None)
or (existing_manifest.get("template_id") if isinstance(existing_manifest, dict) else None)
or f"crafted_{template_id}_bundle"
).strip()
print(f"local_bundle={out_root}")
print(f"public_template_id={public_template_id}")
print(f"template_key={template_key}")
# Auto-generate the static preview image from the bundle's preview file and
# write it into the bundle (+ manifest) BEFORE upload, so a single bundle run
# ships assets/preview.jpg. Non-fatal if it can't run.
if not args.no_capture_preview:
if capture_preview_image(out_root):
print("preview_image_captured=true")
else:
print("preview_image_captured=false")
if not args.upload:
if args.prune_source_after_bundle:
pruned = prune_template_sources(
template_id,
backend_src=backend_src,
frontend_src=frontend_src,
remotion_src=remotion_src,
public_assets_root=args.public_assets_root,
)
print("pruned_source=true")
for path in pruned:
print(f"pruned_path={path}")
print("upload=false")
return
stamp = datetime.now(UTC).strftime("%Y%m%d-%H%M%S")
if args.replace_existing:
prefix = existing_prefix
else:
requested_prefix = (
args.r2_prefix.strip().strip("/")
if isinstance(args.r2_prefix, str) and args.r2_prefix.strip()
else f"crafted-templates/{template_id}-crafted-template-{stamp}"
)
prefix = apply_crafted_r2_namespace(requested_prefix)
deleted = 0
if args.replace_existing:
old_keys = _list_r2_keys(prefix)
for key in old_keys:
if r2_storage.delete_object(key):
deleted += 1
uploaded = []
for p in sorted(out_root.rglob("*")):
if not p.is_file():
continue
rel = p.relative_to(out_root).as_posix()
key = f"{prefix}/{rel}"
r2_storage.upload_file(str(p), key)
uploaded.append(key)
print(f"upload=true")
print(f"r2_prefix={prefix}")
if args.replace_existing:
print("replace_existing=true")
print(f"deleted_existing_files={deleted}")
print(f"uploaded_files={len(uploaded)}")
# Generate + upload summary.json so the dashboard list endpoint reflects
# the new bundle without requiring a follow-up /admin/publish call.
# Composes locally from the same manifest + meta + preview source we
# just bundled, so server and script produce identical artifacts.
if upload_summary_for_bundle(prefix, out_root, manifest):
print(f"summary_uploaded={prefix}/{SUMMARY_OBJECT_NAME}")
else:
print("summary_uploaded=false")
if args.prune_source_after_bundle:
pruned = prune_template_sources(
template_id,
backend_src=backend_src,
frontend_src=frontend_src,
remotion_src=remotion_src,
public_assets_root=args.public_assets_root,
)
print("pruned_source=true")
for path in pruned:
print(f"pruned_path={path}")
if __name__ == "__main__":
main()
|