daggerheart/module/data/pc.mjs

import { getPathValue, getTier } from '../helpers/utils.mjs';
import { MappingField } from './fields.mjs';

const fields = foundry.data.fields;

const attributeField = () =>
    new fields.SchemaField({
        data: new fields.SchemaField({
            value: new fields.NumberField({ initial: 0, integer: true }),
            base: new fields.NumberField({ initial: 0, integer: true }),
            bonus: new fields.NumberField({ initial: 0, integer: true }),
            actualValue: new fields.NumberField({ initial: 0, integer: true }),
            overrideValue: new fields.NumberField({ initial: 0, integer: true })
        }),
        levelMarks: new fields.ArrayField(new fields.NumberField({ nullable: true, initial: null, integer: true })),
        levelMark: new fields.NumberField({ nullable: true, initial: null, integer: true })
    });

const levelUpTier = () => ({
    attributes: new MappingField(new fields.BooleanField()),
    hitPointSlots: new MappingField(new fields.BooleanField()),
    stressSlots: new MappingField(new fields.BooleanField()),
    experiences: new MappingField(new fields.ArrayField(new fields.StringField({}))),
    proficiency: new MappingField(new fields.BooleanField()),
    armorOrEvasionSlot: new MappingField(new fields.StringField({})),
    majorDamageThreshold2: new MappingField(new fields.BooleanField()),
    severeDamageThreshold2: new MappingField(new fields.BooleanField()),
    severeDamageThreshold3: new MappingField(new fields.BooleanField()),
    severeDamageThreshold4: new MappingField(new fields.BooleanField()),
    subclass: new MappingField(
        new fields.SchemaField({
            multiclass: new fields.BooleanField(),
            feature: new fields.StringField({})
        })
    ),
    multiclass: new MappingField(new fields.BooleanField())
});

// const weapon = () => new fields.SchemaField({
//   name: new fields.StringField({}),
//   trait: new fields.StringField({}),
//   range: new fields.StringField({}),
//   damage: new fields.SchemaField({
//     value: new fields.StringField({}),
//     type: new fields.StringField({}),
//   }),
//   feature: new fields.StringField({}),
//   img: new fields.StringField({}),
//   uuid: new fields.StringField({}),
// }, { initial: null, nullable: true });

