smart stockpile receiver

--Receiver

-- Define the inventory to store items in
storage = "occultism:stable_wormhole_0"







configured = true
--------------------------------------------
-- Initialize
local toInventory = peripheral.wrap(storage)
local itemToTransfer = 0
local amountToTransfer = 0
fromInventories = {}
modem = nil
modem2 = nil


-- Connect peripherals
local sides = {"left", "right", "top", "bottom", "front", "back"}
for i, side in ipairs(sides) do
  local device = peripheral.getType(side)
  if device == "modem" then
    rednet.open(side)
    if modem == nil then
    modem = peripheral.wrap(side)
    else
    modem2 = peripheral.wrap(side)
    end
    -- print("Opened rednet on " .. side)
  elseif device then
    local wrapper = peripheral.wrap(side)
    -- print("Wrapped " .. device .. " on " .. side)
  end
end


function checkWireless(modemName)
    for i, side in ipairs(sides) do
        if modemName ~= nil then
            if modemName.isWireless() then
                return true
            else
                return false
            end
        else
            return "noModem"
        end
    end
end

if checkWireless(modem) == "noModem" or checkWireless(modem2) == "noModem" then
    term.clear()
    term.setCursorPos(1,1)
    term.setTextColor(colors.orange)
    print("Missing modems!\n\nNeed one wired and one wireless modem.")
    event, key = os.pullEvent(key)
    term.setTextColor(colors.red)
    print("Rebooting.")
    sleep(3)
    os.reboot()
end


if checkWireless(modem) == false and checkWireless(modem2) == false then
    term.clear()
    term.setCursorPos(1,1)
    term.setTextColor(colors.orange)
    print("No >wireless< modem connected!\n\nConnect one and press any key.")
    event, key = os.pullEvent(key)
    term.setTextColor(colors.red)
    print("Rebooting.")
    sleep(3)
    os.reboot()
elseif checkWireless(modem) == true and checkWireless(modem2) == true then
    term.clear()
    term.setCursorPos(1,1)
    term.setTextColor(colors.orange)
    print("No >wired< modem connected!\n\nConnect one and press any key.")
    event, key = os.pullEvent(key)
    term.setTextColor(colors.red)
    print("Rebooting.")
    sleep(3)
    os.reboot()
end



function scanAndSendInventories(fromInventories)
  local scannedItems = {}
  local sentItems = {}
  for i = 1, #fromInventories do
    local inventory = peripheral.wrap(fromInventories[i])
    for slot, item in pairs(inventory.list()) do
      if not scannedItems[item.name] then
        scannedItems[item.name] = item.count
      else
        scannedItems[item.name] = scannedItems[item.name] + item.count
      end
    end
  end

  for itemName, itemCount in pairs(scannedItems) do
    if not sentItems[itemName] then
      sendRednetMessage(itemName, "stockpileupdate")
      sentItems[itemName] = true
    end
  end
end

-- Function to send rednet message
function sendRednetMessage(itemName, protocoll)
rednet.broadcast("saveItem " .. itemName, protocoll)
end

function moveItems()
  -- Loop through the list of inventories
  for i, fromName in ipairs(fromInventories) do
    local fromInventory = peripheral.wrap(fromName)

    -- Get a list of all items in the inventory
    local items = fromInventory.list()

    -- Loop through the items in the inventory
    for slot, item in pairs(items) do
      -- Check if the item is the specified item to transfer
      if item.name == itemToTransfer then
        local transferred = fromInventory.pushItems(peripheral.getName(toInventory), slot, amountToTransfer)
        amountToTransfer = amountToTransfer - transferred
        if amountToTransfer <= 0 then
          return
        end
      end
    end
  end
end

function split(s, delimiter)
    result = {};
    for match in (s..delimiter):gmatch("(.-)"..delimiter) do
        table.insert(result, match);
    end
    return result;
end

function receive()
    senderID, message, protocol = rednet.receive()
    split(message, " ")
    itemToTransfer = tostring(result[1])
    amountToTransfer = tonumber(result[2])
end

function wholething()
    receive()
    if protocol == "stockpile" then
    term.setTextColor(colors.white)
    print(itemToTransfer, ": ",amountToTransfer)
    moveItems()
    end
end

function updateAvailableItems()
    term.clear()
    term.setCursorPos(1,1)
    print("Scan all attached inventories and send them to my brain?\n")
    term.setTextColor(colors.red)
    print("Hit Enter to confirm.")
    local event, key = os.pullEvent("key")
    if key == keys.enter then
        scanAndSendInventories(fromInventories)
        term.clear()
        term.setCursorPos(1,1)
        term.setTextColor(colors.lime)
        print("Items transmitted!")
        sleep(2)
        os.reboot()
end
end

function retransmitItems()
    event, key = os.pullEvent("key")
    if key == keys.space then
        updateAvailableItems()
    end
end

if configured == false then
    updateAvailableItems()
    local event, key = os.pullEvent("key")
    shell.run("edit ", "startup")
end


--initialize inventories
if checkWireless(modem) == false then
newInventories = modem.getNamesRemote()
else
newInventories = modem2.getNamesRemote()
end

for _, inventoryName in ipairs(newInventories) do
    if inventoryName ~= storage then
            -- print(inventoryName)
      table.insert(fromInventories, inventoryName)
    end
end

---loopy doo
while true do
parallel.waitForAny(retransmitItems, wholething)
end