Function apply_diff

utils/code_utils.py:48–87

Apply a diff to the original code using the specified SEARCH/REPLACE format. Args: original_code: The original source code string. diff_text: The text containing one or more diff blocks. diff_pattern: The regex pattern to identify the SEARCH/REPLACE format. Ret

(
    original_code: str,
    diff_text: str,
    diff_pattern: str = r"<<<<<<< SEARCH\n(.*?)=======\n(.*?)>>>>>>> REPLACE",
)

def apply_diff(
    original_code: str,
    diff_text: str,
    diff_pattern: str = r"<<<<<<< SEARCH\n(.*?)=======\n(.*?)>>>>>>> REPLACE",
) -> str:
    """
    Apply a diff to the original code using the specified SEARCH/REPLACE format.

    Args:
        original_code: The original source code string.
        diff_text: The text containing one or more diff blocks.
        diff_pattern: The regex pattern to identify the SEARCH/REPLACE format.

    Returns:
        The modified code string after applying all diffs.
    """
    # Split into lines for easier processing and replacement
    original_lines = original_code.split("\n")
    result_lines = original_lines.copy()

    # Extract all diff blocks from the provided text
    diff_blocks = extract_diffs(diff_text, diff_pattern)

    # Apply each diff block sequentially
    for search_text, replace_text in diff_blocks:
        search_lines = search_text.split("\n")
        replace_lines = replace_text.split("\n")

        # Find where the search pattern starts in the current version of the code
        # We iterate through the `result_lines` which may have been modified by previous diffs
        for i in range(len(result_lines) - len(search_lines) + 1):
            # Check if the slice of lines matches the search lines exactly
            if result_lines[i : i + len(search_lines)] == search_lines:
                # If a match is found, replace that slice with the replacement lines
                result_lines[i : i + len(search_lines)] = replace_lines
                # Break after the first match to avoid applying the same diff multiple times
                break

    # Join the modified lines back into a single string
    return "\n".join(result_lines)


def extract_diffs(