wonderlandpark/core
2
1
Fork 0
mirror of https://github.com/koreanbots/core.git synced 2025-12-16 14:30:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: added bot list methods

Browse Source
This commit is contained in:
원더 2021-01-09 21:53:46 +09:00
parent 1757b58a6a
commit d17da38615
1 changed files with 33 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
utils/Fetch.ts
View File

@ -11,19 +11,13 @@ const privateKey = fs.readFileSync('./private.key')
const bot = new DataLoader( const bot = new DataLoader(
async (ids: string[]) => async (ids: string[]) =>
(await Promise.all(ids.map(async (el: string) => getBot(el)))).map(row => ({ ...row })), (await Promise.all(ids.map(async (el: string) => getBot(el)))).map(row => ({ ...row }))
{
batchScheduleFn: callback => setTimeout(callback, 100),
}
) )
new DataLoader(async ()=> { return []}) 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 }))
{
batchScheduleFn: callback => setTimeout(callback, 100),
}
) )
const user = new DataLoader( const user = new DataLoader(
@ -36,9 +30,28 @@ const userWithNoBot = new DataLoader(
(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 }))
) )
const botList = new DataLoader( const botListVotes = new DataLoader(
async (argument: botListArgument[]) => async (page: number[]) =>
(await Promise.all(argument.map((el) => getBotList(el.type, el.page, el.query)))).map(row => ({ ...row })) (await Promise.all(page.map((el) => getBotList('VOTE', el)))).map(row => ({ ...row })),
{
batchScheduleFn: callback => setTimeout(callback, 100),
}
)
const botListNew = new DataLoader(
async (page: number[]) =>
(await Promise.all(page.map((el) => getBotList('NEW', el)))).map(row => ({ ...row })),
{
batchScheduleFn: callback => setTimeout(callback, 100),
}
)
const botListTrusted = new DataLoader(
async (page: number[]) =>
(await Promise.all(page.map((el) => getBotList('TRUSTED', el)))).map(row => ({ ...row })),
{
batchScheduleFn: callback => setTimeout(callback, 100),
}
) )
async function getBot(id: string, owners = true): Promise<Bot> { async function getBot(id: string, owners = true): Promise<Bot> {
@ -71,8 +84,8 @@ async function getBot(id: string, owners = true): Promise<Bot> {
'banner', 'banner',
]) ])
.where({ id }) .where({ id })
.orWhere({ vanity: id, trusted: 1 }) .orWhere({ vanity: id, trusted: true })
.orWhere({ vanity: id, partnered: 1 }) .orWhere({ vanity: id, partnered: true })
if (res[0]) { if (res[0]) {
res[0].category = JSON.parse(res[0].category) res[0].category = JSON.parse(res[0].category)
res[0].owners = JSON.parse(res[0].owners) res[0].owners = JSON.parse(res[0].owners)
@ -117,21 +130,18 @@ async function getBotList(type: ListType, page = 1, query?: string):Promise<BotL
} else if (type === 'TRUSTED') { } else if (type === 'TRUSTED') {
count = ( count = (
await knex('bots') await knex('bots')
.where({ trusted: 1 }) .where({ trusted: true })
.count() .count()
)[0]['count(*)'] )[0]['count(*)']
res = await knex('bots') res = await knex('bots')
.where({ trusted: true }) .where({ trusted: true })
.orderBy(await knex.raw('RAND()')) .orderByRaw('RAND()')
.orderBy('votes', 'desc')
.orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select(['id']) .select(['id'])
} else if (type === 'NEW') { } else if (type === 'NEW') {
count = ( count = (
await knex('bots') await knex('bots')
.where({ trusted: 1 })
.count() .count()
)[0]['count(*)'] )[0]['count(*)']
res = await knex('bots') res = await knex('bots')
@ -142,14 +152,12 @@ async function getBotList(type: ListType, page = 1, query?: string):Promise<BotL
} else if (type === 'PARTNERED') { } else if (type === 'PARTNERED') {
count = ( count = (
await knex('bots') await knex('bots')
.where({ partnered: 1 }) .where({ partnered: true })
.count() .count()
)[0]['count(*)'] )[0]['count(*)']
res = await knex('bots') res = await knex('bots')
.where({ partnered: 1 }) .where({ partnered: true })
.orderBy(await knex.raw('RAND()')) .orderByRaw('RAND()')
.orderBy('votes', 'desc')
.orderBy('servers', 'desc')
.limit(16) .limit(16)
.offset(((page ? Number(page) : 1) - 1) * 16) .offset(((page ? Number(page) : 1) - 1) * 16)
.select(['id']) .select(['id'])
@ -217,7 +225,7 @@ async function getBotList(type: ListType, page = 1, query?: string):Promise<BotL
res = [] res = []
} }
return { type, data: await Promise.all(res.map(async el => await getBot(el.id))), currentPage: page, totalPage: Math.ceil(Number(count) / 16) } return { type, data: (await Promise.all(res.map(async el => await getBot(el.id)))).map(r=> ({...r})), currentPage: page, totalPage: Math.ceil(Number(count) / 16) }
} }
export default { bot, user, botList } export default { bot, user, botListVotes, botListNew, botListTrusted }