Merge pull request #184 from austin94/feature/link-to-vs

shana · shana · commit 3df7b6c27e32 · 2016-01-11T14:14:09.000+01:00
diff --git a/src/GitHub.Exports/Extensions/SimpleRepositoryModelExtensions.cs b/src/GitHub.Exports/Extensions/SimpleRepositoryModelExtensions.cs
@@ -1,9 +1,11 @@
 ﻿using System;
+using System.Globalization;
 using System.Linq;
 using System.IO;
 using GitHub.Models;
 using Microsoft.VisualStudio.TeamFoundation.Git.Extensibility;
 using GitHub.Services;
+using GitHub.VisualStudio;
 
 namespace GitHub.Extensions
 {
@@ -30,5 +32,46 @@ public static bool MightContainSolution(this ISimpleRepositoryModel repository)
                       .Any(x => ((x.Attributes.HasFlag(FileAttributes.Directory) || x.Attributes.HasFlag(FileAttributes.Normal)) &&
                                 !x.Name.StartsWith(".", StringComparison.Ordinal) && !x.Name.StartsWith("readme", StringComparison.OrdinalIgnoreCase)));
         }
+
+        public static string CurrentSha(this ISimpleRepositoryModel repository)
+        {
+            if (repository == null)
+                return null;
+
+            var repo = GitService.GitServiceHelper.GetRepo(repository.LocalPath);
+
+            if (repo == null)
+                return null;
+
+            return !repo.Commits.Any() ? null : repo.Commits.ElementAt(0).Sha;
+        }
+
+        public static string BrowserUrl(this ISimpleRepositoryModel repository, IActiveDocument activeDocument)
+        {
+            if (repository == null || activeDocument == null)
+                return null;
+
+            var currentCommitSha = repository.CurrentSha();
+            var currentCloneUrl = repository.CloneUrl;
+            var localPath = repository.LocalPath;
+            var lineTag = "L" + activeDocument.AnchorLine;
+
+            if (string.IsNullOrEmpty(currentCommitSha) || string.IsNullOrEmpty(currentCloneUrl) ||
+                string.IsNullOrEmpty(localPath))
+                return null;
+
+            if (activeDocument.AnchorLine != activeDocument.EndLine)
+            {
+                lineTag += "-L" + activeDocument.EndLine;
+            }
+
+            var outputUri = string.Format(CultureInfo.CurrentCulture, "{0}/blob/{1}{2}#{3}",
+                currentCloneUrl,
+                currentCommitSha,
+                activeDocument.Name.Replace(localPath, "").Replace("\\", "/"),
+                lineTag);
+
+            return outputUri;
+        }
     }
 }
diff --git a/src/GitHub.Exports/Services/IActiveDocument.cs b/src/GitHub.Exports/Services/IActiveDocument.cs
@@ -3,6 +3,9 @@
     public interface IActiveDocument
     {
         string Name { get; }
-        int Line { get; }
+        int AnchorLine { get; }
+        int AnchorColumn { get; }
+        int EndLine { get; }
+        int EndColumn { get; }
     }
 }
diff --git a/src/GitHub.VisualStudio/GitHub.VisualStudio.vsct b/src/GitHub.VisualStudio/GitHub.VisualStudio.vsct
@@ -122,11 +122,21 @@
 
       <Button guid="guidContextMenuSet" id="getLinkCommand" priority="0x0100" type="Button">
         <Parent guid="guidContextMenuSet" id="idContextMenuGroup"/>
+        <CommandFlag>DefaultInvisible</CommandFlag>
+        <CommandFlag>DynamicVisibility</CommandFlag>
         <Strings>
-          <ButtonText>Get Link</ButtonText>
+          <ButtonText>Open in Browser</ButtonText>
         </Strings>
       </Button>
 
+      <Button guid="guidContextMenuSet" id="copyLinkCommand" priority="0x0101" type="Button">
+        <Parent guid="guidContextMenuSet" id="idContextMenuGroup"/>
+        <CommandFlag>DefaultInvisible</CommandFlag>
+        <CommandFlag>DynamicVisibility</CommandFlag>
+        <Strings>
+          <ButtonText>Copy Link to Clipboard</ButtonText>
+        </Strings>
+      </Button>
     </Buttons>
 
   </Commands>
@@ -187,6 +197,7 @@
     <GuidSymbol name="guidContextMenuSet" value="{31057D08-8C3C-4C5B-9F91-8682EA08EC27}">
       <IDSymbol name="idContextMenuGroup" value="0x1010" />
       <IDSymbol name="getLinkCommand" value="0x100" />
+      <IDSymbol name="copyLinkCommand" value="0x101"/>
     </GuidSymbol>
 
     <GuidSymbol name="GUID_XAML_EDITOR" value="{4C87B692-1202-46AA-B64C-EF01FAEC53DA}">
diff --git a/src/GitHub.VisualStudio/GitHubPackage.cs b/src/GitHub.VisualStudio/GitHubPackage.cs
@@ -1,6 +1,8 @@
 ﻿using System;
+using System.Globalization;
 using System.Runtime.InteropServices;
 using GitHub.Extensions;
+using GitHub.Models;
 using GitHub.Services;
 using GitHub.UI;
 using GitHub.VisualStudio.Base;
@@ -60,18 +62,54 @@ protected override void Initialize()
                 var windowFrame = (IVsWindowFrame)window.Frame;
                 ErrorHandler.ThrowOnFailure(windowFrame.Show());
             });
