import os
import shutil

ROOT_ASSETS = os.path.abspath("assets/slike")
TARGET_DIRS = ["ui", "items", "environment", "characters", "animals"]

def flatten_folder(folder_name):
    base_dir = os.path.join(ROOT_ASSETS, folder_name)
    if not os.path.exists(base_dir):
        print(f"⚠️ Folder not found: {folder_name}")
        return

    print(f"🚜 Flattening {folder_name}...")
    moved_count = 0
    
    # Walk bottom-up to handle nested dirs
    for root, dirs, files in os.walk(base_dir, topdown=False):
        if root == base_dir:
            continue # Don't move files already in root

        # SPECIAL EXCEPTION: Skip 'blueprints' folder in 'items'
        if folder_name == "items" and "blueprints" in root:
            continue

        for filename in files:
            if filename.startswith("."): continue
            
            src = os.path.join(root, filename)
            dst = os.path.join(base_dir, filename)
            
            # 1. Handle Collisions (Rename if exists)
            if os.path.exists(dst):
                name, ext = os.path.splitext(filename)
                counter = 1
                while os.path.exists(os.path.join(base_dir, f"{name}_{counter}{ext}")):
                    counter += 1
                dst = os.path.join(base_dir, f"{name}_{counter}{ext}")
            
            # 2. Move
            shutil.move(src, dst)
            moved_count += 1
            
        # 3. Remove empty dir
        try:
            os.rmdir(root)
        except OSError:
            pass # Directory not empty (maybe contains .DS_Store or skipped files)

    print(f"   ✅ Moved {moved_count} images to root of /{folder_name}")

if __name__ == "__main__":
    for d in TARGET_DIRS:
        flatten_folder(d)