import json
import string

def getName(user : str):
    with open("characters.json", "r") as f:
        data = json.load(f)
    
    if user in data:
        return data[user]["Name"]
    else:
        return user

def setUpDict(cmd : dict):
    keys = list(cmd)
    values = list(cmd.values())
    result = ""
    for x in range(len(keys)):
        if x%3 != 2:
            result += keys[x] + ": " + str(values[x]) + "\t"
        else:
            result += keys[x] + ": " + str(values[x]) + "\n"
    return result

def lookUp(data : dict, key : str, cmd : str = ""):
    if key in data:
        if cmd == "":
            return data[key]
        else:
            while cmd[0] == ' ':
                cmd = cmd[1:]
            try:
                cmd = type(data[key])(cmd)
                data[key] = cmd
                return data
            except:
                return "Wrong data type"

def charData(user : str, key : str,cmd : str = ""):
    with open("characters.json", "r") as f:
        data = json.load(f)

    key = string.capwords(key)

    if user in data:
        if key in data[user]:
            if type(data[user][key]) is dict:
                if cmd == "":
                    return setUpDict(data[user][key])
                else:
                    newKey = cmd.split(" ")[0]
                    newcmd = cmd[len(newKey):]

                    lookUpResult = lookUp(data[user][key],newKey,newcmd)

                    if type(lookUpResult) is dict:
                        data[user][key] = lookUpResult
                        with open("characters.json", "w") as f:
                            json.dump(data,f,indent = 4)
                        return "Changed " + data[user]["Name"] + "'s " + key
                    else:
                        return lookUpResult
            else:
                if cmd == "":
                    return data[user][key]
                else:
                    data[user][key] = cmd
                    with open("characters.json", "w") as f:
                            json.dump(data,f,indent = 4)
                    return "Changed " + data[user]["Name"] + "'s " + key +" to " + cmd
        else:
            return "Couldn't find that data. Are you sure you spelled it correctly?"
    else:
        return "You don't have a character. You can make one with !swchar"