From 55497175961a55059c0a15789976a4383a4ef6d0 Mon Sep 17 00:00:00 2001
From: skinmaker1345 <skinmaker1345@gmail.com>
Date: Wed, 22 Jan 2025 23:30:43 +0900
Subject: [PATCH] fix: set state to deleted for deleted bots

---
 pages/bots/[id]/index.tsx |  4 ++--
 types/index.ts            |  1 +
 utils/Query.ts            | 37 +++++++++++++++++++++----------------
 3 files changed, 24 insertions(+), 18 deletions(-)

diff --git a/pages/bots/[id]/index.tsx b/pages/bots/[id]/index.tsx
index af3959d..01e368c 100644
--- a/pages/bots/[id]/index.tsx
+++ b/pages/bots/[id]/index.tsx
@@ -88,7 +88,7 @@ const Bots: NextPage<BotsProps> = ({ data, desc, date, user, theme, csrfToken })
 				) : (
 					<>
 						<div className='w-full pb-2'>
-							{data.state === 'private' ? (
+							{checkBotFlag(data.flags, 'private') ? (
 								<Message type='info'>
 									<h2 className='text-lg font-extrabold'>
 										해당 봇은 특수목적 봇이므로 초대하실 수 없습니다.
@@ -158,7 +158,7 @@ const Bots: NextPage<BotsProps> = ({ data, desc, date, user, theme, csrfToken })
 								</p>
 							</div>
 							<div className='w-full lg:w-1/4'>
-								{data.state === 'ok' && (
+								{(data.state === 'ok' && !checkBotFlag(data.flags, 'private')) && (
 									<LongButton newTab href={`/bots/${router.query.id}/invite`}>
 										<h4 className='whitespace-nowrap'>
 											<i className='fas fa-user-plus text-discord-blurple' /> 초대하기
diff --git a/types/index.ts b/types/index.ts
index efc78ce..6a92f7a 100644
--- a/types/index.ts
+++ b/types/index.ts
@@ -156,6 +156,7 @@ export enum BotFlags {
 	verified = 1 << 4,
 	premium = 1 << 5,
 	hackerthon = 1 << 6,
+	private = 1 << 7,
 }
 
 export enum ServerFlags {
diff --git a/utils/Query.ts b/utils/Query.ts
index 0f41b37..eda62b3 100644
--- a/utils/Query.ts
+++ b/utils/Query.ts
@@ -76,7 +76,12 @@ async function getBot(id: string, topLevel = true): Promise<Bot> {
 
 	if (res) {
 		const discordBot = await get.discord.user.load(res.id)
-		if (!discordBot || Number(discordBot.discriminator) === 0) return null
+		if (!discordBot || Number(discordBot.discriminator) === 0) {
+			knex('bots')
+				.update({ state: 'deleted' })
+				.where({ id })
+				.then((r) => r)
+		}
 		const botMember = (await getMainGuild()
 			?.members?.fetch(res.id)
 			.catch((e) => e)) as GuildMember
@@ -239,41 +244,41 @@ async function getBotList(type: ListType, page = 1, query?: string): Promise<Lis
 	let res: { id: string }[]
 	let count: string | number
 	if (type === 'VOTE') {
-		count = (await knex('bots').whereNot({ state: 'blocked' }).count())[0]['count(*)']
+		count = (await knex('bots').where({ state: 'ok' }).count())[0]['count(*)']
 		res = await knex('bots')
 			.orderBy('votes', 'desc')
 			.orderBy('servers', 'desc')
 			.limit(16)
 			.offset(((page ? Number(page) : 1) - 1) * 16)
 			.select(['id'])
-			.whereNot({ state: 'blocked' })
+			.where({ state: 'ok' })
 	} else if (type === 'TRUSTED') {
-		count = (await knex('bots').where({ trusted: true }).count().whereNot({ state: 'blocked' }))[0][
+		count = (await knex('bots').where({ trusted: true }).count().where({ state: 'ok' }))[0][
 			'count(*)'
 		]
 		res = await knex('bots')
-			.whereNot({ state: 'blocked' })
+			.where({ state: 'ok' })
 			.where({ trusted: true })
 			.orderByRaw('RAND()')
 			.limit(16)
 			.offset(((page ? Number(page) : 1) - 1) * 16)
 			.select(['id'])
-			.whereNot({ state: 'blocked' })
+			.where({ state: 'ok' })
 	} else if (type === 'NEW') {
-		count = (await knex('bots').whereNot({ state: 'blocked' }).count())[0]['count(*)']
+		count = (await knex('bots').where({ state: 'ok' }).count())[0]['count(*)']
 		res = await knex('bots')
 			.orderBy('date', 'desc')
 			.limit(16)
 			.offset(((page ? Number(page) : 1) - 1) * 16)
 			.select(['id'])
-			.whereNot({ state: 'blocked' })
+			.where({ state: 'ok' })
 	} else if (type === 'PARTNERED') {
-		count = (
-			await knex('bots').where({ partnered: true }).andWhereNot({ state: 'blocked' }).count()
-		)[0]['count(*)']
+		count = (await knex('bots').where({ partnered: true }).andWhere({ state: 'ok' }).count())[0][
+			'count(*)'
+		]
 		res = await knex('bots')
 			.where({ partnered: true })
-			.andWhereNot({ state: 'blocked' })
+			.andWhere({ state: 'ok' })
 			.orderByRaw('RAND()')
 			.limit(16)
 			.offset(((page ? Number(page) : 1) - 1) * 16)
@@ -284,12 +289,12 @@ async function getBotList(type: ListType, page = 1, query?: string): Promise<Lis
 		count = (
 			await knex('bots')
 				.where('category', 'like', `%${decodeURI(query)}%`)
-				.andWhereNot({ state: 'blocked' })
+				.andWhere({ state: 'ok' })
 				.count()
 		)[0]['count(*)']
 		res = await knex('bots')
 			.where('category', 'like', `%${decodeURI(query)}%`)
-			.andWhereNot({ state: 'blocked' })
+			.andWhere({ state: 'ok' })
 			.orderBy('votes', 'desc')
 			.orderBy('servers', 'desc')
 			.limit(16)
@@ -299,13 +304,13 @@ async function getBotList(type: ListType, page = 1, query?: string): Promise<Lis
 		if (!query) throw new Error('쿼리가 누락되었습니다.')
 		count = (
 			await knex.raw(
-				'SELECT count(*) FROM bots WHERE `state` != "blocked" AND MATCH(`name`, `intro`, `desc`) AGAINST(? in boolean mode)',
+				'SELECT count(*) FROM bots WHERE `state` = "ok" AND MATCH(`name`, `intro`, `desc`) AGAINST(? in boolean mode)',
 				[decodeURI(query) + '*']
 			)
 		)[0][0]['count(*)']
 		res = (
 			await knex.raw(
-				'SELECT id, votes, MATCH(`name`, `intro`, `desc`) AGAINST(? in boolean mode) as relevance FROM bots WHERE `state` != "blocked" AND MATCH(`name`, `intro`, `desc`) AGAINST(? in boolean mode) ORDER BY relevance DESC, votes DESC LIMIT 8 OFFSET ?',
+				'SELECT id, votes, MATCH(`name`, `intro`, `desc`) AGAINST(? in boolean mode) as relevance FROM bots WHERE `state` = "ok" AND MATCH(`name`, `intro`, `desc`) AGAINST(? in boolean mode) ORDER BY relevance DESC, votes DESC LIMIT 8 OFFSET ?',
 				[decodeURI(query) + '*', decodeURI(query) + '*', ((page ? Number(page) : 1) - 1) * 8]
 			)
 		)[0]