145 lines
3.2 KiB
Lua
145 lines
3.2 KiB
Lua
local util = {}
|
|
local magick = require("magick")
|
|
local db = require("lapis.db")
|
|
local html_escape = require("lapis.html").escape
|
|
|
|
local Avatars = require("models").Avatars
|
|
local Users = require("models").Users
|
|
local Posts = require("models").Posts
|
|
local PostHistory = require("models").PostHistory
|
|
|
|
local babycode = require("lib.babycode")
|
|
|
|
util.TransientUser = {
|
|
is_admin = function (self)
|
|
return false
|
|
end,
|
|
is_mod = function (self)
|
|
return false
|
|
end,
|
|
is_guest = function (self)
|
|
return true
|
|
end,
|
|
is_logged_in_guest = function (self)
|
|
return false
|
|
end,
|
|
username = "Deleted User",
|
|
}
|
|
|
|
function util.get_user_avatar_url(req, user)
|
|
if not user.avatar_id then
|
|
return "/avatars/default.webp"
|
|
end
|
|
return Avatars:find(user.avatar_id).file_path
|
|
end
|
|
|
|
function util.validate_and_create_image(input_image, filename)
|
|
local img = magick.load_image_from_blob(input_image)
|
|
|
|
if not img then
|
|
return false
|
|
end
|
|
|
|
img:strip()
|
|
img:set_gravity("CenterGravity")
|
|
|
|
local width, height = img:get_width(), img:get_height()
|
|
local min_dim = math.min(width, height)
|
|
if min_dim > 256 then
|
|
local ratio = 256.0 / min_dim
|
|
local new_w, new_h = width * ratio, height * ratio
|
|
img:resize(new_w, new_h)
|
|
end
|
|
|
|
width, height = img:get_width(), img:get_height()
|
|
local crop_size = math.min(width, height)
|
|
local x_offset = (width - crop_size) / 2
|
|
local y_offset = (height - crop_size) / 2
|
|
img:crop(crop_size, crop_size, x_offset, y_offset)
|
|
|
|
img:set_format("webp")
|
|
img:set_quality(85)
|
|
|
|
img:write(filename)
|
|
img:destroy()
|
|
return true
|
|
end
|
|
|
|
function util.get_logged_in_user(req)
|
|
if req.session.session_key == nil then
|
|
return nil
|
|
end
|
|
|
|
local session = db.select('* FROM "sessions" WHERE "key" = ? AND "expires_at" > "?" LIMIT 1', req.session.session_key, os.time())
|
|
if #session > 0 then
|
|
return Users:find({id = session[1].user_id})
|
|
end
|
|
|
|
return nil
|
|
end
|
|
|
|
function util.get_logged_in_user_or_transient(req)
|
|
return util.get_logged_in_user(req) or util.TransientUser
|
|
end
|
|
|
|
function util.ntob(v)
|
|
return v ~= 0
|
|
end
|
|
|
|
function util.bton(b)
|
|
return 1 and b or 0
|
|
end
|
|
|
|
function util.stob(s)
|
|
if s == "true" then
|
|
return true
|
|
end
|
|
if s == "false" then
|
|
return false
|
|
end
|
|
end
|
|
|
|
function util.form_bool_to_sqlite(s)
|
|
return util.bton(util.stob(s))
|
|
end
|
|
|
|
function util.is_thread_locked(thread)
|
|
return util.ntob(thread.is_locked)
|
|
end
|
|
|
|
function util.create_post(thread_id, user_id, content)
|
|
db.query("BEGIN")
|
|
local post = Posts:create({
|
|
thread_id = thread_id,
|
|
user_id = user_id,
|
|
current_revision_id = db.NULL,
|
|
})
|
|
|
|
local bb_content = babycode.to_html(content, html_escape)
|
|
|
|
local revision = PostHistory:create({
|
|
post_id = post.id,
|
|
content = bb_content,
|
|
is_initial_revision = true,
|
|
})
|
|
|
|
post:update({current_revision_id = revision.id})
|
|
|
|
db.query("COMMIT")
|
|
return post
|
|
end
|
|
|
|
function util.transfer_and_delete_user(user)
|
|
local deleted_user = Users:find({
|
|
username = "DeletedUser",
|
|
})
|
|
-- this needs to be atomic
|
|
db.query("BEGIN")
|
|
db.query('UPDATE "threads" SET "user_id" = ? WHERE "user_id" = ?', deleted_user.id, user.id)
|
|
db.query('UPDATE "posts" SET "user_id" = ? WHERE "user_id" = ?', deleted_user.id, user.id)
|
|
user:delete() -- uncomment later
|
|
db.query("COMMIT")
|
|
end
|
|
|
|
return util
|