! Detected supported files in Output Smart Bar

wmjordan · Dec 16, 2018 · e19ceb2 · e19ceb2
1 parent e0d0254
commit e19ceb2
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 6 deletions.
diff --git a/Codist/Helpers/ServicesHelper.cs b/Codist/Helpers/ServicesHelper.cs
@@ -28,6 +28,9 @@ private ServicesHelper() {
 		[Import]
 		public IClassificationTypeRegistryService ClassificationTypeRegistry { get; private set; }
 
+		[Import]
+		public Microsoft.VisualStudio.Utilities.IFileExtensionRegistryService FileExtensionRegistry { get; private set; }
+
 		[Import]
 		public IEditorFormatMapService EditorFormatMap { get; private set; }
 

diff --git a/Codist/SmartBars/OutputSmartBar.cs b/Codist/SmartBars/OutputSmartBar.cs
@@ -31,12 +31,14 @@ protected override void AddCommands(CancellationToken cancellationToken) {
 						Process.Start("Explorer.exe", "/select,\"" + s + "\"");
 					}
 				});
-				AddCommand(MyToolBar, KnownImageIds.VisualStudio, "Open file with Visual Studio", ctx => {
-					var s = TryGetPath(View);
-					if (s != null) {
-						CodistPackage.DTE.OpenFile(s, 1, 1);
-					}
-				});
+				if (IsFileTypeRegisteredInVS(t)) {
+					AddCommand(MyToolBar, KnownImageIds.VisualStudio, "Open file with Visual Studio", ctx => {
+						var s = TryGetPath(View);
+						if (s != null && IsFileTypeRegisteredInVS(s)) {
+							CodistPackage.DTE.OpenFile(s, 1, 1);
+						}
+					});
+				}
 			}
 			else if (Directory.Exists(t)) {
 				AddCommand(MyToolBar, KnownImageIds.OpenFolder, "Open folder", ctx => {
@@ -72,6 +74,14 @@ void TryRun(string path) {
 					// ignore
 				}
 			}
+			bool IsFileTypeRegisteredInVS(string fileName) {
+				try {
+					return ServicesHelper.Instance.FileExtensionRegistry.GetContentTypeForExtension(Path.GetExtension(fileName)).TypeName != "UNKNOWN";
+				}
+				catch (ArgumentException) {
+					return false;
+				}
+			}
 		}
 	}
 }