From a9661da14f88fc18750e6363df19631dcc5247a3 Mon Sep 17 00:00:00 2001
From: Katy Coe <djkaty@users.noreply.github.com>
Date: Mon, 27 Jul 2020 22:14:55 +0200
Subject: [PATCH] GUI: Enable drag & drop for all file combinations

---
 Il2CppInspector.GUI/MainWindow.xaml    |  25 +++-
 Il2CppInspector.GUI/MainWindow.xaml.cs | 151 +++++++++++++++++--------
 2 files changed, 123 insertions(+), 53 deletions(-)

diff --git a/Il2CppInspector.GUI/MainWindow.xaml b/Il2CppInspector.GUI/MainWindow.xaml
index a794fc9..4de44f1 100644
--- a/Il2CppInspector.GUI/MainWindow.xaml
+++ b/Il2CppInspector.GUI/MainWindow.xaml
@@ -8,7 +8,8 @@
         mc:Ignorable="d"
         Title="Il2CppInspector" Height="850" Width="1080" Background="White"
         WindowStartupLocation="CenterScreen"
-        ContentRendered="MainWindow_OnContentRendered">
+        ContentRendered="MainWindow_OnContentRendered"
+        Drop="MainWindow_OnDrop">
     <Window.Resources>
         <!-- Our favourite colours -->
         <SolidColorBrush x:Key="MicrosoftBlue" Color="#00A2ED"/>
@@ -92,6 +93,20 @@
         </HierarchicalDataTemplate>
     </Window.Resources>
 
+    <!-- Enable/disable drag & drop dynamically -->
+    <Window.Style>
+        <Style TargetType="{x:Type Window}">
+            <Style.Triggers>
+                <DataTrigger Binding="{Binding ElementName=grdFirstPage, Path=Visibility}" Value="Visible">
+                    <Setter Property="AllowDrop" Value="True"/>
+                </DataTrigger>
+                <DataTrigger Binding="{Binding ElementName=btnSelectBinaryFile, Path=Visibility}" Value="Visible">
+                    <Setter Property="AllowDrop" Value="True"/>
+                </DataTrigger>
+            </Style.Triggers>
+        </Style>
+    </Window.Style>
+
     <!-- Window content -->
     <Grid>
         <!-- Main content -->
@@ -407,20 +422,22 @@
                 <TextBlock TextAlignment="Center">
                     <TextBlock FontSize="22">Option 1</TextBlock>
                     <LineBreak/>
-                    <TextBlock>Select an IL2CPP metadata file</TextBlock>
+                    <TextBlock>Select or drag &amp; drop an IL2CPP metadata file</TextBlock>
+                    <LineBreak/>
+                    <TextBlock FontSize="16">You can drag &amp; drop both the metadata and binary together</TextBlock>
                 </TextBlock>
             </Button>
             <Button Grid.Row="1" Name="btnSelectPackageFile" Style="{StaticResource LightBoxButton}" Margin="100,50,100,100" Click="BtnSelectPackageFile_OnClick">
                 <TextBlock TextAlignment="Center">
                     <TextBlock FontSize="22">Option 2</TextBlock>
                     <LineBreak/>
-                    <TextBlock>Select an APK or IPA file</TextBlock>
+                    <TextBlock>Select or drag &amp; drop an APK or IPA file</TextBlock>
                     <LineBreak/>
                     <TextBlock FontSize="16">Encrypted IPA files are not supported</TextBlock>
                 </TextBlock>
             </Button>
         </Grid>
-        <Button Name="btnSelectBinaryFile" Style="{StaticResource LightBoxButton}" Margin="100" Click="BtnSelectBinaryFile_OnClick" Visibility="Hidden">Select an IL2CPP binary file</Button>
+        <Button Name="btnSelectBinaryFile" Style="{StaticResource LightBoxButton}" Margin="100" Click="BtnSelectBinaryFile_OnClick" Visibility="Hidden">Select or drag &amp; drop an IL2CPP binary file</Button>
 
         <!-- Back button -->
         <Button Name="btnBack" Margin="7" HorizontalAlignment="Left" VerticalAlignment="Bottom" FontSize="18" Width="120" Click="BtnBack_OnClick" Content="&lt;&lt; Back">
