From 5060ec814087eaa5a42e834561358548feb1a909 Mon Sep 17 00:00:00 2001
From: ryutamago <leo.unoki@trili.tech>
Date: Thu, 12 Sep 2024 13:51:49 +0100
Subject: [PATCH] feat(jstzd): implement pull image

---
 Cargo.lock                       |  2 ++
 crates/jstzd/Cargo.toml          |  5 +++-
 crates/jstzd/src/docker/image.rs | 51 ++++++++++++++++++++++++++++++++
 3 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/Cargo.lock b/Cargo.lock
index b90dd1d09..b1e262e59 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2666,7 +2666,9 @@ dependencies = [
  "async-dropper",
  "async-trait",
  "bollard",
+ "env_logger",
  "futures-util",
+ "log",
  "tokio",
 ]
 
diff --git a/crates/jstzd/Cargo.toml b/crates/jstzd/Cargo.toml
index b081273b1..667c3d31c 100644
--- a/crates/jstzd/Cargo.toml
+++ b/crates/jstzd/Cargo.toml
@@ -10,9 +10,12 @@ anyhow.workspace = true
 async-dropper.workspace = true
 async-trait.workspace = true
 bollard.workspace = true
+env_logger.workspace = true
 futures-util.workspace = true
 tokio.workspace = true
+log.workspace = true
 
 [[bin]]
 name = "jstzd"
-path = "src/main.rs"
\ No newline at end of file
+path = "src/main.rs"
+
diff --git a/crates/jstzd/src/docker/image.rs b/crates/jstzd/src/docker/image.rs
index b55179829..1b4489f57 100644
--- a/crates/jstzd/src/docker/image.rs
+++ b/crates/jstzd/src/docker/image.rs
@@ -1,3 +1,12 @@
+use anyhow::Result;
+use bollard::{
+    image::{CreateImageOptions, ListImagesOptions},
+    Docker,
+};
+use futures_util::StreamExt;
+use log::info;
+use std::{collections::HashMap, sync::Arc};
+#[async_trait::async_trait]
 pub trait Image: Sized {
     const LATEST_TAG: &'static str = "latest";
     fn image_name(&self) -> &str;
@@ -15,6 +24,48 @@ pub trait Image: Sized {
     fn exposed_ports(&self) -> &[u16] {
         &[]
     }
+    async fn pull_image(&self, client: Arc<Docker>) -> Result<()> {
+        if Self::image_exists(self, client.clone()).await {
+            info!("Image: {:?} already exists ", self.image_name());
+            return Ok(());
+        }
+        self.create_image(client.clone()).await
+    }
+    async fn image_exists(&self, client: Arc<Docker>) -> bool {
+        let filters = [("reference".to_string(), vec![self.image_uri()])]
+            .into_iter()
+            .collect::<HashMap<_, _>>();
+        let images = &client
+            .list_images(Some(ListImagesOptions::<String> {
+                all: true,
+                filters,
+                ..Default::default()
+            }))
+            .await;
+        match images {
+            Ok(images) => !images.is_empty(),
+            Err(_) => false,
+        }
+    }
+    async fn create_image(&self, client: Arc<Docker>) -> anyhow::Result<()> {
+        let options = Some(CreateImageOptions {
+            from_image: self.image_name(),
+            tag: self.image_tag(),
+            ..Default::default()
+        });
+        let mut stream = client.create_image(options, None, None);
+        while let Some(create_info) = stream.next().await {
+            match create_info {
+                Ok(info) => {
+                    if let Some(status) = info.status {
+                        println!("{:?}", status)
+                    }
+                }
+                Err(e) => return Err(e.into()),
+            }
+        }
+        Ok(())
+    }
 }
 
 pub struct GenericImage {