import csv
import json


def mapType(creatureType):
    """
    maps the slayer's pit type to the foundry system type
    """
    switcher = {
        'Tiere': 'animal',
        'Humanoide': 'humanoid',
        'Konstrukte': 'construct',
        'Magische Wesen': 'magicalEntity',
        'Pflanzenwesen': 'plantBeing',
        'Untote': 'undead'
    }
    return switcher.get(creatureType)


def mapSize(size):
    """
    maps slayer's pits size to foundry system size
    """
    switcher = {
        'wi': 'tiny',
        'kl': 'small',
        'no': 'normal',
        'gr': 'large',
        'ri': 'huge',
        'ge': 'colossal'
    }
    return switcher.get(size)


def reformatCreature(creature):
    """
    reformats from table form into expected foundry style dict
    """
    magic = 0 if (creature['Zauber'] == '') else creature['Zauber']
    targetMagic = 0 if creature['Zielzauber'] == '' else creature['Zielzauber']
    size = mapSize(creature['GK'])
    creatureType = mapType(creature['Gruppe'])
    return {
        'name': creature['Name'],
        'type': 'creature',
        'data': {
            "attributes": {
                "body": {
                    "base": creature['KÖR'],
                    "mod": 0
                },
                "mobility": {
                    "base": creature['AGI'],
                    "mod": 0
                },
                "mind": {
                    "base": creature['GEI'],
                    "mod": 0
                }
            },
            "traits": {
                "strength": {
                    "base": creature['ST'],
                    "mod": 0
                },
                "constitution": {
                    "base": creature['HÄ'],
                    "mod": 0
                },
                "agility": {
                    "base": creature['BE'],
                    "mod": 0
                },
                "dexterity": {
                    "base": creature['GE'],
                    "mod": 0
                },
                "intellect": {
                    "base": creature['VE'],
                    "mod": 0
                },
                "aura": {
                    "base": creature['AU'],
                    "mod": 0
                }
            },
            "combatValues": {
                "hitPoints": {
                    "base": creature['Lebenskraft'],
                    "mod": 0,
                    "value": 0
                },
                "defense": {
                    "base": creature['Abwehr'],
                    "mod": 0
                },
                "initiative": {
                    "base": creature['Initiative'],
                    "mod": 0
                },
                "movement": {
                    "base": creature['Laufen'],
                    "mod": 0
                },
                "meleeAttack": {
                    "base": creature['Schlagen'],
                    "mod": 0
                },
                "rangedAttack": {
                    "base": creature['Schiessen'],
                    "mod": 0
                },
                "spellcasting": {
                    "base": magic,
                    "mod": 0
                },
                "targetedSpellcasting": {
                    "base": targetMagic,
                    "mod": 0
                }
            },
            "baseInfo": {
                "loot": "",
                "foeFactor": creature['GH'],
                "creatureType": creatureType,
                "sizeCategory": size,
                "experiencePoints": creature['EP'],
                "description": creature['Talente']
            }
        }
    }


def decode(filename, source):
    creatures = []
    npcs = []

    with open(filename, newline='\n') as csvFile:
        creatureReader = csv.DictReader(csvFile)
        for row in creatureReader:
            if row['Quelle'] == source:
                if row['Volk'] == '':
                    creatures.append(row)
                else:
                    npcs.append(row)

    return (creatures, npcs)


creatures, npcs = decode('Bestiarium.csv', 'Dungeonslayers Basisbox')

with open('bestiarium.json', 'w') as outfile:
    mappedCreatures = list(map(reformatCreature, creatures))
    json.dump(mappedCreatures, outfile)