local _M = {} -- constant tier ids local tier_id_anonymous = 0 local tier_id_free = 1 -- fallback - remember to keep those updated local anon_limits = { ["tierID"] = tier_id_anonymous, ["tierName"] = "anonymous", ["upload"] = 655360, ["download"] = 655360, ["maxUploadSize"] = 1073741824, ["registry"] = 250 } -- handle request exit when access to portal should be restricted to authenticated users only function _M.exit_access_unauthorized(message) ngx.status = ngx.HTTP_UNAUTHORIZED ngx.header["content-type"] = "text/plain" ngx.say(message or "Portal operator restricted access to authenticated users only") return ngx.exit(ngx.status) end -- handle request exit when access to portal should be restricted to subscription users only function _M.exit_access_forbidden(message) ngx.status = ngx.HTTP_FORBIDDEN ngx.header["content-type"] = "text/plain" ngx.say(message or "Portal operator restricted access to users with active subscription only") return ngx.exit(ngx.status) end function _M.accounts_enabled() return os.getenv("PORTAL_MODULES"):match("a") ~= nil end function _M.get_account_limits() local cjson = require('cjson') if ngx.var.skynet_jwt == "" then return anon_limits end if ngx.var.account_limits == "" then local httpc = require("resty.http").new() -- 10.10.10.70 points to accounts service (alias not available when using resty-http) local res, err = httpc:request_uri("http://10.10.10.70:3000/user/limits?unit=byte", { headers = { ["Cookie"] = "skynet-jwt=" .. ngx.var.skynet_jwt } }) -- fail gracefully in case /user/limits failed if err or (res and res.status ~= ngx.HTTP_OK) then ngx.log(ngx.ERR, "Failed accounts service request /user/limits?unit=byte: ", err or ("[HTTP " .. res.status .. "] " .. res.body)) ngx.var.account_limits = cjson.encode(anon_limits) elseif res and res.status == ngx.HTTP_OK then ngx.var.account_limits = res.body end end return cjson.decode(ngx.var.account_limits) end -- detect whether current user is authenticated function _M.is_authenticated() if not _M.accounts_enabled() then return false end local limits = _M.get_account_limits() return limits.tierID > tier_id_anonymous end -- detect whether current user has active subscription function _M.has_subscription() local limits = _M.get_account_limits() return limits.tierID > tier_id_free end function _M.is_auth_required() -- authentication is required if mode is set to "authenticated" -- or "subscription" (require active subscription to a premium plan) return os.getenv("ACCOUNTS_LIMIT_ACCESS") == "authenticated" or _M.is_subscription_required() end function _M.is_subscription_required() return os.getenv("ACCOUNTS_LIMIT_ACCESS") == "subscription" end function is_access_always_allowed() -- options requests do not attach cookies - should always be available -- requests should not be limited based on accounts if accounts are not enabled return ngx.req.get_method() == "OPTIONS" or not _M.accounts_enabled() end -- check whether access is restricted if portal requires authorization function _M.is_access_unauthorized() if is_access_always_allowed() then return false end -- check if authentication is required and request is not authenticated return _M.is_auth_required() and not _M.is_authenticated() end -- check whether user is authenticated but does not have access to given resources function _M.is_access_forbidden() if is_access_always_allowed() then return false end -- check if active subscription is required and request is from user without it return _M.is_subscription_required() and not _M.has_subscription() end return _M