Fix tests
[oweals/peertube.git] / server / middlewares / oauth.ts
1 import 'express-validator'
2 import * as express from 'express'
3 import * as OAuthServer from 'express-oauth-server'
4
5 import { OAUTH_LIFETIME } from '../initializers'
6 import { logger } from '../helpers'
7
8 const oAuthServer = new OAuthServer({
9   accessTokenLifetime: OAUTH_LIFETIME.ACCESS_TOKEN,
10   refreshTokenLifetime: OAUTH_LIFETIME.REFRESH_TOKEN,
11   model: require('../lib/oauth-model')
12 })
13
14 function authenticate (req: express.Request, res: express.Response, next: express.NextFunction) {
15   oAuthServer.authenticate()(req, res, err => {
16     if (err) {
17       logger.error('Cannot authenticate.', err)
18       return res.sendStatus(500)
19     }
20
21     if (res.statusCode === 401 || res.statusCode === 400 || res.statusCode === 503) return res.end()
22
23     return next()
24   })
25 }
26
27 function token (req: express.Request, res: express.Response, next: express.NextFunction) {
28   return oAuthServer.token()(req, res, next)
29 }
30
31 // ---------------------------------------------------------------------------
32
33 export {
34   authenticate,
35   token
36 }