import os
import json

flags_dir = os.path.join("flags", "1x1")

files = []
for (dirpath, dirnames, filenames) in os.walk(flags_dir):
    files.extend(filenames)
    break

file_codes = [name.replace('.svg', '') for name in files]

country_json = open("country.json")
flags = json.load(country_json)
flags.sort(key=lambda x: x["code"])

with open('country.json', 'w') as output:
    json.dump(flags, output, indent=2, sort_keys=True)

# Check if all files have names
country_codes = [flag["code"] for flag in flags]

all_good = True

for code in file_codes:
  if code not in country_codes:
    print('Code not found in country.json:', code)
    all_good = False

# Check if all countries have files
for code in country_codes:
  if code not in file_codes:
    print('Flag icon not found for:', code)
    all_good = False


if all_good:
  print('All flag icons and country.json are in sync.')