Mercurial Hosting > linkmystyle

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

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
start public repo
author Franklin Schmidt <fschmidt@gmail.com>
date Fri, 11 Jul 2025 20:57:49 -0600
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:8f4df159f06b
1 local Luan = require "luan:Luan.luan"
2 local error = Luan.error
3 local ipairs = Luan.ipairs or error()
4 local range = Luan.range or error()
5 local to_string = Luan.to_string or error()
6 local get_local_only = Luan.get_local_only or error()
7 local set_local_only = Luan.set_local_only or error()
8 local String = require "luan:String.luan"
9 local regex = String.regex or error()
10 local Math = require "luan:Math.luan"
11 local random = Math.random or error()
12 local Table = require "luan:Table.luan"
13 local concat = Table.concat or error()
14 local Html = require "luan:Html.luan"
15 local html_encode = Html.encode or error()
16 local Lucene = require "luan:lucene/Lucene.luan"
17 local lucene_quote = Lucene.quote or error()
18 local Http = require "luan:http/Http.luan"
19 local Db = require "site:/lib/Db.luan"
20 local Utils = require "site:/lib/Utils.luan"
21 local long_or_nil = Utils.long_or_nil or error()
22 local Uploadcare = require "site:/lib/Uploadcare.luan"
23 local uploadcare_url = Uploadcare.url or error()
24 local uploadcare_thumb_url = Uploadcare.thumb_url or error()
25 local Logging = require "luan:logging/Logging.luan"
26 local logger = Logging.logger "User"
27
28
29 local User = {}
30
31 local function from_doc(doc)
32 doc.type == "user" or error "wrong type"
33 return User.new {
34 id = doc.id
35 email = doc.user_email
36 password = doc.password
37 name = doc.user_name
38 code = doc.code
39 registered = doc.user_registered
40 new_email = doc.new_email
41 pic_uuid = doc.pic_uuid
42 pic_filename = doc.pic_filename
43 title = doc.title
44 bio = doc.bio
45 theme_data = doc.theme_data
46 theme_date = doc.theme_date
47 mp_id = doc.mp_id
48 source = doc.source or doc.seller
49 }
50 end
51
52 local function to_doc(user)
53 return {
54 type = "user"
55 id = user.id
56 user_email = user.email
57 password = user.password
58 user_name = user.name
59 code = user.code
60 user_registered = long_or_nil(user.registered)
61 new_email = user.new_email
62 pic_uuid = user.pic_uuid
63 pic_filename = user.pic_filename
64 title = user.title
65 bio = user.bio
66 theme_data = user.theme_data
67 theme_date = long_or_nil(user.theme_date)
68 mp_id = user.mp_id
69 source = user.source
70 }
71 end
72
73 function User.new(user)
74
75 function user.save()
76 local doc = to_doc(user)
77 Db.save(doc)
78 user.id = doc.id
79 end
80
81 function user.reload()
82 return User.get_by_id(user.id) or error(user.id)
83 end
84
85 function user.delete()
86 Db.run_in_transaction( function()
87 local id = user.id
88 Db.delete("icon_user_id:"..id)
89 Db.delete("link_user_id:"..id)
90 Db.delete("pic_user_id:"..id)
91 Db.delete("id:"..id)
92 end )
93 end
94
95 function user.login()
96 local id = to_string(user.id)
97 Http.response.set_persistent_cookie("user",id)
98 Http.response.set_persistent_cookie("password",user.password)
99 Http.request.cookies.user = id
100 Http.request.cookies.password = user.password or error()
101 end
102
103 function user.html_title()
104 return html_encode(user.title or user.name)
105 end
106
107 function user.get_pic_url()
108 local pic_uuid = user.pic_uuid
109 return pic_uuid and uploadcare_thumb_url(pic_uuid)
110 end
111
112 return user
113 end
114
115 local function get_by_id(id)
116 local doc = Db.get_document("id:"..id)
117 return doc and doc.type=="user" and from_doc(doc) or nil
118 end
119 User.get_by_id = get_by_id
120
121 function User.get_by_email(email)
122 local doc = Db.get_document("user_email:"..lucene_quote(email))
123 return doc and from_doc(doc)
124 end
125
126 function User.get_by_name(name)
127 local doc = Db.get_document("user_name:"..lucene_quote(name))
128 return doc and from_doc(doc)
129 end
130
131 function User.search(query,sort,rows)
132 rows = rows or 1000000
133 local users = {}
134 local docs = Db.search(query,1,rows,{sort=sort})
135 for _, doc in ipairs(docs) do
136 users[#users+1] = from_doc(doc)
137 end
138 return users
139 end
140
141 User.name_regex = regex "^[a-zA-Z0-9_-]+$"
142
143 function User.current()
144 local user = get_local_only(User,"current")
145 if user == nil then
146 local id = Http.request.cookies.user
147 local password = Http.request.cookies.password
148 if id == nil or password == nil then
149 user = "nil"
150 else
151 user = get_by_id(id)
152 if user == nil or user.registered == nil or user.password ~= password then
153 user = "nil"
154 end
155 end
156 set_local_only(User,"current",user)
157 end
158 return user ~= "nil" and user or nil
159 end
160
161 function User.current_required()
162 local user = User.current()
163 user or Http.response.send_redirect "/login.html"
164 return user
165 end
166
167 function User.new_code()
168 local t = {}
169 for _ in range(1,6) do
170 t[#t+1] = to_string(random(0,9))
171 end
172 return concat(t)
173 end
174
175 function User.get_background_img_url(data)
176 local uuid = data.background_img_uuid
177 return uuid and uploadcare_url(uuid)
178 end
179
180 return User