123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
import { isArray, isDefined, isString, isBlank } from '../helpers/types'import Config from '../core/config'import normGenerator from './norm'import { createKey } from './KeyStore'
export default class FuseIndex {  constructor({ getFn = Config.getFn } = {}) {    this.norm = normGenerator(3)    this.getFn = getFn    this.isCreated = false
    this.setIndexRecords()  }  setSources(docs = []) {    this.docs = docs  }  setIndexRecords(records = []) {    this.records = records  }  setKeys(keys = []) {    this.keys = keys    this._keysMap = {}    keys.forEach((key, idx) => {      this._keysMap[key.id] = idx    })  }  create() {    if (this.isCreated || !this.docs.length) {      return    }
    this.isCreated = true
    // List is Array<String>    if (isString(this.docs[0])) {      this.docs.forEach((doc, docIndex) => {        this._addString(doc, docIndex)      })    } else {      // List is Array<Object>      this.docs.forEach((doc, docIndex) => {        this._addObject(doc, docIndex)      })    }
    this.norm.clear()  }  // Adds a doc to the end of the index  add(doc) {    const idx = this.size()
    if (isString(doc)) {      this._addString(doc, idx)    } else {      this._addObject(doc, idx)    }  }  // Removes the doc at the specified index of the index  removeAt(idx) {    this.records.splice(idx, 1)
    // Change ref index of every subsquent doc    for (let i = idx, len = this.size(); i < len; i += 1) {      this.records[i].i -= 1    }  }  getValueForItemAtKeyId(item, keyId) {    return item[this._keysMap[keyId]]  }  size() {    return this.records.length  }  _addString(doc, docIndex) {    if (!isDefined(doc) || isBlank(doc)) {      return    }
    let record = {      v: doc,      i: docIndex,      n: this.norm.get(doc)    }
    this.records.push(record)  }  _addObject(doc, docIndex) {    let record = { i: docIndex, $: {} }
    // Iterate over every key (i.e, path), and fetch the value at that key    this.keys.forEach((key, keyIndex) => {      // console.log(key)      let value = this.getFn(doc, key.path)
      if (!isDefined(value)) {        return      }
      if (isArray(value)) {        let subRecords = []        const stack = [{ nestedArrIndex: -1, value }]
        while (stack.length) {          const { nestedArrIndex, value } = stack.pop()
          if (!isDefined(value)) {            continue          }
          if (isString(value) && !isBlank(value)) {            let subRecord = {              v: value,              i: nestedArrIndex,              n: this.norm.get(value)            }
            subRecords.push(subRecord)          } else if (isArray(value)) {            value.forEach((item, k) => {              stack.push({                nestedArrIndex: k,                value: item              })            })          } else {            // If we're here, the `path` is either incorrect, or pointing to a non-string.            // console.error(new Error(`Path "${key}" points to a non-string value. Received: ${value}`))          }        }        record.$[keyIndex] = subRecords      } else if (!isBlank(value)) {        let subRecord = {          v: value,          n: this.norm.get(value)        }
        record.$[keyIndex] = subRecord      }    })
    this.records.push(record)  }  toJSON() {    return {      keys: this.keys,      records: this.records    }  }}
export function createIndex(keys, docs, { getFn = Config.getFn } = {}) {  const myIndex = new FuseIndex({ getFn })  myIndex.setKeys(keys.map(createKey))  myIndex.setSources(docs)  myIndex.create()  return myIndex}
export function parseIndex(data, { getFn = Config.getFn } = {}) {  const { keys, records } = data  const myIndex = new FuseIndex({ getFn })  myIndex.setKeys(keys)  myIndex.setIndexRecords(records)  return myIndex}