Functionality to capture git repository size in usage metrics

Author: StanleyGoldman

src/GitHub.Api/Application/ApplicationManagerBase.cs

@@ -178,6 +178,7 @@ public void SetupGit(GitInstaller.GitInstallationState state)
                 if (Environment.RepositoryPath.IsInitialized)
                 {
                     ConfigureMergeSettings();
+                    CaptureRepoSize();
 
                     GitClient.LfsInstall()
                         .Catch(e =>
@@ -286,6 +287,18 @@ private void ConfigureMergeSettings()
             }).RunWithReturn(true);
         }
 
+        private void CaptureRepoSize()
+        {
+            GitClient.CountObjects()
+                     .Then((success, gitObjects) => {
+                         if (success)
+                         {
+                             UsageTracker.UpdateRepoSize(gitObjects.kilobytes);
+                         }
+                     })
+                     .Start();
+        }
+
         public void RestartRepository()
         {
             if (!Environment.RepositoryPath.IsInitialized)

src/GitHub.Api/Git/GitClient.cs

@@ -38,6 +38,7 @@ public interface IGitClient
         ITask<List<GitLogEntry>> Log(BaseOutputListProcessor<GitLogEntry> processor = null);
         ITask<TheVersion> Version(IOutputProcessor<TheVersion> processor = null);
         ITask<TheVersion> LfsVersion(IOutputProcessor<TheVersion> processor = null);
+        ITask<GitCountObjects> CountObjects(IOutputProcessor<GitCountObjects> processor = null);
         ITask<GitUser> SetConfigNameAndEmail(string username, string email);
     }
 
@@ -98,6 +99,12 @@ public ITask<TheVersion> LfsVersion(IOutputProcessor<TheVersion> processor = nul
                 .Configure(processManager);
         }
 
+        public ITask<GitCountObjects> CountObjects(IOutputProcessor<GitCountObjects> processor = null)
+        {
+            return new GitCountObjectsTask(cancellationToken, processor)
+                .Configure(processManager);
+        }
+
         public ITask<string> GetConfig(string key, GitConfigSource configSource, IOutputProcessor<string> processor = null)
         {
             return new GitConfigGetTask(key, configSource, cancellationToken, processor)

src/GitHub.Api/Git/GitCountObjects.cs

@@ -0,0 +1,23 @@
+using System;
+
+namespace GitHub.Unity
+{
+    [Serializable]
+    public struct GitCountObjects
+    {
+        public static GitCountObjects Default = new GitCountObjects();
+
+        public int objects;
+        public int kilobytes;
+
+        public GitCountObjects(int objects, int kilobytes)
+        {
+            this.objects = objects;
+            this.kilobytes = kilobytes;
+        }
+
+        public int Objects => objects;
+
+        public int Kilobytes => kilobytes;
+    }
+}

src/GitHub.Api/Git/Tasks/GitCountObjectsTask.cs

@@ -0,0 +1,22 @@
+using System.Threading;
+
+namespace GitHub.Unity
+{
+    class GitCountObjectsTask : ProcessTask<GitCountObjects>
+    {
+        private const string TaskName = "git count-objects";
+
+        public GitCountObjectsTask(CancellationToken token, IOutputProcessor<GitCountObjects> processor = null)
+            : base(token, processor ?? new GitCountObjectsProcessor())
+        {
+            Name = TaskName;
+        }
+
+        public override string ProcessArguments
+        {
+            get { return "count-objects"; }
+        }
+        public override TaskAffinity Affinity { get { return TaskAffinity.Exclusive; } }
+        public override string Message { get; set; } = "Counting git objects...";
+    }
+}

src/GitHub.Api/GitHub.Api.csproj

@@ -95,9 +95,11 @@
     <Compile Include="Cache\CachingClasses.cs" />
     <Compile Include="Extensions\ActionExtensions.cs" />
     <Compile Include="Extensions\ListExtensions.cs" />
+    <Compile Include="Git\GitCountObjects.cs" />
     <Compile Include="Git\Tasks\GitCheckoutTask.cs" />
     <Compile Include="Git\GitAheadBehindStatus.cs" />
     <Compile Include="Git\Tasks\GitAheadBehindStatusTask.cs" />
+    <Compile Include="Git\Tasks\GitCountObjectsTask.cs" />
     <Compile Include="Git\Tasks\GitLfsVersionTask.cs" />
     <Compile Include="Git\Tasks\GitVersionTask.cs" />
     <Compile Include="Git\TreeData.cs" />
@@ -116,6 +118,7 @@
     <Compile Include="IO\FileSystem.cs" />
     <Compile Include="Managers\Downloader.cs" />
     <Compile Include="OutputProcessors\GitAheadBehindStatusOutputProcessor.cs" />
+    <Compile Include="OutputProcessors\GitCountObjectsProcessor.cs" />
     <Compile Include="OutputProcessors\LfsVersionOutputProcessor.cs" />
     <Compile Include="OutputProcessors\VersionOutputProcessor.cs" />
     <Compile Include="Helpers\TaskHelpers.cs" />

src/GitHub.Api/Metrics/IUsageTracker.cs

@@ -19,5 +19,6 @@ public interface IUsageTracker
         void IncrementUnityProjectViewContextLfsUnlock();
         void IncrementPublishViewButtonPublish();
         void IncrementApplicationMenuMenuItemCommandLine();
+        void UpdateRepoSize(int kilobytes);
     }
 }

