aider/testsr.py

#!/usr/bin/env python3

import json
import re
import sys

from aider.coders.editblock_coder import (
    DEFAULT_FENCE,
    all_fences,
    find_original_update_blocks,
)


def process_markdown(filename):
    try:
        with open(filename, "r") as file:
            content = file.read()

        # Split the content into sections based on '####' headers
        sections = re.split(r"(?=####\s)", content)

        results = []
        for section in sections:
            if not section.strip():  # Ignore empty sections
                continue
            # Extract the header (if present)
            header = section.split("\n")[0].strip()
            # Get the content (everything after the header)
            content = "\n".join(section.split("\n")[1:]).strip()

            # Process the content with find_original_update_blocks
            try:
                blocks = list(find_original_update_blocks(content, DEFAULT_FENCE))

                # Create a dictionary for this section
                section_result = {"header": header, "blocks": []}

                for block in blocks:
                    if block[0] is None:  # This is a shell command block
                        section_result["blocks"].append({"type": "shell", "content": block[1]})
                    else:  # This is a SEARCH/REPLACE block
                        section_result["blocks"].append(
                            {
                                "type": "search_replace",
                                "filename": block[0],
                                "original": block[1],
                                "updated": block[2],
                            }
                        )

                results.append(section_result)
            except ValueError as e:
                # If an error occurs, add it to the results for this section
                results.append({"header": header, "error": str(e)})

        # Output the results as JSON
        print(json.dumps(results, indent=4))

    except FileNotFoundError:
        print(json.dumps({"error": f"File '{filename}' not found."}, indent=4))
    except Exception as e:
        print(e)


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print(json.dumps({"error": "Usage: python testsr.py <markdown_filename>"}, indent=4))
    else:
        process_markdown(sys.argv[1])