local Adapter    = require('core.chestAdapter18')
local Config     = require('opus.config')
local Equipper   = require('turtle.equipper')
local Util       = require('opus.util')

local fs         = _G.fs
local os         = _G.os
local turtle     = _G.turtle

local STARTUP_FILE = 'usr/autorun/rancher.lua'

local retain = Util.transpose {
	'minecraft:shears',
	'minecraft:wheat',
	'minecraft:diamond_sword',
	'plethora:module:3',
}
local config = {
	animal = 'Cow',
	max_animals = 15,
}
Config.load('rancher', config)

local ANIMALS = {
	Pig   = { min =  0, food = 'minecraft:carrot' },
	Sheep = { min = .5, food = 'minecraft:wheat'  },
	Cow   = { min = .5, food = 'minecraft:wheat'  },
}

local animal = ANIMALS[config.animal]

Equipper.equipLeft('minecraft:diamond_sword')
local sensor = Equipper.equipRight('plethora:sensor')

local chest = Adapter({ side = 'bottom', direction = 'up' }) or error('missing chest')

if not fs.exists(STARTUP_FILE) then
	Util.writeFile(STARTUP_FILE,
		[[os.sleep(1)
shell.openForegroundTab('rancher.lua')]])
	print('Autorun program created: ' .. STARTUP_FILE)
end

local function getAnimalCount()
	local blocks = sensor.sense()

	local grown = 0
	local babies = 0

	Util.filterInplace(blocks, function(v)
		if v.name == config.animal then
			local entity = sensor.getMetaByID(v.id)
			if entity then
				if entity.isChild then
					babies = babies + 1
				else
					grown  = grown  + 1
				end
				return not entity.isChild
			end
		end
	end)

	Util.print('%d grown, %d babies', grown, babies)

	return #blocks
end

local function butcher()
	Equipper.equipRight('minecraft:diamond_sword')
	turtle.select(1)

	turtle.attack()
	for _ = 1, 3 do
		turtle.turnRight()
		turtle.attack()
	end
	Equipper.equipRight('plethora:sensor')

	turtle.eachFilledSlot(function(slot)
		if not retain[slot.name] then
			chest:insert(slot.index, 64)
		end
	end)
end

local function breed()
	turtle.select(1)

	if config.animal == 'Sheep' then
		turtle.place('minecraft:shears')
	end
	turtle.place('minecraft:wheat')
	for _ = 1, 3 do
		turtle.turnRight()
		if config.animal == 'Sheep' then
			turtle.place('minecraft:shears')
		end
		turtle.place('minecraft:wheat')
	end
end

local s, m = turtle.run(function()
	print('Configured animal: ' .. config.animal)

	repeat
		local animalCount = getAnimalCount()
		if animalCount > config.max_animals then
			turtle.setStatus('Butchering')
			butcher()
		elseif turtle.getItemCount(animal.food) == 0 then
			if chest:provide({ name = animal.food, damage = 0 }, 64) == 0 then
				print('Out of ' .. animal.food)
				turtle.setStatus('Out of food')
			end
		else
			turtle.setStatus('Breeding')
			breed()
		end
		os.sleep(5)
	until turtle.isAborted()
end)

if not s and m then
	error(m)
end