Caching

Author: qarmin

postgresql_embedded/build/bundle.rs

@@ -3,14 +3,14 @@
 use anyhow::Result;
 use postgresql_archive::configuration::{custom, theseus};
 use postgresql_archive::repository::github::repository::GitHub;
-use postgresql_archive::{VersionReq, matcher, ExactVersion};
+use postgresql_archive::{ExactVersion, VersionReq, matcher};
 use postgresql_archive::{get_archive, repository};
+use std::env::home_dir;
 use std::fs::File;
 use std::io::Write;
 use std::path::PathBuf;
 use std::str::FromStr;
 use std::{env, fs};
-use std::env::home_dir;
 use url::Url;
 
 /// Stage the PostgreSQL archive when the `bundled` feature is enabled so that
@@ -54,18 +54,30 @@ pub(crate) async fn stage_postgresql_archive() -> Result<()> {
         return Ok(());
     }
 
-    panic!();
-
-
     let (asset_version, archive);
 
-    // Only works when exact version is specified
-    if let Some(exact_version) = version_req.exact_version() {
-        let cached_file_name = home_dir().unwrap_or_else(|| env::current_dir().unwrap_or_default()).join(".theseus")
+    // Only works when exact version is specified and with the `bundled` feature
+    // In runtime, the archive seems to be cached
+    if cfg!(feature = "bundled") && let Some(exact_version) = version_req.exact_version() {
+        println!("Using existing version: {exact_version:?}");
+        let ver = exact_version.to_string();
+        let target_os = if cfg!(target_os = "windows") {
+            "windows"
+        } else if cfg!(target_os = "linux") {
+            "linux"
+        } else {
+            panic!("Unsupported target OS: only windows and linux are supported");
+        };
+        let cached_file_name = home_dir()
+            .unwrap_or_else(|| env::current_dir().unwrap_or_default())
+            .join(".theseus")
             .join("postgresql")
-            .join(format!("postgresql-{}-{}-{}.tar.gz", ver, cfg!(target_os), cfg!(target_arch)));
+            .join(format!("postgresql-{}-{}.tar.gz", ver, target_os));
 
-        println!("Cached file name: {cached_file_name:?} - exists: {}", cached_file_name.exists());
+        println!(
+            "Cached file name: {cached_file_name:?} - exists: {}",
+            cached_file_name.exists()
+        );
         if !cached_file_name.is_file() {
             (asset_version, archive) = get_archive(&releases_url, &version_req).await?;
             fs::create_dir_all(cached_file_name.parent().unwrap())?;