From 3e67b9a0d99c80f09cfd705bc612d9b930f4c8c7 Mon Sep 17 00:00:00 2001
From: Joonas Bergius <joonas@bergi.us>
Date: Tue, 17 Sep 2024 19:06:21 -0500
Subject: [PATCH] refactor: replace dirs crate with etcetera

Signed-off-by: Joonas Bergius <joonas@bergi.us>
---
 testcontainers/Cargo.toml             |  2 +-
 testcontainers/src/core/env/config.rs | 14 +++++++++++---
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/testcontainers/Cargo.toml b/testcontainers/Cargo.toml
index 0895290e..3233f5bb 100644
--- a/testcontainers/Cargo.toml
+++ b/testcontainers/Cargo.toml
@@ -21,9 +21,9 @@ bollard = { version = "0.17.0", features = ["ssl"] }
 bollard-stubs = "=1.45.0-rc.26.0.1"
 bytes = "1.6.0"
 conquer-once = { version = "0.4", optional = true }
-dirs = "5.0.1"
 docker_credential = "1.3.1"
 either = "1.12.0"
+etcetera = "0.8.0"
 futures = "0.3"
 log = "0.4"
 memchr = "2.7.2"
diff --git a/testcontainers/src/core/env/config.rs b/testcontainers/src/core/env/config.rs
index 488b348e..b4313d9b 100644
--- a/testcontainers/src/core/env/config.rs
+++ b/testcontainers/src/core/env/config.rs
@@ -4,7 +4,7 @@ use std::{
     str::FromStr,
 };
 
-use dirs::{home_dir, runtime_dir};
+use etcetera::BaseStrategy;
 
 use crate::core::env::GetEnvValue;
 
@@ -62,7 +62,7 @@ struct TestcontainersProperties {
 #[cfg(feature = "properties-config")]
 impl TestcontainersProperties {
     async fn load() -> Option<Result<Self, ConfigurationError>> {
-        let home_dir = dirs::home_dir()?;
+        let home_dir = home_dir()?;
         let properties_path = home_dir.join(TESTCONTAINERS_PROPERTIES);
 
         let content = tokio::fs::read(properties_path).await.ok()?;
@@ -195,6 +195,14 @@ fn validate_path(path: String) -> Option<String> {
     }
 }
 
+fn home_dir() -> Option<PathBuf> {
+    etcetera::home_dir().ok()
+}
+
+fn runtime_dir() -> Option<PathBuf> {
+    etcetera::choose_base_strategy().ok()?.runtime_dir()
+}
+
 /// Read the Docker authentication configuration in the following order:
 ///
 /// 1. `DOCKER_AUTH_CONFIG` environment variable, unmarshalling the string value from its JSON representation and using it as the Docker config.
@@ -210,7 +218,7 @@ where
             let mut path_to_config = match E::get_env_value("DOCKER_CONFIG").map(PathBuf::from) {
                 Some(path_to_config) => path_to_config,
                 None => {
-                    let home_dir = dirs::home_dir()?;
+                    let home_dir = home_dir()?;
                     home_dir.join(DEFAULT_DOCKER_CONFIG_PATH)
                 }
             };