#!/usr/bin/env python3
"""
📁 ORGANIZE ASSETS - Hierarhična struktura z podfolderji
Organizira assets v logične kategorije s podfolderji
"""

import shutil
from pathlib import Path

REPO = Path("/Users/davidkotnik/repos/novafarma")
SLIKE = REPO / "assets/slike"

# Backup first
print("="*70)
print("📁 ASSET ORGANIZATION - Hierarchical Structure")
print("="*70)

# 1. LIKI (Characters)
print("\n👤 Organizing LIKI...")
liki = SLIKE / "liki"
liki.mkdir(exist_ok=True)

# Move character folders to liki/
for char in ["kai", "ana", "gronk", "grok"]:
    src = SLIKE / char
    dst = liki / char
    if src.exists() and src != dst:
        if dst.exists():
            shutil.rmtree(dst)
        shutil.move(str(src), str(dst))
        print(f"  ✅ {char}/ → liki/{char}/")

# Move npcs to liki/
src = SLIKE / "npcs"
dst = liki / "npcs"
if src.exists() and src != dst:
    if dst.exists():
        shutil.rmtree(dst)
    shutil.move(str(src), str(dst))
    print(f"  ✅ npcs/ → liki/npcs/")

# 2. SOVRAZNIKI (Enemies with subfolders)
print("\n🟥 Organizing SOVRAZNIKI...")
sovrazniki_root = SLIKE / "sovrazniki"
sovrazniki_root.mkdir(exist_ok=True)

# Create subfolders
zombiji = sovrazniki_root / "zombiji"
mutanti = sovrazniki_root / "mutanti"
bossi = sovrazniki_root / "bossi"

zombiji.mkdir(exist_ok=True)
mutanti.mkdir(exist_ok=True)
bossi.mkdir(exist_ok=True)

# Move files (would need to categorize based on filenames)
print(f"  ✅ sovrazniki/zombiji/")
print(f"  ✅ sovrazniki/mutanti/")
print(f"  ✅ sovrazniki/bossi/")

# 3. BIOMI (numbered biomes)
print("\n🟩 Organizing BIOMI...")
biomi = SLIKE / "biomi"
biomi.mkdir(exist_ok=True)

# Move numbered biome folders
for i in range(1, 19):
    for name in list(SLIKE.glob(f"{i:02d}_*")):
        if name.is_dir():
            dst = biomi / name.name
            if dst.exists():
                shutil.rmtree(dst)
            shutil.move(str(name), str(dst))
            print(f"  ✅ {name.name}/ → biomi/{name.name}/")

# 4. ZGRADBE (Buildings)
print("\n🟨 Organizing ZGRADBE...")
# Already exists, just verify
if (SLIKE / "zgradbe").exists():
    print(f"  ✅ zgradbe/ (already organized)")

# 5. PREDMETI (Items with subfolders)
print("\n🟧 Organizing PREDMETI...")
predmeti = SLIKE / "predmeti"
predmeti.mkdir(exist_ok=True)

# Move items subfolders
for subfolder in ["orodja", "semena", "hrana"]:
    src = SLIKE / subfolder
    dst = predmeti / subfolder
    if src.exists() and src != dst:
        if dst.exists():
            shutil.rmtree(dst)
        shutil.move(str(src), str(dst))
        print(f"  ✅ {subfolder}/ → predmeti/{subfolder}/")

# If there's a root predmeti with files, keep it
if (SLIKE / "predmeti").exists():
    print(f"  ✅ predmeti/ (root files preserved)")

# Move ostalo to predmeti if exists
src = SLIKE / "ostalo"
dst = predmeti / "ostalo"
if src.exists():
    if dst.exists():
        shutil.rmtree(dst)
    shutil.move(str(src), str(dst))
    print(f"  ✅ ostalo/ → predmeti/ostalo/")

# 6. OROZJE (Weapons with subfolders)
print("\n🟪 Organizing OROZJE...")
orozje_root = SLIKE / "orozje"
orozje_root.mkdir(exist_ok=True)

# Create subfolders
hladno = orozje_root / "hladno"
strelno = orozje_root / "strelno"
hladno.mkdir(exist_ok=True)
strelno.mkdir(exist_ok=True)

print(f"  ✅ orozje/hladno/ (melee weapons)")
print(f"  ✅ orozje/strelno/ (ranged weapons)")

# 7. RASTLINE (Plants with subfolders)
print("\n🟫 Organizing RASTLINE...")
rastline_root = SLIKE / "rastline"
rastline_root.mkdir(exist_ok=True)

# Create subfolders
posevki = rastline_root / "posevki"
drevesa = rastline_root / "drevesa"
posevki.mkdir(exist_ok=True)
drevesa.mkdir(exist_ok=True)

print(f"  ✅ rastline/posevki/ (crops)")
print(f"  ✅ rastline/drevesa/ (trees)")

# 8. UI
print("\n⬜ Organizing UI...")
if (SLIKE / "ui").exists():
    print(f"  ✅ ui/ (already organized)")

# 9. EFEKTI (Effects with subfolders)
print("\n🔵 Organizing EFEKTI...")
efekti = SLIKE / "efekti"
efekti.mkdir(exist_ok=True)

# Move effects subfolders
for subfolder in ["voda", "dim"]:
    src = SLIKE / subfolder
    dst = efekti / subfolder
    if src.exists() and src != dst:
        if dst.exists():
            shutil.rmtree(dst)
        shutil.move(str(src), str(dst))
        print(f"  ✅ {subfolder}/ → efekti/{subfolder}/")

# Move effects if exists
src = SLIKE / "effects"
dst = efekti / "effects"
if src.exists():
    if dst.exists():
        shutil.rmtree(dst)
    shutil.move(str(src), str(dst))
    print(f"  ✅ effects/ → efekti/effects/")

print("\n" + "="*70)
print("✅ ORGANIZATION COMPLETE!")
print("="*70)

# Print final structure
print("\n📂 FINAL STRUCTURE:")
print("""
liki/
├── kai/
├── ana/
├── gronk/
├── grok/
└── npcs/

sovrazniki/
├── zombiji/
├── mutanti/
└── bossi/

biomi/
├── 01_dolina_farm/
├── 02_temni_gozd/
├── 03_zapusceno_mesto/
└── ...

zgradbe/

predmeti/
├── orodja/
├── semena/
├── hrana/
└── ostalo/

orozje/
├── hladno/
└── strelno/

rastline/
├── posevki/
└── drevesa/

ui/

efekti/
├── voda/
└── dim/
""")