yetangitu/OpenSTF
1
0
Fork 0
mirror of https://github.com/openstf/stf synced 2025-10-04 10:19:30 +02:00
Code Issues Releases Wiki Activity

Save user IP to database.

Browse source
This commit is contained in:
Simo Kinnunen 2014-05-30 11:46:29 +09:00
parent 22fc9e79dd
commit 24c1ba1bda
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
lib/db/api.js
View file

@ -12,6 +12,7 @@ dbapi.close = function(options) {
dbapi.saveUserAfterLogin = function(user) { dbapi.saveUserAfterLogin = function(user) {
return db.run(r.table('users').get(user.email).update({ return db.run(r.table('users').get(user.email).update({
name: user.name name: user.name
, ip: user.ip
, lastLoggedInAt: r.now() , lastLoggedInAt: r.now()
})) }))
.then(function(stats) { .then(function(stats) {
@ -19,6 +20,7 @@ dbapi.saveUserAfterLogin = function(user) {
return db.run(r.table('users').insert({ return db.run(r.table('users').insert({
email: user.email email: user.email
, name: user.name , name: user.name
, ip: user.ip
, group: wireutil.makePrivateChannel() , group: wireutil.makePrivateChannel()
, lastLoggedInAt: r.now() , lastLoggedInAt: r.now()
, createdAt: r.now() , createdAt: r.now()

6
lib/middleware/auth.js
View file

@ -11,7 +11,11 @@ module.exports = function(options) {
, redir = urlutil.removeParam(req.url, 'jwt') , redir = urlutil.removeParam(req.url, 'jwt')
if (data) { if (data) {
// Redirect once to get rid of the token // Redirect once to get rid of the token
dbapi.saveUserAfterLogin(data) dbapi.saveUserAfterLogin({
name: data.name
, email: data.email
, ip: req.ip
})
.then(function() { .then(function() {
req.session.jwt = data req.session.jwt = data
res.redirect(redir) res.redirect(redir)