plethora crafter rename to milo - wip

milo/plugins/turtleLearn.lua

@@ -0,0 +1,175 @@
+local Craft  = require('turtle.craft')
+local itemDB = require('itemDB')
+local Milo   = require('milo')
+local UI     = require('ui')
+local Util   = require('util')
+
+local device = _G.device
+local turtle = _G.turtle
+
+local context = Milo:getContext()
+
+local function getTurtleInventory()
+	local introspectionModule = device['plethora:introspection'] or
+		error('Introspection module not found')
+
+	local list = { }
+	for i = 1,16 do
+		list[i] = introspectionModule.getInventory().getItemMeta(i)
+	end
+	return list
+end
+
+local function learnRecipe()
+	local ingredients = getTurtleInventory()
+
+	if not ingredients then
+		return false, 'No recipe defined'
+	end
+
+	turtle.select(1)
+	if not turtle.craft() then
+		return false, 'Failed to craft'
+	end
+
+	local results = getTurtleInventory()
+	if not results or not results[1] then
+		return false, 'Failed to craft'
+	end
+
+	local maxCount
+	local newRecipe = {
+		ingredients = ingredients,
+	}
+
+	local numResults = 0
+	for _,v in pairs(results) do
+		if v.count > 0 then
+			numResults = numResults + 1
+		end
+	end
+	if numResults > 1 then
+		for _,v1 in pairs(results) do
+			for _,v2 in pairs(ingredients) do
+				if v1.name == v2.name and
+					v1.nbtHash == v2.nbtHash and
+					(v1.damage == v2.damage or
+						(v1.maxDamage > 0 and v2.maxDamage > 0 and
+						 v1.damage ~= v2.damage)) then
+					if not newRecipe.crafingTools then
+						newRecipe.craftingTools = { }
+					end
+					local tool = Util.shallowCopy(v2)
+					if tool.maxDamage > 0 then
+						tool.damage = '*'
+					end
+
+					--[[
+					Turtles can only craft one item at a time using a tool :(
+					]]--
+					maxCount = 1
+
+					newRecipe.craftingTools[Milo:uniqueKey(tool)] = true
+					v1.craftingTool = true
+					break
+				end
+			end
+		end
+	end
+
+	local recipe
+	for _,v in pairs(results) do
+		if not v.craftingTool then
+			recipe = v
+			if maxCount then
+				recipe.maxCount = maxCount
+			end
+			break
+		end
+	end
+
+	if not recipe then
+		debug(results)
+		debug(newRecipe)
+		error('Failed - view system log')
+	end
+
+	newRecipe.count = recipe.count
+
+	local key = Milo:uniqueKey(recipe)
+	if recipe.maxCount ~= 64 then
+		newRecipe.maxCount = recipe.maxCount
+	end
+	for k,ingredient in pairs(Util.shallowCopy(ingredients)) do
+		if ingredient.maxDamage > 0 then
+			-- ingredient.damage = '*'               -- I don't think this is right
+		end
+		ingredients[k] = Milo:uniqueKey(ingredient)
+	end
+
+	context.userRecipes[key] = newRecipe
+	Util.writeTable(Milo.RECIPES_FILE, context.userRecipes)
+	Craft.loadRecipes()
+
+	turtle.emptyInventory()
+
+	return recipe
+end
+
+local turtleLearnWizard = UI.Page {
+	titleBar = UI.TitleBar { title = 'Learn a crafting recipe' },
+	wizard = UI.Wizard {
+		y = 2, ey = -3,
+		pages = {
+			confirmation = UI.Window {
+				index = 1,
+				notice = UI.TextArea {
+					x = 2, ex = -2, y = 2, ey = -2,
+					value =
+[[Place recipe in turtle!]],
+				},
+			},
+		},
+	},
+	notification = UI.Notification { },
+}
+
+function turtleLearnWizard:enable()
+	Milo:pauseCrafting()
+	UI.Page.enable(self)
+end
+
+function turtleLearnWizard:disable()
+	Milo:resumeCrafting()
+	UI.Page.disable(self)
+end
+
+function turtleLearnWizard.wizard.pages.confirmation:validate()
+	local recipe, msg = learnRecipe(self)
+
+	if recipe then
+		local listingPage = UI:getPage('listing')
+		local displayName = itemDB:getName(recipe)
+
+		listingPage.statusBar.filter:setValue(displayName)
+		listingPage.notification:success('Learned: ' .. displayName)
+		listingPage.filter = displayName
+		listingPage:refresh()
+		listingPage.grid:draw()
+
+		return true
+	else
+		turtleLearnWizard.notification:error(msg)
+	end
+end
+
+function turtleLearnWizard:eventHandler(event)
+	if event.type == 'cancel' or event.type == 'accept' then
+		UI:setPage('listing')
+	else
+		return UI.Page.eventHandler(self, event)
+	end
+	return true
+end
+
+context.learnTypes['Turtle crafting'] = turtleLearnWizard