diff src/lib/User.luan @ 0:8f4df159f06b

start public repo
author Franklin Schmidt <fschmidt@gmail.com>
date Fri, 11 Jul 2025 20:57:49 -0600
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/lib/User.luan	Fri Jul 11 20:57:49 2025 -0600
@@ -0,0 +1,180 @@
+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