diff --git a/Il2CppInspector.GUI/MainWindow.xaml.cs b/Il2CppInspector.GUI/MainWindow.xaml.cs
index b6eb5f9..1bfcc1e 100644
--- a/Il2CppInspector.GUI/MainWindow.xaml.cs
+++ b/Il2CppInspector.GUI/MainWindow.xaml.cs
@@ -67,28 +67,32 @@ namespace Il2CppInspectorGUI
         /// Select global metadata file
         /// </summary>
         private async void BtnSelectMetadataFile_OnClick(object sender, RoutedEventArgs e) {
-            var app = (App) Application.Current;
-
             var openFileDialog = new OpenFileDialog {
                 Filter = "IL2CPP global metadata file|global-metadata.dat|All files (*.*)|*.*",
                 CheckFileExists = true
             };
 
             if (openFileDialog.ShowDialog() == true) {
-                areaBusyIndicator.Visibility = Visibility.Visible;
-                grdFirstPage.Visibility = Visibility.Hidden;
+                await LoadMetadataAsync(openFileDialog.FileName);
+            }
+        }
 
-                // Load the metadata file
-                if (await app.LoadMetadataAsync(openFileDialog.FileName)) {
-                    // Metadata loaded successfully
-                    btnSelectBinaryFile.Visibility = Visibility.Visible;
-                    areaBusyIndicator.Visibility = Visibility.Hidden;
-                }
-                else {
-                    areaBusyIndicator.Visibility = Visibility.Hidden;
-                    grdFirstPage.Visibility = Visibility.Visible;
-                    MessageBox.Show(this, app.LastException.Message, "Error", MessageBoxButton.OK, MessageBoxImage.Error);
-                }
+        // Load the metadata file
+        private async Task LoadMetadataAsync(string filename) {
+            var app = (App) Application.Current;
+
+            areaBusyIndicator.Visibility = Visibility.Visible;
+            grdFirstPage.Visibility = Visibility.Hidden;
+
+            if (await app.LoadMetadataAsync(filename)) {
+                // Metadata loaded successfully
+                btnSelectBinaryFile.Visibility = Visibility.Visible;
+                areaBusyIndicator.Visibility = Visibility.Hidden;
+            }
+            else {
+                areaBusyIndicator.Visibility = Visibility.Hidden;
+                grdFirstPage.Visibility = Visibility.Visible;
+                MessageBox.Show(this, app.LastException.Message, "Error", MessageBoxButton.OK, MessageBoxImage.Error);
             }
         }
 
@@ -96,31 +100,36 @@ namespace Il2CppInspectorGUI
         /// Select binary file
         /// </summary>
         private async void BtnSelectBinaryFile_OnClick(object sender, RoutedEventArgs e) {
-            var app = (App) Application.Current;
-
             var openFileDialog = new OpenFileDialog {
                 Filter = "Binary executable file (*.exe;*.dll;*.so;*.bin;*.prx;*.sprx)|*.exe;*.dll;*.so;*.bin;*.prx;*.sprx|All files (*.*)|*.*",
                 CheckFileExists = true
             };
 
             if (openFileDialog.ShowDialog() == true) {
-                txtBusyStatus.Text = "Processing binary...";
-                areaBusyIndicator.Visibility = Visibility.Visible;
-                btnSelectBinaryFile.Visibility = Visibility.Hidden;
+                await LoadBinaryAsync(openFileDialog.FileName);
+            }
+        }
 
-                // Load the binary file
-                if (await app.LoadBinaryAsync(openFileDialog.FileName)) {
-                    // Binary loaded successfully
-                    areaBusyIndicator.Visibility = Visibility.Hidden;
+        // Load the binary file
+        private async Task LoadBinaryAsync(string filename) {
+            var app = (App) Application.Current;
 
-                    lstImages.ItemsSource = app.AppModels;
-                    lstImages.SelectedIndex = 0;
-                }
-                else {
-                    areaBusyIndicator.Visibility = Visibility.Hidden;
-                    btnSelectBinaryFile.Visibility = Visibility.Visible;
-                    MessageBox.Show(this, app.LastException.Message, "Error", MessageBoxButton.OK, MessageBoxImage.Error);
-                }
+            txtBusyStatus.Text = "Processing binary...";
+            areaBusyIndicator.Visibility = Visibility.Visible;
+            btnSelectBinaryFile.Visibility = Visibility.Hidden;
+
+            // Load the binary file
+            if (await app.LoadBinaryAsync(filename)) {
+                // Binary loaded successfully
+                areaBusyIndicator.Visibility = Visibility.Hidden;
+
+                lstImages.ItemsSource = app.AppModels;
+                lstImages.SelectedIndex = 0;
+            }
+            else {
+                areaBusyIndicator.Visibility = Visibility.Hidden;
+                btnSelectBinaryFile.Visibility = Visibility.Visible;
+                MessageBox.Show(this, app.LastException.Message, "Error", MessageBoxButton.OK, MessageBoxImage.Error);
             }
         }
 
