VMF Mod Data: refactoring, adding modification protection

vmf/scripts/mods/vmf/modules/core/mutators/mutators_manager.lua

@@ -48,7 +48,7 @@ end
 
 -- Called after mutator is enabled
 local function on_enabled(mutator)
-  local config = mutator:get_config()
+  local config = mutator:get_internal_data("mutator_config")
   dice_manager.addDice(config.dice)
   set_lobby_data()
   print("[MUTATORS] Enabled " .. mutator:get_name() .. " (" .. tostring(get_index(_mutators, mutator)) .. ")")
@@ -59,7 +59,7 @@ end
 
 -- Called after mutator is disabled
 local function on_disabled(mutator, initial_call)
-  local config = mutator:get_config()
+  local config = mutator:get_internal_data("mutator_config")
 
   -- All mutators run on_disabled on initial call, so there's no need to remove dice and set lobby data
   if not initial_call then
@@ -138,7 +138,7 @@ end
 local function mutator_can_be_enabled(mutator)
 
   -- If conflicting mutators are enabled
-  local mutator_compatibility_config = mutator:get_config().compatibility
+  local mutator_compatibility_config = mutator:get_internal_data("mutator_config").compatibility
   local is_mostly_compatible = mutator_compatibility_config.is_mostly_compatible
   local except = mutator_compatibility_config.except
   for _, other_mutator in ipairs(_mutators) do
@@ -189,7 +189,7 @@ end
 -- Adds mutator names from enable_these_after to the list of mutators that should be enabled after the mutator_name
 local function update_mutators_sequence(mutator)
 
-  local raw_config = mutator:get_config().raw_config
+  local raw_config = mutator:get_internal_data("mutator_config").raw_config
   local enable_before_these = raw_config.enable_before_these
   local enable_after_these = raw_config.enable_after_these
   local mutator_name = mutator:get_name()
@@ -225,8 +225,8 @@ end
 
 -- Uses raw_config to determine if mutators are compatible both ways
 local function is_compatible(mutator, other_mutator)
-  local raw_config = mutator:get_config().raw_config
+  local raw_config = mutator:get_internal_data("mutator_config").raw_config
-  local other_raw_config = other_mutator:get_config().raw_config
+  local other_raw_config = other_mutator:get_internal_data("mutator_config").raw_config
 
   local mutator_name = mutator:get_name()
   local other_mutator_name = other_mutator:get_name()
@@ -274,7 +274,7 @@ end
 local function update_compatibility(mutator)
 
   -- Create default 'compatibility' entry
-  local config = mutator:get_config()
+  local config = mutator:get_internal_data("mutator_config")
   config.compatibility = {}
   local compatibility = config.compatibility
 
@@ -287,7 +287,7 @@ local function update_compatibility(mutator)
 
   for _, other_mutator in ipairs(_mutators) do
 
-    local other_config = other_mutator:get_config()
+    local other_config = other_mutator:get_internal_data("mutator_config")
     local other_mostly_compatible = other_config.compatibility.is_mostly_compatible
     local other_except = other_config.compatibility.except
 
@@ -338,9 +338,9 @@ local function initialize_mutator_config(mutator, _raw_config)
   end
   if raw_config.short_title == "" then raw_config.short_title = nil end
 
-  mutator._data.config = {}
+  rawset(mutator._data, "mutator_config", {})
 
-  local config = mutator._data.config
+  local config = mutator:get_internal_data("mutator_config")
 
   config.dice            = raw_config.dice
   config.short_title     = raw_config.short_title
@@ -376,7 +376,7 @@ function vmf.add_mutator_titles_to_string(mutators, separator, is_short)
   local replace = nil
 
   for _, mutator in ipairs(mutators) do
-    local config = mutator:get_config()
+    local config = mutator:get_internal_data("mutator_config")
     local added_name = (is_short and config.short_title or mutator:get_readable_name())
     if config.title_placement == "before" then
       if before then
@@ -485,7 +485,7 @@ end
 -- Removes all raw_configs which won't be used anymore
 function vmf.mutators_delete_raw_config()
   for _, mutator in ipairs(_mutators) do
-    mutator:get_config().raw_config = nil
+    mutator:get_internal_data("mutator_config").raw_config = nil
   end
 end
 

vmf/scripts/mods/vmf/modules/core/toggling.lua

@@ -8,7 +8,7 @@ local _disabled_mods = vmf:get("disabled_mods_list") or {}
 
 vmf.set_mod_state = function (mod, is_enabled, initial_call)
 
-  mod._data.is_enabled = is_enabled
+  rawset(mod._data, "is_enabled", is_enabled)
 
   if is_enabled then
     mod:enable_all_hooks()

vmf/scripts/mods/vmf/modules/ui/mutators/mutators_gui.lua

