Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
100 changes: 90 additions & 10 deletions src/profiles/installer.rs
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
use super::error::{ProfileError, ProfileResult};
use super::lockfile::ProfileLockfile;
use std::path::{Path, PathBuf};
use walkdir::WalkDir;

/// Installation result: (installed files, sidecar files)
/// - installed: List of relative file paths that were installed
Expand Down Expand Up @@ -78,6 +79,10 @@ pub fn check_all_conflicts(
// Force enabled - will use sidecar
}
None => {
if path_owned_by_profile(lockfile, profile_name, file_path) {
// Directory path owned through tracked child files
continue;
}
// File exists but unknown owner (user's file or orphaned)
if !force {
conflicts.push((file_path.clone(), "unknown".to_string()));
Expand Down Expand Up @@ -204,15 +209,19 @@ impl ProfileInstaller {
true
}
None => {
// Unknown owner
if !force {
if path_owned_by_profile(lockfile, profile_name, file_path) {
// Directory path owned through tracked child files.
false
} else if !force {
// Unknown owner
return Err(ProfileError::FileConflict {
path: file_path.clone(),
owner: "unknown".to_string(),
});
} else {
// Force enabled - use sidecar
true
}
// Force enabled - use sidecar
true
}
}
} else {
Expand All @@ -233,15 +242,13 @@ impl ProfileInstaller {
std::fs::create_dir_all(parent)?;
}

// Copy file
std::fs::copy(&source_path, &final_path)?;
let copied_files =
copy_source_entry(&source_path, &final_path, Path::new(&relative_path))?;

if use_sidecar {
sidecars.push((file_path.clone(), relative_path.clone()));
installed.push(relative_path);
} else {
installed.push(file_path.clone());
sidecars.push((file_path.clone(), relative_path));
}
installed.extend(copied_files);
}

Ok((installed, sidecars))
Expand All @@ -253,3 +260,76 @@ impl Default for ProfileInstaller {
Self::new()
}
}

fn path_owned_by_profile(lockfile: &ProfileLockfile, profile_name: &str, path: &str) -> bool {
let Some(entry) = lockfile.get_profile(profile_name) else {
return false;
};

let prefix = format!("{}/", path.trim_end_matches('/'));
entry
.files
.iter()
.any(|tracked| tracked == path || tracked.starts_with(&prefix))
}

fn copy_source_entry(
source_path: &Path,
dest_path: &Path,
relative_base: &Path,
) -> ProfileResult<Vec<String>> {
if source_path.is_dir() {
return copy_directory_contents(source_path, dest_path, relative_base);
}

std::fs::copy(source_path, dest_path)?;
Ok(vec![normalize_path(relative_base)])
}

fn copy_directory_contents(
source_dir: &Path,
dest_dir: &Path,
relative_base: &Path,
) -> ProfileResult<Vec<String>> {
let mut copied_files = Vec::new();
std::fs::create_dir_all(dest_dir)?;

for entry in WalkDir::new(source_dir).follow_links(true) {
let entry = entry.map_err(|e| ProfileError::IoError(std::io::Error::other(e)))?;
let entry_path = entry.path();
let relative = entry_path.strip_prefix(source_dir).map_err(|_| {
ProfileError::IoError(std::io::Error::other(format!(
"walkdir entry '{}' is outside source '{}'",
entry_path.display(),
source_dir.display()
)))
})?;

if relative.as_os_str().is_empty() {
continue;
}

if relative.components().any(|c| c.as_os_str() == ".git") {
continue;
}

let destination = dest_dir.join(relative);
if entry.file_type().is_dir() {
std::fs::create_dir_all(&destination)?;
continue;
}

if let Some(parent) = destination.parent() {
std::fs::create_dir_all(parent)?;
}

std::fs::copy(entry_path, &destination)?;
copied_files.push(normalize_path(&relative_base.join(relative)));
}

Ok(copied_files)
}

fn normalize_path(path: &Path) -> String {
path.to_string_lossy().replace('\\', "/")
}
118 changes: 118 additions & 0 deletions tests/profiles/test_orchestrator.rs
Original file line number Diff line number Diff line change
Expand Up @@ -263,3 +263,121 @@ fn test_install_profile_conflict_creates_sidecar() {
let content_b = fs::read_to_string(&sidecar_path).unwrap();
assert_eq!(content_b, "# Profile B");
}

#[cfg(unix)]
#[test]
fn test_install_profile_supports_symlinked_directory() {
let temp = tempdir().unwrap();

let profiles_dir = temp.path().join("profiles");
let profile_dir = profiles_dir.join("symlink-profile");
fs::create_dir_all(profile_dir.join("skills")).unwrap();

fs::write(
profile_dir.join("profile.json"),
r#"{"name": "symlink-profile", "version": "1.0.0", "files": []}"#,
)
.unwrap();

let shared_skill_dir = temp.path().join("shared-skills").join("sample-skill");
fs::create_dir_all(&shared_skill_dir).unwrap();
fs::write(shared_skill_dir.join("SKILL.md"), "# Sample Skill").unwrap();

std::os::unix::fs::symlink(
&shared_skill_dir,
profile_dir.join("skills").join("sample-skill"),
)
.unwrap();

let workspace = temp.path().join("workspace");
fs::create_dir_all(&workspace).unwrap();

install_profile(
"symlink-profile",
&profiles_dir,
&workspace,
false,
None,
None,
None,
)
.unwrap();

let installed_skill = workspace.join("skills/sample-skill/SKILL.md");
assert!(installed_skill.exists());
assert_eq!(
fs::read_to_string(installed_skill).unwrap(),
"# Sample Skill"
);

let lockfile = ProfileLockfile::load(&workspace.join(".codanna/profiles.lock.json")).unwrap();
let entry = lockfile.get_profile("symlink-profile").unwrap();
assert!(
entry
.files
.contains(&"skills/sample-skill/SKILL.md".to_string())
);
}

#[cfg(unix)]
#[test]
fn test_force_reinstall_symlinked_directory_overwrites_owned_files() {
let temp = tempdir().unwrap();

let profiles_dir = temp.path().join("profiles");
let profile_dir = profiles_dir.join("symlink-profile");
fs::create_dir_all(profile_dir.join("skills")).unwrap();

fs::write(
profile_dir.join("profile.json"),
r#"{"name": "symlink-profile", "version": "1.0.0", "files": []}"#,
)
.unwrap();

let shared_skill_dir = temp.path().join("shared-skills").join("sample-skill");
fs::create_dir_all(&shared_skill_dir).unwrap();
let skill_file = shared_skill_dir.join("SKILL.md");
fs::write(&skill_file, "# V1").unwrap();

std::os::unix::fs::symlink(
&shared_skill_dir,
profile_dir.join("skills").join("sample-skill"),
)
.unwrap();

let workspace = temp.path().join("workspace");
fs::create_dir_all(&workspace).unwrap();

install_profile(
"symlink-profile",
&profiles_dir,
&workspace,
false,
None,
None,
None,
)
.unwrap();

fs::write(&skill_file, "# V2").unwrap();
install_profile(
"symlink-profile",
&profiles_dir,
&workspace,
true,
None,
None,
None,
)
.unwrap();

assert_eq!(
fs::read_to_string(workspace.join("skills/sample-skill/SKILL.md")).unwrap(),
"# V2"
);
assert!(
!workspace
.join("skills/sample-skill.symlink-profile/SKILL.md")
.exists()
);
}
Loading