export default class DhpPC extends foundry.abstract.TypeDataModel {
    static defineSchema() {
        return {
            resources: new fields.SchemaField({
                health: new fields.SchemaField({
                    value: new fields.NumberField({ initial: 0, integer: true }),
                    min: new fields.NumberField({ initial: 0, integer: true }),
                    max: new fields.NumberField({ initial: 6, integer: true })
                }),
                stress: new fields.SchemaField({
                    value: new fields.NumberField({ initial: 0, integer: true }),
                    min: new fields.NumberField({ initial: 0, integer: true }),
                    max: new fields.NumberField({ initial: 6, integer: true })
                }),
                hope: new fields.SchemaField({
                    value: new fields.NumberField({ initial: -1, integer: true }), // FIXME. Logic is gte and needs -1 in PC/Hope. Change to 0
                    min: new fields.NumberField({ initial: 0, integer: true })
                })
            }),
            bonuses: new fields.SchemaField({
                damage: new fields.ArrayField(
                    new fields.SchemaField({
                        value: new fields.NumberField({ integer: true, initial: 0 }),
                        type: new fields.StringField({ nullable: true }),
                        initiallySelected: new fields.BooleanField(),
                        hopeIncrease: new fields.StringField({ initial: null, nullable: true }),
                        description: new fields.StringField({})
                    })
                )
            }),
            attributes: new fields.SchemaField({
                agility: attributeField(),
                strength: attributeField(),
                finesse: attributeField(),
                instinct: attributeField(),
                presence: attributeField(),
                knowledge: attributeField()
            }),
            proficiency: new fields.SchemaField({
                value: new fields.NumberField({ initial: 1, integer: true }),
                min: new fields.NumberField({ initial: 1, integer: true }),
                max: new fields.NumberField({ initial: 6, integer: true })
            }),
            damageThresholds: new fields.SchemaField({
                minor: new fields.NumberField({ initial: 0, integer: true }),
                major: new fields.NumberField({ initial: 0, integer: true }),
                severe: new fields.NumberField({ initial: 0, integer: true })
            }),
            evasion: new fields.NumberField({ initial: 0, integer: true }),
            // armor: new fields.SchemaField({
            //   value: new fields.NumberField({ initial: 0, integer: true }),
            //   customValue: new fields.NumberField({ initial: null, nullable: true }),
            // }),
            experiences: new fields.ArrayField(
                new fields.SchemaField({
                    id: new fields.StringField({ required: true }),
                    level: new fields.NumberField({ required: true, integer: true }),
                    description: new fields.StringField({}),
                    value: new fields.NumberField({ integer: true, nullable: true, initial: null })
                }),
                {
                    initial: [
                        { id: foundry.utils.randomID(), level: 1, description: '', value: 2 },
                        { id: foundry.utils.randomID(), level: 1, description: '', value: 2 }
                    ]
                }
            ),
            gold: new fields.SchemaField({
                coins: new fields.NumberField({ initial: 0, integer: true }),
                handfulls: new fields.NumberField({ initial: 0, integer: true }),
                bags: new fields.NumberField({ initial: 0, integer: true }),
                chests: new fields.NumberField({ initial: 0, integer: true })
            }),
            pronouns: new fields.StringField({}),
            domainData: new fields.SchemaField({
                maxLoadout: new fields.NumberField({ initial: 2, integer: true }),
                maxCards: new fields.NumberField({ initial: 2, integer: true })
            }),
            levelData: new fields.SchemaField({
                currentLevel: new fields.NumberField({ initial: 1, integer: true }),
                changedLevel: new fields.NumberField({ initial: 1, integer: true }),
                levelups: new MappingField(
                    new fields.SchemaField({
                        level: new fields.NumberField({ required: true, integer: true }),
                        tier1: new fields.SchemaField({
                            ...levelUpTier()
                        }),
                        tier2: new fields.SchemaField(
                            {
                                ...levelUpTier()
                            },
                            { nullable: true, initial: null }
                        ),
                        tier3: new fields.SchemaField(
                            {
                                ...levelUpTier()
                            },
                            { nullable: true, initial: null }
                        )
                    })
                )
            }),
            story: new fields.SchemaField({
                background: new fields.HTMLField(),
                appearance: new fields.HTMLField(),
                connections: new fields.HTMLField(),
                scars: new fields.ArrayField(
                    new fields.SchemaField({
                        name: new fields.StringField({}),
                        description: new fields.HTMLField()
                    })
                )
            }),
            description: new fields.StringField({}),
            //Temporary until new FoundryVersion fix --> See Armor.Mjs DataPreparation
            armorMarks: new fields.SchemaField({
                max: new fields.NumberField({ initial: 6, integer: true }),
                value: new fields.NumberField({ initial: 0, integer: true })
            })
        };
    }

    get canLevelUp() {
        //  return Object.values(this.levels.data).some(x => !x.completed);
        return this.levelData.currentLevel !== this.levelData.changedLevel;
    }

    get tier() {
        return this.#getTier(this.levelData.currentLevel);
    }

    get ancestry() {
        return this.parent.items.find(x => x.type === 'ancestry') ?? null;
    }

    get class() {
        return this.parent.items.find(x => x.type === 'class' && !x.system.multiclass) ?? null;
    }

    get multiclass() {
        return this.parent.items.find(x => x.type === 'class' && x.system.multiclass) ?? null;
    }

    get multiclassSubclass() {
        return this.parent.items.find(x => x.type === 'subclass' && x.system.multiclass) ?? null;
    }

    get subclass() {
        return this.parent.items.find(x => x.type === 'subclass' && !x.system.multiclass) ?? null;
    }

    get subclassFeatures() {
        const subclass = this.subclass;
        const multiclass = this.multiclassSubclass;
        const subclassItems = this.parent.items.filter(x => x.type === 'feature' && x.system.type === 'subclass');
        return {
            subclass: !subclass
                ? {}
                : {
                      foundation: subclassItems.filter(x =>
                          subclass.system.foundationFeature.abilities.some(ability => ability.uuid === x.uuid)
                      ),
                      specialization: subclassItems.filter(x =>
                          subclass.system.specializationFeature.abilities.some(ability => ability.uuid === x.uuid)
                      ),
                      mastery: subclassItems.filter(x =>
                          subclass.system.masteryFeature.abilities.some(ability => ability.uuid === x.uuid)
                      )
                  },
            multiclassSubclass: !multiclass
                ? {}
                : {
                      foundation: subclassItems.filter(x =>
                          multiclass.system.foundationFeature.abilities.some(ability => ability.uuid === x.uuid)
                      ),
                      specialization: subclassItems.filter(x =>
                          multiclass.system.specializationFeature.abilities.some(ability => ability.uuid === x.uuid)
                      ),
                      mastery: subclassItems.filter(x =>
                          multiclass.system.masteryFeature.abilities.some(ability => ability.uuid === x.uuid)
                      )
                  }
        };
    }

