MineClone5/mods/CORE/mcl_enchanting/core.lua

mcl_enchanting.total_weight = 0
mcl_enchanting.all_item_groups = {}

for enchantment, enchantment_def in pairs(mcl_enchanting.enchantments) do
	local all_item_groups = {}
	for primary in pairs(enchantment_def.primary) do
		all_item_groups[primary] = true
		mcl_enchanting.all_item_groups[primary] = true
	end
	for secondary in pairs(enchantment_def.secondary) do
		all_item_groups[secondary] = true
		mcl_enchanting.all_item_groups[secondary] = true
	end
	enchantment_def.all = all_item_groups
	mcl_enchanting.total_weight = mcl_enchanting.total_weight + enchantment_def.weight
end

--[[
minetest.register_on_mods_loaded(function()
	for toolname, tooldef in pairs(minetest.registered_tools) do
		for _, material in pairs(tooldef.materials) do
			local full_name = toolname .. ((material == "") and "" or "_" .. material)
			local old_def = minetest.registered_tools[full_name]
			if not old_def then break end
			mcl_enchanting.all_tools[full_name] = toolname
			for _, enchantment in pairs(tooldef.enchantments) do
				local enchantment_def = mcl_enchanting.enchantments[enchantment]
				for lvl = 1, enchantment_def.max_level do
					local new_def = table.copy(old_def)
					new_def.description = minetest.colorize("#54FCFC", old_def.description) .. "\n" .. mcl_enchanting.get_enchantment_description(enchantment, lvl)
					new_def.inventory_image = old_def.inventory_image .. "^[colorize:violet:50"
					new_def.groups.not_in_creative_inventory = 1
					new_def.texture = old_def.texture or full_name:gsub("%:", "_")
					new_def._original_tool = full_name
					enchantment_def.create_itemdef(new_def, lvl)
					minetest.register_tool(":" .. full_name .. "_enchanted_" .. enchantment .. "_" .. lvl, new_def)
				end
			end
		end
	end
end)
--]]

minetest.register_on_mods_loaded(function()
	for toolname, tooldef in pairs(minetest.registered_tools) do
		-- quick test
		local has_enchantment = false
		for group, groupv in pairs(tooldef.groups) do
			if groupv > 0 and mcl_enchanting.all_item_groups[group] then	
				has_enchantment = true
				break
			end
		end
		if not has_enchantment then
			break
		end
		-- expensive test
		has_enchantment = false
		for enchantment in pairs(mcl_enchanting.enchantments) do
			if mcl_enchanting.item_supports_enchantment(itemname, enchantment) then
				has_enchantment = true
				break
			end
		end
		if not has_enchantment then
			break
		end
		local new_name = toolname .. "_enchanted" 
		tooldef._mcl_enchanting_enchanted_tool = new_name
		local new_def = table.copy(tooldef)
		new_def.inventory_image = old_def.inventory_image .. "^[colorize:violet:50"
		new_def.groups.not_in_creative_inventory = 1
		new_def.groups.enchanted = 1
		new_def.texture = old_def.texture or toolname:gsub("%:", "_")
		new_def._mcl_enchanting_enchanted_tool = new_name
		minetest.register_tool(":" .. new_name, new_def)
	end
end)

function mcl_enchanting.get_enchantments(itemstack)
	return minetest.deserialize(itemstack:get_meta():get_string("mcl_enchanting:enchantments")) or {}
end

function mcl_enchanting.set_enchantments(itemstack, enchantments, data)
	return itemstack:get_meta():set_string("mcl_enchanting:enchantments", minetest.serialize(enchantments))
end

function mcl_enchanting.get_enchantment(itemstack, enchantment, data)
	return (data or mcl_enchanting.get_enchantments(itemstack))[enchantment] or 0
end

function mcl_enchanting.get_enchantment_description(enchantment, level)
	local enchantment_def = mcl_enchanting.enchantments[enchantment]
	return enchantment_def.name .. " " .. (enchantment_def.max_level == 1 and "" or mcl_enchanting.roman_numerals.toRoman(level))
end

function mcl_enchanting.get_enchanted_itemstring(itemname)
	return minetest.registered_items[itemname]._mcl_enchanting_enchanted_tool
end

function mcl_enchanting.item_supports_enchantment(itemname, enchantment)
	if not mcl_enchanting.get_enchanted_itemstring(itemname) then
		return false
	end
	local enchantment_def = mcl_enchanting.enchantments[enchantment]
	for disallow in pairs(enchantment_def.disallow) do
		if minetest.get_item_group(itemname, disallow) > 0 then
			return false
		end
	end
	for group in pairs(enchantment_def.all) do
		if minetest.get_item_group(itemname, group) then
			return true
		end
	end
	return false
end

function mcl_enchanting.can_enchant(itemstack, enchantment, level)
	local enchantment_def = mcl_enchanting.enchantments[enchantment]
	if not enchantment_def then
		return false, "enchantment invalid"
	end
	if itemstack:get_name() == "" then
		return false, "item missing"
	end
	if not mcl_enchanting.item_supports_enchantment(itemdef.name, enchantment) then
		return false, "item not supported"
	end
	if not level then
		return false, "level invalid"
	end
	if level > enchantment_def.max_level then
		return false, "level too high", enchantment_def.max_level
	elseif  level < 1 then
		return false, "level too small", 1
	end
	local item_enchantments = mcl_enchanting.get_enchantments(itemstack)
	local enchantment_level = item_enchantments[enchantment]
	if enchantment_level then
		return false, "incompatible", mcl_enchanting.get_enchantment_description(enchantment, enchantment_level)
	end
	for incompatible in pairs(enchantment_def.incompatible) do
		local incompatible_level = item_enchantments[incompatible]
		if incompatible_level then
			return false, "incompatible", mcl_enchanting.get_enchantment_description(incompatible, incompatible_level)
		end
	end
end

function mcl_enchanting.enchant(itemstack, enchantment, level)
	local enchanted_itemstack = ItemStack(mcl_enchanting.get_enchanted_itemstring(itemstack:get_name()))
	enchanted_itemstack:add_wear(itemstack:get_wear())
	enchanted_itemstack:set_meta(itemstack:get_meta())
	local enchantments = mcl_enchanting.get_enchantments(enchanted_itemstack)
	enchantments[enchantment] = level
	mcl_enchanting.set_enchantments(enchanted_itemstack, enchantments)
	mcl_enchanting.reload_enchantments(enchanted_itemstack, enchantments)
end

function mcl_enchanting.reload_enchantments(itemstack, echantments)
	local itemdef = itemstack:get_definition()
	for enchantment, level in pairs(enchantments) do
		local func = mcl_enchanting.features[enchantment]
		if func then
			func(itemstack, level, itemdef)
		end
	end
end