Find .butane in workspace

butane/tests/common/blog.rs

@@ -4,7 +4,7 @@ use butane::{db::Connection, ForeignKey, Many, ObjectState};
 use chrono::{naive::NaiveDateTime, offset::Utc};
 
 #[cfg(feature = "fake")]
-use fake::{Dummy};
+use fake::Dummy;
 
 #[model]
 #[derive(Debug, Eq, PartialEq)]

butane_cli/Cargo.toml

@@ -22,8 +22,10 @@ sqlite-bundled = ["butane/sqlite-bundled"]
 [dependencies]
 anyhow = "1.0"
 butane = { features = ["default", "pg", "sqlite"], workspace = true }
+cargo_metadata = "0.15"
 chrono = { workspace = true }
 clap = { version = "4.1" }
 quote = { workspace = true }
 serde = { workspace = true }
 serde_json = { workspace = true }
+url = "2.3"

butane_cli/src/lib.rs

@@ -13,8 +13,10 @@ use butane::migrations::{
 };
 use butane::query::BoolExpr;
 use butane::{db, db::Connection, db::ConnectionMethods, migrations};
+use cargo_metadata::MetadataCommand;
 use chrono::Utc;
 use serde::{Deserialize, Serialize};
+use url::Url;
 
 pub type Result<T> = std::result::Result<T, anyhow::Error>;
 
@@ -259,10 +261,43 @@ pub fn get_migrations() -> Result<FsMigrations> {
     Ok(migrations::from_root(root))
 }
 
+pub fn extract_package_directory(
+    package_id: cargo_metadata::PackageId,
+) -> Result<std::path::PathBuf> {
+    let mut file_url = package_id
+        .repr
+        .split_once(&"path+".to_string())
+        .unwrap()
+        .1
+        .to_string();
+    file_url.truncate(file_url.len() - 1);
+    let url = Url::parse(file_url.as_str()).expect("Failed to parse URL");
+    let file_path = url.to_file_path().unwrap();
+    Ok(file_path)
+}
+
 pub fn base_dir() -> Result<PathBuf> {
-    std::env::current_dir()
-        .map(|d| d.join(".butane"))
-        .map_err(|e| e.into())
+    let current_directory = std::env::current_dir()?;
+    let local_butane_dir = current_directory.join(".butane/");
+
+    if !local_butane_dir.exists() {
+        let metadata = MetadataCommand::new().no_deps().exec()?;
+        let workspace_members = metadata.workspace_members;
+
+        // Find the first workspace member with a .butane
+        for member in workspace_members {
+            let package_dir = extract_package_directory(member)?;
+            let member_butane_dir = package_dir.join(".butane/");
+
+            if member_butane_dir.exists() {
+                eprintln!("Using workspace member {:?}", member_butane_dir);
+                return Ok(member_butane_dir);
+            }
+        }
+    }
+
+    // Fallback to the current directory
+    Ok(local_butane_dir)
 }
 
 pub fn handle_error(r: Result<()>) {