    get community() {
        return this.parent.items.find(x => x.type === 'community') ?? null;
    }

    get classFeatures() {
        return this.parent.items.filter(
            x => x.type === 'feature' && x.system.type === SYSTEM.ITEM.featureTypes.class.id && !x.system.multiclass
        );
    }

    get multiclassFeatures() {
        return this.parent.items.filter(
            x => x.type === 'feature' && x.system.type === SYSTEM.ITEM.featureTypes.class.id && x.system.multiclass
        );
    }

    get domains() {
        const classDomains = this.class ? this.class.system.domains : [];
        const multiclassDomains = this.multiclass ? this.multiclass.system.domains : [];
        return [...classDomains, ...multiclassDomains];
    }

    get domainCards() {
        const domainCards = this.parent.items.filter(x => x.type === 'domainCard');
        const loadout = domainCards.filter(x => !x.system.inVault);
        const vault = domainCards.filter(x => x.system.inVault);

        return {
            loadout: loadout,
            vault: vault,
            total: [...loadout, ...vault]
        };
    }

    get armor() {
        return this.parent.items.find(x => x.type === 'armor');
    }

    get activeWeapons() {
        const primaryWeapon = this.parent.items.find(
            x => x.type === 'weapon' && x.system.active && !x.system.secondary
        );
        const secondaryWeapon = this.parent.items.find(
            x => x.type === 'weapon' && x.system.active && x.system.secondary
        );
        return {
            primary: this.#weaponData(primaryWeapon),
            secondary: this.#weaponData(secondaryWeapon),
            burden: this.getBurden(primaryWeapon, secondaryWeapon)
        };
    }

    get inventoryWeapons() {
        const inventoryWeaponFirst = this.parent.items.find(x => x.type === 'weapon' && x.system.inventoryWeapon === 1);
        const inventoryWeaponSecond = this.parent.items.find(
            x => x.type === 'weapon' && x.system.inventoryWeapon === 2
        );
        return {
            first: this.#weaponData(inventoryWeaponFirst),
            second: this.#weaponData(inventoryWeaponSecond)
        };
    }

    get totalAttributeMarks() {
        return Object.keys(this.levelData.levelups).reduce((nr, level) => {
            const nrAttributeMarks = Object.keys(this.levelData.levelups[level]).reduce((nr, tier) => {
                nr += Object.keys(this.levelData.levelups[level][tier]?.attributes ?? {}).length * 2;

                return nr;
            }, 0);

            nr.push(...Array(nrAttributeMarks).fill(Number.parseInt(level)));

            return nr;
        }, []);
    }

    get availableAttributeMarks() {
        const attributeMarks = Object.keys(this.attributes).flatMap(y => this.attributes[y].levelMarks);
        return this.totalAttributeMarks.reduce((acc, attribute) => {
            if (!attributeMarks.findSplice(x => x === attribute)) {
                acc.push(attribute);
            }

            return acc;
        }, []);
    }

    get effects() {
        return this.parent.items.reduce((acc, item) => {
            const effects = item.system.effectData;
            if (effects && !item.system.disabled) {
                for (var key in effects) {
                    const effect = effects[key];
                    for (var effectEntry of effect) {
                        if (!acc[key]) acc[key] = [];
                        acc[key].push({ name: item.name, value: effectEntry });
                    }
                }
            }

            return acc;
        }, {});
    }

    get refreshableFeatures() {
        return this.parent.items.reduce(
            (acc, x) => {
                if (x.type === 'feature' && x.system.refreshData.type) {
                    acc[x.system.refreshData.type].push(x);
                }

                return acc;
            },
            { shortRest: [], longRest: [] }
        );
    }

