------------------------------------------------------------
-- Copyright (c) 2016 tacigar. All rights reserved.
-- https://github.com/tacigar/maidroid
------------------------------------------------------------

local state = {
	WALK_RANDOMLY = 0,
	WALK_TO_PLANT = 1,
	WALK_TO_MOW   = 2,
	PLANT         = 3,
	MOW           = 4,
}

local target_plants = {
	"farming:cotton_8",
	"farming:wheat_8",
}

local _aux = maidroid_core._aux

local FIND_PATH_TIME_INTERVAL = 20
local CHANGE_DIRECTION_TIME_INTERVAL = 20
local MAX_WALK_TIME = 120

-- is_plantable_place reports whether maidroid can plant any seed.
local function is_plantable_place(pos)
	local node = minetest.get_node(pos)
	local lpos = vector.sub(pos, {x = 0, y = -1, z = 0})
	local lnode = minetest.get_node(lpos)
	return node.name == "air"
		and minetest.get_item_group(lnode.name, "wet") > 0
end

-- is_mowable_place reports whether maidroid can mow.
local function is_mowable_place(pos)
	local node = minetest.get_node(pos)
	return maidroid_aux.table.find(target_plants, node.name)
end

do -- register farming core

	local function on_start(self)
		self.object:setacceleration{x = 0, y = -10, z = 0}
		self.object:setvelocity{x = 0, y = 0, z = 0}
		self.state = state.WALK_RANDOMLY
		self.time_counters = {}
		self.path = nil
	end

	local function on_stop(self)

	end

	local searching_range = {x = 5, y = 2, z = 5}

	local function walk_randomly(self, dtime)
		if self.time_counters[1] >= FIND_PATH_TIME_INTERVAL then
			self.time_counters[1] = 0
			self.time_counters[2] = self.time_counters[2] + 1

			if self:has_item_in_main(function(itemname)	return (minetest.get_item_group(itemname, "seed") > 0)) then
				local destination = _aux.search_surrounding(is_plantable_place, searching_range)
				if destination ~= nil then
					local path = minetest.find_path(self.object:getpos(), destination, 10, 1, 1)
					if path ~= nil then -- to walk to plant state.
						to_walk_to_plant(self, path, destination)
						return
					end
				end
			end
			-- if couldn't find path to plant, try to mow.
			local destination = _aux.search_surrounding(is_mowable_place, searching_range)
			if destination ~= nil then
				local path = minetest.find_path(self.object:getpos(), destination, 10, 1, 1)
				if path ~= nil then -- to walk to mow state.
					to_walk_to_mow(self, path, destination)
					return
				end
			end
			-- else do nothing.
			return

		elseif self.time_counters[2] >= CHANGE_DIRECTION_TIME_INTERVAL then
			self.time_counters[1] = self.time_counters[1] + 1
			self.time_counters[2] = 0
			self:change_direction_randomly()
			return
		else
			self.time_counters[1] = self.time_counters[1] + 1
			self.time_counters[2] = self.time_counters[2] + 1
			return
		end
	end

	local function to_walk_randomly(self)
		self.state = state.WALK_RANDOMLY
		self.time_counters[1] = 0
		self.time_counters[2] = 0
		self:change_direction_randomly()
	end

	local function to_walk_to_plant(self, path, destination)
		self.state = state.WALK_TO_PLANT
		self.path = path
		self.destination = destination
		self.time_counters[1] = 0 -- find path interval
		self.time_counters[2] = 0
		self:change_direction(self.path[1])
	end

	local function to_walk_to_mow(self, path, destination)
		self.state = state.WALK_TO_MOW
		self.path = path
		self.destination = destination
		self.time_counters[1] = 0 -- find path interval
		self.time_counters[2] = 0
		self:change_direction(self.path[1])
	end

	local function to_plant(self)
		if self:move_main_to_wield(function(itemname)	return (minetest.get_item_group(itemname, "seed") > 0 end)) then
			self.state = state.PLANT
			self.time_counters[1] = 0
			self.object:set_velocity{x = 0, y = 0, z = 0}
			return
		else
			to_walk_randomly(self)
			return
		end
	end

	local function to_mow(self)
		self.state = state.MOW
		self.time_counters[1] = 0
		self.object:set_velocity{x = 0, y = 0, z = 0}
	end

	local function walk_to_plant_and_mow_common(self, dtime)
		if self.time_counters[2] >= MAX_WALK_TIME then -- time over.
			to_walk_randomly(self)
			return
		end

		if self.time_counters[1] >= FIND_PATH_TIME_INTERVAL then
			self.time_counters[1] = 0
			self.time_counters[2] = self.time_counters[2] + 1
			local path = minetest.find_path(self.object:getpos(), self.destination, 10, 1, 1)
			if path == nil then
				to_walk_randomly(self)
				return
			end
			self.path = path
		end

		-- follow path
		if vector.distance(self.path[1], self.object:getpos()) < 0.01 then
			table.remove(self.path, 1)

			if #self.path == 0 then -- end of path
				if self.state == state.WALK_TO_PLANT then
					to_plant(self)
					return
				elseif self.state == state.WALK_TO_MOW then
					to_mow(self)
					return
				end
			else -- else next step, follow next path.
				self:change_direction(self.path[1])
			end

		else
			-- if maidroid is stopped by obstacles, the maidroid must jump.
			local velocity = self.object:getvelocity()
			if velocity.y == 0 then
				local front_node = self:get_front_node()
				if front_node.name ~= "air" then
					self.object:setvelocity{x = velocity.x, y = 3, z = velocity.z}
				end
			end
		end
	end

	local function plant(self, dtime)
		if self.time_counters[1] >= 5 then
			if is_plantable_place(self.destination) then
				local stack = self:get_wield_item_stack()
				stack:take_item(1)
				self:set_wield_item_stack(stack)
				return
			end
			to_walk_randomly()
			return
		else
			self.time_counters[1] = self.time_counters[1] + 1
		end
	end

	local function mow(self, dtime)
		if self.time_counters[1] >= 5 then
			if is_mowable_place(self.destination) then
				local destnode = minetest.get_node(self.destination)
				minetest.remove_node(self.destination)
				local stacks = minetest.get_node_drops(destnode.name)

				for _, stack in ipairs(stacks) do
					local leftover = self:add_item_to_main(stack)
					minetest.add_item(self.destination, leftover)
				end
			end
			to_walk_randomly()
			return
		else
			self.time_counters[1] = self.time_counters[1] + 1
		end
	end

	local function plant_and_mow_common(self, dtime)
		if self.time_counters[1] >= 5 then
			if (self.state == state.PLANT and not is_plantable_place(self.destination)
			or (self.state == state.MOW   and not is_mowable_place(self.destination)) then


				local inv = self:get_inventory()
				local stacks = inv:get_list("main")

				for idx, stack in ipairs(stacks) do
					local item_name = stack:get_name()
					if minetest.get_item_group(item_name, "seed") > 0 then
						minetest.add_node(self.destination, {name = item_name, param2 = 1})
						stack:take_item(1)
						inv:set_stack("main", idx, stack)
						break
					end
				end
			end
			to_walk(self)
		else
			self.time_count = self.time_count + 1
		end
	end

	local function on_step(self, dtime)
		if self.state == state.WALK_RANDOMLY then
			walk_randomly(self, dtime)
		elseif self.state == state.WALK_TO_PLANT or self.state == state.WALK_TO_MOW then
			walk_to_plant_and_mow_common(self, dtime)
		elseif self.state == state.PLANT or self.state == state.MOW then
			plant_and_mow_common(self, dtime)
		end
	end

	maidroid.register_core("maidroid_core:farming", {
		description      = "maidroid core : farming",
		inventory_image  = "maidroid_core_farming.png",
		on_start         = on_start,
		on_stop          = on_stop,
		on_resume        = on_start,
		on_pause         = on_stop,
		on_step          = on_step,
	})

end -- register farming core