Consume build_scenario via build_info for registry repo and other info

Author: viveksinghggits

scripts/dev/helm_registry_login.sh

@@ -1,31 +1,39 @@
-import subprocess
 import os
+import subprocess
+import sys
+
 import yaml
 
 from lib.base_logger import logger
+from scripts.release.build.build_info import *
 
 CHART_DIR = "helm_chart"
 
-OCI_REGISTRY = "oci://268558157000.dkr.ecr.us-east-1.amazonaws.com/dev/helm-charts"
 
-def run_command(command: list[str], description: str):
+def run_command(command: list[str]):
     try:
-        subprocess.run(command, check=True, text=True, capture_output=False)
-        logger.info(f"Command {' '.join(command)} executed successfully.")
+        # Using capture_output=True to grab stdout/stderr for better error logging.
+        process = subprocess.run(command, check=True, text=True, capture_output=True)
+        logger.info(f"Successfully executed: {' '.join(command)}")
+        if process.stdout:
+            logger.info(process.stdout)
     except subprocess.CalledProcessError as e:
-        logger.error(f"Error executing command: {' '.join(command)}")
-        raise RuntimeError(f"{description} failed.") from e
+        raise RuntimeError(f"Command {' '.join(command)} failed. Stderr: {e.stderr.strip()}") from e
     except FileNotFoundError:
-        raise FileNotFoundError("Error: 'helm' command not found. Ensure Helm CLI is installed and in your PATH.")
+        raise FileNotFoundError(f"Error: {command[0]} command not found. Ensure {command[0]} is installed and in your PATH.")
 
+
+# update_chart_and_get_metadata updates the helm chart's Chart.yaml and sets the version
+# to either evg patch id or commit which is set in OPERATOR_VERSION.
 def update_chart_and_get_metadata(chart_dir: str) -> tuple[str, str]:
     chart_path = os.path.join(chart_dir, "Chart.yaml")
-    version_id = os.environ.get('version_id')
+    version_id = os.environ.get("OPERATOR_VERSION")
     if not version_id:
-        raise ValueError("Error: Environment variable 'version_id' must be set to determine the chart version to publish.")
-        
-    new_version = f"0.0.0+{version_id}"
+        raise ValueError(
+            "Error: Environment variable 'OPERATOR_VERSION' must be set to determine the chart version to publish."
+        )
 
+    new_version = f"0.0.0+{version_id}"
     logger.info(f"New helm chart version will be: {new_version}")
 
     if not os.path.exists(chart_path):
@@ -35,44 +43,76 @@ def update_chart_and_get_metadata(chart_dir: str) -> tuple[str, str]:
         )
 
     try:
-        with open(chart_path, 'r') as f:
+        with open(chart_path, "r") as f:
             data = yaml.safe_load(f)
 
-        chart_name = data.get('name')
+        chart_name = data.get("name")
         if not chart_name:
-             raise ValueError("Chart.yaml is missing required 'name' field.")
+            raise ValueError("Chart.yaml is missing required 'name' field.")
 
-        data['version'] = new_version
-        
-        with open(chart_path, 'w') as f:
-            yaml.safe_dump(data, f, sort_keys=False) 
+        data["version"] = new_version
 
-        logger.info(f"Successfully updated version for chart '{chart_name}' to '{new_version}' before publishing it.")
-        return chart_name, new_version
+        with open(chart_path, "w") as f:
+            yaml.safe_dump(data, f, sort_keys=False)
 
+        logger.info(f"Successfully updated version for chart '{chart_name}' to '{new_version}'.")
+        return chart_name, new_version
     except Exception as e:
         raise RuntimeError(f"Failed to read or update Chart.yaml: {e}")
 
-def publish_helm_chart():
+
+def get_oci_registry(chart_info: HelmChartInfo) -> str:
+    registry = chart_info.registry
+    repo = chart_info.repository
+
+    if not registry:
+        raise ValueError("Error: registry doesn't seem to be set in HelmChartInfo.")
+
+    if not repo:
+        raise ValueError("Error: reposiotry doesn't seem to be set in HelmChartInfo.")
+
+    
+    oci_registry = f"oci://{registry}/{repo}"
+    logger.info(f"Determined OCI Registry: {oci_registry}")
+    return oci_registry
+
+
+def publish_helm_chart(chart_info: HelmChartInfo):
     try:
+        oci_registry = get_oci_registry(chart_info)
         chart_name, chart_version = update_chart_and_get_metadata(CHART_DIR)
-        
         tgz_filename = f"{chart_name}-{chart_version}.tgz"
-        logger.info(f"Packaging chart: {chart_name} with Version: {chart_version}")
 
-        package_command = ["helm", "package", CHART_DIR]
-        run_command(package_command, f"Packaging chart '{CHART_DIR}'")
+        try:
+            logger.info(f"Packaging chart: {chart_name} with Version: {chart_version}")
+            package_command = ["helm", "package", CHART_DIR]
+            run_command(package_command)
+
+            logger.info(f"Pushing chart to registry: {oci_registry}")
+            push_command = ["helm", "push", tgz_filename, oci_registry]
+            run_command(push_command)
 
-        push_command = ["helm", "push", tgz_filename, OCI_REGISTRY]
-        run_command(push_command, f"Pushing '{tgz_filename}' to '{OCI_REGISTRY}'")
+            logger.info(f"Helm Chart {chart_name}:{chart_version} was published successfully!")
+        finally:
+            # Cleanup the local .tgz file regardless of push success/failure
+            if os.path.exists(tgz_filename):
+                logger.info(f"Cleaning up local file: {tgz_filename}")
+                os.remove(tgz_filename)
 
-        if os.path.exists(tgz_filename):
-            logger.info(f"\nCleaning up local file: {tgz_filename}")
-            os.remove(tgz_filename)
-        
-        logger(f"Helm Chart {chart_name}:{chart_version} was published successfully!")
     except (FileNotFoundError, RuntimeError, ValueError) as e:
-        logger.error(f"\Failed publishing the helm chart: {e}")
+        raise Exception(f"Failed publishing the helm chart {e}")
+
+
+def main():
+    build_scenario = os.environ.get("BUILD_SCENARIO")
+    build_info = load_build_info(build_scenario)
+
+    return publish_helm_chart(build_info.helm_charts["mongodb-kubernetes"])
+
 
 if __name__ == "__main__":
-    publish_helm_chart()
+    try:
+        main()
+    except Exception as e:
+        logger.error(f"Failure in the helm publishing process {e}")
+        sys.exit(1)

scripts/publish_helm_chart.py

@@ -69,7 +69,6 @@ def main():
     region = build_info.helm_charts["mongodb-kubernetes"].region
     return helm_registry_login(registry, region)
 
-
 if __name__ == "__main__":
     try:
         main()