local S = minetest.get_translator("better_commands")

better_commands.register_command("enchant", {
    params = S("<selector> <enchantment> [<level>]"),
    description = S("Adds an enchantment to a player's selected item, subject to the same restrictions as an anvil"),
    privs = { give = true },
    func = function(name, param, context)
        local split_param, err = better_commands.parse_params(param)
        if err then return false, better_commands.error(err), 0 end
        if not split_param[1] and split_param[2] then
            return false, nil, 0
        end
        local selector = split_param[1]
        local enchantment = split_param[2] and split_param[2][3]
        if not enchantment then return false, better_commands.error(S("Missing enchantment")), 0 end
        local level_str = split_param[3] and split_param[3][3]
        local level = tonumber(level_str or "1")
        local targets, err = better_commands.parse_selector(selector, context)
        if err or not targets then return false, better_commands.error(err), 0 end
        local count = 0
        local total_count = 0
        local last
        minetest.log("?")
        for _, target in ipairs(targets) do
            if target.is_player and target:is_player() then
                minetest.log("???")
                total_count = total_count + 1
                local itemstack = target:get_wielded_item()
                local can_enchant, errorstring, extra_info = mcl_enchanting.can_enchant(itemstack, enchantment, level)
                if not can_enchant then
                    if errorstring == "enchantment invalid" then
                        err = S("Invalid enchantment '@1'", enchantment)
                    elseif errorstring == "item missing" then
                        err = S("@1 is not holding any item", better_commands.get_entity_name(target))
                    elseif errorstring == "item not supported" then
                        err = S("@1 cannot support that enchantment", itemstack:get_short_description())
                    elseif errorstring == "level invalid" then
                        err = S("Invalid integer '@1'", level_str)
                    elseif errorstring == "level too high" then
                        err = S("@1 is higher than the maximum level of @2 supported by that enchantment", level_str, extra_info)
                    elseif errorstring == "level too small" then
                        err = S("@1 is lower than the minimum level of @2 supported by that enchantment", level_str, extra_info)
                    elseif errorstring == "incompatible" then
                        err = S("@1 can't be combined with @2.",
                                    mcl_enchanting.get_enchantment_description(enchantment, level), extra_info)
                    else
                        err = S("Failed to enchant item.")
                    end
                else
                    count = count + 1
                    last = better_commands.get_entity_name(target)
                    target:set_wielded_item(mcl_enchanting.enchant(itemstack, enchantment, level))
                end
            end
        end
        minetest.log("??")
        if count < 1 then
            if total_count < 1 then
                return false, better_commands.error(S("No player was found")), 0
            else
                return false, better_commands.error(err), 0
            end
        elseif count == 1 then
            return true, S("Applied enchantment @1 to @2's item", mcl_enchanting.get_enchantment_description(enchantment, level), last), 1
        else
            return true, S("Enchanted items of @1 players.", count), count
        end
    end
})

better_commands.register_command("forceenchant", {
    params = S("<selector> <enchantment> [<level>]"),
    description = S("Adds an enchantment to a player's selected item, regardless of whether it would normally be possible"),
    privs = { give = true },
    func = function(name, param, context)
        local split_param, err = better_commands.parse_params(param)
        if err then return false, better_commands.error(err), 0 end
        local selector = split_param[1]
        local enchantment = split_param[2] and split_param[2][3]
        if not enchantment then return false, better_commands.error("Missing enchantment"), 0 end
        local level_str = split_param[3] and split_param[3][3]
        local level = tonumber(level_str or "1")
        local targets, err = better_commands.parse_selector(selector, context)
        if err or not targets then return false, better_commands.error(err), 0 end
        local total_count = 0
        local count = 0
        local last
        for _, target in ipairs(targets) do
            if target.is_player and target:is_player() then
                total_count = total_count + 1
                local itemstack = target:get_wielded_item()
                local _, errorstring = mcl_enchanting.can_enchant(itemstack, enchantment, level)
                if errorstring == "enchantment invalid" then
                    err = S("Invalid enchantment '@1'", enchantment)
                elseif errorstring == "item missing" then
                    err = S("@1 is not holding any item", better_commands.get_entity_name(target))
                elseif errorstring == "item not supported" and not mcl_enchanting.is_enchantable(itemstack:get_name()) then
                    err = S("@1 cannot support that enchantment")
                elseif errorstring == "level invalid" then
                    err = S("Invalid integer '@1'", level_str)
                else
                    target:set_wielded_item(mcl_enchanting.enchant(itemstack, enchantment, level))
                    count = count + 1
                    last = better_commands.get_entity_name(target)
                end
            end
        end
        if count < 1 then
            if total_count < 1 then
                return false, better_commands.error(S("No player was found")), 0
            else
                return false, better_commands.error(err), 0
            end
        elseif count == 1 then
            return true, S("Applied enchantment @1 to @2's item", mcl_enchanting.get_enchantment_description(enchantment, level), last), 1
        else
            return true, S("Enchanted items of @1 players.", count), count
        end
    end
})