joystream/query-node/mappings/bootstrap/index.ts

import { createDBConnection } from '@dzlzv/hydra-processor/lib/db'
import { DatabaseManager, makeDatabaseManager } from '@dzlzv/hydra-db-utils'
import { Connection, getManager, FindConditions } from 'typeorm'

import { bootMembers, IBootstrapMember } from './members'
import { bootWorkers, IBootstrapWorker, IBootstrapWorkers } from './workers'
import { Worker, WorkerType } from 'query-node'

import fs from 'fs'
import path from 'path'

// run bootstrap
init()

// bootstrap flow
async function init() {
  // prepare database and import data
  const [databaseManager, connection] = await createDatabaseManager()

  // escape if db is already initialized
  if (await isDbInitialized(databaseManager)) {
    await connection.close()
    return
  }

  // load import data
  const data = loadData()

  // bootstrap entities
  await bootMembers(databaseManager, data.members)
  await bootWorkers(databaseManager, data.workers)

  await connection.close()
}

async function isDbInitialized(db: DatabaseManager): Promise<boolean> {
  // simple way to check if db is bootstrapped already - check if there is at least 1 storage provider
  const membership = await db.get(Worker, {
    where: {
      type: WorkerType.STORAGE,
    } as FindConditions<Worker>,
  })

  return !!membership
}

async function createDatabaseManager(): Promise<[DatabaseManager, Connection]> {
  // paths in `entities` should be the same as `entities` set in `manifest.yml`
  const entities = ['generated/graphql-server/dist/**/*.model.js']

  // connect to db and create manager
  const connection = await createDBConnection(entities)
  const entityManager = getManager(connection.name)
  const databaseManager = makeDatabaseManager(entityManager)

  return [databaseManager, connection]
}

interface IBootstrapData {
  members: IBootstrapMember[]
  workers: IBootstrapWorkers
}

function loadData(): IBootstrapData {
  return {
    members: JSON.parse(fs.readFileSync(process.env.BOOTSTRAP_DATA_FOLDER + '/members.json').toString()),
    workers: JSON.parse(fs.readFileSync(process.env.BOOTSTRAP_DATA_FOLDER + '/workers.json').toString()),
  }
}