regex.py

import os
import re

def read_generated_code(file_path):
    with open(file_path, 'r') as file:
        return file.read()

def extract_verilog_code(text):
    verilog_pattern=r"(?s)(`include[\s\S]*?endmodule|module\s+\w+\s*#\([\s\S]*?endmodule)"
    #verilog_pattern = r'(?s)module\s+top_module\s*\(.*?\);\s*.*?endmodule'
    matches = re.findall(verilog_pattern, text, re.DOTALL)
    return '\n\n'.join(matches).strip()

def extract_testbench(verilog_content):
    # Regex pattern to capture the entire testbench module
    # This pattern matches the module declaration to 'endmodule' including everything in between
    pattern = re.compile(r'(module\s+(?:.*?\btest\b|.*?\btb\b|.*?\btestbench\b).*?;.*?endmodule)', re.S | re.I)

    # Search for testbench in the file content
    match = pattern.search(verilog_content)
    
    if match: 
        return(match.group(1))
    else:
        return None
    


def save_extracted_code(file_path, extracted_code):
    with open(file_path, 'w') as file:
        file.write(extracted_code)

def main():
    framework_name = os.environ.get('framework_name')  # Set default if not provided
    module_name = os.environ.get('module_name', 'default_module')  # Set default if not provided
    base_path = os.path.join(f'{framework_name}')
    generated_code_path = os.path.join(base_path, "responses", f'{module_name}.txt')
    extracted_code_path = os.path.join(base_path, 'modules', f'{module_name}.v')

    if not os.path.exists(generated_code_path):
        print(f"Error: {generated_code_path} does not exist.")
        return

    generated_code = read_generated_code(generated_code_path)
    verilog_code = extract_verilog_code(generated_code)
    
    if verilog_code:
        os.makedirs(os.path.dirname(extracted_code_path), exist_ok=True)  # Ensure directory exists
        save_extracted_code(extracted_code_path, verilog_code)
        print(f"Extracted Verilog code has been written to {extracted_code_path}")
    else:
        print(f"No Verilog code found in {generated_code_path}")


if __name__ == "__main__":
    main()