civ7-modding-tools/src/nodes/ModifierRequirementNode.ts

Mod generation tool for Civilization 7.

import { TObjectValues } from "../types";
import { REQUIREMENT } from "../constants";

import { BaseNode } from "./BaseNode";
import { ArgumentNode, TArgumentNode } from "./ArgumentNode";
import { TModifierNode } from "./ModifierNode";

export type TModifierRequirementNode = Pick<ModifierRequirementNode,
    "type" |
    "arguments"
>;

export class ModifierRequirementNode extends BaseNode<TModifierRequirementNode> {
    _name = 'Requirement';

    type: TObjectValues<typeof REQUIREMENT> | null = REQUIREMENT.PLAYER_HAS_CIVILIZATION_OR_LEADER_TRAIT;
    arguments: TArgumentNode[] = [];

    constructor(payload: Partial<TModifierRequirementNode> = {}) {
        super();
        this.fill(payload);
    }

    fill = (payload: Partial<TModifierNode> = {}) => {
        for (const [key, value] of Object.entries(payload)) {
            if (this.hasOwnProperty(key)) {
                this[key] = value;
            }
        }
        return this;
    }

    toXmlElement() {
        return {
            _name: this._name,
            _attrs: {
                type: this.type,
            },
            _content: this.arguments.map(item => new ArgumentNode(item).toXmlElement())
        }
    }
}