Update ./routes/utils.ts functions to use object for O(1) lookups

svelte/src/routes/utils.ts

@@ -18,17 +18,11 @@ function getFileUrl(comfyUrl: string, folderType: string, file: any) {
   }
 }
 
-function findFile(filename: string, exts: Array<string>, files: Array<any>) {
-  let fn: any = filename.split('.');
-  fn.pop();
-  fn = fn.join('.');
-  return files.find((f: any) => {
-    const fa = f.name.split('.');
-    const extname = fa.pop().toLowerCase();
-    const name = fa.join('.');
-
-    return name === fn && exts.includes(extname);
-  });
+function findFile(filename: string, exts: Array<string>, filesMap: Array<any>) {
+  if (filesMap.allFiles.hasOwnProperty(filename)) {
+      return filesMap.allFiles[filename];
+  }
+  return false;
 }
 
 function processFile(
@@ -41,11 +35,12 @@ function processFile(
   if (WHITE_EXTS.includes(extname)) {
     file['fileType'] = extname;
     if (extname === 'json') {
-      if (findFile(file.name, IMAGE_EXTS.concat(VIDEO_EXTS), files)) {
+      if (findFile(file.fullPath, IMAGE_EXTS.concat(VIDEO_EXTS), files)) {
         return;
       }
     }
   }
+
   if (IMAGE_EXTS.includes(extname)) {
     file['fileType'] = 'image';
   }
@@ -56,11 +51,15 @@ function processFile(
     return;
   }
 
-  file['url'] = getFileUrl(comfyUrl, folderType, file);
-  if (['image', 'video'].includes(file['fileType'])) {
-    file['previewUrl'] = getFileUrl(comfyUrl, folderType, file);
 
-    let jsonFile = findFile(file.name, JSON_EXTS, files);
+  var url = getFileUrl(comfyUrl, folderType, file);
+  file['url'] = url;
+  if (['image', 'video'].includes(file['fileType'])) {
+    file['previewUrl'] = url;
+    var path = file.fullPath;
+    var jsonFileName = path.substring(0, path.lastIndexOf('.')) || path;
+    var jsonFileName = jsonFileName + ".json";
+    let jsonFile = findFile(jsonFileName, JSON_EXTS, files);
     if (jsonFile) {
       file['url'] = getFileUrl(comfyUrl, folderType, jsonFile);
     }
@@ -98,15 +97,30 @@ export async function fetchFiles(
 
   const res = await fetch(url);
   const ret = await res.json();
-
   let files = ret.files;
   let newFiles: Array<any> = [];
+  var map = {
+    "directories": {},
+    "allFiles": {}
+  };
+
+  var filesMap = files.reduce(function(map, obj) {
+    var dir = (obj.type == 'dir') ? "" : obj.folder_path;
+    var filePath = "/" + dir + obj.name;
+    obj.fullPath = filePath;
+    map.allFiles[filePath] = obj;
+    if (obj.type == 'dir') {
+      map.directories[filePath] = obj;
+    }
+    return map;
+  }, map);
+
   files.forEach((f: any) => {
     let newFile;
     if (f.type === 'dir') {
       newFile = processDir(f);
     } else {
-      newFile = processFile(f, folderType, comfyUrl, files);
+      newFile = processFile(f, folderType, comfyUrl, filesMap);
     }
 
     if (newFile) {