Function remove_duplicate_definition

src/execution/ast.rs:185–203

LLMs tends to re-declare the provided API in code. Those duplicate definitions cause link error, thus those duplicated definitions need to be removed.

(program: &Path)

/// LLMs tends to re-declare the provided API in code.
/// Those duplicate definitions cause link error, thus those duplicated definitions need to be removed.
pub fn remove_duplicate_definition(program: &Path) -> Result<()> {
    let ast = Executor::extract_program_ast(program)?;
    let mut del_ranges = vec![];
    for child in &ast.inner {
        if let Clang::FunctionDecl(fd) = &child.kind {
            let name = fd.get_name();
            if get_func_gadget(&name).is_some() {
                del_ranges.push(fd.range.clone());
            }
        }
    }
    let mut content = std::fs::read_to_string(program)?;
    for sr in del_ranges.iter().rev() {
        let (begin, end) = get_source_code_range(sr)?;
        content.replace_range(begin..end + 1, "");
    }
    std::fs::write(program, content)?;
    Ok(())
}

#[test]
fn test_ast_dump_filter() -> Result<()> {