-- setup is turtle facing apiary, with a (large, maybe dimensional storage) chest beneath the turtle
-- QTRQHPhM
-- RwTbaqpP

local apiary = peripheral.wrap("front")
local chest = peripheral.wrap("bottom")
local maxDrones = 6
local minDrones = 3

bees = {}
stacks = {}
shutdownMsg = nil

function addParent(parent, offspring)
    if not bees[parent] then
    bees[parent] = {
        --name = parent,
        score = 0,
        mutateFrom = {}
    }
    end
end

function addOffspring(offspring, parentss)
    if bees[offspring] then
    for i, parents in ipairs(parentss) do
        table.insert(bees[offspring].mutateFrom, parents)
    end
    else
    bees[offspring] = {
        score = 0,
        mutateFrom = parentss
    }
    end
    for i, parents in ipairs(parentss) do
    for i, parent in ipairs(parents) do
        addParent(parent, offspring)
    end
    end
end




-- produce combinations from 1 or 2 lists
function choose(list, list2)
    local newList = {}
    if list2 then
    for i = 1, #list2 do
        for j = 1, #list do
        if list[j] ~= list[i] then
            table.insert(newList, {list[j], list2[i]})
        end
        end
    end
    else
    for i = 1, #list do
        for j = i, #list do
        if list[i] ~= list[j] then
            table.insert(newList, {list[i], list[j]})
        end
        end
    end
    end
    return newList
end


--bee list from Master Apiarist Database v3.1
--Forestry v2.3.0.7 (683)
--Extra Bees v1.8-dev2
--Magic Bees v2.1.7 (build 78)

addOffspring("Common", choose({"Forest", "Meadows", "Modest", "Wintry", "Tropical", "Marshy", "Water", "Rocky", "Mystical", "Sorcerous", "Unusual", "Attuned"}))
addOffspring("Cultivated", choose({"Common"}, {"Forest", "Meadows", "Modest", "Wintry", "Tropical", "Marshy", "Water", "Rocky", "Mystical", "Sorcerous", "Unusual", "Attuned"}))

