memory-storex.ts

import UrlField from './storage/url-field'
import schemaPatcher from './storage/dexie-schema'
import collections from './old-schema'
import initStorex from './storex'
import inMemoryDb from '@worldbrain/storex-backend-dexie/lib/in-memory'
import { suggestObjects } from './search/suggest'
import { StorageManager } from './types'
import { plugins } from './storex-plugins'
import stemmerSelector from './stemmers'

export default () => {
    const idbImplementation = inMemoryDb()
    return initStorex<StorageManager>({
        stemmerSelector,
        collections,
        schemaPatcher,
        dbName: 'test',
        customFields: [{ key: 'url', field: UrlField }],
        backendPlugins: plugins,
        idbImplementation,
        modifyInstance(storex: StorageManager) {
            const oldMethod = storex.collection.bind(storex)

            storex.collection = (name: string) => ({
                ...oldMethod(name),
                suggestObjects: (query, opts) =>
                    suggestObjects(async () => storex.backend['dexieInstance'])(
                        name,
                        query,
                        opts,
                    ),
            })

            return storex
        },
    })
}