--[[||
  Wow's database API
  This API need:
  -Modem
  -StringUtilsAPI
||]]

--[[
  Error codes:
  0x0: no error
  0x1: nil arg
  0x2: not connected
  0x3: Rednet not opened
  0x4:
  0x5: unknown function
]]
os.loadAPI("StringUtils")
local channel = 644
local modem = peripheral.find("modem")

local function checkopened()
	if not modem.isOpen(channel) then modem.open(channel) end
	return true
end

local function addCheck(sTable, token, Username)
	local cryptoken = StringUtils.encrypt("true", token)
	sTable["tokenUsername"] = Username
	sTable["tokenCheck"] = cryptoken
	return sTable
end

local function send(sTable, token, Username)
  checkopened()
  if sTable == nil then return 0x1 end
  local sTable = addCheck(sTable, token, Username)
  modem.transmit(channel, channel, sTable)
  local x = true
  while x do
  	local event, side, sChan, rChan, msg, distance = os.pullEvent("modem_message")
  	if sChan == channel then
		x = false
  		return msg
	end
  end
end

function database(action, arg1, arg2, arg3, arg4, arg5, arg6)
  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" or type(arg4) ~= "string" or type(arg5) ~= "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, arg4, arg5))
    end

  elseif action == "resetDB" then
   	if type(arg1) ~= "string" or type(arg2) ~= "string" then return 0x1 end
    local TEMP_TABLE = {type = "database", todo = "reset"}
    return(send(TEMP_TABLE, arg1,arg2))

  elseif action == "getinfo" then
    if type(arg1) ~= "string" or type(arg2) ~= "string" or type(arg3) ~= "string" then
      return 0x1
    else
      local TEMP_TABLE = {type = "database", todo = "Userinfo", Username = arg1}
      return(send(TEMP_TABLE, arg2, arg3))
    end

  elseif action == "update" then
  	--arg1: what arg2: new: arg3: who arg4 Username arg5 token
  	if type(arg1) ~= "string" or type(arg2) ~= "string" or type(arg3) ~= "string" or type(arg4) ~= "string"  then return 0x5 end
  	local TEMP_TABLE = {type = "database", todo = "update", toupdate = arg1, newupdate = arg2, who = arg3}
	return(send)
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