addOffspring("Noble", {{"Common", "Cultivated"}})
addOffspring("Majestic", {{"Noble", "Cultivated"}})
addOffspring("Imperial", {{"Noble", "Majestic"}})
addOffspring("Diligent", {{"Common", "Cultivated"}})
addOffspring("Unweary", {{"Diligent", "Cultivated"}})
addOffspring("Industrious", {{"Diligent", "Unweary"}})
addOffspring("Heroic", {{"Steadfast", "Valiant"}})
addOffspring("Sinister", choose({"Cultivated"}, {"Modest", "Tropical"}))
addOffspring("Fiendish", choose({"Sinister"}, {"Cultivated", "Modest", "Tropical"}))
addOffspring("Frugal", choose({"Modest"}, {"Sinister", "Fiendish"}))
addOffspring("Demonic", {{"Sinister", "Fiendish"}})
addOffspring("Austere", {{"Modest", "Frugal"}})
addOffspring("Exotic", {{"Austere", "Tropical"}})
addOffspring("Edenic", {{"Exotic", "Tropical"}})
addOffspring("Spectral", {{"Hermitic", "Ender"}})
addOffspring("Phantasmal", {{"Spectral", "Ender"}})
addOffspring("Icy", {{"Industrious", "Wintry"}})
addOffspring("Glacial", {{"Icy", "Wintry"}})
addOffspring("Vindictive", {{"Monastic", "Demonic"}})
addOffspring("Vengeful", choose({"Vindictive"}, {"Demonic", "Monastic"}))
addOffspring("Avenging", {{"Vengeful", "Vindictive"}})
addOffspring("Leporine", {{"Meadows", "Forest"}})
addOffspring("Merry", {{"Wintry", "Forest"}})
addOffspring("Tipsy", {{"Wintry", "Meadows"}})
addOffspring("Tricky", {{"Sinister", "Common"}})
addOffspring("Rural", {{"Meadows", "Diligent"}})
addOffspring("Secluded", {{"Monastic", "Austere"}})
addOffspring("Hermitic", {{"Monastic", "Secluded"}})
addOffspring("Arid", {{"Meadows", "Modest"}})
addOffspring("Barren", {{"Arid", "Common"}})
addOffspring("Desolate", {{"Arid", "Barren"}})
addOffspring("Decaying", {{"Desolate", "Modest"}})
addOffspring("Skeletal", {{"Desolate", "Frugal"}})
addOffspring("Creepy", {{"Desolate", "Austere"}})
addOffspring("Decomposing", {{"Gnawing", "Common"}})
addOffspring("Tolerant", {{"Rocky", "Diligent"}})
addOffspring("Robust", {{"Rocky", "Tolerant"}})
addOffspring("Resilient", {{"Imperial", "Robust"}})
addOffspring("Corroded", {{"Resilient", "Forest"}})
addOffspring("Tarnished", {{"Resilient", "Marshy"}})
addOffspring("Rusty", {{"Resilient", "Meadows"}})
addOffspring("Leaden", {{"Resilient", "Unweary"}})
addOffspring("Galvanized", {{"Tarnished", "Cultivated"}})
addOffspring("Impregnable", {{"Resilient", "Noble"}})
addOffspring("Invincible", {{"Resilient", "Ender"}})
addOffspring("Glittering", {{"Corroded", "Imperial"}})
addOffspring("Shining", {{"Rusty", "Imperial"}})
addOffspring("Valuable", {{"Glittering", "Shining"}})
addOffspring("Lapis", {{"Resilient", "Water"}})
addOffspring("Emerald", {{"Lapis", "Noble"}})
addOffspring("Ruby", {{"Emerald", "Austere"}})
addOffspring("Sapphire", {{"Emerald", "Ocean"}})
addOffspring("Diamond", {{"Lapis", "Imperial"}})
addOffspring("Unstable", {{"Austere", "Rocky"}})
addOffspring("Nuclear", {{"Unstable", "Rusty"}})
addOffspring("Radioactive", {{"Nuclear", "Glittering"}})
addOffspring("Ancient", {{"Noble", "Diligent"}})
addOffspring("Primeval", {{"Ancient", "Noble"}})
addOffspring("Prehistoric", {{"Primeval", "Majestic"}})
addOffspring("Relic", {{"Prehistoric", "Imperial"}})
addOffspring("Fossilised", {{"Primeval", "Growing"}})
addOffspring("Resinous", {{"Primeval", "Fungal"}})
addOffspring("Oily", {{"Primeval", "Ocean"}})
addOffspring("Distilled", {{"Oily", "Industrious"}})
addOffspring("Refined", {{"Oily", "Distilled"}})
addOffspring("Elastic", {{"Refined", "Resinous"}})
addOffspring("River", {{"Water", "Common"}})
addOffspring("Ocean", {{"Water", "Diligent"}})
addOffspring("Stained", {{"Ebony", "Ocean"}})
addOffspring("Growing", {{"Diligent", "Forest"}})
addOffspring("Thriving", {{"Growing", "Rural"}})
addOffspring("Blooming", {{"Thriving", "Growing"}})
addOffspring("Sweetened", {{"Valiant", "Diligent"}})
addOffspring("Sugary", {{"Sweetened", "Diligent"}})
addOffspring("Ripening", {{"Sugary", "Forest"}})
addOffspring("Fruity", {{"Ripening", "Rural"}})
addOffspring("Farmed", {{"Cultivated", "Rural"}})
addOffspring("Bovine", {{"Rural", "Water"}})
addOffspring("Caffeinated", {{"Tropical", "Rural"}})
addOffspring("Minty", {{"Tropical", "Farmed"}})
addOffspring("Damp", {{"Common", "Marshy"}})
addOffspring("Boggy", {{"Damp", "Marshy"}})
addOffspring("Fungal", {{"Boggy", "Damp"}})
addOffspring("Furious", {{"Embittered", "Sinister"}})
addOffspring("Volcanic", {{"Embittered", "Furious"}})
addOffspring("Malicious", {{"Sinister", "Tropical"}})
addOffspring("Infectious", {{"Malicious", "Tropical"}})
addOffspring("Virulent", {{"Malicious", "Infectious"}})
addOffspring("Viscous", {{"Water", "Exotic"}})
addOffspring("Glutinous", {{"Viscous", "Exotic"}})
addOffspring("Sticky", {{"Viscous", "Glutinous"}})
addOffspring("Corrosive", {{"Virulent", "Sticky"}})
addOffspring("Caustic", {{"Corrosive", "Fiendish"}})
addOffspring("Acidic", {{"Corrosive", "Caustic"}})
addOffspring("Excited", {{"Cultivated", "Valiant"}})
addOffspring("Energetic", {{"Excited", "Valiant"}})
addOffspring("Frigid", {{"Wintry", "Diligent"}})
addOffspring("Absolute", {{"Ocean", "Frigid"}})
addOffspring("Shadowed", {{"Tolerant", "Sinister"}})
addOffspring("Darkened", {{"Shadowed", "Embittered"}})
addOffspring("Abyssal", {{"Shadowed", "Darkened"}})
addOffspring("Maroon", {{"Forest", "Valiant"}})
addOffspring("Saffron", {{"Meadows", "Valiant"}})
addOffspring("Prussian", {{"Water", "Valiant"}})
addOffspring("Natural", {{"Tropical", "Valiant"}})
addOffspring("Ebony", {{"Rocky", "Valiant"}})
addOffspring("Bleached", {{"Wintry", "Valiant"}})
addOffspring("Sepia", {{"Marshy", "Valiant"}})
addOffspring("Amber", {{"Maroon", "Saffron"}})
addOffspring("Turquoise", {{"Natural", "Prussian"}})
addOffspring("Indigo", {{"Maroon", "Prussian"}})
addOffspring("Slate", {{"Ebony", "Bleached"}})
addOffspring("Azure", {{"Prussian", "Bleached"}})
addOffspring("Lavender", {{"Maroon", "Bleached"}})
addOffspring("Lime", {{"Natural", "Bleached"}})
addOffspring("Fuchsia", {{"Indigo", "Lavender"}})
addOffspring("Ashen", {{"Slate", "Bleached"}})
addOffspring("Celebratory", {{"Austere", "Excited"}})
addOffspring("Jaded", {{"Ender", "Relic"}})
addOffspring("Glowering", {{"Furious", "Excited"}})
addOffspring("Hazardous", {{"Austere", "Desolate"}})
addOffspring("Lustered", {{"Resilient", "Unweary"}})
addOffspring("Abnormal", {{"Secluded", "Ender"}})
addOffspring("Spatial", {{"Abnormal", "Hermitic"}})
addOffspring("Quantum", {{"Spatial", "Spectral"}})
addOffspring("Eldritch", choose({"Cultivated"}, {"Mystical", "Sorcerous", "Unusual", "Attuned"}))
addOffspring("Esoteric", {{"Cultivated", "Eldritch"}})
addOffspring("Mysterious", {{"Eldritch", "Esoteric"}})
addOffspring("Arcane", {{"Esoteric", "Mysterious"}})
addOffspring("Charmed", {{"Cultivated", "Eldritch"}})
addOffspring("Enchanted", {{"Eldritch", "Charmed"}})
addOffspring("Supernatural", {{"Charmed", "Enchanted"}})
addOffspring("Ethereal", {{"Arcane", "Supernatural"}})
addOffspring("Watery", {{"Supernatural", "Ethereal"}})
addOffspring("Earthen", {{"Supernatural", "Ethereal"}})
addOffspring("Firey", {{"Supernatural", "Ethereal"}})
addOffspring("Windy", {{"Supernatural", "Ethereal"}})
addOffspring("Pupil", {{"Monastic", "Arcane"}})
addOffspring("Scholarly", {{"Arcane", "Pupil"}})
addOffspring("Savant", {{"Pupil", "Scholarly"}})
addOffspring("Aware", {{"Ethereal", "Attuned"}})
addOffspring("Spirit", choose({"Aware"}, {"MystEtherealical", "Attuned"}))
addOffspring("Soul", {{"Aware", "Spirit"}})
addOffspring("Skulking", {{"Modest", "Eldritch"}})
addOffspring("Ghastly", {{"Skulking", "Ethereal"}})
addOffspring("Spidery", {{"Tropical", "Skulking"}})
addOffspring("Smouldering", {{"Skulking", "Hateful"}})
addOffspring("Timely", {{"Imperial", "Ethereal"}})
addOffspring("Lordly", {{"Imperial", "Timely"}})
addOffspring("Doctoral", {{"Timely", "Lordly"}})
addOffspring("Infernal", {{"Infernal", "ves"}})
addOffspring("Hateful", {{"Infernal", "Eldritch"}})
addOffspring("Spiteful", {{"Infernal", "Hateful"}})
addOffspring("Withering", {{"Demonic", "Spiteful"}})
addOffspring("Oblivion", {{"Oblivion", "ves"}})
addOffspring("Nameless", {{"Ethereal", "Oblivion"}})
addOffspring("Abandoned", {{"Oblivion", "Nameless"}})
addOffspring("Forlorn", {{"Nameless", "Abandoned"}})
addOffspring("Draconic", {{"Imperial", "Abandoned"}})
addOffspring("Ferrous", {{"Common", "Industrious"}})
addOffspring("Auric", {{"Minium", "Plumbum"}})
addOffspring("Cuprum", {{"Industrious", "Meadows"}})
addOffspring("Stannum", {{"Industrious", "Forest"}})
addOffspring("Argentum", {{"Imperial", "Modest"}})
addOffspring("Plumbum", {{"Stannum", "Common"}})
addOffspring("Aluminum", {{"Industrious", "Cultivated"}})
addOffspring("Ardite", {{"Industrious", "Infernal"}})
addOffspring("Cobalt", {{"Imperial", "Infernal"}})
addOffspring("Manyullyn", {{"Ardite", "Cobalt"}})
addOffspring("Diamandi", {{"Austere", "Auric"}})
addOffspring("Esmeraldi", {{"Austere", "Argentum"}})
addOffspring("Apatine", {{"Rural", "Cuprum"}})
addOffspring("Mutable", {{"Unusual", "Eldritch"}})
addOffspring("Transmuting", {{"Unusual", "Mutable"}})
addOffspring("Crumbling", {{"Unusual", "Mutable"}})
addOffspring("Invisible", {{"Mystical", "Mutable"}})
addOffspring("Aer", {{"Windy", "Windy"}})
addOffspring("Ignis", {{"Firey", "Firey"}})
addOffspring("Aqua", {{"Watery", "Watery"}})
addOffspring("Solum", {{"Earthen", "Earthen"}})
addOffspring("Ordered", {{"Ethereal", "Arcane"}})
addOffspring("Chaotic", {{"Ethereal", "Supernatural"}})
addOffspring("Brainy", {{"Skulking", "Pupil"}})
addOffspring("Batty", {{"Skulking", "Windy"}})
addOffspring("Poultry", {{"Common", "Skulking"}})
addOffspring("Beefy", {{"Common", "Skulking"}})
addOffspring("Porcine", {{"Common", "Skulking"}})
addOffspring("Minium", {{"Frugal", "Eldritch"}})




