Add constraint to TDialogPage to ensure that only DialogPage types ar…

…e used
christianhelle · Jun 14, 2019 · 37ca882 · 37ca882
1 parent 2934c65
commit 37ca882
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/src/ApiClientCodeGen.VSIX/Options/IOptionsFactory.cs b/src/ApiClientCodeGen.VSIX/Options/IOptionsFactory.cs
@@ -1,8 +1,11 @@
-namespace ChristianHelle.DeveloperTools.CodeGenerators.ApiClient.Options
+using Microsoft.VisualStudio.Shell;
+
+namespace ChristianHelle.DeveloperTools.CodeGenerators.ApiClient.Options
 {
     public interface IOptionsFactory
     {
-        TOptions Create<TOptions, TDialogPage>() 
-            where TOptions : class;
+        TOptions Create<TOptions, TDialogPage>()
+            where TOptions : class
+            where TDialogPage : DialogPage;
     }
 }
diff --git a/src/ApiClientCodeGen.VSIX/Options/OptionsFactory.cs b/src/ApiClientCodeGen.VSIX/Options/OptionsFactory.cs
@@ -1,9 +1,12 @@
-namespace ChristianHelle.DeveloperTools.CodeGenerators.ApiClient.Options
+using Microsoft.VisualStudio.Shell;
+
+namespace ChristianHelle.DeveloperTools.CodeGenerators.ApiClient.Options
 {
     public class OptionsFactory : IOptionsFactory
     {
-        public TOptions Create<TOptions, TDialogPage>() 
-            where TOptions : class 
+        public TOptions Create<TOptions, TDialogPage>()
+            where TOptions : class
+            where TDialogPage : DialogPage
             => VsPackage.Instance.GetDialogPage(typeof(TDialogPage)) as TOptions;
     }
 }