local max_charge = {50000, 200000, 300000}
local power_usage_per_node = {200, 500, 600}

local S = technic.getter
local mat = technic.materials

core.register_craft({
	output = 'technic:mining_drill',
	recipe = {
		{mat.tin_ingot,             'technic:diamond_drill_head', mat.tin_ingot},
		{'technic:stainless_steel_ingot', 'basic_materials:motor',      'technic:stainless_steel_ingot'},
		{'',                              'technic:red_energy_crystal', mat.copper_ingot},
	}
})
core.register_craft({
	output = 'technic:mining_drill_mk2',
	recipe = {
		{'technic:diamond_drill_head',    'technic:diamond_drill_head',   'technic:diamond_drill_head'},
		{'technic:stainless_steel_ingot', 'technic:mining_drill',         'technic:stainless_steel_ingot'},
		{'',                              'technic:green_energy_crystal', ''},
	}
})
core.register_craft({
	output = 'technic:mining_drill_mk3',
	recipe = {
		{'technic:diamond_drill_head',    'technic:diamond_drill_head',  'technic:diamond_drill_head'},
		{'technic:stainless_steel_ingot', 'technic:mining_drill_mk2',    'technic:stainless_steel_ingot'},
		{'',                              'technic:blue_energy_crystal', ''},
	}
})
for i = 1, 4 do
	core.register_craft({
		output = 'technic:mining_drill_mk3',
		recipe = {
			{'technic:diamond_drill_head',    'technic:diamond_drill_head',   'technic:diamond_drill_head'},
			{'technic:stainless_steel_ingot', 'technic:mining_drill_mk2_'..i, 'technic:stainless_steel_ingot'},
			{'',                              'technic:blue_energy_crystal',  ''},
		}
	})
end

local mining_drill_mode_text = {
	{S("Single node.")},
	{S("3 nodes deep.")},
	{S("3 nodes wide.")},
	{S("3 nodes tall.")},
	{S("3x3 nodes.")},
}

local function get_description(mk, mode)
	local description = "Mining Drill Mk@1"..(mode > 0 and " Mode @2" or "")
	return mode > 0 and S(description, mk, mode) or S(description, mk)
end

local function drill_dig_it0 (pos,player)
	if core.is_protected(pos, player:get_player_name()) then
		core.record_protection_violation(pos, player:get_player_name())
		return
	end
	local node = core.get_node(pos)
	if node.name == "air" or node.name == "ignore" then return end
	if node.name == mat.lava_source then return end
	if node.name == mat.lava_flowing then return end
	if node.name == mat.water_source then core.remove_node(pos) return end
	if node.name == mat.water_flowing then core.remove_node(pos) return end
	local def = core.registered_nodes[node.name]
	if not def then return end
	def.on_dig(pos, node, player)
end

local function drill_dig_it1 (player)
	local dir=player:get_look_dir()
	if math.abs(dir.x)>math.abs(dir.z) then
		if dir.x>0 then return 0 end
		return 1
	end
	if dir.z>0 then return 2 end
	return 3
end

local function drill_dig_it2 (pos,player)
	pos.y=pos.y+1
	drill_dig_it0 (pos,player)
	pos.z=pos.z+1
	drill_dig_it0 (pos,player)
	pos.z=pos.z-2
	drill_dig_it0 (pos,player)
	pos.z=pos.z+1
	pos.y=pos.y-1
	drill_dig_it0 (pos,player)
	pos.z=pos.z+1
	drill_dig_it0 (pos,player)
	pos.z=pos.z-2
	drill_dig_it0 (pos,player)
	pos.z=pos.z+1
	pos.y=pos.y-1
	drill_dig_it0 (pos,player)
	pos.z=pos.z+1
	drill_dig_it0 (pos,player)
	pos.z=pos.z-2
	drill_dig_it0 (pos,player)
end

