From 4f821bfc3785a90410f2ee6b42994431d6f85423 Mon Sep 17 00:00:00 2001
From: wonderlandpark <wonderlandpark@outlook.kr>
Date: Wed, 7 Jul 2021 22:06:39 +0900
Subject: [PATCH] feat(Query): added management queries

---
 utils/Query.ts | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/utils/Query.ts b/utils/Query.ts
index 1efbe80..16cfa2b 100644
--- a/utils/Query.ts
+++ b/utils/Query.ts
@@ -3,7 +3,7 @@ import { TLRU } from 'tlru'
 import DataLoader from 'dataloader'
 import { User as DiscordUser } from 'discord.js'
 
-import { Bot, User, ListType, BotList, TokenRegister, BotFlags, DiscordUserFlags, SubmittedBot } from '@types'
+import { Bot, User, ListType, BotList, TokenRegister, BotFlags, DiscordUserFlags, SubmittedBot, UserSpec } from '@types'
 import { categories, imageSafeHost, SpecialEndPoints, VOTE_COOLDOWN } from './Constants'
 
 import knex from './Knex'
@@ -512,6 +512,19 @@ export const remove = {
 	bot: deleteBot
 }
 
+export const management = {
+	user: {
+		get: async (id: string): Promise<UserSpec> => {
+			const user = await get.user.load(id)
+			const userSpec = await knex('users').select(['email', 'perm']).where({ id })
+			return user && userSpec[0] ? { ...user, ...userSpec[0] } : null
+		},
+		update: async (id: string, { flags }:{ flags: number }) => {
+			await knex('users').update({ flags }).where({ id })
+		}
+	}
+}
+
 export const ratelimit = {
 	image: (ip: string) => {
 		addRequest(ip, imageRateLimit)