fix: always pull Docker images with :latest and :main tags (#439)

* fix: always pull Docker images with :latest and :main tags

* cargo fmt

Author: JohannesHa

crates/worker/src/docker/docker_manager.rs

@@ -65,36 +65,49 @@ impl DockerManager {
     /// Pull a Docker image if it doesn't exist locally
     pub async fn pull_image(&self, image: &str) -> Result<(), DockerError> {
         debug!("Checking if image needs to be pulled: {}", image);
-        if self.docker.inspect_image(image).await.is_err() {
+
+        // Check if the image uses :latest or :main tag
+        let should_always_pull = image.ends_with(":latest") || image.ends_with(":main");
+
+        // Only skip pulling if image exists locally AND it's not a :latest or :main tag
+        if !should_always_pull && self.docker.inspect_image(image).await.is_ok() {
+            debug!("Image {} already exists locally", image);
+            return Ok(());
+        }
+
+        if should_always_pull {
+            info!(
+                "Image {} uses :latest or :main tag, pulling to ensure we have the newest version",
+                image
+            );
+        } else {
             info!("Image {} not found locally, pulling...", image);
+        }
 
-            // Split image name and tag
-            let (image_name, tag) = match image.split_once(':') {
-                Some((name, tag)) => (name, tag),
-                None => (image, "latest"), // Default to latest if no tag specified
-            };
+        // Split image name and tag
+        let (image_name, tag) = match image.split_once(':') {
+            Some((name, tag)) => (name, tag),
+            None => (image, "latest"), // Default to latest if no tag specified
+        };
 
-            let options = CreateImageOptions {
-                from_image: image_name,
-                tag,
-                ..Default::default()
-            };
+        let options = CreateImageOptions {
+            from_image: image_name,
+            tag,
+            ..Default::default()
+        };
 
-            let mut image_stream = self.docker.create_image(Some(options), None, None);
+        let mut image_stream = self.docker.create_image(Some(options), None, None);
 
-            while let Some(info) = image_stream.next().await {
-                match info {
-                    Ok(create_info) => {
-                        debug!("Pull progress: {:?}", create_info);
-                    }
-                    Err(e) => return Err(e),
+        while let Some(info) = image_stream.next().await {
+            match info {
+                Ok(create_info) => {
+                    debug!("Pull progress: {:?}", create_info);
                 }
+                Err(e) => return Err(e),
             }
-
-            info!("Successfully pulled image {}", image);
-        } else {
-            debug!("Image {} already exists locally", image);
         }
+
+        info!("Successfully pulled image {}", image);
         Ok(())
     }