function findAvailableParent(species, score, type)

	for i, stack in pairs(stacks) do
		if stack.beeInfo.displayName == species then
			if type then
				if string.find(stack.name, type) then
					return species, nil, score
				end
			else
				return species, nil, score
			end
		end
	end

	local aScore, aParent, aOffspring
	local beeScore = 99999
	local beeParent = nil
	local beeOffspring = nil

	for i, parents in ipairs(bees[species].mutateFrom) do
		aParent, aOffspring, aScore = findAvailableParent(parents[1], score + 1, type)
		if aScore < beeScore then
			beeScore = aScore
			beeParent = aParent
			beeOffspring = aOffspring
		end
		aParent, aOffspring, aScore = findAvailableParent(parents[2], score + 1, type)
		if aScore < beeScore then
			beeScore = aScore
			beeParent = aParent
			beeOffspring = aOffspring
		end
	end
	if beeOffspring == nil then
		beeOffspring = species
	end
	return beeParent, beeOffspring, beeScore
end


local function findSpecies(species)
	local princessIndex = nil
	local droneIndex = nil

	for i, stack in pairs(stacks) do
		if stack.beeInfo.displayName == species then
			if string.find(stack.name, "Princess") then
				princessIndex = i
			end
			if string.find(stack.name, "Drone") then
				droneIndex = i
			end
		end
	end
	return princessIndex, droneIndex
