import { SYSTEM } from "../config/system.mjs"

export default class OathHammerRegiment extends foundry.abstract.TypeDataModel {
  static defineSchema() {
    const { fields } = foundry.data
    const requiredInteger = { required: true, nullable: false, integer: true }
    const schema = {}

    schema.description = new fields.HTMLField({ required: false, nullable: true, initial: "" })
    schema.notes        = new fields.StringField({ required: false, nullable: true, initial: "" })

    schema.grit = new fields.SchemaField({
      max: new fields.NumberField({ ...requiredInteger, initial: 20, min: 0, max: 200 }),
    })

    schema.armorDice = new fields.SchemaField({
      value:        new fields.NumberField({ ...requiredInteger, initial: 1, min: 0, max: 20 }),
      colorDiceType: new fields.StringField({ required: true, nullable: false, initial: "white", choices: SYSTEM.DICE_COLOR_TYPES }),
    })

    schema.movement = new fields.NumberField({ ...requiredInteger, initial: 60, min: 0, max: 500 })

    // Embedded skill rows: [{name, value, colorDiceType}]
    schema.skills = new fields.ArrayField(new fields.SchemaField({
      name:          new fields.StringField({ required: true, nullable: false, initial: "" }),
      value:         new fields.NumberField({ ...requiredInteger, initial: 2, min: 0, max: 6 }),
      colorDiceType: new fields.StringField({ required: true, nullable: false, initial: "white", choices: SYSTEM.DICE_COLOR_TYPES }),
    }))

    // Embedded attack rows: [{name, damageDice, colorDiceType, ap, special}]
    schema.attacks = new fields.ArrayField(new fields.SchemaField({
      name:          new fields.StringField({ required: true, nullable: false, initial: "" }),
      damageDice:    new fields.NumberField({ ...requiredInteger, initial: 6, min: 0, max: 20 }),
      colorDiceType: new fields.StringField({ required: true, nullable: false, initial: "white", choices: SYSTEM.DICE_COLOR_TYPES }),
      ap:            new fields.NumberField({ ...requiredInteger, initial: 0, min: 0, max: 6 }),
      special:       new fields.StringField({ required: false, nullable: true, initial: "" }),
    }))

    // Embedded trait rows: [{name, description}]
    schema.traits = new fields.ArrayField(new fields.SchemaField({
      name:        new fields.StringField({ required: true, nullable: false, initial: "" }),
      description: new fields.StringField({ required: false, nullable: true, initial: "" }),
    }))

    return schema
  }

  static LOCALIZATION_PREFIXES = ["OATHHAMMER.Regiment"]

  get colorEmoji() {
    return { white: "⬜", red: "🔴", black: "⬛" }[this.armorDice.colorDiceType] ?? "⬜"
  }

  get armorLabel() {
    return `${this.colorEmoji} ${this.armorDice.value}d`
  }
}