feat: add validation for login

server/src/api/routes/auth.js

@@ -11,34 +11,28 @@ import {
   registerFromSession,
 } from '../controllers/authController.js'
 import {
-  checkNewEmail,
-  checkNewUsername,
-  checkPassword,
+  checkUsernameValidators,
+  loginValidators,
+  registerFromSessionValidators,
+  registerValidators,
 } from '../validators/authValidator.js'
-import { noError } from '../../middleware/validator.js'
 
 const router = Router()
 
 router.post(
   '/login',
+  loginValidators,
   passport.authenticate('local', { failureMessage: true }),
   (req, res) => {
     res.json({ message: 'success' })
   }
 )
 
-router.post(
-  '/register',
-  checkNewUsername(),
-  checkNewEmail(),
-  checkPassword(),
-  noError,
-  register
-)
+router.post('/register', registerValidators, register)
 
 router.post('/logout', logout)
 
-router.get('/check-username', checkNewUsername(), noError, (req, res) =>
+router.get('/check-username', checkUsernameValidators, (req, res) =>
   res.json({ available: true })
 )
 router.get('/check-session', checkSession)
@@ -59,8 +53,7 @@ if (enabledMethods.github) {
 
 router.post(
   '/register-from-session',
-  checkNewUsername(),
-  noError,
+  registerFromSessionValidators,
   registerFromSession
 )
 

server/src/api/validators/authValidator.js

@@ -1,10 +1,11 @@
 import { check } from 'express-validator'
 
-import { isValidUsername } from '@linux-odyssey/utils'
+import { isValidUsername, isValidEmail } from '@linux-odyssey/utils'
 import { User } from '@linux-odyssey/models'
 import { passwordPolicy } from '@linux-odyssey/constants'
+import { noError } from '../../middleware/validator.js'
 
-export const checkUsername = () =>
+const checkUsername = () =>
   check('username')
     .notEmpty()
     .isString()
@@ -16,7 +17,7 @@ export const checkUsername = () =>
     })
     .trim()
 
-export const checkNewUsername = () =>
+const checkNewUsername = () =>
   checkUsername().custom(async (username) => {
     if (await User.exists({ username })) {
       throw new Error('Username taken')
@@ -24,9 +25,9 @@ export const checkNewUsername = () =>
     return true
   })
 
-export const checkEmail = () => check('email').notEmpty().isEmail().trim()
+const checkEmail = () => check('email').notEmpty().isString().isEmail().trim()
 
-export const checkNewEmail = () =>
+const checkNewEmail = () =>
   checkEmail().custom(async (email) => {
     if (await User.exists({ email })) {
       throw new Error('Email taken')
@@ -34,5 +35,30 @@ export const checkNewEmail = () =>
     return true
   })
 
-export const checkPassword = () =>
+const checkPassword = () =>
   check('password').notEmpty().isStrongPassword(passwordPolicy)
+
+const checkLogin = () =>
+  check('username')
+    .isString()
+    .notEmpty()
+    .trim()
+    .custom((login) => {
+      if (!isValidUsername(login) && !isValidEmail(login)) {
+        throw new Error('Invalid login')
+      }
+      return true
+    })
+
+export const registerValidators = [
+  checkNewUsername(),
+  checkNewEmail(),
+  checkPassword(),
+  noError,
+]
+
+export const registerFromSessionValidators = [checkNewUsername(), noError]
+
+export const loginValidators = [checkLogin(), noError]
+
+export const checkUsernameValidators = [checkUsername(), noError]