startup

local serverId = 18
local atmTurtleId = 16
local buttons = { }
local window = paintutils.loadImage("window.nfp")
local items = {{2,"RS"},{2,"Coal"},{2,"Lapis"},{5,"Iron"},{10,"Gold"},{50,"Diamond"},{75,"Obsidian"}}
rednet.open("back")

function button(xPos, yPos, xSize, ySize, bColor, tColor, text, id)
  term.setCursorPos(xPos, yPos)
  term.setBackgroundColor(bColor)
  term.setTextColor(tColor)
  for a = 1, ySize + 1 do
  for b = 1, xSize do
    write(" ")
  end
  term.setCursorPos(xPos, yPos + (a - 1))
  end
  term.setCursorPos(((xSize/2) + xPos) - (#text / 2),yPos + (ySize / 2))
  print(text)
  buttons[id] = {xP = xPos, yP = yPos, xS = xSize, yS = ySize}
end

function checkButton(mouseX, mouseY, id)
  if mouseX >= buttons[id].xP and mouseX <= buttons[id].xP + (buttons[id].xS - 1) and mouseY >= buttons[id].yP and mouseY <= buttons[id].yP + (buttons[id].yS - 1) then
    return true
  else
    return false
  end
end

function drawMainScreen()
    term.setBackgroundColor(colors.white)
    term.clear()
    term.setCursorPos(24,1)
    term.setTextColor(colors.black)
    write("ATM")
    button(7,6,15,3,colors.lime,colors.white,"Create Acount","create")
    button(29,6,15,3,colors.lime,colors.white,"Check Balance","check")
    button(7,11,15,3,colors.lime,colors.white,"Withdraw","withdraw")
    button(29,11,15,3,colors.lime,colors.white,"Deposit","deposit")
end

function chooseItem(amount,string)
    amount = tonumber(amount)
    term.setBackgroundColor(colors.white)
    term.clear()
    term.setCursorPos(17,2)
    term.setTextColor(colors.black)
    if string ~= nil then
        write(string)
        term.setCursorPos(17,3)
    end
        write("What item do you want?")
    for a = 1,#items do
        if amount >= items[a][1] then
            if a <= 4 then
                button(3,4+4*a-4,10,3,colors.lime,colors.white,items[a][2],items[a][2])
            else
                button(14,4+4*(a-4)-4,10,3,colors.lime,colors.white,items[a][2],items[a][2])
            end
        else
            if a <= 4 then
                button(3,4+4*a-4,10,3,colors.lightGray,colors.white,items[a][2],items[a][2].."x")
            else
                button(14,4+4*(a-4)-4,10,3,colors.lightGray,colors.white,items[a][2],items[a][2].."x")
            end
        end
    end
    local e,b,x,y = os.pullEvent("mouse_click")
    for a = 1,#items do
        price = items[a][1]
        if checkButton(x,y,items[a][2]) then
            rednet.send(atmTurtleId,tostring(a)..":"..tostring(math.floor(amount/price)))
            rem = amount - math.floor(amount/price) * price
            if rem > 0 then
                return chooseItem(rem,"Given "..tostring(math.floor(amount/price)).." "..items[a][2])
            end
        end
    end
end

function openWindow(action,string,char)
    if action == "get" then
        paintutils.drawImage(window,1,1)
        term.setCursorPos(25 - #string/2,9)
        term.setBackgroundColor(colors.blue)
        print(string)
        term.setCursorPos(6,10)
        term.setBackgroundColor(colors.gray)
        return read(char)
    elseif action == "info" then
        paintutils.drawImage(window,1,1)
        term.setCursorPos(25 - #string/2,9)
        term.setBackgroundColor(colors.blue)
        print(string)
        term.setCursorPos(12,10)
        term.setBackgroundColor(colors.gray)
        write("Press any key to continue")
        os.pullEvent("key")
        drawMainScreen()
    end
end

function card()
    if disk.isPresent("top") then
        return true
    end
    openWindow("info","Please insert a disk")
    return false
end

function getPin(string)
    while true do
        term.setBackgroundColor(colors.white)
        local pin = openWindow("get",string,"*")
        if #pin == 4 then
            return pin
        end
        openWindow("info","Pin must be 4 digits")
    end
end

function createAcount()
    if card() then
    local pin = getPin("Enter a pin")
    drawMainScreen()
    rednet.send(serverId,"create")
    local i,m,d = rednet.receive()
    sleep(1)
    if i == serverId and m == "pin" then
        rednet.send(serverId,pin)
    end
    i,m,d = rednet.receive()
    disk.setLabel("top",m)
    i,m,d = rednet.receive()
    if m == "success" then
        openWindow("info","Success. Your card number is on your disk")
    end
    end
end

function checkBalance(id)
    if card() then
        rednet.send(serverId,"get_balance")
        local i,m,d = rednet.receive()
        sleep(1)
        if i == serverId and m == "id:pin" then
            local pin = getPin("Enter you pin")
            rednet.send(serverId,id..":"..pin)
        end
        rednet.receive()
        local i,m,d = rednet.receive()
        return m
    end
    return "Error"
end

function withdraw(id)
    rednet.send(serverId,"get_balance")
    local i,m,d = rednet.receive()

    rednet.send(serverId,"change_balance")
    local i,m,d = rednet.receive()
    local pin = getPin("Enter you pin")
    local amount = openWindow("get","Enter the amount you want to withdraw")
    if i == serverId and m == "id:pin:amount" then
        sleep(1)
        rednet.send(serverId,id..":"..pin..":-"..amount)
    end
    local i,m,d = rednet.receive()
    if m == "success" then
        chooseItem(amount)
    else
        openWindow("info","Error")
    end
end

drawMainScreen()
local e,button,x,y = os.pullEvent("mouse_click")
if checkButton(x,y,"create") then
createAcount()
elseif checkButton(x,y,"check") then
openWindow("info","Current balance is [ "..checkBalance(disk.getLabel("top")).." ]")
elseif checkButton(x,y,"withdraw") then
withdraw(disk.getLabel("top"))
end