LIKO-12/Peripherals/WEB/init.lua

if (not love.thread) or (not jit) then error("WEB peripherals requires love.thread and luajit") end

local perpath = select(1,...) --The path to the web folder

local bit = require("bit")

local events = require("Engine.events")
local json = require("Engine.JSON")

local hasLuaSec = pcall(require,"ssl")

if hasLuaSec then
  require("ssl"); require(perpath:gsub("/",".").."https")
end

local thread = love.thread.newThread(perpath.."webthread.lua")
local to_channel = love.thread.newChannel()
local from_channel = love.thread.newChannel()

local to_counter = 0
local from_counter = 0

thread:start(to_channel, from_channel)

local function clearFuncsFromTable(t)
  for k,v in pairs(t) do
    if type(v) == "function" then
      t[k] = nil
    elseif type(v) == "table" then
      clearFuncsFromTable(v)
    end
  end
end

return function(config) --A function that creates a new WEB peripheral.
  if not thread:isRunning() then error("Failed to load luajit-request: "..tostring(thread:getError())) end
  
  local timeout = config.timeout or 5
  
  local CPUKit = config.CPUKit
  if not CPUKit then error("WEB Peripheral can't work without the CPUKit passed") end
  
  local indirect = {} --List of functions that requires custom coroutine.resume
  
  local devkit = {}
  
  local WEB, yWEB = {}, {}
  
  function WEB.send(url,args)
    if type(url) ~= "string" then return error("URL must be a string, provided: "..type(url)) end
    local args = args or {}
    if type(args) ~= "table" then return error("Args Must be a table or nil, provided: "..type(args)) end
    
    clearFuncsFromTable(args) --Since JSON can't encode functions !
    args.timeout = timeout
    
    args = json:encode(args)
    
    to_channel:push({url,args})
    to_counter = to_counter + 1
    
    return to_counter --Return the request ID
  end
  
  function WEB.urlEncode(str)
    if type(str) ~= "string" then return error("STR must be a string, provided: "..type(str)) end
    str = str:gsub("\n", "\r\n")
    str = str:gsub("\r\r\n", "\r\n")
    str = str:gsub("([^A-Za-z0-9 %-%_%.])", function(c)
      local n = string.byte(c)
      if n < 128 then
        -- ASCII
        return string.format("%%%02X", n)
      else
        -- Non-ASCII (encode as UTF-8)
        return string.format("%%%02X", 192 + bit.band( bit.arshift(n,6), 31 )) .. 
               string.format("%%%02X", 128 + bit.band( n, 63 ))
      end
    end)
    
    str = str:gsub(" ", "+")
    
    return str
  end
  
  local luasocket_modules = {
    "socket.http", "ltn12", "mime", "socket.smtp", "socket", "socket.url"
  }
  
  for k,v in pairs(luasocket_modules) do require(v) end
  for k,v in pairs(luasocket_modules) do luasocket_modules[v] = k end
  
  function WEB.luasocket(submodule)
    if type(submodule) ~= "string" then return error("Submodule should be a string, provided: "..submodule) end
    submodule = string.lower(submodule)
    if not luasocket_modules[submodule] then return error("Invalid submodule: "..submodule) end
    
    return require(submodule)
  end
  
  function WEB.hasLuaSec()
    return hasLuaSec and true or false
  end
  
  function WEB.luasec(submodule)
    if not hasLuaSec then return error("LuaSec is not supported at the current platform") end
    
    if type(submodule) ~= "string" then return error("Submodule should be a string, provided: "..submodule) end
    submodule = string.lower(submodule)
    if submodule ~= "ssl" and submodule ~= "https" then return error("Invalid submodule: "..submodule) end
    
    if submodule == "https" then
      return require(perpath:gsub("/",".").."https")
    end
    
    return require(submodule)
  end
  
  events:register("love:update",function(dt)
    local result = from_channel:pop()
    if result then
      from_counter = from_counter +1
      local data = json:decode(result)
      CPUKit.triggerEvent("webrequest",from_counter,unpack(data))
    end
  end)

  events:register("love:reboot",function()
    to_channel:clear()
    to_channel:push("shutdown")
  end)

  events:register("love:quit",function()
    to_channel:clear()
    to_channel:push("shutdown")
    thread:wait()
  end)
  
  return WEB, yWEB, devkit
end