|
0
|
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
|
