Implement the UserFormRequiresBinaryFileNameExtractor

Author: MDoerner

Rubberduck.VBEEditor/Utility/ModuleNameFromFileExtractor.cs

@@ -1,7 +1,10 @@
-using System.IO;
+using System.Collections.Generic;
+using System.IO;
 using System.Linq;
 using System.Text;
+using Rubberduck.JunkDrawer.Extensions;
 using Rubberduck.VBEditor.Extensions;
+using Rubberduck.VBEditor.SafeComWrappers;
 
 namespace Rubberduck.VBEditor.Utility
 {
@@ -19,10 +22,9 @@ public string ModuleName(string filename)
                 return null;
             }
 
-            //We cannot read binary files.
-            if(ComponentTypeExtensions.FormBinaryExtension.Equals(Path.GetExtension(filename)))
+            if (!SupportedExtensions.Contains(Path.GetExtension(filename)))
             {
-                return Path.GetFileNameWithoutExtension(filename);
+                return null;
             }
 
             var contents = File.ReadLines(filename, Encoding.Default);
@@ -35,5 +37,10 @@ public string ModuleName(string filename)
             //The format is Attribute VB_Name = "ModuleName"
             return nameLine.Substring("Attribute VB_Name = ".Length + 1, nameLine.Length - "Attribute VB_Name = ".Length - 2);
         }
+
+        private static ICollection<string> SupportedExtensions => 
+            ComponentTypeExtensions.ComponentTypesForExtension(VBEKind.Hosted).Keys
+            .Concat(ComponentTypeExtensions.ComponentTypesForExtension(VBEKind.Standalone).Keys)
+            .ToHashSet();
     }
 }

Rubberduck.VBEEditor/Utility/UserFormRequiredBinaryFileNameExtractor.cs

@@ -0,0 +1,49 @@
+using System.Collections.Generic;
+using System.IO;
+using System.Text;
+using System.Text.RegularExpressions;
+using Rubberduck.VBEditor.Extensions;
+using Rubberduck.VBEditor.SafeComWrappers;
+using Path = System.IO.Path;
+
+namespace Rubberduck.VBEditor.Utility
+{
+    public class UserFormRequiredBinaryFileNameExtractor : IRequiredBinaryFilesFromFileNameExtractor
+    {
+        public ICollection<ComponentType> SupportedComponentTypes => new List<ComponentType>{ComponentType.UserForm};
+
+        public ICollection<string> RequiredBinaryFiles(string fileName, ComponentType componentType)
+        {
+            if (!File.Exists(fileName))
+            {
+                return null;
+            }
+
+            if (!SupportedComponentTypes.Contains(componentType))
+            {
+                return null;
+            }
+
+            if (componentType.FileExtension() != Path.GetExtension(fileName))
+            {
+                return null;
+            }
+
+            var regExPattern = "OleObjectBlob\\s+=\\s+\"([^\"]+)\":";
+            var regEx = new Regex(regExPattern);
+            var contents = File.ReadLines(fileName, Encoding.Default);
+            
+            foreach(var codeLine in contents)
+            {
+                var match = regEx.Match(codeLine);
+                if (match.Success)
+                {
+                    return new List<string>{match.Groups[1].Value};
+                }
+            }
+
+            var fallbackBinaryName = Path.GetFileNameWithoutExtension(fileName) + componentType.BinaryFileExtension();
+            return new List<string>{fallbackBinaryName};
+        }
+    }
+}