From 4bc3367e3fc05a16d8a4ae028efeae8552f663d8 Mon Sep 17 00:00:00 2001
From: DB p <onesound@oooz.net>
Date: Tue, 6 Sep 2022 23:59:33 +0900
Subject: [PATCH] - Add 'Free Space in drive' - when Type "d:\", showing graph
 with free space.

---
 Flow.Launcher.Plugin/Result.cs                |  2 +
 Flow.Launcher/ResultListBox.xaml              |  9 +--
 Flow.Launcher/Themes/Base.xaml                | 15 ++++
 .../Search/ResultManager.cs                   | 70 ++++++++++++++++---
 4 files changed, 82 insertions(+), 14 deletions(-)

diff --git a/Flow.Launcher.Plugin/Result.cs b/Flow.Launcher.Plugin/Result.cs
index 35c491d35..f3f3d2816 100644
--- a/Flow.Launcher.Plugin/Result.cs
+++ b/Flow.Launcher.Plugin/Result.cs
@@ -197,5 +197,7 @@ namespace Flow.Launcher.Plugin
         {
             return AsyncAction?.Invoke(context) ?? ValueTask.FromResult(Action?.Invoke(context) ?? false);
         }
+
+        public int? ProgressBar { get; set; }
     }
 }
diff --git a/Flow.Launcher/ResultListBox.xaml b/Flow.Launcher/ResultListBox.xaml
index 2a7f9c494..4c1b068f8 100644
--- a/Flow.Launcher/ResultListBox.xaml
+++ b/Flow.Launcher/ResultListBox.xaml
@@ -111,7 +111,10 @@
                                 <RowDefinition />
                                 <RowDefinition x:Name="SubTitleRowDefinition" Height="Auto" />
                             </Grid.RowDefinitions>
-
+                            <ProgressBar
+                                x:Name="progressbarResult"
+                                Style="{DynamicResource ProgressBarResult}"
+                                Value="{Binding Result.ProgressBar}" />
                             <TextBlock
                                 x:Name="Title"
                                 VerticalAlignment="Center"
@@ -132,9 +135,7 @@
                                 MinWidth="750"
                                 Style="{DynamicResource ItemSubTitleStyle}"
                                 Text="{Binding Result.SubTitle}"
-                                ToolTip="{Binding ShowSubTitleToolTip}">
-                    
-                            </TextBlock>
+                                ToolTip="{Binding ShowSubTitleToolTip}" />
 
                         </Grid>
 
diff --git a/Flow.Launcher/Themes/Base.xaml b/Flow.Launcher/Themes/Base.xaml
index 087606d04..dbd5b31c1 100644
--- a/Flow.Launcher/Themes/Base.xaml
+++ b/Flow.Launcher/Themes/Base.xaml
@@ -81,6 +81,21 @@
 
 
     <!--  Item Style  -->
+    <Style x:Key="ProgressBarResult" TargetType="{x:Type ProgressBar}">
+        <Setter Property="Height" Value="18" />
+        <Setter Property="Margin" Value="0,0,0,4" />
+        <Setter Property="VerticalAlignment" Value="Center" />
+        <Setter Property="Maximum" Value="100" />
+        <Setter Property="Minimum" Value="0" />
+        <Setter Property="Visibility" Value="Visible" />
+        <Setter Property="Foreground" Value="#26a0da " />
+        <Style.Triggers>
+            <DataTrigger Binding="{Binding Result.ProgressBar}" Value="{x:Null}">
+                <Setter Property="Visibility" Value="Collapsed" />
+            </DataTrigger>
+        </Style.Triggers>
+    </Style>
+
     <Style x:Key="BaseItemTitleStyle" TargetType="{x:Type TextBlock}">
         <Setter Property="Foreground" Value="#FFFFF8" />
         <Setter Property="FontSize" Value="16" />
diff --git a/Plugins/Flow.Launcher.Plugin.Explorer/Search/ResultManager.cs b/Plugins/Flow.Launcher.Plugin.Explorer/Search/ResultManager.cs
index 63310bebd..33771a599 100644
--- a/Plugins/Flow.Launcher.Plugin.Explorer/Search/ResultManager.cs
+++ b/Plugins/Flow.Launcher.Plugin.Explorer/Search/ResultManager.cs
@@ -77,6 +77,43 @@ namespace Flow.Launcher.Plugin.Explorer.Search
             };
         }
 
