startup

--[[

Bank sever - Fork of userAPI by wow



All of the bank's servers and API are made by wow.

All of the bank's graphics are made by Kuruyia.

]]



--Var init

local RP_BOOL = true

if not fs.exists("/StringUtils") then shell.run("pastebin get ad3aUsVw StringUtils") end

os.loadAPI("StringUtils")

local modem = peripheral.find("modem")

local MODEM_CHANNEL = 644



--ProLog 2017 (TM)



local LOG_FILE = "/proLog.log"

local LOG_SCHEM = {}



local function LOGS_GET()

  local OBJECT_UF = fs.open(LOG_FILE, "r")

  local LOGS

  LOGS = textutils.unserialise(OBJECT_UF.readAll())

  if LOGS == nil then

    LOGS = {}

  end

  return LOGS

end



local function proLog(usertoLog, exchangeName, amount)

  --Table creation

  local LOGS = LOGS_GET()

  local newTable = {id = exchangeName, amount = amount}

  if LOGS[usertoLog] == nil then LOGS[usertoLog] = {} end

  local num =  #LOGS[usertoLog]+1

  LOGS[usertoLog][num] = newTable

  local USER_TEMP = fs.open(LOG_FILE, "w")

  USER_TEMP.write(textutils.serialise(LOGS))

  USER_TEMP.close()

end



--Users management

local USER_FILE = "/users"

local DB_SCHEM = {Users = {}, Password = {}, Biolock = {}, AccessLevel = {}, Money = {}}



local function USERS_CREATEDB()

  local USER_TEMP = fs.open(USER_FILE, "w")

  USER_TEMP.write(textutils.serialise(DB_SCHEM))

  USER_TEMP.close()

end



local function USERS_GET()

  local OBJECT_UF = fs.open(USER_FILE, "r")

  local USERS = textutils.unserialise(OBJECT_UF.readAll())

  if USERS == nil then

    USERS_CREATEDB()

    USERS = textutils.unserialise(OBJECT_UF.readAll())

  end

  OBJECT_UF.close()

  return USERS

end



local function USERS_ID(users, tofind)

  local ID = nil

  for i = 1,#users["Users"] do

    if tofind == nil then

      ID = i

    else

      if users["Users"][i] == tofind then

        ID = i

      end

    end

  end

  return ID

end



local function USERS_BIOLOCK(users, tofind)

  local ID = nil

  for i = 1,#users["Biolock"] do

    if tofind == nil then return 0x1

    else

      if users["Biolock"][i] == tofind then

        ID = i

      end

    end

  end

  return ID

end



local function USERS_WHERE(users, tofind)

  local ID = nil

  for i = 1,#users["Where"] do

    if tofind == nil then return 0x1

    else

      if users["Where"][i] == tofind then

        ID = i

      end

    end

  end

  return ID

end



--Encryption and modems

local function SEND(sID, content)

    if type(sID) ~= "number" or content == nil then return 0x1 end

    local Users = USERS_GET()

    --local serialised = textutils.serialise(content)

    --local FinalTable = StringUtils.encrypt(serialised) DO NOT WORK

    modem.transmit(MODEM_CHANNEL, MODEM_CHANNEL, content)

    return 0

end



