Upload FastSeekWpf/App.xaml.cs

FastSeekWpf/App.xaml.cs

@@ -23,10 +23,6 @@ public partial class App : Application
     private volatile bool _shutdown;
     private MainWindow? _mainWindow;
     private Thread? _hotkeyThread;
-    private IntPtr _hotkeyWakeEvent;
-
-    // Console attachment
-    private const int ATTACH_PARENT_PROCESS = -1;
 
     [DllImport("kernel32.dll", SetLastError = true)]
     private static extern bool AttachConsole(int dwProcessId);
@@ -37,66 +33,40 @@ public partial class App : Application
     [DllImport("kernel32.dll", SetLastError = true)]
     private static extern bool FreeConsole();
 
-    [DllImport("kernel32.dll")]
-    private static extern IntPtr GetConsoleWindow();
+    private const int ATTACH_PARENT_PROCESS = -1;
 
     private void OnStartup(object sender, StartupEventArgs e)
     {
-        // --- CLI: Clear cache ---
-        if (e.Args.Contains("--clear-cache"))
+        if (e.Args.Contains("--cli") || e.Args.Contains("--diag"))
         {
-            EnsureConsole();
-            ClearCacheCli();
-            Console.WriteLine("\nPress Enter to exit...");
-            Console.ReadLine();
-            FreeConsole();
+            RunCliMode();
             Shutdown();
             return;
         }
 
-        // --- CLI DIAGNOSTIC MODE ---
-        if (e.Args.Contains("--diag") || e.Args.Contains("--cli"))
+        if (e.Args.Contains("--clear-cache"))
         {
-            EnsureConsole();
-            try
-            {
-                RunDiagnostics();
-            }
-            catch (Exception ex)
-            {
-                Console.ForegroundColor = ConsoleColor.Red;
-                Console.WriteLine($"\nFatal error: {ex}");
-                Console.ResetColor();
-            }
-            Console.WriteLine("\nPress Enter to exit...");
-            Console.ReadLine();
-            FreeConsole();
+            RunClearCache();
             Shutdown();
             return;
         }
 
         Logger.Log("=== App startup ===");
 
-        // Single-instance guard
         _mutex = new Mutex(true, MutexName, out bool createdNew);
         if (!createdNew)
         {
-            Logger.Log("Another instance is running. Sending toggle message.");
             IntPtr hwnd = Win32Api.FindWindowW(null, "FastSeek");
             if (hwnd != IntPtr.Zero)
                 Win32Api.PostMessageW(hwnd, WM_TOGGLE_WINDOW, IntPtr.Zero, IntPtr.Zero);
-            else
-                Logger.Log("Could not find existing window to toggle.");
             Shutdown();
             return;
         }
         _ownsMutex = true;
-        Logger.Log("Mutex acquired. Starting main window.");
 
         _mainWindow = new MainWindow();
         _mainWindow.Show();
 
-        // Start global hotkey thread (Ctrl+Shift+Space = MOD_CONTROL|MOD_SHIFT, VK_SPACE)
         _hotkeyThread = new Thread(HotkeyLoop)
         {
             IsBackground = true,
@@ -108,229 +78,240 @@ public partial class App : Application
 
     private static void EnsureConsole()
     {
-        // Try attaching to parent console first (for dotnet run in PowerShell)
         if (!AttachConsole(ATTACH_PARENT_PROCESS))
-        {
-            // Fallback: allocate a new console
             AllocConsole();
-        }
         Console.OutputEncoding = Encoding.UTF8;
-        // Redirect standard output to the attached console
         var stdout = Console.OpenStandardOutput();
         var writer = new StreamWriter(stdout, Encoding.UTF8) { AutoFlush = true };
         Console.SetOut(writer);
         Console.SetError(writer);
     }
 
-    private static void ClearCacheCli()
+    private static void RunClearCache()
     {
-        PrintBanner("Cache Cleanup");
+        EnsureConsole();
         var cachePath = Path.Combine(
             Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
             "FastSeek", "cache.dat");
-
-        if (!File.Exists(cachePath))
-        {
-            PrintInfo("No cache file found.");
-            PrintPath(cachePath);
-            return;
-        }
-
-        var size = new FileInfo(cachePath).Length;
-        try
+        if (File.Exists(cachePath))
         {
+            var size = new FileInfo(cachePath).Length;
             File.Delete(cachePath);
-            PrintSuccess($"Cache deleted ({FormatBytes(size)})");
+            Console.WriteLine($"Cache cleared ({size:N0} bytes)");
         }
-        catch (Exception ex)
+        else
         {
-            PrintError($"Failed to delete cache: {ex.Message}");
+            Console.WriteLine("No cache found.");
         }
+        Console.WriteLine("Press Enter...");
+        Console.ReadLine();
+        FreeConsole();
     }
 
-    private static void RunDiagnostics()
+    private static void RunCliMode()
     {
-        PrintBanner("FastSeek Diagnostic Mode");
+        EnsureConsole();
+        Console.Clear();
 
-        var sw = Stopwatch.StartNew();
+        // Compact header
+        PrintHeader("FastSeek CLI");
+        Console.WriteLine();
 
-        // 1. Drive discovery
-        PrintSection("1. Drive Discovery");
+        // 1. Drives
         var drives = DriveDiscovery.GetNtfsDrives();
-        PrintMetric("NTFS drives", drives.Count, "found", sw.ElapsedMilliseconds);
-        foreach (var d in drives)
-            PrintDetail($"→ {d.Letter}:  {d.Root}  ({d.DevicePath})");
+        PrintLine(ConsoleColor.Cyan, "Drives", $"{drives.Count} NTFS found");
+        foreach (var d in drives) PrintDetail($"  {d.Letter}: {d.Root}");
         Console.WriteLine();
 
         if (drives.Count == 0)
         {
-            PrintError("No NTFS drives found. Run as Administrator.");
+            PrintError("No NTFS drives. Run as Administrator.");
+            WaitExit();
             return;
         }
 
-        // 2. MFT Scan
-        PrintSection("2. MFT Scan");
-        sw.Restart();
+        // 2. Scan with per-drive debug
+        PrintLine(ConsoleColor.Cyan, "Scan", "Reading MFT...");
         var index = new IndexStore();
         long totalRecords = 0;
+        var sw = Stopwatch.StartNew();
+
         foreach (var drive in drives)
         {
             try
             {
                 using var reader = new MftReader(drive);
-                var scan = reader.ScanDirect();
-                if (scan == null) scan = reader.Scan();
+                ScanResult? scan = null;
+                string method = "direct";
+
+                try { scan = reader.ScanDirect(); }
+                catch (Exception ex1)
+                {
+                    PrintDetail($"  {drive.Letter}: direct scan failed: {ex1.Message}");
+                    method = "fallback";
+                }
+
+                if (scan == null)
+                {
+                    try { scan = reader.Scan(); method = "fallback"; }
+                    catch (Exception ex2)
+                    {
+                        PrintError($"  {drive.Letter}: fallback scan failed: {ex2.Message}");
+                        continue;
+                    }
+                }
+
                 index.PopulateFromScan(scan, drive.Root);
                 totalRecords += scan.Records.Count;
-                PrintDetail($"→ {drive.Letter}: {scan.Records.Count:N0} records");
+                PrintLine(ConsoleColor.Green, $"  {drive.Letter}", $"{scan.Records.Count:N0} records ({method})");
             }
             catch (Exception ex)
             {
-                PrintDetail($"→ {drive.Letter}: ERROR - {ex.Message}");
+                PrintError($"  {drive.Letter}: {ex.Message}");
             }
         }
+
         index.CompleteIndex();
         sw.Stop();
 
-        double recPerMs = totalRecords / Math.Max(sw.ElapsedMilliseconds, 1);
-        PrintMetric("Total records", totalRecords, "records", sw.ElapsedMilliseconds);
-        PrintMetric("Scan speed", recPerMs, "rec/ms", null);
-        PrintMetric("Final index", index.Count, "entries", null);
+        PrintLine(ConsoleColor.White, "Index", $"{index.Count:N0} entries  |  {sw.ElapsedMilliseconds}ms  |  {totalRecords / Math.Max(sw.ElapsedMilliseconds, 1):N0} rec/ms");
         Console.WriteLine();
 
-        // 3. Cache save
-        PrintSection("3. Cache Save");
-        sw.Restart();
+        // 3. Cache
+        var cacheSw = Stopwatch.StartNew();
         CacheManager.SaveCache(index);
-        sw.Stop();
-        var cachePath = Path.Combine(
-            Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
-            "FastSeek", "cache.dat");
+        cacheSw.Stop();
+        var cachePath = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "FastSeek", "cache.dat");
         long cacheSize = File.Exists(cachePath) ? new FileInfo(cachePath).Length : 0;
-        PrintMetric("Save time", sw.ElapsedMilliseconds, "ms", null);
-        PrintMetric("Cache size", cacheSize, "bytes", null);
-        PrintDetail($"→ {FormatBytes(cacheSize)} ({cacheSize:N0} bytes)");
+        PrintLine(ConsoleColor.Cyan, "Cache", $"saved {FormatBytes(cacheSize)} in {cacheSw.ElapsedMilliseconds}ms");
         Console.WriteLine();
 
-        // 4. Cache load
-        PrintSection("4. Cache Load");
-        sw.Restart();
-        var loadedCache = CacheManager.LoadCache();
-        IndexStore? loadedIndex = null;
-        if (loadedCache != null)
-            loadedIndex = IndexStore.FromCache(loadedCache);
-        sw.Stop();
-        PrintMetric("Load time", sw.ElapsedMilliseconds, "ms", null);
-        PrintMetric("Entries", loadedIndex?.Count ?? 0, "entries", null);
-        Console.WriteLine();
-
-        // 5. Search benchmark
-        PrintSection("5. Search Benchmark");
-        Console.WriteLine("   Running 1,000 random queries...");
+        // 4. Benchmark
+        PrintLine(ConsoleColor.Cyan, "Benchmark", "1,000 random queries...");
+        var benchSw = Stopwatch.StartNew();
         var rnd = new Random(42);
-        var names = loadedIndex?.NameCache ?? index.NameCache;
-        var queries = Enumerable.Range(0, 1000)
-            .Select(_ =>
-            {
-                if (names.Count == 0) return "test";
-                var name = names[rnd.Next(names.Count)];
-                if (name.Length < 2) return name;
-                int start = rnd.Next(name.Length - 1);
-                int len = rnd.Next(1, Math.Min(name.Length - start, 6));
-                return name.Substring(start, len).ToLowerInvariant();
-            }).ToList();
-
+        var names = index.NameCache;
         var exclusions = CacheManager.LoadExclusions();
-        sw.Restart();
-        long resultCount = 0;
-        foreach (var q in queries)
+        long benchResults = 0;
+
+        for (int i = 0; i < 1000; i++)
         {
-            var r = SearchEngine.Search(loadedIndex ?? index, q, 50, false, exclusions);
-            resultCount += r.Count;
+            if (names.Count == 0) break;
+            var name = names[rnd.Next(names.Count)];
+            if (name.Length < 2) continue;
+            int start = rnd.Next(name.Length - 1);
+            int len = rnd.Next(1, Math.Min(name.Length - start, 6));
+            var q = name.Substring(start, len).ToLowerInvariant();
+            var r = SearchEngine.Search(index, q, 50, false, exclusions);
+            benchResults += r.Count;
         }
-        sw.Stop();
+        benchSw.Stop();
 
-        double avgMs = sw.ElapsedMilliseconds / 1000.0;
-        double avgResults = resultCount / 1000.0;
-        PrintMetric("Total time", sw.ElapsedMilliseconds, "ms", null);
-        PrintMetric("Avg/query", avgMs, "ms", null);
-        PrintMetric("Avg results", avgResults, "results/query", null);
-        PrintMetric("Throughput", 1000.0 / Math.Max(sw.ElapsedMilliseconds, 1) * 1000, "queries/sec", null);
+        double qps = 1000.0 / Math.Max(benchSw.ElapsedMilliseconds, 1) * 1000;
+        PrintLine(ConsoleColor.White, "Result", $"{benchSw.ElapsedMilliseconds}ms total  |  {qps:N0} qps  |  {benchResults / 1000.0:F1} avg results");
         Console.WriteLine();
-
-        // Memory
-        PrintSection("6. Memory Usage");
-        GC.Collect();
-        GC.WaitForPendingFinalizers();
-        GC.Collect();
-        long memBytes = GC.GetTotalMemory(false);
-        PrintMetric("Working set", memBytes, "bytes", null);
-        PrintDetail($"→ {FormatBytes(memBytes)}");
+        PrintDivider();
         Console.WriteLine();
 
-        // Log
-        PrintSection("7. Log File");
-        PrintPath(Path.Combine(
-            Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
-            "FastSeek", "log.txt"));
+        // 5. Interactive search
+        if (index.Count == 0)
+        {
+            PrintError("INDEX IS EMPTY — GUI will show no results.");
+            PrintInfo("Possible causes:");
+            PrintDetail("  • Not running as Administrator");
+            PrintDetail("  • MFT direct read blocked (try chkdsk)");
+            PrintDetail("  • FSCTL_ENUM_USN_DATA not supported on this volume");
+            Console.WriteLine();
+        }
+
+        PrintHeader("Interactive Search (type 'exit' to quit)");
         Console.WriteLine();
 
-        PrintSuccess("Diagnostic complete.");
+        while (true)
+        {
+            Console.ForegroundColor = ConsoleColor.DarkGray;
+            Console.Write("> ");
+            Console.ResetColor();
+            var query = Console.ReadLine()?.Trim();
+            if (string.IsNullOrEmpty(query)) continue;
+            if (query.Equals("exit", StringComparison.OrdinalIgnoreCase)) break;
+
+            var searchSw = Stopwatch.StartNew();
+            var results = SearchEngine.Search(index, query, 20, false, exclusions);
+            searchSw.Stop();
+
+            Console.ForegroundColor = ConsoleColor.DarkGray;
+            Console.WriteLine($"  {results.Count} results  |  {searchSw.Elapsed.TotalMilliseconds:F3}ms");
+            Console.ResetColor();
+
+            foreach (var r in results.Take(10))
+            {
+                var color = r.Kind switch
+                {
+                    ResultKind.App => ConsoleColor.Green,
+                    ResultKind.Document => ConsoleColor.Yellow,
+                    ResultKind.Image => ConsoleColor.Magenta,
+                    ResultKind.Video => ConsoleColor.Red,
+                    ResultKind.Audio => ConsoleColor.Blue,
+                    ResultKind.Archive => ConsoleColor.DarkYellow,
+                    ResultKind.Folder => ConsoleColor.Cyan,
+                    _ => ConsoleColor.Gray
+                };
+                Console.ForegroundColor = color;
+                Console.Write($"  [{GetBadgeLabel(r.Kind),-4}] ");
+                Console.ResetColor();
+                Console.WriteLine($"{r.Name}  →  {r.FullPath}");
+            }
+            if (results.Count > 10)
+            {
+                Console.ForegroundColor = ConsoleColor.DarkGray;
+                Console.WriteLine($"  ... and {results.Count - 10} more");
+                Console.ResetColor();
+            }
+            Console.WriteLine();
+        }
+
+        WaitExit();
     }
 
-    // --- Pretty-print helpers ---
-    private static void PrintBanner(string title)
+    private static void WaitExit()
     {
-        int width = 50;
-        string line = new('═', width);
-        int pad = (width - title.Length) / 2;
         Console.WriteLine();
-        Console.ForegroundColor = ConsoleColor.Cyan;
-        Console.WriteLine($"╔{line}╗");
-        Console.WriteLine($"║{new string(' ', pad)}{title}{new string(' ', width - pad - title.Length)}║");
-        Console.WriteLine($"╚{line}╝");
+        Console.ForegroundColor = ConsoleColor.DarkGray;
+        Console.WriteLine("Press Enter to exit...");
         Console.ResetColor();
-        Console.WriteLine();
+        Console.ReadLine();
+        FreeConsole();
     }
 
-    private static void PrintSection(string title)
+    // --- Formatting helpers ---
+    private static void PrintHeader(string text)
     {
-        Console.ForegroundColor = ConsoleColor.Yellow;
-        Console.WriteLine($"▶ {title}");
+        Console.ForegroundColor = ConsoleColor.Cyan;
+        Console.WriteLine($"━━━ {text} ━━━");
         Console.ResetColor();
     }
 
-    private static void PrintMetric(string label, double value, string unit, long? ms)
+    private static void PrintDivider()
     {
-        Console.Write("   ");
         Console.ForegroundColor = ConsoleColor.DarkGray;
-        Console.Write($"{label,-16}");
+        Console.WriteLine(new string('─', 60));
         Console.ResetColor();
-        Console.Write(": ");
-        Console.ForegroundColor = ConsoleColor.White;
-        Console.Write(value.ToString("N0").PadLeft(12));
-        Console.ResetColor();
-        Console.Write($" {unit}");
-        if (ms.HasValue)
-        {
-            Console.ForegroundColor = ConsoleColor.DarkGray;
-            Console.Write($"  ({ms.Value} ms)");
-            Console.ResetColor();
-        }
-        Console.WriteLine();
     }
 
-    private static void PrintDetail(string text)
+    private static void PrintLine(ConsoleColor color, string label, string value)
     {
         Console.ForegroundColor = ConsoleColor.DarkGray;
-        Console.WriteLine($"   {text}");
+        Console.Write($"{label,-10}: ");
+        Console.ForegroundColor = color;
+        Console.WriteLine(value);
         Console.ResetColor();
     }
 
-    private static void PrintSuccess(string text)
+    private static void PrintDetail(string text)
     {
-        Console.ForegroundColor = ConsoleColor.Green;
-        Console.WriteLine($"✓ {text}");
+        Console.ForegroundColor = ConsoleColor.DarkGray;
+        Console.WriteLine(text);
         Console.ResetColor();
     }
 
@@ -348,13 +329,6 @@ public partial class App : Application
         Console.ResetColor();
     }
 
