|
2
|
1 local Luan = require "luan:Luan.luan"
|
|
|
2 local error = Luan.error
|
|
|
3 local ipairs = Luan.ipairs or error()
|
|
3
|
4 local range = Luan.range or error()
|
|
2
|
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()
|
|
3
|
8 local String = require "luan:String.luan"
|
|
|
9 local sub_string = String.sub 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()
|
|
46
|
14 local Html = require "luan:Html.luan"
|
|
|
15 local html_encode = Html.encode or error()
|
|
117
|
16 local url_encode = Html.url_encode or error()
|
|
2
|
17 local Lucene = require "luan:lucene/Lucene.luan"
|
|
|
18 local lucene_quote = Lucene.quote or error()
|
|
|
19 local Http = require "luan:http/Http.luan"
|
|
|
20 local Db = require "site:/lib/Db.luan"
|
|
3
|
21 local run_in_transaction = Db.run_in_transaction or error()
|
|
4
|
22 local Chat = require "site:/lib/Chat.luan"
|
|
30
|
23 local chat_search = Chat.search or error()
|
|
33
|
24 local Utils = require "site:/lib/Utils.luan"
|
|
|
25 local list_to_set = Utils.list_to_set or error()
|
|
|
26 local set_to_list = Utils.set_to_list or error()
|
|
88
|
27 local base_url = Utils.base_url or error()
|
|
2
|
28 local Logging = require "luan:logging/Logging.luan"
|
|
|
29 local logger = Logging.logger "User"
|
|
|
30
|
|
|
31
|
|
|
32 local User = {}
|
|
|
33
|
|
|
34 local function from_doc(doc)
|
|
|
35 doc.type == "user" or error "wrong type"
|
|
|
36 return User.new {
|
|
|
37 id = doc.id
|
|
|
38 email = doc.user_email
|
|
|
39 password = doc.password
|
|
40
|
40 was_notified = doc.was_notified=="true"
|
|
|
41 notify_email = doc.notify_email
|
|
|
42 multi_notify = doc.multi_notify=="true"
|
|
41
|
43 voice_url = doc.voice_url
|
|
46
|
44 name = doc.name
|
|
2
|
45 }
|
|
|
46 end
|
|
|
47
|
|
|
48 local function to_doc(user)
|
|
|
49 return {
|
|
|
50 type = "user"
|
|
|
51 id = user.id
|
|
|
52 user_email = user.email
|
|
|
53 password = user.password
|
|
40
|
54 was_notified = user.was_notified and "true" or nil
|
|
|
55 notify_email = user.notify_email
|
|
|
56 multi_notify = user.multi_notify and "true" or nil
|
|
41
|
57 voice_url = user.voice_url
|
|
46
|
58 name = user.name
|
|
2
|
59 }
|
|
|
60 end
|
|
|
61
|
|
|
62 function User.new(user)
|
|
|
63
|
|
|
64 function user.save()
|
|
|
65 local doc = to_doc(user)
|
|
|
66 Db.save(doc)
|
|
|
67 user.id = doc.id
|
|
|
68 end
|
|
|
69
|
|
40
|
70 function user.reload()
|
|
|
71 return User.get_by_id(user.id) or error(user.id)
|
|
|
72 end
|
|
|
73
|
|
4
|
74 function user.delete()
|
|
|
75 run_in_transaction( function()
|
|
|
76 local id = user.id
|
|
30
|
77 local chats = chat_search("chat_user_ids:"..id)
|
|
4
|
78 for _, chat in ipairs(chats) do
|
|
|
79 chat.delete()
|
|
|
80 end
|
|
|
81 Db.delete("id:"..id)
|
|
|
82 end )
|
|
2
|
83 end
|
|
|
84
|
|
|
85 function user.login()
|
|
|
86 local id = to_string(user.id)
|
|
|
87 Http.response.set_persistent_cookie("user",id)
|
|
|
88 Http.response.set_persistent_cookie("password",user.password)
|
|
|
89 Http.request.cookies.user = id
|
|
|
90 Http.request.cookies.password = user.password or error()
|
|
|
91 end
|
|
|
92
|
|
30
|
93 function user.last_update()
|
|
|
94 local chats = chat_search( "chat_user_ids:"..user.id, "chat_updated desc", 1 )
|
|
|
95 local n = #chats
|
|
|
96 if n == 0 then
|
|
|
97 return 0
|
|
|
98 elseif n == 1 then
|
|
|
99 return chats[1].updated
|
|
|
100 else error() end
|
|
|
101 end
|
|
|
102
|
|
33
|
103 function user.chatting_with_ids()
|
|
|
104 local my_id = user.id
|
|
|
105 local user_ids = list_to_set{}
|
|
|
106 local chats = chat_search( "chat_user_ids:"..my_id )
|
|
|
107 for _, chat in ipairs(chats) do
|
|
|
108 for _, user_id in ipairs(chat.user_ids) do
|
|
|
109 user_ids[user_id] = true
|
|
|
110 end
|
|
|
111 end
|
|
|
112 user_ids[my_id] = false
|
|
|
113 return set_to_list(user_ids)
|
|
|
114 end
|
|
|
115
|
|
46
|
116 function user.name_html()
|
|
|
117 return html_encode(user.name or user.email)
|
|
|
118 end
|
|
|
119
|
|
88
|
120 function user.login_url()
|
|
117
|
121 return base_url().."/login.html?email="..url_encode(user.email).."&password="..user.password
|
|
88
|
122 end
|
|
|
123
|
|
2
|
124 return user
|
|
|
125 end
|
|
|
126
|
|
|
127 local function get_by_id(id)
|
|
|
128 local doc = Db.get_document("id:"..id)
|
|
|
129 return doc and doc.type=="user" and from_doc(doc) or nil
|
|
|
130 end
|
|
|
131 User.get_by_id = get_by_id
|
|
|
132
|
|
3
|
133 local password_chars = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
|
|
134 do
|
|
|
135 local t = {}
|
|
|
136 for i in range(1,#password_chars) do
|
|
|
137 t[#t+1] = sub_string(password_chars,i,i)
|
|
|
138 end
|
|
|
139 password_chars = t
|
|
|
140 end
|
|
|
141
|
|
|
142 local function new_password()
|
|
|
143 local n = #password_chars
|
|
|
144 local t = {}
|
|
|
145 for _ in range(1,10) do
|
|
|
146 t[#t+1] = password_chars[random(n)]
|
|
|
147 end
|
|
|
148 return concat(t)
|
|
|
149 end
|
|
|
150
|
|
4
|
151 local function get_by_email(email)
|
|
|
152 local doc = Db.get_document("user_email:"..lucene_quote(email))
|
|
|
153 return doc and from_doc(doc)
|
|
|
154 end
|
|
|
155 User.get_by_email = get_by_email
|
|
|
156
|
|
3
|
157 function User.get_or_create_by_email(email)
|
|
|
158 return run_in_transaction( function()
|
|
4
|
159 local user = get_by_email(email)
|
|
|
160 if user == nil then
|
|
|
161 user = User.new{
|
|
3
|
162 email = email
|
|
|
163 password = new_password()
|
|
40
|
164 notify_email = email
|
|
3
|
165 }
|
|
|
166 user.save()
|
|
|
167 end
|
|
4
|
168 return user
|
|
3
|
169 end )
|
|
2
|
170 end
|
|
|
171
|
|
|
172 function User.search(query,sort,rows)
|
|
|
173 rows = rows or 1000000
|
|
|
174 local users = {}
|
|
|
175 local docs = Db.search(query,1,rows,{sort=sort})
|
|
|
176 for _, doc in ipairs(docs) do
|
|
|
177 users[#users+1] = from_doc(doc)
|
|
|
178 end
|
|
|
179 return users
|
|
|
180 end
|
|
|
181
|
|
3
|
182 local function current()
|
|
2
|
183 local user = get_local_only(User,"current")
|
|
|
184 if user == nil then
|
|
|
185 local id = Http.request.cookies.user
|
|
|
186 local password = Http.request.cookies.password
|
|
|
187 if id == nil or password == nil then
|
|
|
188 user = "nil"
|
|
|
189 else
|
|
|
190 user = get_by_id(id)
|
|
3
|
191 if user == nil or user.password ~= password then
|
|
2
|
192 user = "nil"
|
|
|
193 end
|
|
|
194 end
|
|
|
195 set_local_only(User,"current",user)
|
|
|
196 end
|
|
|
197 return user ~= "nil" and user or nil
|
|
|
198 end
|
|
3
|
199 User.current = current
|
|
2
|
200
|
|
|
201 function User.current_required()
|
|
3
|
202 local user = current()
|
|
2
|
203 user or Http.response.send_redirect "/login.html"
|
|
|
204 return user
|
|
|
205 end
|
|
|
206
|
|
|
207 return User
|
