From 854745c670dbf0e7ee9ae30a4512b5f7a75868c6 Mon Sep 17 00:00:00 2001
From: Tamo <tamo@meilisearch.com>
Date: Wed, 16 Aug 2023 18:41:05 +0200
Subject: [PATCH] wip: starts working on importing the snapshots

---
 index-scheduler/src/index_mapper/mod.rs |  2 +-
 index-scheduler/src/lib.rs              | 51 ++++++++++++++++++-------
 2 files changed, 38 insertions(+), 15 deletions(-)

diff --git a/index-scheduler/src/index_mapper/mod.rs b/index-scheduler/src/index_mapper/mod.rs
index 18aed42b0..7850966cf 100644
--- a/index-scheduler/src/index_mapper/mod.rs
+++ b/index-scheduler/src/index_mapper/mod.rs
@@ -61,7 +61,7 @@ pub struct IndexMapper {
     pub(crate) index_stats: Database<UuidCodec, SerdeJson<IndexStats>>,
 
     /// Path to the folder where the LMDB environments of each index are.
-    base_path: PathBuf,
+    pub(crate) base_path: PathBuf,
     /// The map size an index is opened with on the first time.
     index_base_map_size: usize,
     /// The quantity by which the map size of an index is incremented upon reopening, in bytes.
diff --git a/index-scheduler/src/lib.rs b/index-scheduler/src/lib.rs
index 7e285f2a0..56cf11d92 100644
--- a/index-scheduler/src/lib.rs
+++ b/index-scheduler/src/lib.rs
@@ -683,24 +683,46 @@ impl IndexScheduler {
 
                                     let snapshot_id = path.strip_prefix("/snapshots/snapshot-").unwrap();
                                     let snapshot_dir =
-                                        PathBuf::from(format!("~/zk-snapshots/{}", snapshot_id));
-
-                                    // TODO: everything
+                                        PathBuf::from(format!("{}/zk-snapshots/{}", env!("HOME"), snapshot_id));
 
                                     // 1. TODO: Ensure the snapshot version file is the same as our version.
 
-                                    // 2. Download and import the index-scheduler database
-                                    log::info!("Importing the index scheduler.");
-                                    let tasks =
+                                    // 2. Download all the databases
+                                    let tasks_file = tempfile::NamedTempFile::new_in(run.env.path()).unwrap();
+
+                                    log::info!("Downloading the index scheduler database.");
+                                    let tasks_snapshot =
                                             snapshot_dir.join("tasks.mdb");
+                                    std::fs::copy(tasks_snapshot, tasks_file).unwrap();
+
+
+                                    log::info!("Downloading the indexes databases");
+                                    let indexes_files = tempfile::TempDir::new_in(&run.index_mapper.base_path).unwrap();
+                                    let mut indexes = Vec::new();
+
+                                    let dst = snapshot_dir.join("indexes");
+                                    let mut indexes_snapshot = tokio::fs::read_dir(&dst).await.unwrap();
+                                    while let Some(file) = indexes_snapshot.next_entry().await.unwrap() {
+                                        let uuid = file.file_name().as_os_str().to_str().unwrap().to_string();
+                                        log::info!("\tDownloading the index {}", uuid.to_string());
+                                        std::fs::copy(dst.join(&uuid), indexes_files.path().join(&uuid)).unwrap();
+                                        indexes.push(uuid);
+                                    }
+
+                                    // 3. Lock the index-mapper and close all the env
+                                    // TODO: continue here
+
+
+
+                                    // run.env.close();
+
+                                    // 4. Move all the databases
+
+                                    // 5. Unlock the index-mapper
+
+                                    // 2. Download and import the index-scheduler database
 
                                     // 3. Snapshot every indexes
-                                    log::info!("Importing the indexes");
-                                    let dst = snapshot_dir.join("indexes");
-                                    let mut indexes = tokio::fs::read_dir(dst).await.unwrap();
-                                    while let Some(uuid) = indexes.next_entry().await.unwrap() {
-                                        // TODO: Import the index
-                                    }
                                 }
                                 _ => (),
                             },
@@ -733,9 +755,10 @@ impl IndexScheduler {
                                         .await
                                         .unwrap();
 
-                                    tokio::fs::create_dir_all("~/zk-snapshots").await.unwrap();
+                                    let zk_snapshots = format!("{}/zk-snapshots", env!("HOME"));
+                                    tokio::fs::create_dir_all(&zk_snapshots).await.unwrap();
                                     let snapshot_dir =
-                                        PathBuf::from(format!("~/zk-snapshots/{snapshot_id}"));
+                                        PathBuf::from(format!("{zk_snapshots}/{snapshot_id}"));
                                     tokio::fs::create_dir(&snapshot_dir).await.unwrap();
 
                                     // 1. Snapshot the version file.