Function collect_tree

atomic-core/src/output/repo/tree.rs:783–857

Collect children from the tree under a given path. This function traverses the TREE table to find all files and directories under the specified parent path, applying the given options for filtering. # Arguments `txn` - Transaction providing tree access `parent_path` - Path to start from (empty for root) `options` - Collection options # Returns A `TreeCollectResult` containing all matching ite

(
    txn: &T,
    parent_path: &str,
    options: TreeCollectOptions,
)

/// let src_result = collect_tree(&txn, "", options)?;
/// ```
pub fn collect_tree<T: TreeTxnT + GraphTxnT>(
    txn: &T,
    parent_path: &str,
    options: TreeCollectOptions,
) -> Result<TreeCollectResult, PristineError> {
    let mut result = TreeCollectResult::new();

    // Track directories we've seen to avoid duplicates
    let mut seen_directories: HashSet<String> = HashSet::new();

    // Iterate over all tree entries
    for entry in txn.iter_tree()? {
        let (path, inode) = entry?;

        // Skip if path doesn't start with parent_path
        if !parent_path.is_empty() && !path.starts_with(parent_path) {
            continue;
        }

        // Check filters
        if !options.should_include(&path) {
            result.record_skipped();
            continue;
        }

        // Check depth
        if options.exceeds_depth(&path) {
            result.record_skipped();
            continue;
        }

        // Collect parent directories if needed
        if options.collect_directories {
            let mut current = String::new();
            for component in path.split('/') {
                if !current.is_empty() {
                    current.push('/');
                }
                current.push_str(component);

                // Don't add the file itself as a directory
                if current == path {
                    break;
                }

                // Check if we should include this directory
                if !seen_directories.contains(&current)
                    && options.should_include(&current)
                    && !options.exceeds_depth(&current)
                {
                    seen_directories.insert(current.clone());
                    // We don't have the inode for intermediate directories
                    // In a full implementation, we'd look this up
                    result.add_directory(TreeItem::directory(&current, Inode::ROOT));
                }
            }
        }