attackTrigger: call scripts when a unit attacks another with weapons of a given type or weapons of a given material.

scripts/attackTrigger.lua

@@ -0,0 +1,141 @@
+--attackTrigger.lua
+--author expwnent
+--triggers scripts when a unit attacks another with a weapon type, a weapon of a particular material
+
+local eventful = require 'plugins.eventful'
+eventful.enableEvent(eventful.eventType.UNIT_ATTACK,1)
+
+itemTriggers = itemTriggers or {}
+materialTriggers = materialTriggers or {}
+itemMaterialTriggers = itemMaterialTriggers or {}
+
+local function processTrigger(command, attacker, defender)
+ local command2 = {}
+ for i,arg in ipairs(command) do
+  if arg == '\\ATTACKER' then
+   command2[i] = '' .. attacker.id
+  elseif arg == '\\DEFENDER' then
+   command2[i] = '' .. defender.id
+  else
+   command2[i] = arg
+  end
+ end
+ print(dfhack.run_command(table.unpack(command2)))
+end
+
+eventful.onUnitAttack.attackTrigger = function(attacker,defender,wound)
+ attacker = df.unit.find(attacker)
+ defender = df.unit.find(defender)
+ 
+ if not attacker then
+  return
+ end
+ 
+ local attackerWeapon
+ for _,item in ipairs(attacker.inventory) do
+  if item.mode == df.unit_inventory_item.T_mode.Weapon then
+   attackerWeapon = item.item
+   break
+  end
+ end
+ 
+ if not attackerWeapon then
+  return
+ end
+ 
+ local weaponType = attackerWeapon.subtype.subtype --attackerWeapon.subtype.id
+ 
+ if itemTriggers[weaponType] then
+  for _,command in pairs(itemTriggers[weaponType]) do
+   processTrigger(command,attacker,defender)
+  end
+ end
+ 
+-- if materialTriggers[attackerWeapon.mat_type] and materialTriggers[attackerWeapon.mat_type][attackerWeapon.mat_index] then
+ if materialTriggers[attackerWeapon.mat_index] then
+  for _,command in pairs(materialTriggers[attackerWeapon.mat_index]) do
+   processTrigger(command,attacker,defender)
+  end
+ end
+ 
+-- if itemMaterialTriggers[weaponType] and itemMaterialTriggers[weaponType][attackerWeapon.mat_type] and itemMaterialTriggers[weaponType][attackerWeapon.mat_type][attackerWeapon.mat_index] then
+--  for _,command in pairs(itemMaterialTriggers[weaponType][attackerWeapon.mat_type][attackerWeapon.mat_index]) do
+--   processTrigger(command,attacker,defender)
+--  end
+-- end
+end
+
+local args = {...}
+
+local i = 1
+local command
+local weaponType
+local material
+while i <= #args do
+ if command then
+  table.insert(command, args[i])
+  i = i+1
+ else
+  if args[i] == '-weaponType' then
+   weaponType = args[i+1]
+   i = i+2
+  elseif args[i] == '-material' then
+   material = args[i+1]
+   i = i+2
+  elseif args[i] == '-command' then
+   command = {}
+   i = i+1
+  elseif args[i] == '-clear' then
+   itemTriggers = {}
+   materialTriggers = {}
+   i = i+1
+  else
+   error('Invalid arguments to attackTrigger.lua.')
+  end
+ end
+end
+
+if weaponType then
+ local temp
+ for _,itemdef in ipairs(df.global.world.raws.itemdefs.weapons) do
+  if itemdef.id == weaponType then
+   temp = itemdef.subtype
+   break
+  end
+ end
+ if not temp then
+  error 'Could not find weapon type.'
+ end
+ weaponType = temp
+end
+
+if material then
+ local i = 0
+ while true do
+  local mat = dfhack.matinfo.decode(0,i)
+  if not mat then
+   error 'Could not find material.'
+  end
+  if mat.inorganic.id == material then
+   material = mat.index
+   break
+  end
+  i = i+1
+ end
+end
+
+if material then
+ if not materialTriggers[material] then
+  materialTriggers[material] = {}
+ end
+ table.insert(materialTriggers[material],command)
+-- table.insert(materialTriggers[material],function() 
+--  print(dfhack.run_command(command))
+-- end)
+elseif weaponType then
+ if not itemTriggers[weaponType] then
+  itemTriggers[weaponType] = {}
+ end
+ table.insert(itemTriggers[weaponType],command)
+end
+