import os
import shutil
import json


def write_field(f, key, value):
    f.write(key + ": " + value + "\n")


def parse_art_json(output_directory, filename, json_file, threed = False, animation = False):
    json_data = json.load(json_file)

    year = None

    if "date" in json_data:
        if "-" in json_data["date"]:
            year = int(json_data["date"].split('-')[0])
        else:
            year = int(json_data["date"])

    base_directory = os.path.join(art_output_directory, str(year))
    if not os.path.exists(base_directory):
        os.mkdir(base_directory)

    with open(os.path.join(base_directory, filename) + '.md', 'w') as f:
        f.write('---\n')

        if "title" in json_data:
            write_field(f, 'title', json_data["title"])

        if "date" in json_data:
            if "-" in json_data["date"]:
                write_field(f, 'date', json_data["date"])
            else:
                write_field(f, 'date', str(json_data["date"]) + '-01-01')
                write_field(f, 'excludefeed', "true")

        write_field(f, 'layout', 'art-detail')

        if threed:
            write_field(f, 'filename', '/3d/' + filename + '.glb')
            if "camera-orbit" in json_data:
                write_field(f, 'orbit', json_data["camera-orbit"])
            if "camera-target" in json_data:
                write_field(f, 'target', json_data["camera-target"])
            if "camera-fov" in json_data:
                write_field(f, 'fov', json_data["camera-fov"])
        else:
            if animation:
                write_field(f, 'filename', '/animation/' + filename + '.webm')
            else:
                write_field(f, 'filename', '/art/' + filename + '.webp')

        if "alt_text" in json_data:
            write_field(f, 'alt_text',
                        "\"" + json_data["alt_text"].replace('\n', '').replace('"', '\\"') + "\"")

        write_field(f, 'slug', filename)

        if threed:
            write_field(f, 'threed', 'true')

        if animation:
            write_field(f, 'animation', 'true')

        characters = []
        if "characters" in json_data:
            f.write("characters:\n")
            for character in json_data["characters"]:
                f.write("- " + character + "\n")
                characters.append(character)

        tags = []
        if "tags" in json_data:
            f.write("arttags:\n")
            for tag in json_data["tags"]:
                f.write("- " + tag.lower() + "\n")
                tags.append(tag.lower())

        if "nsfw" in json_data:
            write_field(f, 'nsfw', str(json_data["nsfw"]).lower())

        if "mastodon_url" in json_data:
            write_field(f, 'mastodon_url', json_data["mastodon_url"])

        if "pixiv_url" in json_data:
            write_field(f, 'pixiv_url', json_data["pixiv_url"])

        if "newgrounds_url" in json_data:
            write_field(f, 'newgrounds_url', json_data["newgrounds_url"])

        if "program" in json_data:
            write_field(f, 'program', json_data["program"])

        f.write('---\n')

        if "description" in json_data:
            f.write(json_data["description"])
            f.write('\n')

        return (year, characters, tags)


art_data_directory = '../art'
threed_data_directory = '../3d'
animation_data_directory = '../animation'
art_output_directory = '../content/art'

shutil.rmtree(art_output_directory)
os.mkdir(art_output_directory)

collected_years = set()
year_stats = {}
total_art = 0
character_stats = {}
tag_stats = {}
new_banner = ""
comissions_enabled = False

for filename in os.listdir(art_data_directory):
    f = os.path.join(art_data_directory, filename)

    if os.path.isfile(f):
        filename_without_ext = os.path.splitext(filename)[0]

        if filename_without_ext == ".DS_Store":
            continue

        with open(f, "r") as file:
            year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file)

            if year in year_stats:
                year_stats[year] += 1
            else:
                year_stats[year] = 1

            for character in characters:
                if character in character_stats:
                    character_stats[character] += 1
                else:
                    character_stats[character] = 1

            for tag in tags:
                if tag in tag_stats:
                    tag_stats[tag] += 1
                else:
                    tag_stats[tag] = 1

            collected_years.add(year)
            total_art += 1

