From 7cce595d0b4325077ef8585e04a7696802d94466 Mon Sep 17 00:00:00 2001
From: Adam Dyess <adam.dyess@canonical.com>
Date: Thu, 7 Nov 2024 08:31:42 -0600
Subject: [PATCH] Update the snap version of via upstream .go-version (#762)

---
 .../hack/update-component-versions.py         | 25 +++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/build-scripts/hack/update-component-versions.py b/build-scripts/hack/update-component-versions.py
index 71f7b71d7..88eee17d8 100755
--- a/build-scripts/hack/update-component-versions.py
+++ b/build-scripts/hack/update-component-versions.py
@@ -15,13 +15,17 @@
 import sys
 import yaml
 from pathlib import Path
+import re
 import util
+import urllib.request
+
 
 logging.basicConfig(level=logging.INFO)
 
 LOG = logging.getLogger(__name__)
 
 DIR = Path(__file__).absolute().parent
+SNAPCRAFT = DIR.parent.parent / "snap/snapcraft.yaml"
 COMPONENTS = DIR.parent / "components"
 CHARTS = DIR.parent.parent / "k8s" / "manifests" / "charts"
 
@@ -125,6 +129,8 @@ def update_component_versions(dry_run: bool):
         if not dry_run:
             Path(path).write_text(version.strip() + "\n")
 
+    update_go_version(dry_run)
+
     for component, pull_helm_chart in [
         ("bitnami/contour", pull_contour_chart),
         ("metallb", pull_metallb_chart),
@@ -134,6 +140,25 @@ def update_component_versions(dry_run: bool):
             pull_helm_chart()
 
 
+def update_go_version(dry_run: bool):
+    k8s_version = (COMPONENTS / "kubernetes/version").read_text().strip()
+    url = f"https://raw.githubusercontent.com/kubernetes/kubernetes/refs/tags/{k8s_version}/.go-version"
+    with urllib.request.urlopen(url) as response:
+        go_version = response.read().decode("utf-8").strip()
+
+    LOG.info("Upstream go version is %s", go_version)
+    go_snap = f'go/{".".join(go_version.split(".")[:2])}/stable'
+    snapcraft_yaml = SNAPCRAFT.read_text()
+    if f"- {go_snap}" in snapcraft_yaml:
+        LOG.info("snapcraft.yaml already contains go version %s", go_snap)
+        return
+
+    LOG.info("Update go snap version to %s in %s", go_snap, SNAPCRAFT)
+    if not dry_run:
+        updated = re.sub(r"- go/\d+\.\d+/stable", f"- {go_snap}", snapcraft_yaml)
+        SNAPCRAFT.write_text(updated)
+
+
 def main():
     parser = argparse.ArgumentParser(
         "update-component-versions.py", usage=USAGE, description=DESCRIPTION