/* global WIKI */ import express from 'express' import ExpressBrute from 'express-brute' import BruteKnex from '../helpers/brute-knex.mjs' import { find, isEmpty, set } from 'lodash-es' import path from 'node:path' import { DateTime } from 'luxon' export default function () { const router = express.Router() const bruteforce = new ExpressBrute(new BruteKnex({ createTable: true, knex: WIKI.db.knex }), { freeRetries: 5, minWait: 5 * 60 * 1000, // 5 minutes maxWait: 60 * 60 * 1000, // 1 hour failCallback: (req, res, next) => { res.status(401).send('Too many failed attempts. Try again later.') } }) /** * Login form */ router.get('/login', async (req, res, next) => { // -> Bypass Login if (WIKI.config.auth.autoLogin && !req.query.all) { const stg = await WIKI.db.authentication.query().orderBy('order').first() const stgInfo = find(WIKI.data.authentication, ['key', stg.strategyKey]) if (!stgInfo.useForm) { return res.redirect(`/login/${stg.key}`) } } // -> Show Login res.sendFile(path.join(WIKI.ROOTPATH, 'assets/index.html')) }) /** * Social Strategies Login */ router.get('/login/:strategy', async (req, res, next) => { try { await WIKI.db.users.login({ strategy: req.params.strategy }, { req, res }) } catch (err) { next(err) } }) /** * Social Strategies Callback */ router.all('/login/:strategy/callback', async (req, res, next) => { if (req.method !== 'GET' && req.method !== 'POST') { return next() } try { const authResult = await WIKI.db.users.login({ strategy: req.params.strategy }, { req, res }) res.cookie('jwt', authResult.jwt, { expires: DateTime.now().plus({ years: 1 }).toJSDate() }) const loginRedirect = req.cookies['loginRedirect'] if (loginRedirect === '/' && authResult.redirect) { res.clearCookie('loginRedirect') res.redirect(authResult.redirect) } else if (loginRedirect) { res.clearCookie('loginRedirect') res.redirect(loginRedirect) } else if (authResult.redirect) { res.redirect(authResult.redirect) } else { res.redirect('/') } } catch (err) { next(err) } }) /** * Logout */ router.get('/logout', async (req, res, next) => { const redirURL = await WIKI.db.users.logout({ req, res }) req.logout((err) => { if (err) { return next(err) } res.clearCookie('jwt') res.redirect(redirURL) }) }) /** * Register form */ router.get('/register', async (req, res, next) => { set(res.locals, 'pageMeta.title', 'Register') const localStrg = await WIKI.db.authentication.getStrategy('local') if (localStrg.selfRegistration) { res.sendFile(path.join(WIKI.ROOTPATH, 'assets/index.html')) } else { next(new WIKI.Error.AuthRegistrationDisabled()) } }) /** * Verify */ router.get('/verify/:token', bruteforce.prevent, async (req, res, next) => { try { const usr = await WIKI.db.userKeys.validateToken({ kind: 'verify', token: req.params.token }) await WIKI.db.users.query().patch({ isVerified: true }).where('id', usr.id) req.brute.reset() if (WIKI.config.auth.enforce2FA) { res.redirect('/login') } else { const result = await WIKI.db.users.refreshToken(usr) res.cookie('jwt', result.token, { expires: DateTime.now().plus({ years: 1 }).toJSDate() }) res.redirect('/') } } catch (err) { next(err) } }) /** * Reset Password */ router.get('/login-reset/:token', bruteforce.prevent, async (req, res, next) => { try { const usr = await WIKI.db.userKeys.validateToken({ kind: 'resetPwd', token: req.params.token }) if (!usr) { throw new Error('Invalid Token') } req.brute.reset() const changePwdContinuationToken = await WIKI.db.userKeys.generateToken({ userId: usr.id, kind: 'changePwd' }) const bgUrl = !isEmpty(WIKI.config.auth.loginBgUrl) ? WIKI.config.auth.loginBgUrl : '/_assets/img/splash/1.jpg' res.render('login', { bgUrl, hideLocal: WIKI.config.auth.hideLocal, changePwdContinuationToken }) } catch (err) { next(err) } }) /** * JWT Public Endpoints */ router.get('/.well-known/jwk.json', function (req, res, next) { res.json(WIKI.config.certs.jwk) }) router.get('/.well-known/jwk.pem', function (req, res, next) { res.send(WIKI.config.certs.public) }) return router }