119 lines
3.5 KiB
Lua
119 lines
3.5 KiB
Lua
local class = require "middleclass"
|
|
local plugin = require "bunkerweb.plugin"
|
|
local session = require "resty.session"
|
|
local utils = require "bunkerweb.utils"
|
|
|
|
local sessions = class("sessions", plugin)
|
|
|
|
function sessions:initialize(ctx)
|
|
-- Call parent initialize
|
|
plugin.initialize(self, "sessions", ctx)
|
|
-- Check if random cookie name and secrets are already generated
|
|
local is_random = {
|
|
"SESSIONS_SECRET",
|
|
"SESSIONS_NAME",
|
|
}
|
|
self.randoms = {}
|
|
for _, var in ipairs(is_random) do
|
|
if self.variables[var] == "random" then
|
|
local data, _ = self.datastore:get("storage_sessions_" .. var)
|
|
if data then
|
|
self.randoms[var] = data
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
function sessions:set()
|
|
if self.is_loading or self.kind ~= "http" then
|
|
return self:ret(true, "set not needed")
|
|
end
|
|
local checks = {
|
|
["IP"] = self.ctx.bw.remote_addr,
|
|
["USER_AGENT"] = self.ctx.bw.http_user_agent or "",
|
|
}
|
|
self.ctx.bw.sessions_checks = {}
|
|
for check, value in pairs(checks) do
|
|
if self.variables["SESSIONS_CHECK_" .. check] == "yes" then
|
|
table.insert(self.ctx.bw.sessions_checks, { check, value })
|
|
end
|
|
end
|
|
return self:ret(true, "success")
|
|
end
|
|
|
|
function sessions:init()
|
|
if self.is_loading or self.kind ~= "http" then
|
|
return self:ret(true, "init not needed")
|
|
end
|
|
-- Get redis vars
|
|
local redis_vars = {
|
|
["USE_REDIS"] = "",
|
|
["REDIS_HOST"] = "",
|
|
["REDIS_PORT"] = "",
|
|
["REDIS_DATABASE"] = "",
|
|
["REDIS_SSL"] = "",
|
|
["REDIS_TIMEOUT"] = "",
|
|
["REDIS_KEEPALIVE_IDLE"] = "",
|
|
["REDIS_KEEPALIVE_POOL"] = "",
|
|
}
|
|
for k, _ in pairs(redis_vars) do
|
|
local value, err = utils.get_variable(k, false)
|
|
if value == nil then
|
|
return self:ret(false, "can't get " .. k .. " variable : " .. err)
|
|
end
|
|
redis_vars[k] = value
|
|
end
|
|
-- Init configuration
|
|
local config = {
|
|
secret = self.variables["SESSIONS_SECRET"],
|
|
cookie_name = self.variables["SESSIONS_NAME"],
|
|
idling_timeout = tonumber(self.variables["SESSIONS_IDLING_TIMEOUT"]),
|
|
rolling_timeout = tonumber(self.variables["SESSIONS_ROLLING_TIMEOUT"]),
|
|
absolute_timeout = tonumber(self.variables["SESSIONS_ABSOLUTE_TIMEOUT"]),
|
|
}
|
|
if self.variables["SESSIONS_SECRET"] == "random" then
|
|
if self.randoms["SESSIONS_SECRET"] then
|
|
config.secret = self.randoms["SESSIONS_SECRET"]
|
|
else
|
|
config.secret = utils.rand(16)
|
|
local ok, err = self.datastore:set("storage_sessions_SESSIONS_SECRET", config.secret)
|
|
if not ok then
|
|
self.logger:log(ngx.ERR, "error from datastore:set : " .. err)
|
|
end
|
|
end
|
|
end
|
|
if self.variables["SESSIONS_NAME"] == "random" then
|
|
if self.randoms["SESSIONS_NAME"] then
|
|
config.cookie_name = self.randoms["SESSIONS_NAME"]
|
|
else
|
|
config.cookie_name = utils.rand(16)
|
|
local ok, err = self.datastore:set("storage_sessions_SESSIONS_NAME", config.cookie_name)
|
|
if not ok then
|
|
self.logger:log(ngx.ERR, "error from datastore:set : " .. err)
|
|
end
|
|
end
|
|
end
|
|
if redis_vars["USE_REDIS"] ~= "yes" then
|
|
config.storage = "cookie"
|
|
else
|
|
config.storage = "redis"
|
|
config.redis = {
|
|
prefix = "sessions_",
|
|
connect_timeout = tonumber(redis_vars["REDIS_TIMEOUT"]),
|
|
send_timeout = tonumber(redis_vars["REDIS_TIMEOUT"]),
|
|
read_timeout = tonumber(redis_vars["REDIS_TIMEOUT"]),
|
|
keepalive_timeout = tonumber(redis_vars["REDIS_KEEPALIVE_IDLE"]),
|
|
pool = "bw-redis",
|
|
pool_size = tonumber(redis_vars["REDIS_KEEPALIVE_POOL"]),
|
|
ssl = redis_vars["REDIS_SSL"] == "yes",
|
|
host = redis_vars["REDIS_HOST"],
|
|
port = tonumber(redis_vars["REDIS_PORT"]),
|
|
database = tonumber(redis_vars["REDIS_DATABASE"]),
|
|
}
|
|
end
|
|
session.init(config)
|
|
return self:ret(true, "sessions init successful")
|
|
end
|
|
|
|
return sessions
|