reconciler(install): find latest version of extension if not supplied

tembo-io · Jun 27, 2024 · f6c81d5 · f6c81d5
1 parent b6a42cc
commit f6c81d5
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 13 deletions.
diff --git a/conductor/Cargo.lock b/conductor/Cargo.lock
diff --git a/tembo-operator/Cargo.lock b/tembo-operator/Cargo.lock
diff --git a/tembo-operator/Cargo.toml b/tembo-operator/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 name = "controller"
 description = "Tembo Operator for Postgres"
-version = "0.49.1"
+version = "0.49.2"
 edition = "2021"
 default-run = "controller"
 license = "Apache-2.0"

diff --git a/tembo-operator/src/extensions/install.rs b/tembo-operator/src/extensions/install.rs
@@ -5,6 +5,7 @@ use crate::{
         kubernetes_queries::{add_trunk_install_to_status, remove_trunk_installs_from_status},
         types::{TrunkInstall, TrunkInstallStatus},
     },
+    trunk::get_latest_trunk_project_version,
     Context,
 };
 use k8s_openapi::{api::core::v1::Pod, apimachinery::pkg::apis::meta::v1::ObjectMeta};
@@ -300,18 +301,29 @@ async fn execute_extension_install_command(
     // Handle the case where version is None
     let version = match &ext.version {
         None => {
-            error!(
-                "Installing extension {} into {}: missing version",
+            warn!(
+                "Version for extension {} not specified in {}, will fetch latest version",
                 ext.name, coredb_name
             );
-            return Ok(TrunkInstallStatus {
-                name: ext.name.clone(),
-                version: None,
-                error: true,
-                loading: false,
-                error_message: Some("Missing version".to_string()),
-                installed_to_pods: Some(vec![pod_name.to_string()]),
-            });
+
+            match get_latest_trunk_project_version(&ext.name).await {
+                Ok(latest_version) => latest_version,
+                Err(_) => {
+                    error!(
+                        "Failed to get latest version for extension {} in {}",
+                        ext.name, coredb_name
+                    );
+
+                    return Ok(TrunkInstallStatus {
+                        name: ext.name.clone(),
+                        version: None,
+                        error: true,
+                        loading: false,
+                        error_message: Some("Missing version".to_string()),
+                        installed_to_pods: Some(vec![pod_name.to_string()]),
+                    });
+                }
+            }
         }
         Some(version) => version.clone(),
     };