.

PHI-CL/control/infinity.lua

@@ -0,0 +1,44 @@
+local main = {}
+local chest_filter = {}
+local hide = {
+    ['parameters'] = true,
+    ['spawnables'] = true,
+}
+
+do
+    for _, q in pairs(prototypes.quality) do
+        chest_filter[q.name] = {}
+    end
+
+    local j = 1
+
+    for _, i in pairs(prototypes.item) do
+        if not (i.hidden or (i.subgroup and i.subgroup.name and hide[i.subgroup.name])) then
+            for _, q in pairs(prototypes.quality) do
+                table.insert(chest_filter[q.name], {index = j, name = i.name, quality = q.name, mode = 'exactly', count = i.stack_size})
+            end
+
+            j = j + 1
+        end
+    end
+end
+
+function main.build(event)
+    if event.entity.type ~= 'infinity-container' and event.entity.type ~= 'infinity-cargo-wagon' then
+        return
+    end
+
+    if event.entity.name == 'super-infinity-chest' then
+        event.entity.remove_unfiltered_items = true
+        event.entity.infinity_container_filters = chest_filter[event.entity.quality.name]
+        event.entity.get_inventory(defines.inventory.chest).set_bar()
+    end
+
+    if event.entity.name == 'super-infinity-cargo-wagon' then
+        event.entity.remove_unfiltered_items = true
+        event.entity.infinity_container_filters = chest_filter[event.entity.quality.name]
+        event.entity.get_inventory(defines.inventory.cargo_wagon).set_bar()
+    end
+end
+
+return main