Function save_learnings_to_context_file

atomic-agent/src/learnings.rs:364–431

Save learnings to the agent's context file. If the file doesn't exist, creates it with the learnings section. If the file exists but has no learnings section, appends the section. If the file exists with a learnings section, merges new learnings (deduplicating by finding text) and rewrites the section. # Arguments `repo_root` — Path to the repository root `agent_name` — Agent registry key (dete

(
    repo_root: &Path,
    agent_name: &str,
    learnings: &Learnings,
)

///
/// Returns an error if the file cannot be read or written.
pub fn save_learnings_to_context_file(
    repo_root: &Path,
    agent_name: &str,
    learnings: &Learnings,
) -> Result<SaveResult, std::io::Error> {
    if learnings.is_empty() {
        return Ok(SaveResult {
            file_path: context_file_path(repo_root, agent_name),
            added: 0,
            skipped_duplicates: 0,
        });
    }

    let file_path = context_file_path(repo_root, agent_name);

    // Ensure parent directory exists (for .atomic/learnings.md)
    if let Some(parent) = file_path.parent() {
        std::fs::create_dir_all(parent)?;
    }

    // Read existing content (or empty string if file doesn't exist)
    let existing_content = std::fs::read_to_string(&file_path).unwrap_or_default();

    // Dedup against existing learnings
    let original_count = context_file_learning_count(learnings);
    let deduped = dedup_learnings(&existing_content, learnings);
    let new_count = context_file_learning_count(&deduped);
    let skipped = original_count - new_count;

    if new_count == 0 {
        return Ok(SaveResult {
            file_path,
            added: 0,
            skipped_duplicates: skipped,
        });
    }

    // Format the new learnings as individual items and merge by heading
    let new_items = collect_new_items(&deduped);

    // Build the updated file content
    let updated_content = if existing_content.contains(SECTION_START) {
        // Merge into existing headings (no duplicate ### Repo / ### Workflow)
        merge_items_into_section(&existing_content, &new_items)
    } else if existing_content.is_empty() {
        let new_markdown = format_learnings_markdown(&deduped);
        // New file — create with just the learnings section
        build_new_section(&new_markdown)
    } else {
        // Existing file without a learnings section — append
        let new_markdown = format_learnings_markdown(&deduped);
        let mut content = existing_content.clone();
        if !content.ends_with('\n') {
            content.push('\n');
        }
        content.push('\n');
        content.push_str(&build_new_section(&new_markdown));
        content