end

local function breedSpecies(parent1, parent2)
	local princess1 = nil
	local princess2 = nil
	local drone1 = nil
	local drone2 = nil

	princess1, drone1 = findSpecies(parent1)
	princess2, drone2 = findSpecies(parent2)

	if not (princess1 and drone2) then
		princess1 = princess2
		drone2 = drone1
	end

	print("Breeding " .. parent1 .. " with " .. parent2)

	if princess1 and drone2 then
		chest.pushItem("up", princess1, 1) --todo check against description
		turtle.drop()
		chest.pushItem("up", drone2, 1) --todo check against description
		turtle.drop()
		return true
	end
	return false
end

local function findExtraBee(type, msg)
	local counts = {}
	local bestcount = 0
	local bestspecies = nil

	for i, stack in pairs(stacks) do
		if string.find(stack.name, type) then
			if counts[stack.beeInfo.displayName] then
				counts[stack.beeInfo.displayName] = counts[stack.beeInfo.displayName] + 1
			else
				counts[stack.beeInfo.displayName] = 1
			end
			if counts[stack.beeInfo.displayName] > bestcount then
				bestcount = counts[stack.beeInfo.displayName]
				bestspecies = stack.beeInfo.displayName
			end
		end
	end
	if bestcount < 2 then
		shutdownMsg = "Need more " .. type .. "s!";
	else
		if msg == "Used" or bestcount > maxDrones then
			print(msg .. " Extra " .. bestspecies .. " " .. type)
		end
	end
	return bestspecies, bestcount
