canary.py を追加

canary.py

@@ -0,0 +1,94 @@
+import argparse
+import re
+import subprocess
+
+VERSION_FILE = "VERSION"
+
+
+def update_sdk_version(version_content):
+    updated_content = []
+    sdk_version_updated = False
+    new_version = None
+
+    for line in version_content:
+        line = line.strip()  # 前後の余分なスペースや改行を削除
+        if line.startswith("MOMO_VERSION="):
+            version_match = re.match(
+                r"MOMO_VERSION=(\d{4}\.\d+\.\d+)(-canary\.(\d+))?", line
+            )
+            if version_match:
+                major_minor_patch = version_match.group(1)
+                canary_suffix = version_match.group(2)
+                if canary_suffix is None:
+                    new_version = f"{major_minor_patch}-canary.0"
+                else:
+                    canary_number = int(version_match.group(3))
+                    new_version = f"{major_minor_patch}-canary.{canary_number + 1}"
+
+                updated_content.append(f"MOMO_VERSION={new_version}")
+                sdk_version_updated = True
+            else:
+                updated_content.append(line)
+        else:
+            updated_content.append(line)
+
+    if not sdk_version_updated:
+        raise ValueError("MOMO_VERSION not found in VERSION file.")
+
+    return updated_content, new_version
+
+
+def write_version_file(filename, updated_content, dry_run):
+    if dry_run:
+        print(f"Dry run: The following changes would be written to {filename}:")
+        for line in updated_content:
+            print(line.strip())
+    else:
+        with open(filename, "w") as file:
+            file.write("\n".join(updated_content) + "\n")
+        print(f"{filename} updated.")
+
+
+def git_operations(new_version, dry_run):
+    commit_message = f"[canary] Update VERSION to {new_version}"
+
+    if dry_run:
+        print(f"Dry run: Would execute git commit -am '{commit_message}'")
+        print(f"Dry run: Would execute git tag {new_version}")
+        print("Dry run: Would execute git push")
+        print(f"Dry run: Would execute git push origin {new_version}")
+    else:
+        print(f"Executing: git commit -am '{commit_message}'")
+        subprocess.run(["git", "commit", "-am", commit_message], check=True)
+
+        print(f"Executing: git tag {new_version}")
+        subprocess.run(["git", "tag", new_version], check=True)
+
+        print("Executing: git push")
+        subprocess.run(["git", "push"], check=True)
+
+        print(f"Executing: git push origin {new_version}")
+        subprocess.run(["git", "push", "origin", new_version], check=True)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Update VERSION file and push changes with git."
+    )
+    parser.add_argument(
+        "--dry-run", action="store_true", help="Perform a dry run without making any changes."
+    )
+    args = parser.parse_args()
+
+    # Read and update the VERSION file
+    with open(VERSION_FILE, "r") as file:
+        version_content = file.readlines()
+    updated_version_content, new_version = update_sdk_version(version_content)
+    write_version_file(VERSION_FILE, updated_version_content, args.dry_run)
+
+    # Perform git operations
+    git_operations(new_version, args.dry_run)
+
+
+if __name__ == "__main__":
+    main()