local function USERS_CREATE(name, pass, biolock, accesslevel)



  local oUSERS = USERS_GET()

  local fUSERS = oUSERS

  local lastID = USERS_ID(oUSERS)



  --Check if user already exist because or AL will not fuck my script ;(

  if USERS_ID(oUSERS, name) ~= nil then return 0x1 end

  if accesslevel > 5 or accesslevel < 1 then return 0x1 end

  if lastID == nil then lastID = 1 else lastID = lastID + 1 end

  if name == nil or pass == nil or accesslevel == nil then

    return 0x1

  end

  if biolock == nil then biolock = "unknown" end

  --end of check



  fUSERS["Users"][lastID] = name

  fUSERS["Password"][lastID] = StringUtils.SHA1(pass)

  fUSERS["Biolock"][lastID] = biolock

  fUSERS["Money"][lastID] = 50

  fUSERS["AccessLevel"][lastID] = accesslevel

  local TO_SAVE = fs.open(USER_FILE,"w")

  TO_SAVE.write(textutils.serialise(fUSERS))

  TO_SAVE.close()

  return 0

end



local function USERS_UPDATE(toUpdate, userID, newUpdate)

    if type(toUpdate) ~= "string" or type(userID) ~= "number" or newUpdate == nil then return 0x1 end

    local oUSERS = USERS_GET()

    --Check if id exist

    local lastID = USERS_ID(oUSERS)

    if userID > lastID then return 0x2 end



    if toUpdate == "password" then

    if type(newUpdate) ~= "string" then return 0x1 end

    oUSERS["Password"][userID] = newUpdate

    local TO_SAVE = fs.open(USER_FILE,"w")

      TO_SAVE.write(textutils.serialise(oUSERS))

      TO_SAVE.close()

    return 0

  elseif toUpdate == "Money" then

    if type(newUpdate) ~= "number" then return 0x1 end

    oUSERS["Money"][userID] = newUpdate

    local TO_SAVE = fs.open(USER_FILE,"w")

      TO_SAVE.write(textutils.serialise(oUSERS))

      TO_SAVE.close()

    return 0

    elseif toUpdate == "accesslevel" then

        if type(newUpdate) ~= "number" then return 0x1 end

        oUSERS["AccessLevel"][userID] = newUpdate

        local TO_SAVE = fs.open(USER_FILE,"w")

        TO_SAVE.write(textutils.serialise(oUSERS))

        TO_SAVE.close()

        return 0

    end



end



--EXEC



local function execute(sTable,sID)

    local uLevel = -1

  local uID = 0

    if sTable.type ~= "ping" then

        if sTable["fromUsername"] == nil or sTable["fromPassword"] == nil then return SEND(sID, 0x3) end

        --Check account

        local uTable = USERS_GET()

        uID = USERS_ID(uTable, sTable.fromUsername)

        if uID == nil then return SEND(sID, 0x2) end

    --Encryption password is: "1" (Encryption of the SHA-1 password with a password and decryption server side)

    if StringUtils.decrypt(sTable.fromPassword, uTable["Password"][uID]) == "1" then

      print(uTable["Users"][uID].." Logged")

      uLevel = uTable["AccessLevel"][uID]

    else

      return SEND(sID, 0x5)

    end

  end

  if sTable.type == "ping" then

    SEND(sID,0x0)



  elseif sTable.type == "transaction" then

    --if you see this: buy Life.exe or Life.bin if you are on Linux. Thank you ! :p

    if sTable["amount"] == nil or sTable["to"] == nil then return SEND(sID, 0x1) end

    local uTable = USERS_GET()

    local toID = USERS_ID(uTable, sTable.to)

    if toID == nil then SEND(sID, 0x9) end

    --Check money

    if uTable["Money"][uID] >= sTable["amount"] then else return SEND(sID, 0x8) end

    --Transfering

    USERS_UPDATE("Money", uID, uTable["Money"][uID] - sTable["amount"])

    USERS_UPDATE("Money", toID, uTable["Money"][toID] + sTable["amount"])

    proLog(uTable["Users"][uID], uTable["Users"][toID], sTable["amount"])

    SEND(sID, 0x0)



  elseif sTable.type == "database" then

    if sTable.todo == "createUser" then

        if uLevel < 4 then return SEND(sID, 0x6) end

      if sTable.Username == nil or sTable.Password == nil or sTable.Biolock == nil or type(sTable.accesslevel) == number then

        print("User creation failed")

        SEND(sID, 0x1)

      else

        if USERS_CREATE(sTable.Username,sTable.Password,sTable.Biolock,sTable.accesslevel) then

          print("User Created")

          SEND(sID, 0x0)

        else

          print("Failed to create")

          SEND(sID, 0x7)

        end

      end



      elseif sTable.todo == "findBL" then

            if uLevel ~= 0 and uLevel ~= 5 then return SEND(sID, "Bad accessLevel") end

            if type(sTable.toFind) ~= "string" then return SEND(sID, 0x1) end

            local uTable = USERS_GET()

            local uID = USERS_BIOLOCK(uTable, sTable.toFind)

            if uID == nil then return SEND(sID, 0x0) end -- 0x0 <- Unknown user

            return SEND(sID, uTable["AccessLevel"][uID])



    elseif sTable.todo == "reset" then

        if uLevel < 4 then return SEND(sID, "Bad accessLevel") end

      USERS_CREATEDB()

      print("Database reseted") --Pls add a Terminal confirm

      SEND(sID, 0x0)



      elseif sTable.todo == "Userinfo" then

      local TUserTable = USERS_GET()

      local Tid = uID

      if Tid == nil then SEND(sID, 0x2) else

        local Tbiolock = TUserTable["Biolock"][Tid]

        local TAccessL = TUserTable["AccessLevel"][Tid]

        local TMoney = TUserTable["Money"][Tid]

        local TEMP_TABLE = {id = Tid, biolockid = Tbiolock, accesslevel = TAccessL, money = TMoney}

        SEND(sID, TEMP_TABLE)

      end

    end

    else

        SEND(sID, 0x1)

  end

end



--Starting

print("Modem state: "..tostring(modem.isOpen(MODEM_CHANNEL)))

if modem.isOpen(MODEM_CHANNEL) ~= true then print("Modem opened\nID: ".. os.computerID()) modem.open(MODEM_CHANNEL) end



local function receiveo()

  while RP_BOOL do

    local event, modemSide, sID, rID, msg, distance = os.pullEvent("modem_message")

    if type(msg) ~= "table" then SEND(sID, "Please use API") else

      execute(msg,sID)

    end

  end

end



parallel.waitForAll(receiveo)