end

local function purebreed(species)
	local princessCount = 0
	local droneCount = 0

	--count how many i have

	for i, stack in pairs(stacks) do
		if stack.beeInfo.displayName == species then
			if string.find(stack.name, "Princess") then
				princessCount = princessCount + stack.qty
			end
			if string.find(stack.name, "Drone") then
				droneCount = droneCount + stack.qty
			end
		end
	end

	if princessCount >= 1 and droneCount >= minDrones then
		return true
	end

	print("Attempting to purebreed " .. species)

	if princessCount >= 1 and droneCount >= 1 then
		breedSpecies(species, species)
		return false
	end

	--find something to breed it with, an available parent, or an extra bee

	local parent1, offspring, score
	if(princessCount >= 1) then
		parent1, offspring, score = findAvailableParent(species, 0, "Drone")
		if score == 99999 then
			parent1, score = findExtraBee("Drone", "Used")
		else
			local princess1, drone1 = findSpecies(parent1)
			if princess1 and drone2 and not purebreed(parent1) then
				return false
			end
		end
	else
		parent1, offspring, score = findAvailableParent(species, 0, "Princess")
		if score == 99999 then
			parent1, score = findExtraBee("Princess", "Used")
		else
			local princess1, drone1 = findSpecies(parent1)
			if princess1 and drone2 and not purebreed(parent1) then
				return false
			end
		end
	end
	if shutdownMsg then
		return false
	end

	breedSpecies(species, parent1)
	return false
