47f10301c8
* Set common package.json sideEffects: false * Configure SWC to modularize lodash imports * Import specific lodash functions instead of _ * Add an eslint rule to avoid full lodash import
54 lines
1.6 KiB
TypeScript
54 lines
1.6 KiB
TypeScript
import * as admin from 'firebase-admin'
|
|
|
|
import { initAdmin } from './script-init'
|
|
initAdmin()
|
|
|
|
import { getValues } from '../utils'
|
|
import { User } from '../../../common/user'
|
|
import { batchedWaitAll } from '../../../common/util/promise'
|
|
import { Contract } from '../../../common/contract'
|
|
import { updateWordScores } from '../update-recommendations'
|
|
import { computeFeed } from '../update-feed'
|
|
import { getFeedContracts, getTaggedContracts } from '../get-feed-data'
|
|
import { CATEGORY_LIST } from '../../../common/categories'
|
|
|
|
const firestore = admin.firestore()
|
|
|
|
async function updateFeed() {
|
|
console.log('Updating feed')
|
|
|
|
const contracts = await getValues<Contract>(firestore.collection('contracts'))
|
|
const feedContracts = await getFeedContracts()
|
|
const users = await getValues<User>(
|
|
firestore.collection('users').where('username', '==', 'JamesGrugett')
|
|
)
|
|
|
|
await batchedWaitAll(
|
|
users.map((user) => async () => {
|
|
console.log('Updating recs for', user.username)
|
|
await updateWordScores(user, contracts)
|
|
console.log('Updating feed for', user.username)
|
|
await computeFeed(user, feedContracts)
|
|
})
|
|
)
|
|
|
|
console.log('Updating feed categories!')
|
|
|
|
await batchedWaitAll(
|
|
users.map((user) => async () => {
|
|
for (const category of CATEGORY_LIST) {
|
|
const contracts = await getTaggedContracts(category)
|
|
const feed = await computeFeed(user, contracts)
|
|
await firestore
|
|
.collection(`private-users/${user.id}/cache`)
|
|
.doc(`feed-${category}`)
|
|
.set({ feed })
|
|
}
|
|
})
|
|
)
|
|
}
|
|
|
|
if (require.main === module) {
|
|
updateFeed().then(() => process.exit())
|
|
}
|