import json
import os

from untitled import model

SAVE_FOLDER = os.path.join("untitled", "saves")


def save(save: model.Save, folder=None):
    folder = folder or SAVE_FOLDER
    file_name = save.cat.name + ".kitten"
    save_file = os.path.join(folder, file_name)

    data = save.to_dict()

    os.makedirs(folder, exist_ok=True)

    with open(save_file, "w") as f:
        json.dump(data, f)


def load(name, folder=None):
    folder = folder or SAVE_FOLDER
    file_name = name + ".kitten"
    save_file = os.path.join(folder, file_name)

    with open(save_file) as f:
        data = json.load(f)

    save = model.Save.from_dict(data)

    return save