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 = os.path.abspath(sys.argv[2])
    target_subdir = os.path.join(work_dir, "A Rendre")

    os.makedirs(target_subdir, exist_ok=True)

    pattern = re.compile(r"^Copie(\d+)\.json$")
    copies_map = defaultdict(list)

    # 1. Collect all data
    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}")

    # 2. Check constraints and symlink files
    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}")

            # Determine source: must contain both files
            source_folder = None

            # Check Bnot first
            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
            # Fallback to Anot
            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:
                print(f"Skipping ID {copie_id}: Files missing in both Anot and {bnot_dir}.")
                continue

            # Create destination directory
            dest_folder_name = safe_name if len(ids) == 1 else f"{safe_name} ({copie_id})"
            dest_path = os.path.join(target_subdir, dest_folder_name)
            os.makedirs(dest_path, exist_ok=True)

            print(f"Linking '{source_folder}' -> '{dest_path}'")

            # Link configuration: (source_filename, dest_filename)
            links = [
                ("Concat.jpg", f"{safe_name}.jpg"),
                ("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)
                    os.symlink(src_file, dst_link)
                except Exception as e:
                    print(f"Error linking {src_name} for {dest_folder_name}: {e}")

if __name__ == "__main__":
    main()