view src/lib/User.luan @ 3:b016e4b7c8da default tip

add read_me
author Franklin Schmidt <fschmidt@gmail.com>
date Sat, 12 Jul 2025 12:51:36 -0600
parents 8f4df159f06b
children
line wrap: on
line source

local Luan = require "luan:Luan.luan"
local error = Luan.error
local ipairs = Luan.ipairs or error()
local range = Luan.range or error()
local to_string = Luan.to_string or error()
local get_local_only = Luan.get_local_only or error()
local set_local_only = Luan.set_local_only or error()
local String = require "luan:String.luan"
local regex = String.regex or error()
local Math = require "luan:Math.luan"
local random = Math.random or error()
local Table = require "luan:Table.luan"
local concat = Table.concat or error()
local Html = require "luan:Html.luan"
local html_encode = Html.encode or error()
local Lucene = require "luan:lucene/Lucene.luan"
local lucene_quote = Lucene.quote or error()
local Http = require "luan:http/Http.luan"
local Db = require "site:/lib/Db.luan"
local Utils = require "site:/lib/Utils.luan"
local long_or_nil = Utils.long_or_nil or error()
local Uploadcare = require "site:/lib/Uploadcare.luan"
local uploadcare_url = Uploadcare.url or error()
local uploadcare_thumb_url = Uploadcare.thumb_url or error()
local Logging = require "luan:logging/Logging.luan"
local logger = Logging.logger "User"


local User = {}

local function from_doc(doc)
	doc.type == "user" or error "wrong type"
	return User.new {
		id = doc.id
		email = doc.user_email
		password = doc.password
		name = doc.user_name
		code = doc.code
		registered = doc.user_registered
		new_email = doc.new_email
		pic_uuid = doc.pic_uuid
		pic_filename = doc.pic_filename
		title = doc.title
		bio = doc.bio
		theme_data = doc.theme_data
		theme_date = doc.theme_date
		mp_id = doc.mp_id
		source = doc.source or doc.seller
	}
end

local function to_doc(user)
	return {
		type = "user"
		id = user.id
		user_email = user.email
		password = user.password
		user_name = user.name
		code = user.code
		user_registered = long_or_nil(user.registered)
		new_email = user.new_email
		pic_uuid = user.pic_uuid
		pic_filename = user.pic_filename
		title = user.title
		bio = user.bio
		theme_data = user.theme_data
		theme_date = long_or_nil(user.theme_date)
		mp_id = user.mp_id
		source = user.source
	}
end

function User.new(user)

	function user.save()
		local doc = to_doc(user)
		Db.save(doc)
		user.id = doc.id
	end

	function user.reload()
		return User.get_by_id(user.id) or error(user.id)
	end

	function user.delete()
		Db.run_in_transaction( function()
			local id = user.id
			Db.delete("icon_user_id:"..id)
			Db.delete("link_user_id:"..id)
			Db.delete("pic_user_id:"..id)
			Db.delete("id:"..id)
		end )
	end

	function user.login()
		local id = to_string(user.id)
		Http.response.set_persistent_cookie("user",id)
		Http.response.set_persistent_cookie("password",user.password)
		Http.request.cookies.user = id
		Http.request.cookies.password = user.password or error()
	end

	function user.html_title()
		return html_encode(user.title or user.name)
	end

	function user.get_pic_url()
		local pic_uuid = user.pic_uuid
		return pic_uuid and uploadcare_thumb_url(pic_uuid)
	end

	return user
end

local function get_by_id(id)
	local doc = Db.get_document("id:"..id)
	return doc and doc.type=="user" and from_doc(doc) or nil
end
User.get_by_id = get_by_id

function User.get_by_email(email)
	local doc = Db.get_document("user_email:"..lucene_quote(email))
	return doc and from_doc(doc)
end

function User.get_by_name(name)
	local doc = Db.get_document("user_name:"..lucene_quote(name))
	return doc and from_doc(doc)
end

function User.search(query,sort,rows)
	rows = rows or 1000000
	local users = {}
	local docs = Db.search(query,1,rows,{sort=sort})
	for _, doc in ipairs(docs) do
		users[#users+1] = from_doc(doc)
	end
	return users
end

User.name_regex = regex "^[a-zA-Z0-9_-]+$"

function User.current()
	local user = get_local_only(User,"current")
	if user == nil then
		local id = Http.request.cookies.user
		local password = Http.request.cookies.password
		if id == nil or password == nil then
			user = "nil"
		else
			user = get_by_id(id)
			if user == nil or user.registered == nil or user.password ~= password then
				user = "nil"
			end
		end
		set_local_only(User,"current",user)
	end
	return user ~= "nil" and user or nil
end

function User.current_required()
	local user = User.current()
	user or Http.response.send_redirect "/login.html"
	return user
end

function User.new_code()
	local t = {}
	for _ in range(1,6) do
		t[#t+1] = to_string(random(0,9))
	end
	return concat(t)
end

function User.get_background_img_url(data)
	local uuid = data.background_img_uuid
	return uuid and uploadcare_url(uuid)
end

return User