+            ServiceProvider.AddDynamicMenuItem(GuidList.guidContextMenuSet, PkgCmdIDList.getLinkCommand,
+                IsValidGithubRepo, 
+                OpenRepoInBrowser);
+            ServiceProvider.AddDynamicMenuItem(GuidList.guidContextMenuSet, PkgCmdIDList.copyLinkCommand,
+                IsValidGithubRepo,
+                CopyRepoLinkToClipboard);
 
-            ServiceProvider.AddTopLevelMenuItem(GuidList.guidContextMenuSet, PkgCmdIDList.getLinkCommand, (s, e) =>
-            {
-                var ap = ServiceProvider.GetExportedValue<IActiveDocument>();
-                var name = ap.Name;
-                var line = ap.Line;
-                System.Windows.Forms.MessageBox.Show(name + " : " + line);
-                
-            });
             base.Initialize();
         }
 
+        private void CopyRepoLinkToClipboard()
+        {
+            if (!IsValidGithubRepo()) return;
+
+            var activeDocument = ServiceProvider.GetExportedValue<IActiveDocument>();
+            var activeRepo = ServiceProvider.GetExportedValue<ITeamExplorerServiceHolder>().ActiveRepo;
+            var outputUri = activeRepo?.BrowserUrl(activeDocument);
+
+            if (string.IsNullOrEmpty(outputUri)) return;
+
+            System.Windows.Clipboard.SetText(outputUri);
+        }
+
+        private bool IsValidGithubRepo()
+        {
+            var cloneUrl = ServiceProvider.GetExportedValue<ITeamExplorerServiceHolder>().ActiveRepo?.CloneUrl;
+
+            if (string.IsNullOrEmpty(cloneUrl))
+                return false;
+
+            return cloneUrl.Host == "github.com";
+        }
+
+        private void OpenRepoInBrowser()
+        {
+            if (!IsValidGithubRepo()) return;
+
+            var activeDocument = ServiceProvider.GetExportedValue<IActiveDocument>();
+            var activeRepo = ServiceProvider.GetExportedValue<ITeamExplorerServiceHolder>().ActiveRepo;
+
+            var outputUri = activeRepo?.BrowserUrl(activeDocument);
+
+            if (string.IsNullOrEmpty(outputUri)) return;
+
+            var vsBrowserProvider = ServiceProvider.GetExportedValue<IVisualStudioBrowser>();
+            vsBrowserProvider.OpenUrl(new Uri(outputUri));
+        }
+
         void StartFlow(UIControllerFlow controllerFlow)
         {
             var uiProvider = ServiceProvider.GetExportedValue<IUIProvider>();
diff --git a/src/GitHub.VisualStudio/PkgCmdID.cs b/src/GitHub.VisualStudio/PkgCmdID.cs
@@ -12,5 +12,6 @@ static class PkgCmdIDList
         public const int refreshCommand = 0x302;
         public const int pullRequestCommand = 0x310;
         public const int getLinkCommand = 0x100;
+        public const int copyLinkCommand = 0x101;
     };
 }
diff --git a/src/GitHub.VisualStudio/Services/ActiveDocument.cs b/src/GitHub.VisualStudio/Services/ActiveDocument.cs
@@ -12,25 +12,32 @@ namespace GitHub.VisualStudio
     class ActiveDocument : IActiveDocument
     {
         public string Name { get; private set; }
-        public int Line { get; private set; }
-        public int Column { get; private set; }
+        public string ShortName { get; private set; }
+        public int AnchorLine { get; private set; }
+        public int AnchorColumn { get; private set; }
+        public int EndLine { get; private set; }
+        public int EndColumn { get; private set; }
 
         [ImportingConstructor]
         public ActiveDocument([Import(typeof(SVsServiceProvider))] IServiceProvider serviceProvider)
         {
-            Line = Column = -1;
+            AnchorLine = AnchorColumn = EndLine = EndColumn = -1;
             Name = Services.Dte2?.ActiveDocument?.FullName;
+            ShortName = Services.Dte2?.ActiveDocument?.Name;
+
             var textManager = serviceProvider.GetService(typeof(SVsTextManager)) as IVsTextManager;
             Debug.Assert(textManager != null, "No SVsTextManager service available");
             if (textManager == null)
                 return;
             IVsTextView view;
-            int line, col;
+            int anchorLine, anchorCol, endLine, endCol;
             if (ErrorHandler.Succeeded(textManager.GetActiveView(0, null, out view)) &&
-                ErrorHandler.Succeeded(view.GetCaretPos(out line, out col)))
+                ErrorHandler.Succeeded(view.GetSelection(out anchorLine, out anchorCol, out endLine, out endCol)))
             {
-                Line = line;
-                Column = col;
+                AnchorLine = anchorLine + 1;
+                AnchorColumn = anchorCol + 1;
+                EndLine = endLine + 1;
+                EndColumn = endCol + 1;
             }
         }
     }

Original file line number	Diff line number	Diff line change
`@@ -3,6 +3,9 @@`
`3`	`3`	`public interface IActiveDocument`
`4`	`4`	`{`
`5`	`5`	`string Name { get; }`
`6`		`- int Line { get; }`
	`6`	`+ int AnchorLine { get; }`
	`7`	`+ int AnchorColumn { get; }`
	`8`	`+ int EndLine { get; }`
	`9`	`+ int EndColumn { get; }`
`7`	`10`	`}`
`8`	`11`	`}`
Original file line number	Diff line number	Diff line change
`@@ -12,5 +12,6 @@ static class PkgCmdIDList`
`12`	`12`	`public const int refreshCommand = 0x302;`
`13`	`13`	`public const int pullRequestCommand = 0x310;`
`14`	`14`	`public const int getLinkCommand = 0x100;`
	`15`	`+ public const int copyLinkCommand = 0x101;`
`15`	`16`	`};`
`16`	`17`	`}`