+        private static string toReadableSize(long pDrvSize, int pi)
+        {
+            int mok = 0;
+            double drvSize = pDrvSize;
+            string Space = "Byte";
+            string returnStr = "";
+
+            while (drvSize > 1024.0)
+            {
+                drvSize /= 1024.0;
+                mok++;
+            }
+
+            if (mok == 1)
+                Space =  "KB";
+            else if (mok == 2)
+                Space = " MB";
+            else if (mok == 3)
+                Space = " GB";
+            else if (mok == 4)
+                Space = " TB";
+
+            if (mok != 0)
+                if (pi == 1)
+                    returnStr = string.Format("{0:F1}{1}", drvSize, Space);
+                else if (pi == 2)
+                    returnStr = string.Format("{0:F2}{1}", drvSize, Space);
+                else if (pi == 3)
+                    returnStr = string.Format("{0:F3}{1}", drvSize, Space);
+                else
+                    returnStr = string.Format("{0}{1}", Convert.ToInt32(drvSize), Space);
+            else
+                returnStr = string.Format("{0}{1}", Convert.ToInt32(drvSize), Space);
+
+            return returnStr;
+        }
+
         internal static Result CreateOpenCurrentFolderResult(string path, bool windowsIndexed = false)
         {
             var retrievedDirectoryPath = FilesFolders.ReturnPreviousDirectoryIfIncompleteString(path);
@@ -86,19 +123,32 @@ namespace Flow.Launcher.Plugin.Explorer.Search
                 Path.DirectorySeparatorChar
             }, StringSplitOptions.None).Last();
 
-            if (retrievedDirectoryPath.EndsWith(":\\"))
-            {
-                var driveLetter = path.Substring(0, 1).ToUpper();
-                folderName = driveLetter + " drive";
-            }
-
             var title = "Open current directory";
 
             if (retrievedDirectoryPath != path)
                 title = "Open " + folderName;
 
-
             var subtitleFolderName = folderName;
+            var subtitle = $"Use > to search within {subtitleFolderName}, " +
+                           $"* to search for file extensions or >* to combine both searches.";
+
+
+
+
+
+            int? progressBar = null;
+            if (retrievedDirectoryPath.EndsWith(":\\"))
+            {
+                title = ""; // hide title when use progress bar,
+                var driveLetter = path.Substring(0, 1).ToUpper();
+                folderName = driveLetter + " drive";
+                var driveName = driveLetter + ":\\";
+                DriveInfo drv = new DriveInfo(driveLetter);
+                subtitle = toReadableSize(drv.AvailableFreeSpace, 2) + " free of " + toReadableSize(drv.TotalSize, 2);
+                double UsingSize = ((Convert.ToDouble(drv.TotalSize) - Convert.ToDouble(drv.AvailableFreeSpace)) / Convert.ToDouble(drv.TotalSize) * 100);
+                progressBar = Convert.ToInt32(UsingSize);
+            }
+
 
             // ie. max characters can be displayed without subtitle cutting off: "Program Files (x86)"
             if (folderName.Length > 19)
@@ -107,11 +157,11 @@ namespace Flow.Launcher.Plugin.Explorer.Search
             return new Result
             {
                 Title = title,
-                SubTitle = $"Use > to search within {subtitleFolderName}, " +
-                           $"* to search for file extensions or >* to combine both searches.",
+                SubTitle = subtitle,
                 AutoCompleteText = GetPathWithActionKeyword(retrievedDirectoryPath, ResultType.Folder),
                 IcoPath = retrievedDirectoryPath,
                 Score = 500,
+                ProgressBar = progressBar,
                 Action = c =>
                 {
                     Context.API.OpenDirectory(retrievedDirectoryPath);
@@ -208,4 +258,4 @@ namespace Flow.Launcher.Plugin.Explorer.Search
         Folder,
         File
     }
-}
\ No newline at end of file
+}