-    private static void PrintPath(string path)
-    {
-        Console.ForegroundColor = ConsoleColor.DarkGray;
-        Console.WriteLine($"   {path}");
-        Console.ResetColor();
-    }
-
     private static string FormatBytes(long bytes)
     {
         if (bytes < 1024) return $"{bytes} B";
@@ -363,37 +337,39 @@ public partial class App : Application
         return $"{bytes / (1024.0 * 1024 * 1024):F2} GB";
     }
 
+    private static string GetBadgeLabel(ResultKind kind) => kind switch
+    {
+        ResultKind.App => "APP",
+        ResultKind.Document => "DOC",
+        ResultKind.Image => "IMG",
+        ResultKind.Video => "VID",
+        ResultKind.Audio => "AUD",
+        ResultKind.Archive => "ZIP",
+        ResultKind.Folder => "DIR",
+        _ => "FILE"
+    };
+
     private void OnExit(object sender, ExitEventArgs e)
     {
         _shutdown = true;
-        if (_hotkeyWakeEvent != IntPtr.Zero)
-            SetEvent(_hotkeyWakeEvent);
-        _hotkeyThread?.Join(1000);
+        _hotkeyThread?.Join(500);
         if (_ownsMutex)
             _mutex?.ReleaseMutex();
         _mutex?.Dispose();
-        if (_hotkeyWakeEvent != IntPtr.Zero)
-            CloseHandle(_hotkeyWakeEvent);
         Logger.Log("=== App exit ===");
     }
 
     private void HotkeyLoop(object? param)
     {
         IntPtr targetHwnd = (IntPtr)(param ?? IntPtr.Zero);
-        if (targetHwnd == IntPtr.Zero)
-        {
-            Logger.Log("HotkeyLoop: targetHwnd is zero, exiting.");
-            return;
-        }
+        if (targetHwnd == IntPtr.Zero) return;
 
         uint modifiers = 0x0002 | 0x0004;
         if (!Win32Api.RegisterHotKey(IntPtr.Zero, 1, modifiers, 0x20))
         {
-            int err = Marshal.GetLastWin32Error();
-            Logger.Log($"RegisterHotKey failed: error {err}. Hotkey will not work.");
+            Logger.Log($"RegisterHotKey failed: {Marshal.GetLastWin32Error()}");
             return;
         }
-        Logger.Log("Hotkey registered: Ctrl+Shift+Space");
 
         try
         {
@@ -403,10 +379,7 @@ public partial class App : Application
                 if (result == 0 || result == -1) break;
 
                 if (msg.message == Win32Api.WM_HOTKEY)
-                {
-                    Logger.Log("Hotkey pressed, toggling window.");
                     Win32Api.PostMessageW(targetHwnd, WM_TOGGLE_WINDOW, IntPtr.Zero, IntPtr.Zero);
-                }
                 else
                 {
                     Win32Api.TranslateMessage(ref msg);
@@ -417,16 +390,6 @@ public partial class App : Application
         finally
         {
             Win32Api.UnregisterHotKey(IntPtr.Zero, 1);
-            Logger.Log("Hotkey unregistered.");
         }
     }
-
-    [DllImport("kernel32.dll", SetLastError = true, CharSet = CharSet.Unicode)]
-    private static extern IntPtr CreateEventW(IntPtr lpEventAttributes, bool bManualReset, bool bInitialState, string? lpName);
-
-    [DllImport("kernel32.dll", SetLastError = true)]
-    private static extern bool SetEvent(IntPtr hEvent);
-
-    [DllImport("kernel32.dll", SetLastError = true)]
-    private static extern bool CloseHandle(IntPtr hObject);
 }