1 files changed, 58 insertions, 39 deletions

diff --git a/viewer/src/services/navigation.ts b/viewer/src/services/navigation.ts
index fa17990..fb01169 100644
--- a/viewer/src/services/navigation.ts
+++ b/viewer/src/services/navigation.ts
@@ -1,7 +1,7 @@
 /* ldgallery - A static generator which turns a collection of tagged
 --             pictures into a searchable web gallery.
 --
--- Copyright (C) 2019-2020  Guillaume FOUET
+-- Copyright (C) 2019-2022  Guillaume FOUET
 --
 -- This program is free software: you can redistribute it and/or modify
 -- it under the terms of the GNU Affero General Public License as
@@ -17,14 +17,43 @@
 -- along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
-export default class Navigation {
+import { DirectoryItem, DownloadableItem, Item } from '@/@types/gallery';
+import { ItemType } from '@/@types/itemType';
+import {
+  faFile,
+  faFileAlt,
+  faFileAudio,
+  faFilePdf,
+  faBook,
+  faFileVideo,
+  faFolder,
+  faHome,
+  faImage,
+  IconDefinition,
+} from '@fortawesome/free-solid-svg-icons';
+import { isDirectory } from './itemGuards';
 
+const ICON_BY_TYPE: Record<ItemType, IconDefinition> = {
+  directory: faFolder,
+  picture: faImage,
+  plaintext: faFileAlt,
+  markdown: faFileAlt,
+  pdf: faFilePdf,
+  epub: faBook,
+  video: faFileVideo,
+  audio: faFileAudio,
+  other: faFile,
+};
+
+// ---
+
+export const useNavigation = () => {
   // Searches for an item by path from a root item (navigation)
-  public static searchCurrentItemPath(root: Gallery.Item, path: string): Gallery.Item[] {
+  function searchCurrentItemPath(root: Item, path: string): Item[] {
     if (path === root.path) return [root];
-    if (root.properties.type === "directory" && path.startsWith(root.path)) {
+    if (isDirectory(root) && path.startsWith(root.path)) {
       const itemChain = root.properties.items
-        .map(item => this.searchCurrentItemPath(item, path))
+        .map(item => searchCurrentItemPath(item, path))
         .find(itemChain => itemChain.length > 0);
       if (itemChain) return [root, ...itemChain];
     }
@@ -32,49 +61,39 @@ export default class Navigation {
   }
 
   // Normalize a string to lowercase, no-accents
-  public static normalize(value: string) {
+  function normalize(value: string) {
     return value
-      .normalize("NFD")
-      .replace(/[\u0300-\u036f]/g, "")
+      .normalize('NFD')
+      .replace(/[\u0300-\u036f]/g, '')
       .toLowerCase();
   }
 
-  // Checks if the type of an item matches
-  public static checkType(item: Gallery.Item | null, type: Gallery.ItemType): boolean {
-    return item?.properties.type === type ?? false;
-  }
-
-  public static getLastDirectory(itemPath: Gallery.Item[]): Gallery.Directory {
+  function getLastDirectory(itemPath: Item[]): DirectoryItem {
     for (let idx = itemPath.length - 1; idx >= 0; idx--) {
       const item = itemPath[idx];
-      if (Navigation.checkType(item, "directory")) return item as Gallery.Directory;
+      if (isDirectory(item)) return item;
     }
-    throw new Error("No directory found");
+    throw new Error('No directory found');
   }
 
-  // Sort a list of items, moving the directories to the beginning of the list
-  public static directoriesFirst(items: Gallery.Item[]) {
-    return [
-      ...items
-        .filter(child => Navigation.checkType(child, "directory"))
-        .sort((a, b) => a.title.localeCompare(b.title)),
-
-      ...items
-        .filter(child => !Navigation.checkType(child, "directory")),
-    ];
+  // Get the icon for an item
+  function getIcon(item: Item): IconDefinition {
+    if (item.path.length <= 1) return faHome;
+    return ICON_BY_TYPE[item.properties.type];
   }
 
-  // Get the icon for an item
-  public static getIcon(item: Gallery.Item): string {
-    if (item.path.length <= 1) return "home";
-    switch (item.properties.type) {
-      case "picture":
-        return "image";
-      case "directory":
-        return "folder";
-      case "other":
-      default:
-        return "file";
-    }
+  // Get the file name of an item, without its cache timestamp
+  function getFileName(item: Item): string {
+    if (isDirectory(item)) return item.title;
+    const timeStamped = (item as DownloadableItem).properties.resource.split('/').pop() ?? '';
+    return timeStamped.split('?')[0];
   }
-}
+
+  return {
+    searchCurrentItemPath,
+    normalize,
+    getLastDirectory,
+    getIcon,
+    getFileName,
+  };
+};