dbAPI

--[[||
  Wow's database API
  This API need:
  -Rednet (Opened)
  -StringUtilsAPI
||]]

--[[
  Error codes:
  0x0: no error
  0x1: nil arg
  0x2: not connected
  0x3: Rednet not opened
  0x4: Bad ID
  0x5: unknown function
]]
os.loadAPI("StringUtils")
local id = 78
function findServerID()
  --id = rednet.lookup("HomePI", "main")
  return id
end

local function send(sTable)
  if sTable == nil then return 0x1 end
  rednet.send(id, sTable)
  sID, msg = rednet.receive()
  if sID ~= findServerID() then return 0x4 end
  return msg
end

function database(action, arg1, arg2, arg3, arg4, arg5)
  if action == "createUser" then
    if arg1 == nil or arg2 == nil or arg3 == nil or arg4 == nil or type(arg4) ~= "number" or type(arg1) ~= "string" or type(arg2) ~= "string" or type(arg3) ~= "string" then
      return 0x1
    else
      local TEMP_TABLE = {type = "database", todo = action, Username = arg1, Password = arg2, Biolock = arg3, accesslevel = arg4}
      local lol = textutils.serialise(TEMP_TABLE)
      return(send(TEMP_TABLE))
    end
  elseif action == "resetDB" then
    local TEMP_TABLE = {type = "database", todo = "reset"}
    return(send(TEMP_TABLE))
  elseif action == "getinfo" then
    if type(arg1) ~= "string" then
      return 0x1
    else
      local TEMP_TABLE = {type = "database", todo = "Userinfo", Username = arg1}
      return(send(TEMP_TABLE))
    end
  else
    return 0x5
  end
end

function newToken(user, pass)
  if type(user) ~= "string" or type(pass) ~= "string" then return 0x1 end
  local rMath = tostring(math.random(1,0xBEEF))
  local encPass = StringUtils.encrypt(tostring(rMath),StringUtils.SHA1(pass))
  local TEMP_TABLE = {type = "token", todo = "createToken", math = rMath, Username = user, enc = encPass}
  return send(TEMP_TABLE)
end

function getToken(user, pass)
  if type(user) ~= "string" or type(pass) ~= "string" then return 0x1 end
  local rMath = tostring(math.random(1,0xBEEF))
  local encPass = StringUtils.encrypt(tostring(rMath),StringUtils.SHA1(pass))
  local TEMP_TABLE = {type = "token", todo = "getToken", math = rMath, Username = user, enc = encPass}
  return send(TEMP_TABLE)
end