    #weaponData(weapon) {
        return weapon
            ? {
                  name: weapon.name,
                  trait: CONFIG.daggerheart.ACTOR.abilities[weapon.system.trait].name, //Should not be done in data?
                  range: CONFIG.daggerheart.GENERAL.range[weapon.system.range],
                  damage: {
                      value: weapon.system.damage.value,
                      type: CONFIG.daggerheart.GENERAL.damageTypes[weapon.system.damage.type]
                  },
                  feature: CONFIG.daggerheart.ITEM.weaponFeatures[weapon.system.feature],
                  img: weapon.img,
                  uuid: weapon.uuid
              }
            : null;
    }

    prepareDerivedData() {
        this.resources.hope.max = 6 - this.story.scars.length;
        if (this.resources.hope.value >= this.resources.hope.max) {
            this.resources.hope.value = Math.max(this.resources.hope.max - 1, 0);
        }

        for (var attributeKey in this.attributes) {
            const attribute = this.attributes[attributeKey];

            attribute.levelMark = attribute.levelMarks.find(x => this.isSameTier(x)) ?? null;

            const actualValue = attribute.data.base + attribute.levelMarks.length + attribute.data.bonus;
            attribute.data.actualValue = actualValue;
            attribute.data.value = attribute.data.overrideValue
                ? attribute.data.overrideValue
                : attribute.data.actualValue;
        }

        this.evasion = this.class?.system?.evasion ?? 0;
        // this.armor.value = this.activeArmor?.baseScore ?? 0;
        this.damageThresholds = this.class?.system?.damageThresholds ?? { minor: 0, major: 0, severe: 0 };

        this.applyLevels();
        this.applyEffects();
    }

    applyLevels() {
        let healthBonus = 0,
            stressBonus = 0,
            proficiencyBonus = 0,
            evasionBonus = 0,
            armorBonus = 0,
            minorThresholdBonus = 0,
            majorThresholdBonus = 0,
            severeThresholdBonus = 0;
        let experienceBonuses = {};
        let advancementFirst = null,
            advancementSecond = null;
        for (var level in this.levelData.levelups) {
            var levelData = this.levelData.levelups[level];
            for (var tier in levelData) {
                var tierData = levelData[tier];
                if (tierData) {
                    healthBonus += Object.keys(tierData.hitPointSlots).length;
                    stressBonus += Object.keys(tierData.stressSlots).length;
                    proficiencyBonus += Object.keys(tierData.proficiency).length;
                    advancementFirst =
                        Object.keys(tierData.subclass).length > 0 && level >= 5 && level <= 7
                            ? { ...tierData.subclass[0], tier: getTier(Number.parseInt(level), true) }
                            : advancementFirst;
                    advancementSecond =
                        Object.keys(tierData.subclass).length > 0 && level >= 8 && level <= 10
                            ? { ...tierData.subclass[0], tier: getTier(Number.parseInt(level), true) }
                            : advancementSecond;

                    for (var index in Object.keys(tierData.experiences)) {
                        for (var experienceKey in tierData.experiences[index]) {
                            var experience = tierData.experiences[index][experienceKey];
                            experienceBonuses[experience] = experienceBonuses[experience]
                                ? experienceBonuses[experience] + 1
                                : 1;
                        }
                    }

                    evasionBonus += Object.keys(tierData.armorOrEvasionSlot).filter(
                        x => tierData.armorOrEvasionSlot[x] === 'evasion'
                    ).length;
                    armorBonus += Object.keys(tierData.armorOrEvasionSlot).filter(
                        x => tierData.armorOrEvasionSlot[x] === 'armor'
                    ).length;

                    majorThresholdBonus += Object.keys(tierData.majorDamageThreshold2).length * 2;
                    severeThresholdBonus += Object.keys(tierData.severeDamageThreshold2).length * 2;
                    severeThresholdBonus += Object.keys(tierData.severeDamageThreshold3).length * 3;
                    severeThresholdBonus += Object.keys(tierData.severeDamageThreshold4).length * 4;
                }
            }
        }

        this.resources.health.max += healthBonus;
        this.resources.stress.max += stressBonus;
        this.proficiency.value += proficiencyBonus;
        this.evasion += evasionBonus;
        this.armorMarks = {
            max: this.armor ? this.armor.system.marks.max + armorBonus : 0,
            value: this.armor ? this.armor.system.marks.value : 0
        };
        this.damageThresholds.minor += minorThresholdBonus;
        this.damageThresholds.major += majorThresholdBonus;
        this.damageThresholds.severe += severeThresholdBonus;

        this.experiences = this.experiences.map(x => ({ ...x, value: x.value + (experienceBonuses[x.id] ?? 0) }));

        const subclassFeatures = this.subclassFeatures;
        if (advancementFirst) {
            if (advancementFirst.multiclass) {
                this.multiclassSubclass.system[`${advancementFirst.feature}Feature`].unlocked = true;
                this.multiclassSubclass.system[`${advancementFirst.feature}Feature`].tier = advancementFirst.tier;
                subclassFeatures.multiclassSubclass[advancementFirst.feature].forEach(x => (x.system.disabled = false));
            } else {
                this.subclass.system[`${advancementFirst.feature}Feature`].unlocked = true;
                this.subclass.system[`${advancementFirst.feature}Feature`].tier = advancementFirst.tier;
                subclassFeatures.subclass[advancementFirst.feature].forEach(x => (x.system.disabled = false));
            }
        }
        if (advancementSecond) {
            if (advancementSecond.multiclass) {
                this.multiclassSubclass.system[`${advancementSecond.feature}Feature`].unlocked = true;
                this.multiclassSubclass.system[`${advancementSecond.feature}Feature`].tier = advancementSecond.tier;
                subclassFeatures.multiclassSubclass[advancementSecond.feature].forEach(
                    x => (x.system.disabled = false)
                );
            } else {
                this.subclass.system[`${advancementSecond.feature}Feature`].unlocked = true;
                this.subclass.system[`${advancementSecond.feature}Feature`].tier = advancementSecond.tier;
                subclassFeatures.subclass[advancementSecond.feature].forEach(x => (x.system.disabled = false));
            }
        }

        //General progression
        for (var i = 0; i < this.levelData.currentLevel; i++) {
            const tier = getTier(i + 1);
            if (tier !== 'tier0') {
                this.domainData.maxLoadout = Math.min(this.domainData.maxLoadout + 1, 5);
                this.domainData.maxCards += 1;
            }

            switch (tier) {
                case 'tier1':
                    this.damageThresholds.severe += 2;
                    break;
                case 'tier2':
                    this.damageThresholds.major += 1;
                    this.damageThresholds.severe += 3;
                    break;
                case 'tier3':
                    this.damageThresholds.major += 2;
                    this.damageThresholds.severe += 4;
                    break;
            }
        }
    }

    applyEffects() {
        const effects = this.effects;
        for (var key in effects) {
            const effectType = effects[key];
            for (var effect of effectType) {
                switch (key) {
                    case SYSTEM.EFFECTS.effectTypes.health.id:
                        this.resources.health.max += effect.value.valueData.value;
                        break;
                    case SYSTEM.EFFECTS.effectTypes.stress.id:
                        this.resources.stress.max += effect.value.valueData.value;
                        break;
                    case SYSTEM.EFFECTS.effectTypes.damage.id:
                        this.bonuses.damage.push({
                            value: getPathValue(effect.value.valueData.value, this),
                            type: 'physical',
                            description: effect.name,
                            hopeIncrease: effect.value.valueData.hopeIncrease,
                            initiallySelected: effect.value.initiallySelected,
                            appliesOn: effect.value.appliesOn
                        });
                }
            }
        }
    }

    getBurden(primary, secondary) {
        const twoHanded =
            primary?.system?.burden === 'twoHanded' ||
            secondary?.system?.burden === 'twoHanded' ||
            (primary?.system?.burden === 'oneHanded' && secondary?.system?.burden === 'oneHanded');
        const oneHanded =
            !twoHanded && (primary?.system?.burden === 'oneHanded' || secondary?.system?.burden === 'oneHanded');

        return twoHanded ? 'twoHanded' : oneHanded ? 'oneHanded' : null;
    }

    isSameTier(level) {
        return this.#getTier(this.levelData.currentLevel) === this.#getTier(level);
    }

    #getTier(level) {
        if (level >= 8) return 3;
        else if (level >= 5) return 2;
        else if (level >= 2) return 1;
        else return 0;
    }
}