package management

forestry/beeInfo.lua

@@ -0,0 +1,176 @@
+_G.requireInjector(_ENV)
+
+local Event = require('event')
+local UI = require('ui')
+local Util = require('util')
+
+local chest = peripheral.wrap('bottom')
+
+local data
+local monitor = UI.Device({
+  deviceType = 'monitor',
+  textScale = .5
+})
+
+UI:setDefaultDevice(monitor)
+
+local breedingPage = UI.Page({
+  titleBar = UI.TitleBar(),
+  grid = UI.Grid({
+    columns = {
+      { heading  = '  ', key = 'chance' },
+      { heading = 'Princess', key = 'princess', },
+      { heading = 'Drone', key = 'drone' },
+      { heading = 'Result', key = 'result', },
+    },
+    y = 2, ey = -8,
+    sortColumn = 'result',
+    autospace = true
+  }),
+  specialConditions = UI.Window({
+    backgroundColor = colors.red,
+    y = -7,
+    height = 2
+  }),
+  buttons = UI.Window({
+    y = monitor.height - 4,
+    width = monitor.width,
+    height = 5,
+    backgroundColor = colors.gray,
+    prevButton = UI.Button({
+      event = 'previous',
+      x = 2,
+      y = 2,
+      height = 3,
+      width = 5,
+      text = ' < '
+    }),
+    resetButton = UI.Button({
+      event = 'clear',
+      x = 8,
+      y = 2,
+      height = 3,
+      width = monitor.width - 14,
+      text = 'Clear'
+    }),
+    nextButton = UI.Button({
+      event = 'next',
+      x = monitor.width - 5,
+      y = 2,
+      height = 3,
+      width = 5,
+      text = ' > '
+    })
+  })
+})
+
+function breedingPage:getBreedingData()
+  self.grid.values = { }
+  local stacks = chest.getAllStacks(false)
+  local stack = stacks[1]
+  self.titleBar.title = stack.individual.displayName
+  if stack.individual.active then
+  end
+  for _,d in pairs(data) do
+    if d.allele1 == stack.individual.displayName or
+       d.allele2 == stack.individual.displayName then
+      local ind = ''
+      if d.specialConditions then
+        ind = '*'
+      end
+      table.insert(self.grid.values, {
+        princess = d.allele1 .. ind,
+        drone = d.allele2,
+        result = d.result,
+        chance = d.chance .. '%',
+        specialConditions = d.specialConditions
+      })
+    end
+  end
+  self.grid.index = 1
+  self.grid:adjustWidth()
+  self.grid:update()
+  self:draw()
+  self:sync()
+end
+
+function breedingPage.specialConditions:draw()
+  local selected = self.parent.grid:getSelected()
+  if selected and selected.specialConditions then
+    local sc = ''
+    if selected.specialConditions then
+      for _,v in ipairs(selected.specialConditions) do
+        if sc ~= '' then
+          sc = sc .. ', '
+        end
+        sc = sc .. v
+      end
+    end
+    self:clear()
+    self:setCursorPos(2, 1)
+    self:print(sc)
+  else
+    self:clear(colors.red)
+  end
+end
+
+function breedingPage.grid:draw()
+  UI.Grid.draw(self)
+  self.parent.specialConditions:draw()
+end
+
+function breedingPage:eventHandler(event)
+  if event.type == 'next' then
+    self.grid:setPage(self.grid:getPage() + 1)
+  elseif event.type == 'previous' then
+    self.grid:setPage(self.grid:getPage() - 1)
+  elseif event.type == 'clear' then
+    self.grid:setTable({})
+    self.grid:draw()
+  elseif event.type == 'grid_focus_row' then
+    self.specialConditions:draw()
+  else
+    return UI.Page.eventHandler(self, event)
+  end
+  return false
+end
+
+Event.on('turtle_inventory', function()
+  local slot = turtle.selectSlotWithQuantity(1)
+
+  if slot then
+    turtle.dropDown()
+    breedingPage:getBreedingData()
+    turtle.suckDown()
+    turtle.drop()
+  end
+  
+end)
+
+if not fs.exists('.bee.data') then
+  local p = peripheral.wrap("back")
+  local data = p.getBeeBreedingData()
+  local t = { }
+  for _,d in pairs(data) do
+    d = Util.shallowCopy(d)
+    if type(d.specialConditions) == 'string' then
+      if d.specialConditions == '[]' then
+        d.specialConditions = ''
+      end
+    end
+    if #d.specialConditions == 0 then
+      d.specialConditions = nil
+    else
+      d.specialConditions = Util.shallowCopy(d.specialConditions)
+    end
+    table.insert(t, d)
+  end
+  Util.writeTable('.bee.data', t)
+else
+  data = Util.readTable('.bee.data')
+end
+
+UI:setPage(breedingPage)
+
+UI:pullEvents()
+