Rewrite document store to keep all docs: filter and path selection without recreation. Much faster sorting and filtering.

cista-front/src/utils/index.ts

@@ -57,44 +57,40 @@ export function getFileExtension(filename: string) {
     return '' // No hay extensión
   }
 }
-export function getFileType(extension: string): string {
-  const videoExtensions = ['mp4', 'avi', 'mkv', 'mov']
-  const imageExtensions = ['jpg', 'jpeg', 'png', 'gif']
-  const pdfExtensions = ['pdf']
-
-  if (videoExtensions.includes(extension)) {
-    return 'video'
-  } else if (imageExtensions.includes(extension)) {
-    return 'image'
-  } else if (pdfExtensions.includes(extension)) {
-    return 'pdf'
-  } else {
-    return 'unknown'
-  }
+interface FileTypes {
+  [key: string]: string[]
 }
 
-const collator = new Intl.Collator('en', { sensitivity: 'base', numeric: true, usage: 'search' })
+const filetypes: FileTypes = {
+  video: ['avi', 'mkv', 'mov', 'mp4', 'webm'],
+  image: ['avif', 'gif', 'jpg', 'jpeg', 'png', 'webp', 'svg'],
+  pdf: ['pdf'],
+}
 
+export function getFileType(name: string): string {
+  const ext = name.split('.').pop()?.toLowerCase()
+  if (!ext || ext.length === name.length) return 'unknown'
+  return Object.keys(filetypes).find(type => filetypes[type].includes(ext)) || 'unknown'
+}
+
+// Prebuilt for fast & consistent sorting
+export const collator = new Intl.Collator('en', { sensitivity: 'base', numeric: true, usage: 'search' })
+
+// Preformat document names for faster search
 export function haystackFormat(str: string) {
   const based = str.normalize('NFKD').replace(/[\u0300-\u036f]/g, '').toLowerCase()
   return '^' + based + '$'
 }
 
-export function localeIncludes(haystack: string, based: string, words: string[]) {
+
+// Preformat search string for faster search
+export function needleFormat(query: string) {
+  const based = query.normalize('NFKD').replace(/[\u0300-\u036f]/g, '').toLowerCase()
+  return {based, words: based.split(/\W+/)}
+}
+
+// Test if haystack includes needle
+export function localeIncludes(haystack: string, filter: { based: string, words: string[] }) {
+  const {based, words} = filter
   return haystack.includes(based) || words && words.every(word => haystack.includes(word))
 }
-
-export function buildCorpus(data: any[]) {
-  return data.map(item => [haystackFormat(item.name), item])
-}
-
-export function search(corpus: [string, any][], search: string) {
-  const based = search.normalize('NFKD').replace(/[\u0300-\u036f]/g, '').toLowerCase()
-  const words = based.split(/\W+/)
-  const ret = []
-  for (const [haystack, item] of corpus) {
-    if (localeIncludes(haystack, based, words))
-      ret.push(item)
-  }
-  return ret
-}