ATM Pull API

--ATM version 0.0.6--
--Must be called pullapi.lua--

local chests = {"minecraft:chest_0"}

local dropper = "minecraft:dropper_0"
local junk = "minecraft:dropper_1"

local coin = "minecraft:diamond"

local d = peripheral.wrap(dropper)
local j = peripheral.wrap(junk)


function getStored()
    local count = 0
    for k, v in pairs(chests) do
        cs = peripheral.wrap(v)
        for k2, v2 in pairs(cs.list()) do
            if v2.name == coin then
                count = count + v2.count
            else
                j.pullItems(v, k2, v2.count)
            end
        end
    end
    return count
end

function deposit()
    local count = 0
    for k, v in pairs(chests) do
        for i = 1, 9 do
            local tab = d.getItemDetail(i)
            if tab ~= nil then
                if tab.name == coin then
                    local deped = d.pushItems(v, i, 64)
                    count = count + deped
                else
                    d.pushItems(junk, i, 64)
                end
            end
        end
    end
    if count > 0 then
        return true, count
    else
        return false, "Diamonds enterd must be more than 0"
    end
end

function withdraw(amount)
    if amount > 576 then
        return false, "amount must be at maximum 9 stacks or 576 items"
    end
    if amount > getStored() then
        return false, "not enough money in storage"
    end

    local count = 0
    for k, v in pairs(chests) do
        cs = peripheral.wrap(v)
        for i = 1, 54 do
            tab = cs.getItemDetail(i)
            if tab ~= nil then
                if tab.name == coin then
                    if (count + tab.count) > amount then
                        d.pullItems(v, i, (amount - count))
                        return true
                    else
                        d.pullItems(v, i, tab.count)
                        count = count + tab.count
                    end
                    if count == amount then
                        return true
                    end
                else
                    j.pullItems(chests, i, 64)
                end
            end
        end
    end
end