added disease

Author: Zt-freak

castor.json

@@ -2,7 +2,7 @@
   "ArchiveName": "zzz-ArluqTools",
   "RepoName": "ZtModArchive/ArluqTools",
   "Author": "Apodemus",
-  "Version": "v2.2",
+  "Version": "v2.3",
   "Type": "package",
   "License": "MIT License",
   "Description": "This is a lua library that is an abstraction  layer around ZT2's libraries.",

scripts/services/AnimalService.lua

@@ -167,6 +167,29 @@ function AnimalService.delete (animal)
     deleteEntity(animal)
 end
 
+---- DISEASE ----
+
+--- Get if animal has a disease
+--- @param animal animal
+--- @return bool
+function AnimalService.isDiseased (animal)
+    return animal:BFG_GET_ATTR_BOOLEAN("b_Disease")
+end
+
+
+--- Give animal a disease
+--- @param animal animal
+--- @param diseaseName string
+function AnimalService.giveDisease (animal, diseaseName)
+    local diseaseManager = queryObject("ZTDiseaseMgr")
+    local animalName = getName(animal)
+    local diseaseSeed = {
+        key = diseaseName,
+        val = animalName
+    }
+    diseaseManager:ZT_FORCE_DISEASE(diseaseSeed)
+end
+
 ---- ENVIROMENT ----
 
 --- Get animal enviroment suitability