import os
import sys
import json
import re
from collections import defaultdict

def main():
    if len(sys.argv) < 3:
        print("Usage: python giving_names.py <directory_path> <Bnot/BGnot>")
        sys.exit(1)

    work_dir = os.path.abspath(sys.argv[1])
    bnot_dir = sys.argv[2]
    target_subdir = os.path.join(work_dir, "A Rendre")
    os.makedirs(target_subdir, exist_ok=True)

    # --- 1. Load the expected names list ---
    expected_names = set()
    names_path = os.path.join(work_dir, "names")
    if not os.path.exists(names_path):
        names_path = "names" # Fallback to current dir

    if os.path.exists(names_path):
        with open(names_path, 'r', encoding='utf-8') as f:
            expected_names = {line.strip() for line in f if line.strip()}
    else:
        print(f"Warning: 'names' file not found in {work_dir} or current directory.")

    # --- 2. Existing Collection Logic ---
    pattern = re.compile(r"^Copie(\d+)\.json$")
    copies_map = defaultdict(list)
    assigned_names = set() # To track which names were successfully linked

    for filename in os.listdir(work_dir):
        match = pattern.match(filename)
        if match:
            copie_id = match.group(1)
            json_path = os.path.join(work_dir, filename)
            try:
                with open(json_path, 'r', encoding='utf-8') as f:
                    data = json.load(f)
                    name = data.get("name", "Unknown").strip()
                    copies_map[name].append(copie_id)
            except Exception as e:
                print(f"Error processing {filename}: {e}")

    # --- 3. Process and Link ---
    for name, ids in copies_map.items():
        if name == "Unknown":
            print(f"ALERT: 'Unknown' name found for IDs: {', '.join(ids)}")
        elif len(ids) > 1:
            print(f"ALERT: Name '{name}' assigned to multiple IDs: {', '.join(ids)}")

        safe_name = re.sub(r'[<>:"/\\|?*]', '', name).strip()

        for copie_id in ids:
            path_b = os.path.join(work_dir, f"{bnot_dir}/Copie{copie_id}")
            path_a = os.path.join(work_dir, f"Anot/Copie{copie_id}")

            source_folder = None
            if os.path.exists(os.path.join(path_b, "Concat.jpg")) and os.path.exists(os.path.join(path_b, "score.json")):
                source_folder = path_b
            elif os.path.exists(os.path.join(path_a, "Concat.jpg")) and os.path.exists(os.path.join(path_a, "score.json")):
                source_folder = path_a

            if not source_folder:
                continue

            # If we reached here, the link is possible
            assigned_names.add(name)

            dest_folder_name = f"{safe_name} ({copie_id})"
            dest_path = os.path.join(target_subdir, dest_folder_name)
            os.makedirs(dest_path, exist_ok=True)

            links = [("Concat.jpg", f"{safe_name}.jpg"),("Concat_F.pdf", f"{safe_name}.pdf"), ("score.json", "score.json")]
            for src_name, dst_name in links:
                src_file = os.path.join(source_folder, src_name)
                dst_link = os.path.join(dest_path, dst_name)
                try:
                    if os.path.lexists(dst_link): os.remove(dst_link)
                    if os.path.exists(src_file): os.symlink(src_file, dst_link)
                except Exception as e:
                    print(f"Error linking {src_name} for {dest_folder_name}: {e}")

    # --- 4. Print Unassigned Names ---
    unassigned = expected_names - assigned_names
    if unassigned:
        print("\n" + "!" * 40)
        print("NAMES FROM LIST NOT ASSIGNED:")
        for n in sorted(unassigned):
            print(f" - {n}")
        print("!" * 40)

if __name__ == "__main__":
    main()
