feat: added Bot list method

This commit is contained in:
원더 2021-01-09 19:30:09 +09:00
parent fb624ef25c
commit d18184c389

View File

@ -1,11 +1,10 @@
import DataLoader from 'dataloader' import DataLoader from 'dataloader'
import * as Query from './Query'
import fetch from 'node-fetch'
import jwt from 'jsonwebtoken'
import fs from 'fs' import fs from 'fs'
import { Bot, ListType, User } from '../types' import { Bot, ListType, User, BotList } from '../types'
import knex from './Knex' import knex from './Knex'
import { cats } from './Constants' import { cats } from './Constants'
import { botListArgument } from './Yup'
import { ReactText } from 'react'
const publicPem = fs.readFileSync('./public.pem') const publicPem = fs.readFileSync('./public.pem')
const privateKey = fs.readFileSync('./private.key') const privateKey = fs.readFileSync('./private.key')
@ -18,6 +17,7 @@ const bot = new DataLoader(
} }
) )
new DataLoader(async ()=> { return []})
const botWithNoUser = new DataLoader( const botWithNoUser = new DataLoader(
async (ids: string[]) => async (ids: string[]) =>
(await Promise.all(ids.map(async (el: string) => getBot(el, false)))).map(row => ({ ...row })), (await Promise.all(ids.map(async (el: string) => getBot(el, false)))).map(row => ({ ...row })),
@ -28,18 +28,17 @@ const botWithNoUser = new DataLoader(
const user = new DataLoader( const user = new DataLoader(
async (ids: string[]) => async (ids: string[]) =>
(await Promise.all(ids.map((el: string) => getUser(el)))).map(row => ({ ...row })), (await Promise.all(ids.map((el: string) => getUser(el)))).map(row => ({ ...row }))
{
batchScheduleFn: callback => setTimeout(callback, 100),
}
) )
const userWithNoBot = new DataLoader( const userWithNoBot = new DataLoader(
async (ids: string[]) => async (ids: string[]) =>
(await Promise.all(ids.map((el: string) => getUser(el, false)))).map(row => ({ ...row })), (await Promise.all(ids.map((el: string) => getUser(el, false)))).map(row => ({ ...row }))
{ )
batchScheduleFn: callback => setTimeout(callback, 100),
} const botList = new DataLoader(
async (argument: botListArgument[]) =>
(await Promise.all(argument.map((el) => getBotList(el.type, el.page, el.query)))).map(row => ({ ...row }))
) )
async function getBot(id: string, owners = true): Promise<Bot> { async function getBot(id: string, owners = true): Promise<Bot> {
@ -104,9 +103,9 @@ async function getUser(id: string, bots = true): Promise<User> {
return res[0] || null return res[0] || null
} }
async function getBotList(type:ListType, page = 1, query?: string) { async function getBotList(type: ListType, page = 1, query?: string):Promise<BotList> {
let res let res: { id: string }[]
let count let count: ReactText
if (type === 'VOTE') { if (type === 'VOTE') {
count = (await knex('bots').count())[0]['count(*)'] count = (await knex('bots').count())[0]['count(*)']
res = await knex('bots') res = await knex('bots')
@ -114,33 +113,7 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('servers', 'desc') .orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'partnered',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} else if (type === 'TRUSTED') { } else if (type === 'TRUSTED') {
count = ( count = (
await knex('bots') await knex('bots')
@ -154,33 +127,7 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('servers', 'desc') .orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'partnered',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} else if (type === 'NEW') { } else if (type === 'NEW') {
count = ( count = (
await knex('bots') await knex('bots')
@ -191,33 +138,7 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('date', 'desc') .orderBy('date', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'partnered',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} else if (type === 'PARTNERED') { } else if (type === 'PARTNERED') {
count = ( count = (
await knex('bots') await knex('bots')
@ -231,33 +152,7 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('servers', 'desc') .orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'partnered',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} else if (type === 'CATEGORY') { } else if (type === 'CATEGORY') {
if (!query) throw new Error('쿼리가 누락되었습니다.') if (!query) throw new Error('쿼리가 누락되었습니다.')
if (!cats.includes(query)) throw new Error('알 수 없는 카테고리입니다.') if (!cats.includes(query)) throw new Error('알 수 없는 카테고리입니다.')
@ -272,33 +167,7 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('servers', 'desc') .orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'partnered',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} else if (type === 'SEARCH') { } else if (type === 'SEARCH') {
if (!query) throw new Error('쿼리가 누락되었습니다.') if (!query) throw new Error('쿼리가 누락되었습니다.')
try { try {
@ -321,32 +190,7 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('servers', 'desc') .orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} catch (e) { } catch (e) {
count = ( count = (
await knex('bots') await knex('bots')
@ -366,43 +210,14 @@ async function getBotList(type:ListType, page = 1, query?: string) {
.orderBy('servers', 'desc') .orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select([ .select(['id'])
'id',
'owners',
'lib',
'prefix',
'votes',
'servers',
'intro',
'desc',
'web',
'git',
'url',
'category',
'status',
'name',
'avatar',
'tag',
'verified',
'trusted',
'discord',
'boosted',
'state',
'vanity',
'bg',
'banner',
])
} }
} else { } else {
count = 1 count = 1
res = [] res = []
} }
res.map(el => {
el.category = JSON.parse(el.category)
return el
})
return { type, data: res, currentPage: page, totalPage: Math.ceil(count / 16) } return { type, data: await Promise.all(res.map(async el => await getBot(el.id))), currentPage: page, totalPage: Math.ceil(Number(count) / 16) }
} }
export default { bot, user } export default { bot, user, botList }