local function drill_dig_it3 (pos,player)
	pos.y=pos.y+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x-2
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	pos.y=pos.y-1
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x-2
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	pos.y=pos.y-1
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x-2
	drill_dig_it0 (pos,player)
end

local function drill_dig_it4 (pos,player)
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x-2
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	pos.z=pos.z+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x-2
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	pos.z=pos.z-2
	drill_dig_it0 (pos,player)
	pos.x=pos.x+1
	drill_dig_it0 (pos,player)
	pos.x=pos.x-2
	drill_dig_it0 (pos,player)
end

local function cost_to_use(drill_type, mode)
	local mult
	if mode == 1 then
		mult = 1
	elseif mode <= 4 then
		mult = 3
	else
		mult = 9
	end
	return power_usage_per_node[drill_type] * mult
end

local function drill_dig_it(pos, player, mode)
	if mode == 1 then
		drill_dig_it0(pos, player)
	end

	if mode == 2 then -- 3 deep
		local dir = drill_dig_it1(player)
		if dir == 0 then -- x+
			drill_dig_it0(pos, player)
			pos.x = pos.x + 1
			drill_dig_it0(pos, player)
			pos.x = pos.x + 1
			drill_dig_it0(pos, player)
		end
		if dir == 1 then  -- x-
			drill_dig_it0(pos, player)
			pos.x=pos.x-1
			drill_dig_it0 (pos,player)
			pos.x=pos.x-1
			drill_dig_it0 (pos,player)
		end
		if dir==2 then  -- z+
			drill_dig_it0 (pos,player)
			pos.z=pos.z+1
			drill_dig_it0 (pos,player)
			pos.z=pos.z+1
			drill_dig_it0 (pos,player)
		end
		if dir==3 then  -- z-
			drill_dig_it0 (pos,player)
			pos.z=pos.z-1
			drill_dig_it0 (pos,player)
			pos.z=pos.z-1
			drill_dig_it0 (pos,player)
		end
	end

	if mode==3 then -- 3 wide
		local dir = drill_dig_it1(player)
		if dir==0 or dir==1 then -- x
			drill_dig_it0 (pos,player)
			pos.z=pos.z+1
			drill_dig_it0 (pos,player)
			pos.z=pos.z-2
			drill_dig_it0 (pos,player)
		end
		if dir==2 or dir==3 then  -- z
			drill_dig_it0 (pos,player)
			pos.x=pos.x+1
			drill_dig_it0 (pos,player)
			pos.x=pos.x-2
			drill_dig_it0 (pos,player)
		end
	end

	if mode==4 then -- 3 tall, selected in the middle
		drill_dig_it0 (pos,player)
		pos.y=pos.y-1
		drill_dig_it0 (pos,player)
		pos.y=pos.y-1
		drill_dig_it0 (pos,player)
	end

	if mode==5 then -- 3 x 3
		local dir=player:get_look_dir()
		if math.abs(dir.y)<0.5 then
			dir=drill_dig_it1(player)
				if dir==0 or dir==1 then -- x
					drill_dig_it2(pos,player)
				end
				if dir==2 or dir==3 then  -- z
					drill_dig_it3(pos,player)
				end
		else
		drill_dig_it4(pos,player)
		end
	end

	core.sound_play("mining_drill", {pos = pos, gain = 1.0, max_hear_distance = 10}, true)
end

local function pos_is_pointable(pos)
	local node = core.get_node(pos)
	local nodedef = core.registered_nodes[node.name]
	return nodedef and nodedef.pointable
end

local function mining_drill_mk2_setmode(user,itemstack)
	local player_name = user:get_player_name()
	local meta = itemstack:get_meta()
	local mode = meta:get_int("mode")
	if mode == 0 then
		core.chat_send_player(player_name, S("Use while sneaking to change Mining Drill Mk@1 modes.", 2))
	end
	mode = mode < 4 and mode + 1 or 1
	core.chat_send_player(player_name, get_description(2, mode)..": "..mining_drill_mode_text[mode][1])
    itemstack:set_name("technic:mining_drill_mk2_"..mode)
	meta:set_int("mode", mode)
	return itemstack
