userAPI: Serveur

--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


--Users management
local USER_FILE = "/users"
local DB_SCHEM = {Users = {}, Password = {}, Biolock = {}, AccessLevel = {}, Token= {}, Where= {}}

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_TOKEN(users, tofind)
  local ID = nil
  for i = 1,#users["Token"] do
    if tofind == nil then return 0x1
    else
      if users["Token"][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["Token"][lastID] = "unknown"
  fUSERS["AccessLevel"][lastID] = accesslevel
  fUSERS["Where"][lastID] = "unknown"
  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 == "token" then
        oUSERS["Token"][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




--Token functions

local function TOKEN_GEN()
    local TOKEN_LENGHT = 25
    local TOKEN = ""
    local TOKEN_TABLE = {"a","b","c","d","e","f","g","h","i","j","k","l","m","n","o","p","q","r","s","t","u","v","w","x","y","z","0","1","2","3","4","5","6","7","8","9","#", "$", "*", "!"}
 local x    = 0
 while TOKEN_LENGHT > 0 do
  x=x+1
        local mRandom = math.random(1,#TOKEN_TABLE)
        TOKEN = TOKEN..TOKEN_TABLE[mRandom]
        TOKEN_LENGHT = TOKEN_LENGHT - 1
  print("Generating Token: "..tostring(x).."/25")
    end
    return TOKEN
end

local function TOKEN_CREATE(id)
    local USERS = USERS_GET()
    if type(id) ~= "number" then return 0x1 end
    local newToken = TOKEN_GEN()
    --Check if token already exist
    local YES = false
    while YES do
        if USERS_TOKEN(USERS, newToken) == nil then YES = false else
        newToken = TOKEN_GEN() end
    end
    --Write the new token
    local retu = USERS_UPDATE("token", id, newToken)
    if retu then
        return newToken
    else
        return retu
    end
end

local function execute(sTable,sID)
    local uLevel = 0
    if sTable.type == "ping" then
        return SEND(sID,0x0)
    end

    if sTable.todo ~= "getToken" and sTable.todo ~= "createToken" then
        if sTable["tokenUsername"] == nil or sTable["tokenCheck"] == nil then return SEND(sID, 0x3) end
        --Check account
        local uTable = USERS_GET()
        local uID = USERS_ID(uTable, sTable["tokenUsername"])
        if uID == nil then return SEND(sID, 0x2) end
        local uTOKEN = uTable["Token"][uID]
        if StringUtils.decrypt(sTable["tokenCheck"], uTOKEN) ~= "true" then return SEND(sID, 0x4) end
        uLevel = uTable["AccessLevel"][uID]
    end

  if sTable.type == "token" then
    if sTable.todo == "createToken" then
      if type(sTable.enc) ~= "string" or type(sTable.Username) ~= "string" or type(sTable.math) ~= "string" then SEND(sID, 0x1) else
      local oTable = USERS_GET()
      local uID = USERS_ID(oTable,sTable.Username)
      if uID == nil then SEND(sID, 0x2) return false end
      local SHA1Pass = oTable["Password"][uID]
      local crypted = StringUtils.decrypt(sTable.enc, SHA1Pass)
      if crypted == sTable.math then
        --He got it we need to give a token to him
        local Token = TOKEN_CREATE(uID)
        SEND(sID, Token)
        print("Token created for: "..sTable["Username"])
        print("New Token hash: "..StringUtils.SHA1(Token))
      else
        SEND(sID, 0x5)
      end
      end

    elseif sTable.todo == "getToken" then
      if type(sTable.enc) ~= "string" or type(sTable.Username) ~= "string" or type(sTable.math) ~= "string" then SEND(sID, "nil entry") else
      local oTable = USERS_GET()
      local uID = USERS_ID(oTable,sTable.Username)
      if uID == nil then SEND(sID, 0x2) return false end
      local SHA1Pass = oTable["Password"][uID]
      local crypted = StringUtils.decrypt(sTable.enc, SHA1Pass)
      if crypted == sTable.math then
        --He got it we need to give a token to him
        SEND(sID, oTable["Token"][uID])
      else
        SEND(sID, 0x5)
      end
      end
    end

  elseif sTable.type == "ping" then
    SEND(sID, "true")
    log(sID, sTable, "ping")

  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
    if uLevel ~= 0 and uLevel ~= 5 then return SEND(sID, 0x6) end
      if sTable.Username == nil then
      SEND(sID, 0x1)

      else
        local TUserTable = USERS_GET()
        local Tid = USERS_ID(TUserTable, sTable.Username)
        if Tid == nil then SEND(sID, 0x2) else
            local Tbiolock = TUserTable["Biolock"][Tid]
            local TAccessL = TUserTable["AccessLevel"][Tid]
            local TEMP_TABLE = {id = Tid, biolockid = Tbiolock, accesslevel = TAccessL}
            SEND(sID, TEMP_TABLE)

        end
      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)