Show how many comments there are on current document

Author: jcansdale

src/GitHub.InlineReviews/CommentsMargin.cs

@@ -1,10 +1,10 @@
 using System;
 using System.Windows;
+using GitHub.Services;
 using Microsoft.VisualStudio.Text.Editor;
 using GitHub.InlineReviews.Views;
 using GitHub.InlineReviews.ViewModels;
 using GitHub.InlineReviews.Commands;
-using System.Windows.Input;
 
 namespace GitHub.InlineReviews
 {
@@ -29,9 +29,10 @@ internal class CommentsMargin : IWpfTextViewMargin
         /// Initializes a new instance of the <see cref="ToggleCommentsMargin"/> class for a given <paramref name="textView"/>.
         /// </summary>
         /// <param name="textView">The <see cref="IWpfTextView"/> to attach the margin to.</param>
-        public CommentsMargin(IWpfTextView textView, IEnableInlineCommentsCommand enableInlineCommentsCommand)
+        public CommentsMargin(IWpfTextView textView, IEnableInlineCommentsCommand enableInlineCommentsCommand,
+            IPullRequestSessionManager sessionManager)
         {
-            var viewModel = new CommentsMarginViewModel(enableInlineCommentsCommand);
+            var viewModel = new CommentsMarginViewModel(sessionManager, textView.TextBuffer, enableInlineCommentsCommand);
             visualElement = new CommentsMarginView { DataContext = viewModel, ClipToBounds = true };
         }
 

src/GitHub.InlineReviews/CommentsMarginProvider.cs

@@ -1,8 +1,9 @@
 using System;
 using System.ComponentModel.Composition;
-using GitHub.InlineReviews.Commands;
 using Microsoft.VisualStudio.Utilities;
 using Microsoft.VisualStudio.Text.Editor;
+using GitHub.Services;
+using GitHub.InlineReviews.Commands;
 
 namespace GitHub.InlineReviews
 {
@@ -18,11 +19,13 @@ namespace GitHub.InlineReviews
     internal sealed class CommentsMarginFactory : IWpfTextViewMarginProvider
     {
         readonly IEnableInlineCommentsCommand enableInlineCommentsCommand;
+        readonly IPullRequestSessionManager sessionManager;
 
         [ImportingConstructor]
-        public CommentsMarginFactory(IEnableInlineCommentsCommand enableInlineCommentsCommand)
+        public CommentsMarginFactory(IEnableInlineCommentsCommand enableInlineCommentsCommand, IPullRequestSessionManager sessionManager)
         {
             this.enableInlineCommentsCommand = enableInlineCommentsCommand;
+            this.sessionManager = sessionManager;
         }
 
         /// <summary>
@@ -35,7 +38,7 @@ public CommentsMarginFactory(IEnableInlineCommentsCommand enableInlineCommentsCo
         /// </returns>
         public IWpfTextViewMargin CreateMargin(IWpfTextViewHost wpfTextViewHost, IWpfTextViewMargin marginContainer)
         {
-            return new CommentsMargin(wpfTextViewHost.TextView, enableInlineCommentsCommand);
+            return new CommentsMargin(wpfTextViewHost.TextView, enableInlineCommentsCommand, sessionManager);
         }
     }
 }

src/GitHub.InlineReviews/ViewModels/CommentsMarginViewModel.cs

@@ -1,15 +1,44 @@
 using System.Windows.Input;
+using GitHub.Services;
+using GitHub.Extensions;
+using Microsoft.VisualStudio.Text;
+using ReactiveUI;
+using Task = System.Threading.Tasks.Task;
 
 namespace GitHub.InlineReviews.ViewModels
 {
-    public class CommentsMarginViewModel
+    public class CommentsMarginViewModel : ReactiveObject
     {
-        public CommentsMarginViewModel(ICommand enableInlineComments)
+        readonly IPullRequestSessionManager sessionManager;
+        readonly ITextBuffer textBuffer;
+
+        int commentsInFile;
+
+        public CommentsMarginViewModel(IPullRequestSessionManager sessionManager, ITextBuffer textBuffer, ICommand enableInlineComments)
         {
+            this.sessionManager = sessionManager;
+            this.textBuffer = textBuffer;
+
             EnableInlineComments = enableInlineComments;
+            InitializeAsync().Forget();
+        }
+
+        async Task InitializeAsync()
+        {
+            await sessionManager.EnsureInitialized();
+            var relativePath = sessionManager.GetRelativePath(textBuffer);
+            if (relativePath != null)
+            {
+                var sessionFile = await sessionManager.CurrentSession.GetFile(relativePath);
+                CommentsInFile = sessionFile?.InlineCommentThreads?.Count ?? -1;
+            }
         }
 
-        public int CommentsInFile { get; } = 777;
+        public int CommentsInFile
+        {
+            get { return commentsInFile; }
+            private set { this.RaiseAndSetIfChanged(ref commentsInFile, value); }
+        }
 
         public ICommand EnableInlineComments { get; }
     }

src/GitHub.InlineReviews/Views/CommentsMarginView.xaml

@@ -24,7 +24,7 @@
                 </Button.Content>
             </Button>
             <Hyperlink>
-                <Run Text="{Binding CommentsInFile, Mode=OneWay}"/>
+                <Run Text="{Binding CommentsInFile}"/>
             </Hyperlink>
         </TextBlock>
     </Grid>