#!/usr/bin/env python3
import os
import argparse

def replace_content(path, old_string, new_string, preview, verbose):
    try:
        with open(path, 'r', encoding='utf-8') as f:
            content = f.read()

        if old_string in content:
            new_content = content.replace(old_string, new_string)
            
            print_verbose(f"Replacing content in: {path}", verbose)
                
            if not preview:
                with open(path, 'w', encoding='utf-8') as f:
                    f.write(new_content)

    except UnicodeDecodeError as e:
        print_verbose(f"Warning: Unicode decode error encountered in file {path}. Skipping file.", verbose)

def print_verbose(content,verbose):
    if verbose:
        print(content)
    

def process_directory(base_path, old_string, new_string, recursive, folder, files, content, preview, verbose, hidden):
    # Eine Liste, um die Pfade der umzubenennenden Ordner zu speichern
    directories_to_rename = []

    for root, dirs, filenames in os.walk(base_path):
        # Wenn "hidden" nicht gesetzt ist, versteckte Dateien/Ordner aus der Liste entfernen
        if not hidden:
            dirs[:] = [d for d in dirs if not d.startswith(".")]
            filenames = [f for f in filenames if not f.startswith(".")]

        if content:
            for f in filenames:
                replace_content(os.path.join(root, f), old_string, new_string, preview, verbose)

        if files:
            for f in filenames:
                if old_string in f:
                    old_path = os.path.join(root, f)
                    new_path = os.path.join(root, f.replace(old_string, new_string))
                    print_verbose(f"Renaming file from: {old_path} to: {new_path}",verbose)
                    if not preview:
                        os.rename(old_path, new_path)

        # Pfade von zu ändernden Ordnern speichern
        if folder:
            for d in dirs:
                if old_string in d:
                    old_path = os.path.join(root, d)
                    new_path = os.path.join(root, d.replace(old_string, new_string))
                    directories_to_rename.append((old_path, new_path))

        if not recursive:
            break

    # Ordnernamen ändern nach dem os.walk() Durchlauf
    for old_path, new_path in directories_to_rename:
        print_verbose(f"Renaming directory from: {old_path} to: {new_path}",verbose)
        if not preview:
            os.rename(old_path, new_path)

def main():
    parser = argparse.ArgumentParser(
        description="Replace strings in directories and files."
    )

    # positional args
    parser.add_argument(
        'paths',
        nargs='+',
        help="Paths in which replacements should be made."
    )
    parser.add_argument(
        'old_string',
        help="The string to be replaced."
    )

    # options with short and long flags
    parser.add_argument(
        '-n', '--new-string',
        dest='new_string',
        default="",
        help="The string to replace with. Default is empty string."
    )
    parser.add_argument(
        '-r', '--recursive',
        action='store_true',
        help="Replace in all subdirectories and files."
    )
    parser.add_argument(
        '-F', '--folder',
        action='store_true',
        help="Replace in folder names."
    )
    parser.add_argument(
        '-f', '--files',
        action='store_true',
        help="Replace in file names."
    )
    parser.add_argument(
        '-c', '--content',
        action='store_true',
        help="Replace inside file contents."
    )
    parser.add_argument(
        '-p', '--preview',
        action='store_true',
        help="Preview changes without replacing."
    )
    parser.add_argument(
        '-v', '--verbose',
        action='store_true',
        help="Verbose mode."
    )
    parser.add_argument(
        '-H', '--hidden',
        action='store_true',
        help="Apply to hidden files and folders."
    )

    args = parser.parse_args()


    # Use os.path.expanduser to expand the tilde to the home directory
    expanded_paths = [os.path.expanduser(path) for path in args.paths]

    for path in expanded_paths:
        print_verbose(f"Replacing in path: {path}",args.verbose)
        process_directory(path, args.old_string, args.new_string, args.recursive, args.folder, args.files, args.content, args.preview, args.verbose, args.hidden)

if __name__ == "__main__":
    main()