#104 - Refactored search history to use IndexedDB

This commit is contained in:
Simon Cambier
2022-10-22 16:37:31 +02:00
parent 8696f1381d
commit 9eed978e8a
9 changed files with 75 additions and 72 deletions

View File

@@ -1,9 +1,39 @@
import type { TFile } from 'obsidian' import type { TFile } from 'obsidian'
import type { IndexedDocument } from './globals' import type { IndexedDocument } from './globals'
import { database } from './database'
class CacheManager { class CacheManager {
private documentsCache: Map<string, IndexedDocument> = new Map() private documentsCache: Map<string, IndexedDocument> = new Map()
private writeInterval = 10_000 // In milliseconds /**
* Show an empty input field next time the user opens Omnisearch modal
*/
private nextQueryIsEmpty = true
public async addToSearchHistory(query: string): Promise<void> {
if (!query) {
this.nextQueryIsEmpty = true
return
}
this.nextQueryIsEmpty = false
let history = await database.searchHistory.toArray()
history = history.filter(s => s.query !== query).reverse()
history.unshift({ query })
history = history.slice(0, 10)
await database.searchHistory.clear()
await database.searchHistory.bulkAdd(history)
}
public async getSearchHistory(): Promise<ReadonlyArray<string>> {
const data = (await database.searchHistory.toArray())
.reverse()
.map(o => o.query)
console.log(this.nextQueryIsEmpty)
if (this.nextQueryIsEmpty) {
data.unshift('')
}
console.log(data)
return data
}
public async updateDocument(path: string, note: IndexedDocument) { public async updateDocument(path: string, note: IndexedDocument) {
this.documentsCache.set(path, note) this.documentsCache.set(path, note)

View File

@@ -2,6 +2,7 @@
import { debounce } from 'obsidian' import { debounce } from 'obsidian'
import { toggleInputComposition } from 'src/globals' import { toggleInputComposition } from 'src/globals'
import { createEventDispatcher, tick } from 'svelte' import { createEventDispatcher, tick } from 'svelte'
import { cacheManager } from "../cache-manager"
export let initialValue = '' export let initialValue = ''
export let placeholder = '' export let placeholder = ''
@@ -24,22 +25,25 @@
} }
const debouncedOnInput = debounce(() => { const debouncedOnInput = debounce(() => {
// If typing a query and not executing it,
// the next time we open the modal, the search field will be empty
cacheManager.addToSearchHistory('')
dispatch('input', value) dispatch('input', value)
}, 250) }, 200)
</script> </script>
<div class="omnisearch-input-container"> <div class="omnisearch-input-container">
<div class="omnisearch-input-field"> <div class="omnisearch-input-field">
<input <input
bind:value
bind:this="{elInput}" bind:this="{elInput}"
on:input="{debouncedOnInput}" bind:value
on:compositionstart="{_ => toggleInputComposition(true)}"
on:compositionend="{_ => toggleInputComposition(false)}"
type="text"
class="prompt-input" class="prompt-input"
on:compositionend="{_ => toggleInputComposition(false)}"
on:compositionstart="{_ => toggleInputComposition(true)}"
on:input="{debouncedOnInput}"
{placeholder} {placeholder}
spellcheck="false"/> spellcheck="false"
type="text"/>
</div> </div>
<slot></slot> <slot></slot>
</div> </div>

View File

@@ -1,4 +1,4 @@
<script lang="ts" context="module"> <script context="module" lang="ts">
let lastSearch = '' let lastSearch = ''
</script> </script>
@@ -19,7 +19,6 @@
import ResultItemInFile from './ResultItemInFile.svelte' import ResultItemInFile from './ResultItemInFile.svelte'
import { Query } from 'src/search/query' import { Query } from 'src/search/query'
import { openNote } from 'src/tools/notes' import { openNote } from 'src/tools/notes'
import { saveSearchHistory } from '../search/search-history'
export let modal: OmnisearchInFileModal export let modal: OmnisearchInFileModal
export let parent: OmnisearchVaultModal | null = null export let parent: OmnisearchVaultModal | null = null
@@ -54,7 +53,7 @@
lastSearch = searchQuery lastSearch = searchQuery
} }
selectedIndex = 0 selectedIndex = 0
scrollIntoView() await scrollIntoView()
})() })()
$: { $: {
@@ -112,7 +111,6 @@
evt?: MouseEvent | KeyboardEvent evt?: MouseEvent | KeyboardEvent
): Promise<void> { ): Promise<void> {
if (note) { if (note) {
await saveSearchHistory()
modal.close() modal.close()
if (parent) parent.close() if (parent) parent.close()
@@ -145,9 +143,9 @@
</script> </script>
<InputSearch <InputSearch
value="{searchQuery}"
on:input="{e => (searchQuery = e.detail)}" on:input="{e => (searchQuery = e.detail)}"
placeholder="Omnisearch - File" /> placeholder="Omnisearch - File"
value="{searchQuery}"/>
<ModalContainer> <ModalContainer>
{#if groupedOffsets.length && note} {#if groupedOffsets.length && note}

View File

@@ -1,6 +1,6 @@
<script lang="ts"> <script lang="ts">
import { MarkdownView, Notice, TFile } from 'obsidian' import { MarkdownView, Notice, TFile } from 'obsidian'
import { onMount, onDestroy, tick } from 'svelte' import { onDestroy, onMount, tick } from 'svelte'
import InputSearch from './InputSearch.svelte' import InputSearch from './InputSearch.svelte'
import ModalContainer from './ModalContainer.svelte' import ModalContainer from './ModalContainer.svelte'
import { eventBus, type ResultNote } from 'src/globals' import { eventBus, type ResultNote } from 'src/globals'
@@ -10,9 +10,9 @@
import { OmnisearchInFileModal, type OmnisearchVaultModal } from 'src/components/modals' import { OmnisearchInFileModal, type OmnisearchVaultModal } from 'src/components/modals'
import ResultItemVault from './ResultItemVault.svelte' import ResultItemVault from './ResultItemVault.svelte'
import { Query } from 'src/search/query' import { Query } from 'src/search/query'
import { saveSearchHistory, searchHistory } from 'src/search/search-history'
import { settings } from '../settings' import { settings } from '../settings'
import * as NotesIndex from '../notes-index' import * as NotesIndex from '../notes-index'
import { cacheManager } from "../cache-manager"
export let modal: OmnisearchVaultModal export let modal: OmnisearchVaultModal
let selectedIndex = 0 let selectedIndex = 0
@@ -21,8 +21,9 @@
let previousQuery: string let previousQuery: string
let resultNotes: ResultNote[] = [] let resultNotes: ResultNote[] = []
let query: Query let query: Query
$: selectedNote = resultNotes[selectedIndex]
$: selectedNote = resultNotes[selectedIndex]
$: searchQuery = previousQuery
$: if (searchQuery) { $: if (searchQuery) {
updateResults() updateResults()
} else { } else {
@@ -31,8 +32,7 @@
onMount(async () => { onMount(async () => {
await NotesIndex.refreshIndex() await NotesIndex.refreshIndex()
previousQuery = searchHistory[historySearchIndex] previousQuery = (await cacheManager.getSearchHistory())[historySearchIndex]
searchQuery = previousQuery
eventBus.enable('vault') eventBus.enable('vault')
eventBus.on('vault', 'enter', openNoteAndCloseModal) eventBus.on('vault', 'enter', openNoteAndCloseModal)
eventBus.on('vault', 'create-note', createNoteAndCloseModal) eventBus.on('vault', 'create-note', createNoteAndCloseModal)
@@ -49,18 +49,21 @@
eventBus.disable('vault') eventBus.disable('vault')
}) })
function prevSearchHistory() { async function prevSearchHistory() {
if (++historySearchIndex >= searchHistory.length) { // Filter out the empty string, if it's there
const history = (await cacheManager.getSearchHistory()).filter(s => s)
if (++historySearchIndex >= history.length) {
historySearchIndex = 0 historySearchIndex = 0
} }
searchQuery = searchHistory[historySearchIndex] previousQuery = history[historySearchIndex]
} }
function nextSearchHistory() { async function nextSearchHistory() {
const history = (await cacheManager.getSearchHistory()).filter(s => s)
if (--historySearchIndex < 0) { if (--historySearchIndex < 0) {
historySearchIndex = searchHistory.length ? searchHistory.length - 1 : 0 historySearchIndex = history.length ? history.length - 1 : 0
} }
searchQuery = searchHistory[historySearchIndex] previousQuery = history[historySearchIndex]
} }
async function updateResults() { async function updateResults() {
@@ -91,12 +94,11 @@
} }
function saveCurrentQuery() { function saveCurrentQuery() {
searchHistory.unshift(searchQuery) cacheManager.addToSearchHistory(searchQuery)
} }
function openSearchResult(note: ResultNote, newPane = false) { function openSearchResult(note: ResultNote, newPane = false) {
saveCurrentQuery() saveCurrentQuery()
saveSearchHistory()
openNote(note, newPane) openNote(note, newPane)
} }

View File

@@ -6,18 +6,13 @@ class OmnisearchCache extends Dexie {
{ path: string; hash: string; size: number; text: string }, { path: string; hash: string; size: number; text: string },
string string
> >
documents!: Dexie.Table< searchHistory!: Dexie.Table<{ id?: number; query: string }, number>
{ document: IndexedDocument; path: string; mtime: number },
string
>
minisearch!: Dexie.Table<string>
constructor() { constructor() {
super(app.appId + '_omnisearch') super(app.appId + '_omnisearch')
this.version(2).stores({ this.version(3).stores({
pdf: 'path, hash, size, text', pdf: 'path, hash, size, text',
documents: 'path, mtime, document', searchHistory: '++id, query',
minisearch: 'data',
}) })
} }
} }

View File

@@ -39,7 +39,7 @@ export async function getPDFFiles(): Promise<IndexedDocument[]> {
input.push( input.push(
NotesIndex.processQueue(async () => { NotesIndex.processQueue(async () => {
const doc = await fileToIndexedDocument(file) const doc = await fileToIndexedDocument(file)
cacheManager.updateDocument(file.path, doc) await cacheManager.updateDocument(file.path, doc)
data.push(doc) data.push(doc)
}) })
) )

View File

@@ -12,8 +12,6 @@ export const highlightClass = 'suggestion-highlight omnisearch-highlight'
export const eventBus = new EventBus() export const eventBus = new EventBus()
export const historyFilePath = `${app.vault.configDir}/plugins/omnisearch/historyCache.json`
export const EventNames = { export const EventNames = {
ToggleExcerpts: 'toggle-excerpts', ToggleExcerpts: 'toggle-excerpts',
} as const } as const

View File

@@ -8,7 +8,6 @@ import { loadSettings, settings, SettingsTab, showExcerpt } from './settings'
import { eventBus, EventNames } from './globals' import { eventBus, EventNames } from './globals'
import { registerAPI } from '@vanakat/plugin-api' import { registerAPI } from '@vanakat/plugin-api'
import api from './tools/api' import api from './tools/api'
import { loadSearchHistory } from './search/search-history'
import { isFilePlaintext } from './tools/utils' import { isFilePlaintext } from './tools/utils'
import * as NotesIndex from './notes-index' import * as NotesIndex from './notes-index'
import * as FileLoader from './file-loader' import * as FileLoader from './file-loader'
@@ -17,7 +16,6 @@ export default class OmnisearchPlugin extends Plugin {
async onload(): Promise<void> { async onload(): Promise<void> {
await cleanOldCacheFiles() await cleanOldCacheFiles()
await loadSettings(this) await loadSettings(this)
await loadSearchHistory()
// Initialize minisearch // Initialize minisearch
await Search.initSearchEngine() await Search.initSearchEngine()
@@ -125,6 +123,7 @@ async function cleanOldCacheFiles() {
`${app.vault.configDir}/plugins/omnisearch/notesCache.json`, `${app.vault.configDir}/plugins/omnisearch/notesCache.json`,
`${app.vault.configDir}/plugins/omnisearch/notesCache.data`, `${app.vault.configDir}/plugins/omnisearch/notesCache.data`,
`${app.vault.configDir}/plugins/omnisearch/searchIndex.data`, `${app.vault.configDir}/plugins/omnisearch/searchIndex.data`,
`${app.vault.configDir}/plugins/omnisearch/historyCache.json`,
`${app.vault.configDir}/plugins/omnisearch/pdfCache.data`, `${app.vault.configDir}/plugins/omnisearch/pdfCache.data`,
] ]
for (const item of toDelete) { for (const item of toDelete) {

View File

@@ -1,23 +0,0 @@
import { historyFilePath } from '../globals'
export let searchHistory: string[] = []
export async function loadSearchHistory(): Promise<void> {
if (await app.vault.adapter.exists(historyFilePath)) {
try {
searchHistory = JSON.parse(await app.vault.adapter.read(historyFilePath))
// Keep the last 100 searches
searchHistory = searchHistory.slice(0, 100)
} catch (e) {
console.trace('Could not load search history from the file')
console.error(e)
searchHistory = []
}
} else {
searchHistory = []
}
}
export async function saveSearchHistory(): Promise<void> {
await app.vault.adapter.write(historyFilePath, JSON.stringify(searchHistory))
}