opus-apps/milo/plugins/demandCraft.lua

local Craft  = require('turtle.craft')
local itemDB = require('itemDB')
local Milo   = require('milo')
local UI     = require('ui')
local Util   = require('util')

local colors = _G.colors

local demandCrafting = { }

local craftPage = UI.Page {
  titleBar = UI.TitleBar { },
  wizard = UI.Wizard {
    y = 2, ey = -2,
    pages = {
      quantity = UI.Window {
        index = 1,
        text = UI.Text {
          x = 6, y = 3,
          value = 'Quantity',
        },
        count = UI.TextEntry {
          x = 15, y = 3, width = 10,
          limit = 6,
          value = 1,
        },
        ejectText = UI.Text {
          x = 6, y = 4,
          value = 'Eject',
        },
        eject = UI.Chooser {
          x = 15, y = 4, width = 7,
          value = true,
          nochoice = 'No',
          choices = {
            { name = 'Yes', value = true },
            { name = 'No', value = false },
          },
        },
      },
      resources = UI.Window {
        index = 2,
        grid = UI.ScrollingGrid {
          y = 2, ey = -2,
          columns = {
            { heading = 'Name',  key = 'displayName' },
            { heading = 'Total', key = 'total'      , width = 5 },
            { heading = 'Used',  key = 'used'       , width = 5 },
            { heading = 'Need',  key = 'need'       , width = 5 },
          },
          sortColumn = 'displayName',
        },
      },
    },
  },
}

function craftPage:enable(item)
  self.item = item
  self:focusFirst()
  self.titleBar.title = itemDB:getName(item)
--  self.wizard.pages.quantity.eject.value = true
  UI.Page.enable(self)
end

function craftPage.wizard.pages.resources.grid:getDisplayValues(row)
  local function dv(v)
    if v == 0 then
      return ''
    end
    return Util.toBytes(v)
  end
  row = Util.shallowCopy(row)
  row.total = Util.toBytes(row.total)
  row.used = dv(row.used)
  row.need = dv(row.need)
  return row
end

function craftPage.wizard.pages.resources.grid:getRowTextColor(row, selected)
  if row.need > 0 then
    return colors.orange
  end
  return UI.Grid:getRowTextColor(row, selected)
end

function craftPage.wizard:eventHandler(event)
  if event.type == 'nextView' then
    local count = tonumber(self.pages.quantity.count.value)
    if not count or count <= 0 then
      self.pages.quantity.count.backgroundColor = colors.red
      self.pages.quantity.count:draw()
      return false
    end
    self.pages.quantity.count.backgroundColor = colors.black
  end
  return UI.Wizard.eventHandler(self, event)
end

function craftPage.wizard.pages.resources:enable()
  local items = Milo:listItems()
  local count = tonumber(self.parent.quantity.count.value)
  local recipe = Craft.findRecipe(craftPage.item)
  if recipe then
    local ingredients = Craft.getResourceList4(recipe, items, count)
    for _,v in pairs(ingredients) do
      v.displayName = itemDB:getName(v)
    end
    self.grid:setValues(ingredients)
  else
    self.grid:setValues({ })
  end
  return UI.Window.enable(self)
end

function craftPage:eventHandler(event)
  if event.type == 'cancel' then
    UI:setPreviousPage()

  elseif event.type == 'accept' then
    local key = Milo:uniqueKey(self.item)
    demandCrafting[key] = Util.shallowCopy(self.item)
    demandCrafting[key].count = tonumber(self.wizard.pages.quantity.count.value)
    demandCrafting[key].ocount = demandCrafting[key].count
    demandCrafting[key].forceCrafting = true
    demandCrafting[key].eject = self.wizard.pages.quantity.eject.value == true
    UI:setPreviousPage()
  else
    return UI.Page.eventHandler(self, event)
  end
  return true
end

local demandCraftingTask = {
  priority = 20,
}

function demandCraftingTask:cycle(context)
  local demandCrafted = { }

  -- look directly at the adapter import activity to determine
  -- if the item was imported into storage from any source.
  -- The item does NOT need to come from the machine that did
  -- the crafting.
  for _,key in pairs(Util.keys(demandCrafting)) do
    local item = demandCrafting[key]

    local imported = context.inventoryAdapter.activity[key]
    if imported then
      item.crafted = math.min(imported, item.count)
      item.count = math.max(0, item.count - item.crafted)
      context.inventoryAdapter.activity[key] = imported - item.crafted
    end
    demandCrafted[key] = item
  end

  if Util.size(demandCrafted) > 0 then
    Milo:craftItems(demandCrafted)
  end

  for _,key in pairs(Util.keys(demandCrafting)) do
    local item = demandCrafting[key]
    if item.crafted then
      item.count = math.max(0, item.count - item.crafted)
      if item.count <= 0 then
        item.statusCode = 'success'
        demandCrafting[key] = nil
        if item.eject then
          Milo:eject(item, item.ocount)
        end
      end
    end
  end
end

UI:addPage('craft', craftPage)
Milo:registerTask(demandCraftingTask)