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())
}