gridsome/lib/store/NodeIndex.js

A JAMstack framework for building blazing fast websites with Vue.js

const { Collection } = require('lokijs')
const { SyncWaterfallHook } = require('tapable')

class NodeIndex {
  constructor (app) {
    this.app = app

    this.hooks = {
      addEntry: new SyncWaterfallHook(['entry', 'node', 'collection'])
    }

    this.hooks.addEntry.tap('BelongsToProcessor', require('./processNodeReferences'))

    this.index = new Collection('NodeIndex', {
      indices: ['typeName', 'id'],
      unique: ['uid'],
      disableMeta: true,
      adaptiveBinaryIndices: false
    })
  }

  addEntry (node, collection) {
    const options = {
      typeName: node.internal.typeName,
      uid: node.$uid,
      id: node.id
    }

    const entry = this.hooks.addEntry.call(
      options,
      node,
      collection
    )

    this.index.insert(entry)
  }

  updateEntry (node, collection) {
    const oldEntry = this.index.by('uid', node.$uid)

    const options = {
      $loki: oldEntry.$loki,
      typeName: node.internal.typeName,
      uid: node.$uid,
      id: node.id
    }

    const entry = this.hooks.addEntry.call(
      options,
      node,
      collection
    )

    this.index.update(entry)
  }

  removeEntry (node) {
    this.index.findAndRemove({ uid: node.$uid })
  }

  getEntry (uid) {
    return this.index.by('uid', uid) || null
  }

  getChain () {
    return this.index.chain()
  }
}

module.exports = NodeIndex