63 lines
2.2 KiB
Ruby
63 lines
2.2 KiB
Ruby
require "base64"
|
|
require "zlib"
|
|
require "json"
|
|
|
|
def set_session(request, response, key, val, uid = nil)
|
|
session = request.cookies["session"]
|
|
session = session.nil? ? "{}" : Zlib::Inflate.inflate(Base64.decode64(session))
|
|
session = JSON.parse(session)
|
|
session[key] = val
|
|
compressed = Zlib::Deflate.deflate(JSON.generate(session))
|
|
encoded = Base64.encode64(compressed)
|
|
response.set_cookie("session",
|
|
value: encoded,
|
|
path: "/",
|
|
expires: Time.now + 360 * 24 * 60 * 60)
|
|
begin
|
|
DB["UPDATE SignedInUsers SET last_used_at = CURRENT_TIMESTAMP WHERE code = ?", uid].update if uid
|
|
rescue Sequel::Error => e
|
|
File.write("log/main.log", "DB Error: #{e.message} when updating last_used_at for #{uid}\n", mode: "a")
|
|
end
|
|
rescue JSON::ParserError, Zlib::Error
|
|
response.delete_cookie("session")
|
|
end
|
|
|
|
def get_session(request, response, key, uid = nil)
|
|
session = request.cookies["session"]
|
|
session = session.nil? ? "{}" : Zlib::Inflate.inflate(Base64.decode64(session))
|
|
session = JSON.parse(session)
|
|
begin
|
|
DB["UPDATE SignedInUsers SET last_used_at = CURRENT_TIMESTAMP WHERE code = ?", uid].update if uid
|
|
rescue Sequel::Error => e
|
|
File.write("log/main.log", "DB Error: #{e.message} when updating last_used_at for #{uid}\n", mode: "a")
|
|
end
|
|
session[key]
|
|
rescue JSON::ParserError, Zlib::Error
|
|
response.delete_cookie("session")
|
|
""
|
|
end
|
|
|
|
def get_session_all(request, response)
|
|
session = request.cookies["session"]
|
|
session = session.nil? ? "{}" : Zlib::Inflate.inflate(Base64.decode64(session))
|
|
JSON.parse(session)
|
|
rescue JSON::ParserError, Zlib::Error
|
|
response.delete_cookie("session")
|
|
""
|
|
end
|
|
|
|
def rm_session(request, response, key)
|
|
session = request.cookies["session"]
|
|
session = session.nil? ? "{}" : Zlib::Inflate.inflate(Base64.decode64(session))
|
|
session = JSON.parse(session)
|
|
session.delete(key)
|
|
compressed = Zlib::Deflate.deflate(JSON.generate(session))
|
|
encoded = Base64.encode64(compressed)
|
|
response.set_cookie("session",
|
|
value: encoded,
|
|
path: "/",
|
|
expires: Time.now + 360 * 24 * 60 * 60)
|
|
rescue JSON::ParserError, Zlib::Error
|
|
response.delete_cookie("session")
|
|
end
|