0
0

Refactoring

This commit is contained in:
こぴなたみぽ 2017-09-16 14:30:44 +09:00
parent 86858586cf
commit bbfac657fb
2 changed files with 23 additions and 17 deletions

View File

@ -1,6 +1,6 @@
import * as express from 'express';
import App from './models/app';
import User from './models/user';
import{ default as User, IUser } from './models/user';
import AccessToken from './models/access-token';
import isNativeToken from './common/is-native-token';
@ -13,7 +13,7 @@ export interface IAuthContext {
/**
* Authenticated user
*/
user: any;
user: IUser;
/**
* Weather if the request is via the User-Native Token or not
@ -25,11 +25,15 @@ export default (req: express.Request) => new Promise<IAuthContext>(async (resolv
const token = req.body['i'] as string;
if (token == null) {
return resolve({ app: null, user: null, isSecure: false });
return resolve({
app: null,
user: null,
isSecure: false
});
}
if (isNativeToken(token)) {
const user = await User
const user: IUser = await User
.findOne({ token: token });
if (user === null) {
@ -56,6 +60,10 @@ export default (req: express.Request) => new Promise<IAuthContext>(async (resolv
const user = await User
.findOne({ _id: accessToken.user_id });
return resolve({ app: app, user: user, isSecure: false });
return resolve({
app: app,
user: user,
isSecure: false
});
}
});

View File

@ -2,7 +2,7 @@ import * as http from 'http';
import * as websocket from 'websocket';
import * as redis from 'redis';
import config from '../conf';
import User from './models/user';
import { default as User, IUser } from './models/user';
import AccessToken from './models/access-token';
import isNativeToken from './common/is-native-token';
@ -26,7 +26,7 @@ module.exports = (server: http.Server) => {
return;
}
const user = await authenticate(connection, request.resourceURL.query.i);
const user = await authenticate(request.resourceURL.query.i);
if (user == null) {
connection.send('authentication-failed');
@ -56,7 +56,11 @@ module.exports = (server: http.Server) => {
});
};
function authenticate(connection: websocket.connection, token: string): Promise<any> {
/**
*
* @param token
*/
function authenticate(token: string): Promise<IUser> {
if (token == null) {
return Promise.resolve(null);
}
@ -64,8 +68,7 @@ function authenticate(connection: websocket.connection, token: string): Promise<
return new Promise(async (resolve, reject) => {
if (isNativeToken(token)) {
// Fetch user
// SELECT _id
const user = await User
const user: IUser = await User
.findOne({
token: token
});
@ -81,13 +84,8 @@ function authenticate(connection: websocket.connection, token: string): Promise<
}
// Fetch user
// SELECT _id
const user = await User
.findOne({ _id: accessToken.user_id }, {
fields: {
_id: true
}
});
const user: IUser = await User
.findOne({ _id: accessToken.user_id });
resolve(user);
}