@@ -125,7 +125,7 @@ local function offset_function_callback(ui_scenegraph_, style, content, ui_rende
   -- Find out if mutator can be enabled.
   local can_be_enabled = true
 
-  local mutator_compatibility_config = mutator:get_config().compatibility
+  local mutator_compatibility_config = mutator:get_internal_data("mutator_config").compatibility
   local is_mostly_compatible = mutator_compatibility_config.is_mostly_compatible
   local except = mutator_compatibility_config.except
 

vmf/scripts/mods/vmf/modules/vmf_mod_data.lua

@@ -0,0 +1,53 @@
+-- Defining VMFMod class.
+VMFMod = class(VMFMod)
+
+-- Creating mod data table when object of vmf mod is instantiated.
+function VMFMod:init(mod_name)
+  self._data = {}
+  setmetatable(self._data, {
+    __newindex = function(t_, k)
+      self:warning("Attempt to change internal mod data value (\"%s\"). Changing internal mod data is forbidden.", k)
+    end
+  })
+  rawset(self._data, "name",          mod_name)
+  rawset(self._data, "readable_name", mod_name)
+  rawset(self._data, "is_enabled",    true)
+  rawset(self._data, "is_togglable",  false)
+  rawset(self._data, "is_mutator",    false)
+end
+
+-- #####################################################################################################################
+-- ##### VMFMod ########################################################################################################
+-- #####################################################################################################################
+
+--[[
+  Universal function for getting any internal mod data. Returned table values shouldn't be modified, because it lead to
+  unexpected VMF behaviour.
+  * key [string]: data entry name
+--]]
+function VMFMod:get_internal_data(key)
+  return self._data[key]
+end
+
+
+--[[
+  Predefined functions for getting specific internal mod data.
+--]]
+function VMFMod:get_name()
+  return self._data.name
+end
+function VMFMod:get_readable_name()
+  return self._data.readable_name
+end
+function VMFMod:get_description()
+  return self._data.description
+end
+function VMFMod:is_enabled()
+  return self._data.is_enabled
+end
+function VMFMod:is_togglable()
+    return self._data.is_togglable
+end
+function VMFMod:is_mutator()
+  return self._data.is_mutator
+end

vmf/scripts/mods/vmf/modules/vmf_mod_manager.lua

@@ -17,16 +17,9 @@ local function create_mod(mod_name)
 
   table.insert(_mods_unloading_order, 1, mod_name)
 
-  local mod = VMFMod:new()
+  local mod = VMFMod:new(mod_name)
   _mods[mod_name] = mod
 
-  mod._data = {}
-  mod._data.name = mod_name
-  mod._data.readable_name = mod_name
-  mod._data.is_enabled = true
-  mod._data.is_togglable = false
-  mod._data.is_mutator = false
-
   return mod
 end
 
@@ -96,42 +89,6 @@ function get_mod(mod_name)
   return _mods[mod_name]
 end
 
--- #####################################################################################################################
--- ##### VMFMod ########################################################################################################
--- #####################################################################################################################
-
-VMFMod = class(VMFMod)
-
--- DATA
-
-function VMFMod:get_name()
-  return self._data.name
-end
-
-function VMFMod:get_readable_name()
-  return self._data.readable_name
-end
-
-function VMFMod:get_description()
-  return self._data.description
-end
-
-function VMFMod:is_enabled()
-  return self._data.is_enabled
-end
-
-function VMFMod:is_togglable()
-    return self._data.is_togglable
-end
-
-function VMFMod:is_mutator()
-  return self._data.is_mutator
-end
-
-function VMFMod:get_config()
-  return self._data.config
-end
-
 -- #####################################################################################################################
 -- ##### VMF Initialization ############################################################################################
 -- #####################################################################################################################
@@ -152,11 +109,11 @@ function vmf.initialize_mod_data(mod, mod_data)
   end
 
   if mod_data.name then
-    mod._data.readable_name = mod_data.name
+    rawset(mod._data, "readable_name", mod_data.name)
   end
-  mod._data.description  = mod_data.description
+  rawset(mod._data, "description",  mod_data.description)
-  mod._data.is_togglable = mod_data.is_togglable or mod_data.is_mutator
+  rawset(mod._data, "is_togglable", mod_data.is_togglable or mod_data.is_mutator)
-  mod._data.is_mutator   = mod_data.is_mutator
+  rawset(mod._data, "is_mutator",   mod_data.is_mutator)
 
   if mod_data.is_mutator then
     vmf.register_mod_as_mutator(mod, mod_data.mutator_settings)

vmf/scripts/mods/vmf/vmf_loader.lua

@@ -11,6 +11,7 @@ local vmf_mod_object = {}
 -- #####################################################################################################################
 
 function vmf_mod_object:init()
+	dofile("scripts/mods/vmf/modules/vmf_mod_data")
 	dofile("scripts/mods/vmf/modules/vmf_mod_manager")
 	dofile("scripts/mods/vmf/modules/core/safe_calls")
 	dofile("scripts/mods/vmf/modules/core/events")