mods/my_mod/rules/tesla_trooper.yaml

tesla_trooper:
  kind: infantry
  inherits: ^infantry_base
  tags: [elite, coalition_common]
  traits:
    info:
      name: Tesla Trooper
    health:
      max: 80
      armor: flak
    mobile:
      speed: 3.0
      locomotor: foot
    turret:
      weapon: ::tesla_gun
    buildable:
      cost: 500
      time: "10s"
      built_at: [tag:barracks]

tesla_gun:
  kind: weapon
  damage: 50
  cooldown: "0.5s"
  range: 6.0
  projectile: ::tesla_bolt
  warhead: ::tesla_aoe

tesla_aoe:
  kind: warhead
  spread: 3.0
  falloff: linear
  versus:
    none:     100%

  

id: shield_lib
name: Shield Library
version: "1.0.0"
framework: true
imports:
- traits.yaml
scripts:
- scripts/shield.kts

mods/shield_lib/traits.yaml

shield:
  kind: trait
  description: Wraps the carrier in an energy shield that absorbs incoming damage before health is reduced, then recharges after a delay.
  script: scripts/shield.kts
  applies_to: [structure, vehicle]
  params:
    strength:
      type: Int
      required: true
      description: Maximum hit-points of the shield.
    recharge_delay:
      type: Duration
      default: "3s"
      description: Time after the shield is fully depleted before it begins recharging.

trait.on("engine:attached") { ctx, _ ->
    ctx.vars["shield_hp"] = ctx.params.int("strength")
}


trait.on("core:damaging", priority = -10) { ctx, ev ->
    val shieldHp = ctx.vars.int("shield_hp")
    if (shieldHp == 0) return@on    

    val amount = ev.int("amount")
    val absorbed = if (shieldHp < amount) shieldHp else amount
    ctx.vars["shield_hp"] = shieldHp - absorbed
    ev["amount"] = amount - absorbed  

    if (ctx.vars.int("shield_hp") == 0) {
        ctx.emit("shield_broken")  
        val timer = ctx.schedule(ctx.params.duration("recharge_delay"), "recharge")
        ctx.vars["recharge_timer"] = timer
    }
}

trait.on("shield_lib:recharge") { ctx, _ ->
    ctx.vars["shield_hp"] = ctx.params.int("strength")
}