end

local function targetSpecies(species)
	local parent1
	local parent2 = nil

	--find the highest level bee i have that is a parent to species
	--offspring would be the next step after parent on the way to species

	local parent1, offspring, score = findAvailableParent(species, 0, nil)
	if score == 0 then
		--found species
		return true
	end
	if score == 99999 then
		shutdownMsg = "No available parents for " .. species
		return false
	end

	print("Targeting Species " .. offspring)

	local princess1, drone1 = findSpecies(parent1)
	local princess2, drone2

	--find which bee i need to breed parent1 with to get offspring, showing preference for bees that i own

	for i, parents in ipairs(bees[offspring].mutateFrom) do
		if parents[1] == parent1 then
			princess2, drone2 = findSpecies(parents[2])
			if (princess1 and drone2) or (princess2 and drone1) then
				parent2 = parents[2]
				break
			end
			if not parent2 then
				parent2 = parents[2]
			end
			if princess2 or drone2 then
				parent2 = parents[2]
			end
		end
		if parents[2] == parent1 then
			princess2, drone2 = findSpecies(parents[1])
			if (princess1 and drone2) or (princess2 and drone1) then
				parent2 = parents[1]
				break
			end
			if not parent2 then
				parent2 = parents[1]
			end
			if princess2 or drone2 then
				parent2 = parents[1]
			end
		end

	end

	princess2, drone2 = findSpecies(parent2)

	if (princess1 and drone2) or (princess2 and drone1) then
		--i have both bees i need so breed it now
		if princess1 and drone1 and not purebreed(parent1) then
			return false
		end
		if princess2 and drone2 and not purebreed(parent2) then
			return false
		end
		breedSpecies(parent1, parent2)
		return false
	end

	--i have the species but not princess or drone, so purebreed it
	if princess2 or drone2 then
		purebreed(parent2)
		return false
	end

	--i dont have the species i need to breed with parent, so target it
	targetSpecies(parent2)
	return false
end

local function dropExtraDrones()

	local species, count = findExtraBee("Drone", "Dropping")
	shutdownMsg = nil

	if count <= maxDrones then
		return
	end

	for i, stack in pairs(stacks) do
		if stack.beeInfo.displayName == species and string.find(stack.name, "Drone") then
			chest.pushItem("up", i, 1)
			turtle.dropUp()
			count = count - 1
			if count <= maxDrones then
				return
			end
		end
	end

end


local function main(species)

	for slot = 1, 16 do
		if turtle.getItemCount(slot) > 0 then
			turtle.select(slot)
			turtle.dropDown()
		end
	end
	turtle.select(1)

	local apiarystacks

	while(1) do

		--test for breeding in progress
		apiarystacks = apiary.getAllStacks()
		while apiarystacks[1] do
			sleep(1)
			while(turtle.suck()) do
				turtle.dropDown()
			end

			apiarystacks = apiary.getAllStacks()
		end

		--clear apiary
		while(turtle.suck()) do
			turtle.dropDown()
		end

		stacks = chest.getAllStacks()

		if targetSpecies(species) then
			if purebreed(species) then
				shutdownMsg = "Breeding complete: " .. species
			end
		end

		if(shutdownMsg) then
			print(shutdownMsg)
			break
		end

		dropExtraDrones()

		sleep(1)
	end
end

local species = ...
if species == nil then
	print("Usage")
	print("   programname species")
	return
end
if bees[species] == nil then
	print("Species does not exist in database")
end

main(species)