| 1 | using EnvDTE; |
|---|
| 2 | using Microsoft.VisualStudio.Editor; |
|---|
| 3 | using Microsoft.VisualStudio.Shell; |
|---|
| 4 | using Microsoft.VisualStudio.Shell.Interop; |
|---|
| 5 | using Microsoft.VisualStudio.Text; |
|---|
| 6 | using Microsoft.VisualStudio.Text.Editor; |
|---|
| 7 | using Microsoft.VisualStudio.TextManager.Interop; |
|---|
| 8 | using System; |
|---|
| 9 | using System.IO; |
|---|
| 10 | |
|---|
| 11 | namespace LLVM.ClangFormat |
|---|
| 12 | { |
|---|
| 13 | internal sealed class Vsix |
|---|
| 14 | { |
|---|
| 15 | /// <summary> |
|---|
| 16 | /// Returns the currently active view if it is a IWpfTextView. |
|---|
| 17 | /// </summary> |
|---|
| 18 | public static IWpfTextView GetCurrentView() |
|---|
| 19 | { |
|---|
| 20 | // The SVsTextManager is a service through which we can get the active view. |
|---|
| 21 | var textManager = (IVsTextManager)Package.GetGlobalService(typeof(SVsTextManager)); |
|---|
| 22 | IVsTextView textView; |
|---|
| 23 | textManager.GetActiveView(1, null, out textView); |
|---|
| 24 | |
|---|
| 25 | // Now we have the active view as IVsTextView, but the text interfaces we need |
|---|
| 26 | // are in the IWpfTextView. |
|---|
| 27 | return VsToWpfTextView(textView); |
|---|
| 28 | } |
|---|
| 29 | |
|---|
| 30 | public static bool IsDocumentDirty(Document document) |
|---|
| 31 | { |
|---|
| 32 | var textView = GetDocumentView(document); |
|---|
| 33 | var textDocument = GetTextDocument(textView); |
|---|
| 34 | return textDocument?.IsDirty == true; |
|---|
| 35 | } |
|---|
| 36 | |
|---|
| 37 | public static IWpfTextView GetDocumentView(Document document) |
|---|
| 38 | { |
|---|
| 39 | var textView = GetVsTextViewFrompPath(document.FullName); |
|---|
| 40 | return VsToWpfTextView(textView); |
|---|
| 41 | } |
|---|
| 42 | |
|---|
| 43 | public static IWpfTextView VsToWpfTextView(IVsTextView textView) |
|---|
| 44 | { |
|---|
| 45 | var userData = (IVsUserData)textView; |
|---|
| 46 | if (userData == null) |
|---|
| 47 | return null; |
|---|
| 48 | Guid guidWpfViewHost = DefGuidList.guidIWpfTextViewHost; |
|---|
| 49 | object host; |
|---|
| 50 | userData.GetData(ref guidWpfViewHost, out host); |
|---|
| 51 | return ((IWpfTextViewHost)host).TextView; |
|---|
| 52 | } |
|---|
| 53 | |
|---|
| 54 | public static IVsTextView GetVsTextViewFrompPath(string filePath) |
|---|
| 55 | { |
|---|
| 56 | // From http://stackoverflow.com/a/2427368/4039972 |
|---|
| 57 | var dte2 = (EnvDTE80.DTE2)Package.GetGlobalService(typeof(SDTE)); |
|---|
| 58 | var sp = (Microsoft.VisualStudio.OLE.Interop.IServiceProvider)dte2; |
|---|
| 59 | var serviceProvider = new Microsoft.VisualStudio.Shell.ServiceProvider(sp); |
|---|
| 60 | |
|---|
| 61 | IVsUIHierarchy uiHierarchy; |
|---|
| 62 | uint itemID; |
|---|
| 63 | IVsWindowFrame windowFrame; |
|---|
| 64 | if (VsShellUtilities.IsDocumentOpen(serviceProvider, filePath, Guid.Empty, |
|---|
| 65 | out uiHierarchy, out itemID, out windowFrame)) |
|---|
| 66 | { |
|---|
| 67 | // Get the IVsTextView from the windowFrame. |
|---|
| 68 | return VsShellUtilities.GetTextView(windowFrame); |
|---|
| 69 | } |
|---|
| 70 | return null; |
|---|
| 71 | } |
|---|
| 72 | |
|---|
| 73 | public static ITextDocument GetTextDocument(IWpfTextView view) |
|---|
| 74 | { |
|---|
| 75 | ITextDocument document; |
|---|
| 76 | if (view != null && view.TextBuffer.Properties.TryGetProperty(typeof(ITextDocument), out document)) |
|---|
| 77 | return document; |
|---|
| 78 | return null; |
|---|
| 79 | } |
|---|
| 80 | |
|---|
| 81 | public static string GetDocumentParent(IWpfTextView view) |
|---|
| 82 | { |
|---|
| 83 | ITextDocument document = GetTextDocument(view); |
|---|
| 84 | if (document != null) |
|---|
| 85 | { |
|---|
| 86 | return Directory.GetParent(document.FilePath).ToString(); |
|---|
| 87 | } |
|---|
| 88 | return null; |
|---|
| 89 | } |
|---|
| 90 | |
|---|
| 91 | public static string GetDocumentPath(IWpfTextView view) |
|---|
| 92 | { |
|---|
| 93 | return GetTextDocument(view)?.FilePath; |
|---|
| 94 | } |
|---|
| 95 | } |
|---|
| 96 | } |
|---|
| 97 | |
|---|