strixonomy_refactor/

apply.rs

use crate::error::{RefactorError, Result};
use crate::model::{FileChange, RefactorPlan};
use std::fs;
use std::path::{Path, PathBuf};
use strixonomy_core::{
    atomic_write, canonical_workspace_root, read_to_string_capped, validate_workspace_scope_any,
    MAX_FILE_BYTES,
};

/// Validate every path in a refactor plan is within at least one workspace root.
pub fn validate_refactor_plan_paths_any(
    workspace_roots: &[PathBuf],
    plan: &RefactorPlan,
) -> Result<()> {
    for change in &plan.changes {
        validate_workspace_scope_any(&change.path, workspace_roots)
            .map_err(RefactorError::Invalid)?;
    }
    Ok(())
}

/// Validate every path in a refactor plan is within the workspace jail.
pub fn validate_refactor_plan_paths(workspace_root: &Path, plan: &RefactorPlan) -> Result<()> {
    let root = canonical_workspace_root(workspace_root).map_err(RefactorError::Invalid)?;
    validate_refactor_plan_paths_any(std::slice::from_ref(&root), plan)
}

/// Returns true when client-submitted plan matches a server re-preview byte-for-byte.
pub fn plans_equivalent(server: &RefactorPlan, client: &RefactorPlan) -> bool {
    if server.changes.len() != client.changes.len() {
        return false;
    }
    let mut server_sorted: Vec<&FileChange> = server.changes.iter().collect();
    let mut client_sorted: Vec<&FileChange> = client.changes.iter().collect();
    server_sorted.sort_by_key(|c| &c.path);
    client_sorted.sort_by_key(|c| &c.path);
    for (s, c) in server_sorted.iter().zip(client_sorted.iter()) {
        if s.path != c.path || s.preview_text != c.preview_text {
            return false;
        }
    }
    true
}

struct FileBackup {
    path: PathBuf,
    previous: Option<String>,
    created: bool,
}

fn backup_file(path: &Path) -> std::io::Result<FileBackup> {
    let created = !path.exists();
    let previous = if created {
        None
    } else {
        Some(read_to_string_capped(path, MAX_FILE_BYTES).map_err(std::io::Error::other)?)
    };
    Ok(FileBackup { path: path.to_path_buf(), previous, created })
}

fn restore_backup(backup: &FileBackup) -> std::io::Result<()> {
    if backup.created {
        if backup.path.exists() {
            fs::remove_file(&backup.path)?;
        }
    } else if let Some(content) = &backup.previous {
        atomic_write(&backup.path, content)?;
    }
    Ok(())
}

/// Apply a refactor plan to disk. When `preview_only` is true, no files are written.
pub fn apply_refactor_plan(
    plan: &RefactorPlan,
    preview_only: bool,
    workspace_root: &Path,
) -> Result<()> {
    let root = canonical_workspace_root(workspace_root).map_err(RefactorError::Invalid)?;
    apply_refactor_plan_checked(plan, preview_only, Some(std::slice::from_ref(&root))).map(|_| ())
}

/// Apply plan and return count of files written.
pub fn apply_refactor_plan_checked(
    plan: &RefactorPlan,
    preview_only: bool,
    workspace_roots: Option<&[PathBuf]>,
) -> Result<usize> {
    apply_refactor_plan_checked_with_overrides(plan, preview_only, workspace_roots, None)
}

/// Apply plan with optional open-buffer overrides used for compare-and-swap.
pub fn apply_refactor_plan_checked_with_overrides(
    plan: &RefactorPlan,
    preview_only: bool,
    workspace_roots: Option<&[PathBuf]>,
    document_overrides: Option<&std::collections::HashMap<PathBuf, String>>,
) -> Result<usize> {
    if let Some(roots) = workspace_roots {
        validate_refactor_plan_paths_any(roots, plan)?;
    }
    if preview_only {
        return Ok(0);
    }
    let mut written = 0usize;
    let mut backups: Vec<FileBackup> = Vec::new();
    for change in &plan.changes {
        if change.preview_text == change.original_text {
            continue;
        }
        let backup = backup_file(&change.path)?;
        // Compare-and-swap against the same source the preview used (buffer or disk).
        let current = effective_current_text(&change.path, &backup, document_overrides);
        if current != change.original_text {
            rollback_backups(&backups)?;
            return Err(RefactorError::Invalid(format!(
                "file changed since preview: {}",
                change.path.display()
            )));
        }
        if let Err(e) = atomic_write(&change.path, &change.preview_text) {
            rollback_backups(&backups)?;
            return Err(RefactorError::Io(e));
        }
        backups.push(backup);
        written += 1;
    }
    if written == 0 && !plan.changes.is_empty() {
        rollback_backups(&backups)?;
        return Err(RefactorError::Invalid("no files changed".to_string()));
    }
    Ok(written)
}

fn effective_current_text(
    path: &Path,
    backup: &FileBackup,
    document_overrides: Option<&std::collections::HashMap<PathBuf, String>>,
) -> String {
    if let Some(overrides) = document_overrides {
        if let Some(text) = overrides.get(path) {
            return text.clone();
        }
        if let Ok(canonical) = path.canonicalize() {
            if let Some(text) = overrides.get(&canonical) {
                return text.clone();
            }
        }
    }
    backup.previous.clone().unwrap_or_default()
}

fn rollback_backups(backups: &[FileBackup]) -> Result<()> {
    let mut restore_errors = Vec::new();
    for b in backups.iter().rev() {
        if let Err(e) = restore_backup(b) {
            restore_errors.push(format!("{}: {e}", b.path.display()));
        }
    }
    if restore_errors.is_empty() {
        Ok(())
    } else {
        Err(RefactorError::Invalid(format!(
            "refactor rollback failed for: {}",
            restore_errors.join("; ")
        )))
    }
}

pub fn plan_touches_path(plan: &RefactorPlan, path: &Path) -> bool {
    plan.changes.iter().any(|c| c.path == path)
}