PeerTube/server/core/middlewares/auth.ts

113 lines
3.3 KiB
TypeScript
Raw Normal View History

2021-08-27 14:32:44 +02:00
import express from 'express'
2018-12-26 10:36:24 +01:00
import { Socket } from 'socket.io'
import { HttpStatusCode, HttpStatusCodeType, ServerErrorCodeType } from '@peertube/peertube-models'
import { getAccessToken } from '@server/lib/auth/oauth-model.js'
import { RunnerModel } from '@server/models/runner/runner.js'
import { logger } from '../helpers/logger.js'
import { handleOAuthAuthenticate } from '../lib/auth/oauth.js'
2016-03-21 11:56:33 +01:00
function authenticate (req: express.Request, res: express.Response, next: express.NextFunction) {
handleOAuthAuthenticate(req, res)
.then((token: any) => {
res.locals.oauth = { token }
res.locals.authenticated = true
return next()
})
.catch(err => {
logger.info('Cannot authenticate.', { err })
return res.fail({
status: err.status,
message: 'Token is invalid',
type: err.name
})
})
}
2018-12-26 10:36:24 +01:00
function authenticateSocket (socket: Socket, next: (err?: any) => void) {
2020-11-19 08:58:34 +01:00
const accessToken = socket.handshake.query['accessToken']
2018-12-26 10:36:24 +01:00
logger.debug('Checking access token in runner.')
2018-12-26 10:36:24 +01:00
2019-04-23 09:50:57 +02:00
if (!accessToken) return next(new Error('No access token provided'))
2021-03-03 15:22:38 +01:00
if (typeof accessToken !== 'string') return next(new Error('Access token is invalid'))
2019-04-23 09:50:57 +02:00
2018-12-26 10:36:24 +01:00
getAccessToken(accessToken)
.then(tokenDB => {
const now = new Date()
if (!tokenDB || tokenDB.accessTokenExpiresAt < now || tokenDB.refreshTokenExpiresAt < now) {
return next(new Error('Invalid access token.'))
}
2021-03-03 15:22:38 +01:00
socket.handshake.auth.user = tokenDB.User
2018-12-26 10:36:24 +01:00
return next()
})
2020-01-31 16:56:52 +01:00
.catch(err => logger.error('Cannot get access token.', { err }))
2018-12-26 10:36:24 +01:00
}
function authenticatePromise (options: {
req: express.Request
res: express.Response
errorMessage?: string
errorStatus?: HttpStatusCodeType
errorType?: ServerErrorCodeType
}) {
const { req, res, errorMessage = 'Not authenticated', errorStatus = HttpStatusCode.UNAUTHORIZED_401, errorType } = options
2021-02-03 09:33:05 +01:00
return new Promise<void>(resolve => {
// Already authenticated? (or tried to)
2020-06-17 10:55:40 +02:00
if (res.locals.oauth?.token.User) return resolve()
if (res.locals.authenticated === false) {
return res.fail({
status: errorStatus,
type: errorType,
message: errorMessage
})
}
authenticate(req, res, () => resolve())
})
}
function optionalAuthenticate (req: express.Request, res: express.Response, next: express.NextFunction) {
if (req.header('authorization')) return authenticate(req, res, next)
res.locals.authenticated = false
return next()
}
2016-03-21 11:56:33 +01:00
// ---------------------------------------------------------------------------
function authenticateRunnerSocket (socket: Socket, next: (err?: any) => void) {
const runnerToken = socket.handshake.auth['runnerToken']
logger.debug('Checking runner token in socket.')
if (!runnerToken) return next(new Error('No runner token provided'))
if (typeof runnerToken !== 'string') return next(new Error('Runner token is invalid'))
RunnerModel.loadByToken(runnerToken)
.then(runner => {
if (!runner) return next(new Error('Invalid runner token.'))
socket.handshake.auth.runner = runner
return next()
})
.catch(err => logger.error('Cannot get runner token.', { err }))
}
// ---------------------------------------------------------------------------
2017-05-15 22:22:03 +02:00
export {
authenticate,
2018-12-26 10:36:24 +01:00
authenticateSocket,
2022-06-22 14:03:50 +02:00
authenticatePromise,
optionalAuthenticate,
authenticateRunnerSocket
2017-05-15 22:22:03 +02:00
}