Method extract_code_blocks

src/context/builder.rs:379–422

Extracts code blocks for the entry-point nodes.

(
        &self,
        entry_points: &[Node],
        options: &BuildContextOptions,
        file_cache: &mut HashMap<String, Option<String>>,
    )

    /// Extracts code blocks for the entry-point nodes.
    fn extract_code_blocks(
        &self,
        entry_points: &[Node],
        options: &BuildContextOptions,
        file_cache: &mut HashMap<String, Option<String>>,
    ) -> Vec<CodeBlock> {
        debug_assert!(
            options.max_code_blocks > 0,
            "max_code_blocks must be positive"
        );
        debug_assert!(
            options.max_code_block_size > 0,
            "max_code_block_size must be positive"
        );
        let mut blocks: Vec<CodeBlock> = Vec::new();

        for node in entry_points {
            if blocks.len() >= options.max_code_blocks {
                break;
            }

            if let Some(code) = self.get_code_cached(node, file_cache) {
                let truncated = if code.len() > options.max_code_block_size {
                    let prefix =
                        crate::text::utf8_prefix_at_or_before(&code, options.max_code_block_size);
                    // Prefer a line boundary within the truncated prefix.
                    let end = prefix.rfind('\n').unwrap_or(prefix.len());
                    format!("{}...", &prefix[..end])
                } else {
                    code
                };

                blocks.push(CodeBlock {
                    content: truncated,
                    file_path: node.file_path.clone(),
                    start_line: node.start_line,
                    end_line: node.end_line,
                    node_id: Some(node.id.clone()),
                });
            }
        }

        blocks
    }

    /// Merges code blocks from the same file that are adjacent or overlapping.
    /// Two blocks are "adjacent" if the gap between them is <= 5 lines.