for filename in os.listdir(threed_data_directory):
    f = os.path.join(threed_data_directory, filename)

    if os.path.isfile(f):
        filename_without_ext = os.path.splitext(filename)[0]

        with open(f, "r") as file:
            year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file, True)

            if year in year_stats:
                year_stats[year] += 1
            else:
                year_stats[year] = 1

            for character in characters:
                if character in character_stats:
                    character_stats[character] += 1
                else:
                    character_stats[character] = 1

            for tag in tags:
                if tag in tag_stats:
                    tag_stats[tag] += 1
                else:
                    tag_stats[tag] = 1

            collected_years.add(year)
            total_art += 1

for filename in os.listdir(animation_data_directory):
    f = os.path.join(animation_data_directory, filename)

    if os.path.isfile(f):
        filename_without_ext = os.path.splitext(filename)[0]

        with open(f, "r") as file:
            year, characters, tags = parse_art_json(art_output_directory, filename_without_ext, file, False, True)

            if year in year_stats:
                year_stats[year] += 1
            else:
                year_stats[year] = 1

            for character in characters:
                if character in character_stats:
                    character_stats[character] += 1
                else:
                    character_stats[character] = 1

            for tag in tags:
                if tag in tag_stats:
                    tag_stats[tag] += 1
                else:
                    tag_stats[tag] = 1

            collected_years.add(year)
            total_art += 1

for year in collected_years:
    with open(os.path.join(art_output_directory, str(year), '_index.md'), 'w') as f:
        f.write('---\n')

        write_field(f, 'title', str(year))
        write_field(f, 'layout', 'gallery')
        write_field(f, 'json', 'art')
        write_field(f, 'thumbnails', 'true')
        write_field(f, 'selectedyear', str(year))

        f.write('---\n')

with open('../data/art-config.json', 'r') as f:
    json_data = json.load(f)

    featured_pieces = json_data["featured"]

    if "new-banner" in json_data:
        new_banner = json_data["new-banner"]

    comissions_enabled = json_data["commissions"]

with open(art_output_directory + '/_index.md', 'w') as f:
    f.write('---\n')

    write_field(f, 'title', 'Art')
    write_field(f, 'layout', 'art')

    write_field(f, 'summary', 'My personal art gallery.')

    write_field(f, 'new_banner', new_banner)
    write_field(f, 'commissions', str(comissions_enabled).lower())

    f.write('aliases:\n')
    f.write('- /gallery\n')

    f.write('featured:\n')
    for piece_name in featured_pieces:
        piece_path = os.path.join(art_data_directory, piece_name + ".json")
        with open(piece_path, 'r') as piecef:
            json_data = json.load(piecef)

            f.write("- filename: " + piece_name + '.webp\n')
            f.write('  date: ' + json_data["date"] + '\n')
            f.write('  title: ' + json_data["title"] + '\n')

    f.write('years:\n')
    for year in reversed(list(collected_years)):
        f.write('- ' + str(year) + '\n')

    f.write('---\n')

os.mkdir(art_output_directory + "/stats")

with open(art_output_directory + '/stats/_index.md', 'w') as f:
    f.write('---\n')

    write_field(f, 'title', 'Stats')
    write_field(f, 'layout', 'art-stats')

    write_field(f, 'total', str(total_art))

    f.write('years:\n')
    for year, num in reversed(sorted(year_stats.items(), key=lambda t: t[1])):
        f.write('- year: ' + str(year) + '\n')
        f.write('  num: ' + str(num) + '\n')

    f.write('characters:\n')
    for name, num in sorted(character_stats.items(), key=lambda t: t[1], reverse=True)[:10]:
        f.write('- name: ' + str(name) + '\n')
        f.write('  num: ' + str(num) + '\n')

    f.write('tags:\n')
    for name, num in sorted(tag_stats.items(), key=lambda t: t[1]):
        f.write('- name: ' + str(name) + '\n')
        f.write('  num: ' + str(num) + '\n')

    f.write('---\n')