chore: partial release script to handle sdk-platform-java

Author: suztomo

.github/release/partial_release.py

@@ -38,12 +38,20 @@ def main(ctx):
 @main.command()
 @click.option(
     "--artifact-ids",
-    required=True,
+    required=False,
     type=str,
     help="""
     Artifact IDs whose version needs to update, separated by comma.
     """,
 )
+@click.option(
+    "--sdk-platform-java",
+    is_flag=True,
+    default=False,
+    help="""
+    If set, bump versions for all modules in sdk-platform-java.
+    """,
+)
 @click.option(
     "--versions",
     required=False,
@@ -53,18 +61,29 @@ def main(ctx):
     The path to the versions.txt.
     """,
 )
-def bump_snapshot_version(artifact_ids: str, versions: str) -> None:
+def bump_snapshot_version(artifact_ids: str, sdk_platform_java: bool, versions: str) -> None:
+    if sdk_platform_java:
+        artifact_ids = _get_sdk_platform_java_artifacts(artifact_ids)
     bump_version(artifact_ids, "snapshot", versions)
 
+
 @main.command()
 @click.option(
     "--artifact-ids",
-    required=True,
+    required=False,
     type=str,
     help="""
     Artifact IDs whose version needs to update, separated by comma.
     """,
 )
+@click.option(
+    "--sdk-platform-java",
+    is_flag=True,
+    default=False,
+    help="""
+    If set, bump versions for all modules in sdk-platform-java.
+    """,
+)
 @click.option(
     "--version-type",
     required=False,
@@ -83,9 +102,45 @@ def bump_snapshot_version(artifact_ids: str, versions: str) -> None:
     The path to the versions.txt.
     """,
 )
-def bump_released_version(artifact_ids: str, version_type: str, versions: str) -> None:
+def bump_released_version(
+    artifact_ids: str, sdk_platform_java: bool, version_type: str, versions: str
+) -> None:
+    if sdk_platform_java:
+        artifact_ids = _get_sdk_platform_java_artifacts(artifact_ids)
     bump_version(artifact_ids, version_type, versions)
 
+
+def _get_sdk_platform_java_artifacts(artifact_ids: str) -> str:
+    sdk_artifacts = [
+        "gapic-generator-java",
+        "api-common",
+        "gax",
+        "gax-grpc",
+        "gax-httpjson",
+        "proto-google-common-protos",
+        "grpc-google-common-protos",
+        "proto-google-iam-v1",
+        "grpc-google-iam-v1",
+        "proto-google-iam-v2beta",
+        "grpc-google-iam-v2beta",
+        "proto-google-iam-v2",
+        "grpc-google-iam-v2",
+        "proto-google-iam-v3",
+        "grpc-google-iam-v3",
+        "proto-google-iam-v3beta",
+        "grpc-google-iam-v3beta",
+        "google-cloud-core",
+        "google-cloud-shared-dependencies",
+        "google-iam-policy",
+        "gapic-showcase",
+        "proto-gapic-showcase-v1beta1",
+        "grpc-gapic-showcase-v1beta1",
+    ]
+    if artifact_ids:
+        return ",".join(sdk_artifacts + artifact_ids.split(","))
+    return ",".join(sdk_artifacts)
+
+
 def bump_version(artifact_ids: str, version_type: str, versions: str) -> None:
     target_artifact_ids = set(artifact_ids.split(","))
     version_enum = _parse_type_or_raise(version_type)
@@ -130,7 +185,8 @@ def bump_version(artifact_ids: str, version_type: str, versions: str) -> None:
                 f"{artifact_id}:{major}.{minor}.{patch}:{major}.{minor}.{patch}"
             )
     with open(versions, "w") as versions_file:
-        versions_file.writelines("\n".join(newlines))
+        for line in newlines:
+            versions_file.write(f"{line}\n")
 
 
 def _parse_type_or_raise(version_type: str) -> VersionType:

generation/apply_versions.sh

@@ -21,25 +21,38 @@ if [[ -z "$versions_file" || -z "$column_name" ]]; then
   echo "Usage: $0 path/to/versions.txt (released|current)"
   exit 1
 fi
-if [[ "$column_name" == "released" ]]; then 
+if [[ "$column_name" == "released" ]]; then
   column_index=2
 elif [[ "$column_name" == "current" ]]; then
   column_index=3
-elif "$column_name" != "current" ]]; then
+else
   echo "Error: column_name must be either 'released' or 'current'"
   exit 1
 fi
 
+SED_SCRIPT=$(mktemp)
+trap "rm -f $SED_SCRIPT" EXIT
 
-SED_OPTIONS=""
-
-# The second column is 
 for KV in $(cut -f1,"${column_index}" -d: $versions_file |grep -v "#"); do
   K=${KV%:*}; V=${KV#*:}
-  echo Key:$K, Value:$V;
-  SED_OPTIONS="$SED_OPTIONS -e /x-version-update:$K:current/{s|<version>.*<\/version>|<version>$V<\/version>|;}"
+  echo "Key:$K, Value:$V";
+  # Pattern 1: XML <version> tags and property tags (e.g., <google-cloud-shared-dependencies.version>)
+  echo "/x-version-update:$K:\(current\|released\)/{s|>\([^<]*\)<|>$V<|;}" >> "$SED_SCRIPT"
+  # Pattern 2: YAML _VERSION: '...' or "..."
+  echo "/x-version-update:$K:\(current\|released\)/{s|: [\'\"][^\'\"]*[\'\"]|: \'$V\'|;}" >> "$SED_SCRIPT"
+  # Pattern 3: Java static final String VERSION = "..."
+  echo "/x-version-update-start:$K:\(current\|released\)/{n;s|\".*\"|\"$V\"|;}" >> "$SED_SCRIPT"
 done
 
+
 echo "Running sed command. It may take few minutes."
-find . -maxdepth 3 -name pom.xml |sort --dictionary-order |xargs sed -i.bak $SED_OPTIONS
-find . -maxdepth 3 -name pom.xml.bak |xargs rm
+# Including pom.xml, *.yaml, and Version.java files.
+FILES=$(find . \( -name pom.xml -o -name "*.yaml" -o -name "Version.java" \) | sort --dictionary-order)
+TOTAL_FILES=$(echo "$FILES" | grep -c .)
+CURRENT_FILE=0
+while read -r FILE; do
+  CURRENT_FILE=$((CURRENT_FILE + 1))
+  echo -ne "Progress: $CURRENT_FILE/$TOTAL_FILES ($((CURRENT_FILE * 100 / TOTAL_FILES))%)\r"
+  sed -i -f "$SED_SCRIPT" "$FILE"
+done <<< "$FILES"
+echo