fix(#20): validator middleware returns 400 res

src/api/controllers/bot.ts

@@ -2,10 +2,10 @@ import { Delete, Get, Middleware, Post, Router } from "@discordx/koa"
 import { Client } from "discordx"
 import { Context } from "koa"
 import { injectable } from "tsyringe"
-import validator, { Joi } from 'koa-context-validator'
+import { Joi } from 'koa-context-validator'
 
 import { BaseController } from "@utils/classes"
-import { authenticated, botOnline } from "@api/middlewares"
+import { authenticated, botOnline, validator } from "@api/middlewares"
 import { BaseGuildTextChannel, BaseGuildVoiceChannel, Channel, ChannelType, Guild as DGuild, GuildTextBasedChannel, NewsChannel, PermissionsBitField, User as DUser } from "discord.js"
 import { generalConfig } from "@config"
 import { isInMaintenance, setMaintenance } from "@utils/functions"

src/api/controllers/database.ts

@@ -1,14 +1,14 @@
 import { Get, Middleware, Post, Router } from "@discordx/koa"
 import { injectable } from "tsyringe"
 import { Context } from "koa"
-import validator, { Joi } from "koa-context-validator"
+import { Joi } from "koa-context-validator"
 
 import { Database } from "@services"
 import { BaseController } from "@utils/classes"
 import { formatDate } from "@utils/functions"
 
 import { databaseConfig } from "@config"
-import { authenticated } from "@api/middlewares"
+import { authenticated, validator } from "@api/middlewares"
 
 @Router({ options: { prefix: "/database" } })
 @Middleware(

src/api/controllers/stats.ts

@@ -1,11 +1,11 @@
+import { injectable } from "tsyringe"
 import { Get, Middleware, Router } from "@discordx/koa"
+import { Joi } from "koa-context-validator"
 import { Context } from "koa"
 
 import { BaseController } from "@utils/classes"
-import { injectable } from "tsyringe"
 import { Stats } from "@services"
-import { authenticated } from "@api/middlewares"
-import validator, { Joi } from "koa-context-validator"
+import { authenticated, validator } from "@api/middlewares"
 
 @Router({ options: { prefix: '/stats' }})
 @Middleware(

src/api/middlewares/index.ts

@@ -1,3 +1,4 @@
 export * from './log'
 export * from './botOnline'
-export * from './authenticated'
+export * from './authenticated'
+export * from './validator'

src/api/middlewares/validator.ts

@@ -0,0 +1,47 @@
+import Joi, { ValidationOptions } from "joi";
+import { Context, Next } from "koa";
+
+type InputSchema = {
+    query?: Joi.ObjectSchema
+    headers?: Joi.ObjectSchema
+    body?: Joi.ObjectSchema
+    params?: Joi.ObjectSchema
+}
+
+const isKeyOnContext = (key:string) => ['params'].includes(key)
+
+export const validator = (inputSchema: InputSchema, options: ValidationOptions = {}) => (ctx: Context, next: Next) => {
+
+    const validateOptions = {
+        ...options,
+        context: {
+            ...ctx,
+            ...options.context
+        },
+    }
+    const keys = Object.keys(inputSchema)
+
+    for (const key of keys) {
+
+        const targetSchema = inputSchema[key as keyof typeof inputSchema]
+        if (!targetSchema) continue
+
+        let source, value: string
+        if (isKeyOnContext(key)) {
+            source = ctx
+            value = ctx[key]
+        } else {
+            source = ctx.request
+            value = ctx.request[key as keyof typeof ctx.request]
+        }
+
+        const validatedValue = targetSchema.validate(value, validateOptions)
+        if (validatedValue.error) {
+            ctx.status = 400
+            ctx.body = validatedValue.error.message
+            return
+        }
+    }
+
+    next()
+}