AutoBee OC

---- AutoBee for OpenComputers ----

local running = true
if pcall(function() dofile("autobeeCore.lua") end) == false then
  if pcall(function() dofile("autobee/autobeeCore.lua") end) == false then
    print("Missing core library, fetching from pastebin.")
    shell.execute("pastebin get Vvckgdst autobeeCore.lua")
  end
end

-- Peripheral check
function peripheralCheck()
  local apiary = nil
  for address, componentType in pairs(component.list()) do
    -- this is the search for 1.7.10 Forestry apiaries
    if string.find(componentType, "apiculture") and componentType:sub(21,21) == "0" then
      apiary = address
    -- this is the search for 1.10.2 Forestry apiaries and bee houses
    elseif componentType == "bee_housing" then
      apiary = address
      print("WARNING: OpenComputers on 1.10.2 does not support necessary APIs. See https://github.com/jetpack-maniac/autobee for more information.")
      os.exit()
    end
  end
  if apiary == nil then
    print("No apiaries found.  Closing program.")
    os.exit()
  else
    dependencyCheck(component.proxy(apiary))
  end
end

-- looks at a device and determines if it's a valid apiary, returns true or false
function isApiary(address)
  if address == nil then return false end
  device = component.type(address)
  -- 1.10.2 Apiaries
  if string.find(device, "bee_housing") then -- TODO discriminate apiaries versus bee houses
    return true
  -- 1.7.10 Apiaries
  elseif string.find(device, "apiculture") and device:sub(21,21) == "0" then
    return true
  else
  return false
  end
end

-- examines peripherals and returns a valid apiary or nil
function findApiary()
  local devices = component.list()
  for address, device in pairs(devices) do
    if isApiary(address) then
      return component.proxy(address)
    else
      return false
    end
  end
end

-- Version Check
if pcall(function() component = require("component") end) == true then
  if component.proxy ~= nil then
    version = "OpenComputers"
    keyboard = require("keyboard")
    event = require("event")
    term = require("term")
    peripheralCheck()
    print("AutoBee running.")
    print("Hold Ctrl+W to stop. Hold Ctrl+L to clear terminal.")
  else
    error("This version is for OpenComputers.  See https://github.com/jetpack-maniac/autobee for more details.")
  end
end

local apiaryTimerIDs = {}

--These next two functions are callbacks to enable OC events to work properly
function deviceConnect(event, address)
  addDevice(address)
end

function deviceDisconnect(event, address)
  removeDevices()
  initDevices()
end

function removeDevices()
  for timerID, address in pairs(apiaryTimerIDs) do
    event.cancel(address)
  end
  apiaryTimerIDs = {}
end

function addDevice(address)
  if address == nil then return false end
  if isApiary(address) == true then
    local apiary = Apiary(component.proxy(address), address)
    apiaryTimerIDs[address] = event.timer(delay, function() apiary.checkApiary() end, math.huge)
    print(size(apiaryTimerIDs).." apiaries connected.")
    return true
  end
  return false
end

function initDevices()
  --Aux ignore in case the program crashed and listeners are still active
  event.ignore("component_available",deviceConnect)
  event.ignore("component_removed",deviceDisconnect)
  local devices = component.list()
  for address, device in pairs(devices) do
    addDevice(address)
  end
  event.listen("component_added",deviceConnect)
  event.listen("component_removed",deviceDisconnect)
end

----------------------
-- The main loop
----------------------

if running == true then initDevices() end -- inits devices upon first run

while running do
  if version == "OpenComputers" then
    if keyboard.isKeyDown(keyboard.keys.w) and keyboard.isControlDown() then
      event.ignore("component_available",deviceConnect)
      event.ignore("component_removed",deviceDisconnect)
      removeDevices()
      print("AutoBee: Interrupt detected. Closing program.")
      break
    end
    if keyboard.isKeyDown(keyboard.keys.l) and keyboard.isControlDown() then
      term.clear()
      print("AutoBee running.")
      print("Hold Ctrl+W to stop. Hold Ctrl+L to clear terminal.")
      print(size(apiaryTimerIDs).." apiaries connected.")
    end
    os.sleep(delay)
  end
end