Fix broken things

pulsejet · Nov 11, 2017 · bcea4d7 · bcea4d7
1 parent 21052c2
commit bcea4d7
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 19 deletions.
diff --git a/AUTHORS b/AUTHORS
@@ -1,2 +1,5 @@
 Daniel Aleksandersen (da2x)
-Mark Young (tip2tail)
+Mark Young (tip2tail)
+
+Search Engine Functionality:
+RadialApps (radialapps)
diff --git a/EdgeDeflector/EdgeDeflector.csproj b/EdgeDeflector/EdgeDeflector.csproj
@@ -61,6 +61,7 @@
     <Reference Include="System" />
     <Reference Include="System.Core" />
     <Reference Include="System.Web" />
+    <Reference Include="System.Windows.Forms" />
     <Reference Include="System.Xml.Linq" />
     <Reference Include="System.Data.DataSetExtensions" />
     <Reference Include="Microsoft.CSharp" />

diff --git a/EdgeDeflector/Program.cs b/EdgeDeflector/Program.cs
@@ -164,24 +164,9 @@ static string RewriteMsEdgeUriSchema(string uri)
             Regex rgx = new Regex(msedge_protocol_pattern);
             string new_uri = rgx.Replace(uri, string.Empty);
 
-            if (engine == "Google")
-            {
-                int index = new_uri.IndexOf("&");
-                if (index > 0)
-                    new_uri = new_uri.Substring(0, index);
-                new_uri = new_uri.Replace("bing.com/search?q=", "google.com/search?q=");
-            }
-            else if (engine == "DuckDuckGo")
-            {
-                int index = new_uri.IndexOf("&");
-                if (index > 0)
-                    new_uri = new_uri.Substring(0, index);
-                new_uri = new_uri.Replace("bing.com/search?q=", "duckduckgo.com/?q=");
-            }
-
             if (IsHttpUri(new_uri))
             {
-                return new_uri;
+                return replaceSearchEngine(new_uri, engine);
             }
 
             // May be new-style Cortana URI - try and split out
@@ -191,12 +176,29 @@ static string RewriteMsEdgeUriSchema(string uri)
                 if (IsHttpUri(cortanaUri))
                 {
                     // Correctly form the new URI before returning
-                    return cortanaUri;
+                    return replaceSearchEngine(cortanaUri, engine);
                 }
             }
 
             // defer fallback to web browser
-            return "http://" + new_uri;
+            return "http://" + replaceSearchEngine(new_uri, engine);
+        }
+
+        static string replaceSearchEngine(string new_uri, string engine)
+        {
+            int index = new_uri.IndexOf("&");
+            if (index > 0) new_uri = new_uri.Substring(0, index);
+
+            if (engine == "Google")
+            {
+                return new_uri.Replace("bing.com/search?q=", "google.com/search?q=");
+            }
+            else if (engine == "DuckDuckGo")
+            {
+                return new_uri.Replace("bing.com/search?q=", "duckduckgo.com/?q=");
+            }
+
+            return new_uri;
         }
 
         static void OpenUri(string uri)