From 58504c0a4c868060e64bba4227233002d9cd95ae Mon Sep 17 00:00:00 2001
From: Samuel Colvin <s@muelcolvin.com>
Date: Sat, 10 Feb 2024 20:28:30 +0000
Subject: [PATCH] fix bump_npm.py

---
 uprev_npm_packages.py => bump_npm.py | 31 +++++++++++++++++-----------
 1 file changed, 19 insertions(+), 12 deletions(-)
 rename uprev_npm_packages.py => bump_npm.py (66%)

diff --git a/uprev_npm_packages.py b/bump_npm.py
similarity index 66%
rename from uprev_npm_packages.py
rename to bump_npm.py
index e317b427..22e424b1 100644
--- a/uprev_npm_packages.py
+++ b/bump_npm.py
@@ -1,16 +1,19 @@
+from __future__ import annotations
+
 import json
 import re
 from pathlib import Path
 
 
-def replace_package_json(package_json: Path, new_version: str, deps: bool = False) -> None:
+def replace_package_json(package_json: Path, new_version: str, deps: bool = False) -> tuple[Path, str]:
     content = package_json.read_text()
     content, r_count = re.subn(r'"version": *".*?"', f'"version": "{new_version}"', content, count=1)
     assert r_count == 1 , f'Failed to update version in {package_json}, expect replacement count 1, got {r_count}'
     if deps:
         content, r_count = re.subn(r'"(@pydantic/.+?)": *".*?"', fr'"\1": "{new_version}"', content)
         assert r_count == 1, f'Failed to update version in {package_json}, expect replacement count 1, got {r_count}'
-    package_json.write_text(content)
+
+    return package_json, content
 
 
 def main():
@@ -21,28 +24,32 @@ def main():
 
     rest, patch_version = old_version.rsplit('.', 1)
     new_version = f'{rest}.{int(patch_version) + 1}'
-    replace_package_json(fastui_package_json, new_version)
     bootstrap_package_json = this_dir / 'src/npm-fastui-bootstrap/package.json'
-    replace_package_json(bootstrap_package_json, new_version, deps=True)
     prebuilt_package_json = this_dir / 'src/npm-fastui-prebuilt/package.json'
-    replace_package_json(prebuilt_package_json, new_version)
+    to_update: list[tuple[Path, str]] = [
+        replace_package_json(fastui_package_json, new_version),
+        replace_package_json(bootstrap_package_json, new_version, deps=True),
+        replace_package_json(prebuilt_package_json, new_version),
+    ]
 
     python_init = this_dir / 'src/python-fastui/fastui/__init__.py'
     python_content = python_init.read_text()
     python_content, r_count = re.subn(r"(_PREBUILT_VERSION = )'.+'", fr"\1'{new_version}'", python_content)
     assert r_count == 1, f'Failed to update version in {python_init}, expect replacement count 1, got {r_count}'
-    python_init.write_text(python_content)
+    to_update.append((python_init, python_content))
 
-    files = fastui_package_json, bootstrap_package_json, prebuilt_package_json, python_init
-    files = '\n'.join(str(f.relative_to(this_dir)) for f in files)
-    print(f"""\
-Updated version from `{old_version}` to `{new_version}` in:
+    # logic is finished, no update all files
+    print(f'Updating files:')
+    for package_json, content in to_update:
+        print(f'  {package_json.relative_to(this_dir)}')
+        package_json.write_text(content)
 
-{files}
+    print(f"""
+Bumped from `{old_version}` to `{new_version}` in {len(to_update)} files.
 
 To publish the new version, run:
 
-    npm --workspaces publish
+> npm --workspaces publish
 """)