# Code to modify the birdnet_analysis.py and restore the Processed folder
# The last 15 wav files are stored in a dynamic manner in the Processed folder

def read_file(file_path):
    with open(file_path, 'r') as file:
        return file.readlines()

def write_file(file_path, content):
    with open(file_path, 'w') as file:
        file.writelines(content)

def process_code(lines):
    processed_lines = []
    for line in lines:
        # Add the import statement for shutil if it's not already there
        if 'import os.path' in line and 'import shutil' not in lines:
            processed_lines.append('import shutil\n')
        processed_lines.append(line)

        # Modify the process_file function to include the conf parameter
        if 'def process_file(file_name, report_queue):' in line:
            processed_lines[-1] = 'def process_file(file_name, report_queue, conf):\n'

        # Modify the handle_reporting_queue function to include the conf parameter and the new logic
        if 'def handle_reporting_queue(queue):' in line:
            processed_lines[-1] = 'def handle_reporting_queue(queue, conf):\n'
        if 'os.remove(file.file_name)' in line:
            indent = ' ' * (len(line) - len(line.lstrip()))
            processed_lines.append(f"{indent}# Move the file to the 'Processed' folder\n")
            processed_lines.append(f"{indent}processed_dir = os.path.join(conf['RECS_DIR'], 'Processed')\n")
            processed_lines.append(f"{indent}if not os.path.exists(processed_dir):\n")
            processed_lines.append(f"{indent}    os.makedirs(processed_dir)\n")
            processed_lines.append(f"{indent}shutil.move(file.file_name, processed_dir)\n")
            processed_lines.append(f"{indent}\n")
            processed_lines.append(f"{indent}# Maintain the file count in the 'Processed' folder\n")
            processed_lines.append(f"{indent}maintain_file_count(processed_dir)\n")
            processed_lines.append(f"{indent}\n")
            continue  # Skip the original line that removes the file

        # Add the new maintain_file_count function at the end of the file
        if 'if __name__' in line and not any('def maintain_file_count' in l for l in processed_lines):
            processed_lines.append('\n')
            processed_lines.append('def maintain_file_count(directory, max_files=15):\n')
            processed_lines.append('    files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith(\'.wav\')]\n')
            processed_lines.append('    files.sort(key=lambda x: os.path.getmtime(x))\n')
            processed_lines.append('\n')
            processed_lines.append('    while len(files) > max_files:\n')
            processed_lines.append('        os.remove(files.pop(0))\n')
            processed_lines.append('\n')

    return processed_lines

# Paths to the original and new code files
original_code_path = '/home/pi/BirdNET-Pi/scripts/birdnet_analysis.py'
new_code_path = '/home/pi/BirdNET-Pi/scripts/birdnet_analysis.py'

# Read the original code
original_lines = read_file(original_code_path)

# Process the code
modified_lines = process_code(original_lines)

# Write the modified code to a new file
write_file(new_code_path, modified_lines)

print(f"The code has been modified and saved to {new_code_path}")