DocuMentor.py

import os

# Configuration option to check for existing .md files before creating new ones
check_for_existing_md = True  # Set to False to disable this check

# Mapping of file extensions to markdown code fence language identifiers
extension_to_lang = {
    '.php': 'php', '.py': 'python', '.js': 'javascript', '.java': 'java', '.c': 'c',
    '.cpp': 'cpp', '.cs': 'csharp', '.ts': 'typescript', '.swift': 'swift', '.go': 'go',
    '.rb': 'ruby', '.lua': 'lua', '.groovy': 'groovy', '.scala': 'scala', '.kt': 'kotlin',
    '.rs': 'rust', '.dart': 'dart', '.sh': 'bash', '.bat': 'bat', '.pl': 'perl',
    '.h': 'c', '.m': 'objc', '.sql': 'sql', '.html': 'html', '.css': 'css',
    '.sass': 'sass', '.scss': 'scss', '.less': 'less', '.json': 'json', '.xml': 'xml',
    '.yml': 'yaml', '.yaml': 'yaml', '.md': 'markdown', '.ini': 'ini', '.cfg': 'ini',
    '.conf': 'ini'
    # Add more as needed
}

def is_binary(file_path):
    try:
        with open(file_path, 'rb') as file:
            chunk = file.read(1024)  # Read the first 1024 bytes
            if b'\0' in chunk:
                return True
            chunk.decode('utf-8')
        return False
    except UnicodeDecodeError:
        return True

def is_code_file(file_path):
    _, ext = os.path.splitext(file_path)
    return ext in extension_to_lang

def get_language_for_extension(file_path):
    _, ext = os.path.splitext(file_path)
    return extension_to_lang.get(ext, '')

def create_clone_directory(source_dir, target_dir, clone_mode):
    if not os.path.exists(target_dir):
        os.makedirs(target_dir)

    for root, dirs, files in os.walk(source_dir):
        for dir_ in dirs:
            os.makedirs(os.path.join(target_dir, os.path.relpath(os.path.join(root, dir_), source_dir)), exist_ok=True)

        for file in files:
            source_file_path = os.path.join(root, file)
            relative_path = os.path.relpath(root, source_dir)
            target_file_path = os.path.join(target_dir, relative_path, file + ".md")

            if check_for_existing_md and os.path.exists(target_file_path):
                continue

            if is_binary(source_file_path):
                open(target_file_path, "w").close()
            else:
                with open(source_file_path, "r", encoding='utf-8', errors='replace') as source_file, open(target_file_path, "w", encoding='utf-8') as target_file:
                    if clone_mode == 2 and is_code_file(source_file_path):
                        lang = get_language_for_extension(source_file_path)
                        target_file.write(f"Please analyze the following code to identify its purpose and functionality. Include inline comments to explain important functions, methods, and processes. Highlight how it integrates with other parts of the application or any external services it interacts with.\n\n```{lang}\n")
                        target_file.write(source_file.read())
                        target_file.write("\n```\n\nFocus on the role of key components and how they contribute to the overall operation of the code. Thank you!")
                    else:
                        target_file.write(source_file.read())



def generate_directory_trees(target_dir):
    basic_tree = []
    advanced_tree = []

    for root, dirs, files in os.walk(target_dir):
        level = root.replace(target_dir, '').count(os.sep)
        indent = ' ' * 4 * (level)
        basic_tree.append('{}{}/'.format(indent, os.path.basename(root)))
        advanced_tree.append('{}{}/'.format(indent, os.path.basename(root)))
        subindent = ' ' * 4 * (level + 1)
        for f in files:
            advanced_tree.append('{}{}'.format(subindent, f))

    with open(os.path.join(target_dir, ".directory-tree-basic.md"), "w") as basic_file:
        basic_file.write('\n'.join(basic_tree))

    with open(os.path.join(target_dir, ".directory-tree-advanced.md"), "w") as advanced_file:
        advanced_file.write('\n'.join(advanced_tree))

def main():
    global check_for_existing_md
    check_for_existing_md_input = input("Check for existing .md files before creating new ones? (yes/no): ").strip().lower()
    check_for_existing_md = check_for_existing_md_input == 'yes'

    print("Enter the path to the codebase directory (use a format compatible with your operating system):")
    source_dir = input().strip().strip('"')
    print("Enter the directory path to create the clone (use a format compatible with your operating system):")
    target_dir = input().strip().strip('"')

    clone_mode = int(input("Choose cloning mode - 1: Just clone and add md, 2: Clone, add md and wrap code in quotes (1/2): ").strip())

    create_clone_directory(source_dir, target_dir, clone_mode)

    create_directory_tree = input("Create a Folder structure tree markdown file? (yes/no): ").strip().lower()
    if create_directory_tree == 'yes':
        generate_directory_trees(target_dir) 

    print("Clone creation complete!")

if __name__ == "__main__":
    main()