@@ -128,31 +137,36 @@ namespace Il2CppInspectorGUI
         /// Select APK or IPA package file
         /// </summary>
         private async void BtnSelectPackageFile_OnClick(object sender, RoutedEventArgs e) {
-            var app = (App) Application.Current;
-
             var openFileDialog = new OpenFileDialog {
                 Filter = "Android/iOS Application Package (*.apk;*.ipa;*.zip)|*.apk;*.ipa;*.zip|All files (*.*)|*.*",
                 CheckFileExists = true
             };
 
             if (openFileDialog.ShowDialog() == true) {
-                txtBusyStatus.Text = "Extracting package...";
-                areaBusyIndicator.Visibility = Visibility.Visible;
-                grdFirstPage.Visibility = Visibility.Hidden;
+                await LoadPackageAsync(openFileDialog.FileName);
+            }
+        }
 
-                // Load the package
-                if (await app.LoadPackageAsync(openFileDialog.FileName)) {
-                    // Package loaded successfully
-                    areaBusyIndicator.Visibility = Visibility.Hidden;
+        // Load the package file
+        private async Task LoadPackageAsync(string filename) {
+            var app = (App) Application.Current;
 
-                    lstImages.ItemsSource = app.AppModels;
-                    lstImages.SelectedIndex = 0;
-                }
-                else {
-                    areaBusyIndicator.Visibility = Visibility.Hidden;
-                    grdFirstPage.Visibility = Visibility.Visible;
-                    MessageBox.Show(this, app.LastException.Message, "Error", MessageBoxButton.OK, MessageBoxImage.Error);
-                }
+            txtBusyStatus.Text = "Extracting package...";
+            areaBusyIndicator.Visibility = Visibility.Visible;
+            grdFirstPage.Visibility = Visibility.Hidden;
+
+            // Load the package
+            if (await app.LoadPackageAsync(filename)) {
+                // Package loaded successfully
+                areaBusyIndicator.Visibility = Visibility.Hidden;
+
+                lstImages.ItemsSource = app.AppModels;
+                lstImages.SelectedIndex = 0;
+            }
+            else {
+                areaBusyIndicator.Visibility = Visibility.Hidden;
+                grdFirstPage.Visibility = Visibility.Visible;
+                MessageBox.Show(this, app.LastException.Message, "Error", MessageBoxButton.OK, MessageBoxImage.Error);
             }
         }
 
@@ -479,6 +493,45 @@ namespace Il2CppInspectorGUI
                 }
             }
         }
+
+        /// <summary>
+        /// Surface to handle dropped files
+        /// </summary>
+        private async void MainWindow_OnDrop(object sender, DragEventArgs e) {
+            if (e.Data.GetDataPresent(DataFormats.FileDrop)) {
+                string[] files = (string[]) e.Data.GetData(DataFormats.FileDrop);
+
+                if (grdFirstPage.Visibility == Visibility.Visible) {
+                    // Metadata or APK/IPA
+                    if (files.Length == 1) {
+                        switch (files[0].ToLower()) {
+                            case var s when s.EndsWith(".dat"):
+                                await LoadMetadataAsync(s);
+                                break;
+
+                            case var s when s.EndsWith(".apk") || s.EndsWith(".ipa") || s.EndsWith(".zip"):
+                                await LoadPackageAsync(s);
+                                break;
+                        }
+                    }
+                    // Metadata and binary
+                    else if (files.Length == 2) {
+                        var metadataIndex = files[0].ToLower().EndsWith(".dat") ? 0 : 1;
+                        var binaryIndex = 1 - metadataIndex;
+
+                        await LoadMetadataAsync(files[metadataIndex]);
+
+                        // Only load binary if metadata was successful
+                        if (btnSelectBinaryFile.Visibility == Visibility.Visible)
+                            await LoadBinaryAsync(files[binaryIndex]);
+                    }
+                }
+                // Binary (on 2nd page)
+                else if (btnSelectBinaryFile.Visibility == Visibility.Visible)
+                    if (files.Length == 1)
+                        await LoadBinaryAsync(files[0]);
+            }
+        }
     }
 
     // Replacement for TreeViewItem that includes checkbox state