From 84b98654f477efee4efaa5dc3f3ae0eb25916a97 Mon Sep 17 00:00:00 2001
From: Guy Moreau <gmyx@gpws.ca>
Date: Thu, 26 Dec 2019 15:01:06 -0500
Subject: [PATCH] add support for modules

---
 prototypes/entity/assembling-machines.lua | 9 +++++++++
 prototypes/item/assembling-machines.lua   | 7 +++++++
 2 files changed, 16 insertions(+)

diff --git a/prototypes/entity/assembling-machines.lua b/prototypes/entity/assembling-machines.lua
index 85efad6..d913f2d 100644
--- a/prototypes/entity/assembling-machines.lua
+++ b/prototypes/entity/assembling-machines.lua
@@ -55,6 +55,15 @@ local rocketAssembling = util.table.deepcopy(assemblingMachine1)
 rocketAssembling.name = "rocket-assembling-machine"
 rocketAssembling.localised_name = nil
 
+--support for modules
+rocketAssembling.module_specification = {module_slots = 4}
+rocketAssembling.allowed_effects = {
+  "consumption",
+  "speed",
+  "productivity",
+  "pollution"
+}
+
 rocketAssembling.icon      = data.raw["item"][rocketAssembling.name].icon
 rocketAssembling.icon_size = data.raw["item"][rocketAssembling.name].icon_size
 rocketAssembling.icons     = util.table.deepcopy(data.raw["item"][rocketAssembling.name].icons)
diff --git a/prototypes/item/assembling-machines.lua b/prototypes/item/assembling-machines.lua
index 94e9ef2..d9fbf96 100644
--- a/prototypes/item/assembling-machines.lua
+++ b/prototypes/item/assembling-machines.lua
@@ -4,6 +4,13 @@ data:extend{{
   icon = "__MoreScience__/graphics/icons/assembling-machine-4.png",
   icon_size = 64,
   --flags = {},
+  module_specification = {module_slots = 4},
+  allowed_effects = {
+        "consumption",
+        "speed",
+        "productivity",
+        "pollution"
+  },
   subgroup = "ms-science-" .. require("prototypes/settings").rocketParts.subgroup,
   order = "a[" .. require("prototypes/settings").rocketParts.subgroup .. "]-a-a[assembling-machine]",
   place_result = "rocket-assembling-machine",