end

local function mining_drill_mk3_setmode(user,itemstack)
	local player_name = user:get_player_name()
	local meta = itemstack:get_meta()
	local mode = meta:get_int("mode")
	if mode == 0 then
		core.chat_send_player(player_name, S("Use while sneaking to change Mining Drill Mk@1 modes.", 3))
	end
	mode = mode < 5 and mode + 1 or 1
	core.chat_send_player(player_name, get_description(3, mode)..": "..mining_drill_mode_text[mode][1])
    itemstack:set_name("technic:mining_drill_mk3_"..mode)
	meta:set_int("mode", mode)
	return itemstack
end

local function mining_drill_mk2_handler(itemstack, user, pointed_thing)
	local keys = user:get_player_control()
	local meta = itemstack:get_meta()
	local mode = meta:get_int("mode")
	if mode == 0 or keys.sneak then
		return mining_drill_mk2_setmode(user, itemstack)
	end
	if pointed_thing.type ~= "node" or not pos_is_pointable(pointed_thing.under) then
		return
	end
	local charge_to_take = cost_to_use(2, mode)
	if technic.use_charge(itemstack, charge_to_take) then
		local pos = core.get_pointed_thing_position(pointed_thing, false)
		drill_dig_it(pos, user, mode)
	end
	return itemstack
end

local function mining_drill_mk3_handler(itemstack, user, pointed_thing)
	local keys = user:get_player_control()
	local meta = itemstack:get_meta()
	local mode = meta:get_int("mode")
	if mode == 0 or keys.sneak then
		return mining_drill_mk3_setmode(user, itemstack)
	end
	if pointed_thing.type ~= "node" or not pos_is_pointable(pointed_thing.under) then
		return
	end
	local charge_to_take = cost_to_use(3, mode)
	if technic.use_charge(itemstack, charge_to_take) then
		local pos = core.get_pointed_thing_position(pointed_thing, false)
		drill_dig_it(pos, user, mode)
	end
	return itemstack
end

-- register Mining Drill Mk1
technic.register_power_tool("technic:mining_drill", {
	description = get_description(1, 0),
	inventory_image = "technic_mining_drill.png",
	max_charge = max_charge[1],
	on_use = function(itemstack, user, pointed_thing)
		if pointed_thing.type ~= "node" or not pos_is_pointable(pointed_thing.under) then
			return itemstack
		end
		local charge_to_take = cost_to_use(1, 1)
		if technic.use_charge(itemstack, charge_to_take) then
			local pos = core.get_pointed_thing_position(pointed_thing, false)
			drill_dig_it(pos, user, 1)
		end
		return itemstack
	end,
})

do -- register Mining Drill Mk2
	local inventory_image = "technic_mining_drill_mk2.png"
	for i = 0, 4 do
		local overlay = i > 0 and "^technic_tool_mode"..i..".png" or ""
		technic.register_power_tool("technic:mining_drill_mk2"..(i > 0 and ("_"..i) or ""), {
			description = get_description(2, i),
			inventory_image = inventory_image .. overlay,
			wield_image = inventory_image,
			max_charge = max_charge[2],
			groups = i > 0 and {not_in_creative_inventory=1} or nil,
			on_use = mining_drill_mk2_handler,
		})
	end
end

do -- register Mining Drill Mk3
	local inventory_image = "technic_mining_drill_mk3.png"
	for i = 0, 5 do
		local overlay = i > 0 and "^technic_tool_mode"..i..".png" or ""
		technic.register_power_tool("technic:mining_drill_mk3"..(i > 0 and ("_"..i) or ""), {
			description = get_description(3, i),
			inventory_image = inventory_image .. overlay,
			wield_image = inventory_image,
			max_charge = max_charge[3],
			groups = i > 0 and {not_in_creative_inventory=1} or nil,
			on_use = mining_drill_mk3_handler,
		})
	end
end
