Allow category_id in register_factoid to be nil

1 files changed, 4 insertions, 7 deletions

diff --git a/init.lua b/init.lua
index 45121df..4ba6f88 100644
--- a/init.lua
+++ b/init.lua
@@ -270,16 +270,13 @@ end
 
 function doc.sub.items.register_factoid(category_id, factoid_type, factoid_generator)
 	local ftable = { fgen = factoid_generator, ftype = factoid_type }
-	if category_id == "nodes" then
-		table.insert(factoid_generators.nodes, ftable)
+	if category_id == "nodes" or category_id == "tools" or category_id == "craftitems" then
+		table.insert(factoid_generators[category_id], ftable)
 		return true
-	elseif category_id == "tools" then
+	elseif category_id == nil then
+		table.insert(factoid_generators.nodes, ftable)
 		table.insert(factoid_generators.tools, ftable)
-		return true
-	elseif category_id == "craftitems" then
 		table.insert(factoid_generators.craftitems, ftable)
-		return true
-	else
 		return false
 	end
 end