src/GitHub.Api/Metrics/UsageModel.cs

@@ -42,6 +42,7 @@ public class Measures
         public int UnityProjectViewContextLfsUnlock { get; set; }
         public int PublishViewButtonPublish { get; set; }
         public int ApplicationMenuMenuItemCommandLine { get; set; }
+        public int GitRepoSize { get; set; }
     }
 
     class UsageModel

src/GitHub.Api/Metrics/UsageTracker.cs

@@ -236,6 +236,13 @@ public void IncrementApplicationMenuMenuItemCommandLine()
             usageLoader.Save(usage);
         }
 
+        public void UpdateRepoSize(int kilobytes)
+        {
+            var usage = usageLoader.Load(userId);
+            usage.GetCurrentMeasures(appVersion, unityVersion, instanceId).GitRepoSize = kilobytes;
+            usageLoader.Save(usage);
+        }
+
         public bool Enabled
         {
             get

src/GitHub.Api/OutputProcessors/GitCountObjectsProcessor.cs

@@ -0,0 +1,24 @@
+namespace GitHub.Unity
+{
+    public class GitCountObjectsProcessor : BaseOutputProcessor<GitCountObjects>
+    {
+        public override void LineReceived(string line)
+        {
+            if (line == null)
+            {
+                return;
+            }
+
+            //2488 objects, 4237 kilobytes
+
+            var proc = new LineParser(line);
+
+            var objects = int.Parse(proc.ReadUntilWhitespace());
+            proc.ReadUntil(',');
+            proc.SkipWhitespace();
+            var kilobytes = int.Parse(proc.ReadUntilWhitespace());
+
+            RaiseOnEntry(new GitCountObjects(objects, kilobytes));
+        }
+    }
+}

src/tests/TestUtils/Helpers/AssertExtensions.cs

@@ -114,5 +114,11 @@ public static void AssertNotEqual(this GitStatus gitStatus, GitStatus other)
             Action action = () => gitStatus.AssertEqual(other);
             action.ShouldThrow<AssertionException>();
         }
+
+        public static void AssertEqual(this GitCountObjects gitStatus, GitCountObjects other)
+        {
+            gitStatus.Objects.Should().Be(other.Objects, "Objects should be equal");
+            gitStatus.Kilobytes.Should().Be(other.Kilobytes, "KilobytesS should be equal");
+        }
     }
 }