From f6875acc729fe7cdd09da49bc03fefcd75fd7b6a Mon Sep 17 00:00:00 2001
From: Scott Idem <stidem@gmail.com>
Date: Fri, 23 Jan 2026 16:56:10 -0500
Subject: [PATCH] Enhance: Support flexible hash prefix length and organized
 caching

- Updated organized path logic to accept variable prefix length.
- Refactored IPC handlers to pass hash_prefix_length from frontend.
- Standardized parameter structures for check-cache and download-to-cache.
---
 src/main/file_handlers.ts | 23 ++++++++++++++---------
 1 file changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/main/file_handlers.ts b/src/main/file_handlers.ts
index 0fead89..7b6a732 100644
--- a/src/main/file_handlers.ts
+++ b/src/main/file_handlers.ts
@@ -7,23 +7,28 @@ import { expandPath } from './file_utils';
 let endpoints_in_progress: string[] = [];
 
 export function registerFileHandlers() {
-  function get_legacy_hashed_path(root: string, hash: string) {
-    return path.join(expandPath(root), `${hash}.file`);
+  // Flexible organization: [root]/[prefix_len-char-prefix]/[hash].file
+  function get_organized_hashed_path(root: string, hash: string, prefix_len: number = 2) {
+    const expanded_root = expandPath(root);
+    const prefix = hash.substring(0, Math.max(1, Math.min(prefix_len, 8)));
+    const dir = path.join(expanded_root, prefix);
+    if (!fs.existsSync(dir)) fs.mkdirSync(dir, { recursive: true });
+    return path.join(dir, `${hash}.file`);
   }
 
-  ipcMain.handle('native:check-cache', async (event, { cache_root, hash }) => {
-    const full_path = get_legacy_hashed_path(cache_root, hash);
+  ipcMain.handle('native:check-cache', async (event, { cache_root, hash, hash_prefix_length = 2 }) => {
+    const full_path = get_organized_hashed_path(cache_root, hash, hash_prefix_length);
     return fs.existsSync(full_path);
   });
 
-  ipcMain.handle('native:download-to-cache', async (event, { url, cache_root, hash, api_key, account_id }) => {
-    const full_path = get_legacy_hashed_path(cache_root, hash);
+  ipcMain.handle('native:download-to-cache', async (event, { url, cache_root, hash, api_key, account_id, hash_prefix_length = 2 }) => {
+    const full_path = get_organized_hashed_path(cache_root, hash, hash_prefix_length);
     const tmp_path = `${full_path}.tmp`;
 
     if (endpoints_in_progress.includes(url)) return { success: true, status: 'in_progress' };
     if (fs.existsSync(full_path)) return { success: true, path: full_path, status: 'exists' };
 
-    console.log(`Native: Downloading ${hash} -> ${full_path}`);
+    console.log(`Native: Organized Download (${hash_prefix_length} chars) -> ${full_path}`);
 
     try {
       endpoints_in_progress.push(url);
@@ -54,9 +59,9 @@ export function registerFileHandlers() {
     }
   });
 
-  ipcMain.handle('native:launch-from-cache', async (event, { cache_root, hash, temp_root, filename }) => {
+  ipcMain.handle('native:launch-from-cache', async (event, { cache_root, hash, temp_root, filename, hash_prefix_length = 2 }) => {
     try {
-      const source = get_legacy_hashed_path(cache_root, hash);
+      const source = get_organized_hashed_path(cache_root, hash, hash_prefix_length);
       const expanded_temp = expandPath(temp_root);
       const target = path.join(expanded_temp, filename);
       if (!fs.existsSync(expanded_temp)) fs.mkdirSync(expanded_temp, { recursive: true });