wabbajack/Wabbajack.Lib/Metrics.cs

using System;
using System.Threading.Tasks;
using Microsoft.Win32;
using Wabbajack.Common;

namespace Wabbajack.Lib
{
    public class Metrics
    {
        public const string Downloading = "downloading";
        public const string BeginInstall = "begin_install";
        public const string FinishInstall = "finish_install";
        private static readonly AsyncLock _creationLock = new AsyncLock();

        public static async ValueTask<string> GetMetricsKey()
        {
            using var _ = await _creationLock.WaitAsync();
            if (!Utils.HaveEncryptedJson(Consts.MetricsKeyHeader))
            {
                if (!Utils.HaveRegKeyMetricsKey())
                {
                    // When there's no file or regkey
                    var key = Utils.MakeRandomKey();
                    await key.ToEcryptedJson(Consts.MetricsKeyHeader);
                    using RegistryKey regKey = Registry.CurrentUser.CreateSubKey(@"Software\Wabbajack", RegistryKeyPermissionCheck.Default)!;
                    regKey.SetValue("x-metrics-key", key);
                    return key;
                }

                // If there is no file but a registry key, create the file and transfer the data from the registry key
                using (RegistryKey regKey = Registry.CurrentUser.OpenSubKey(@"Software\Wabbajack", RegistryKeyPermissionCheck.Default)!)
                {
                    string key = (string)regKey.GetValue(Consts.MetricsKeyHeader)!;
                    await key.ToEcryptedJson(Consts.MetricsKeyHeader);
                    return key;
                }
            }

            if (Utils.HaveRegKeyMetricsKey())
            {
                // When there's a file and a regkey
                using RegistryKey regKey = Registry.CurrentUser.OpenSubKey(@"Software\Wabbajack", RegistryKeyPermissionCheck.Default)!;
                return (string)regKey.GetValue(Consts.MetricsKeyHeader)!;
            }

            // If there's a regkey and a file, return regkey
            using (RegistryKey regKey = Registry.CurrentUser.CreateSubKey(@"Software\Wabbajack", RegistryKeyPermissionCheck.Default)!)
            {
                try
                {
                    string key = await Utils.FromEncryptedJson<string>(Consts.MetricsKeyHeader)!;
                    regKey.SetValue("x-metrics-key", key);
                    return key;
                }
                catch (Exception)
                {
                    // Probably an encryption error
                    await Utils.DeleteEncryptedJson(Consts.MetricsKeyHeader);
                    return await GetMetricsKey();
                }

            }
        }
        /// <summary>
        /// This is all we track for metrics, action, and value. The action will be like
        /// "downloaded", the value "Joe's list".
        /// </summary>
        /// <param name="action"></param>
        /// <param name="value"></param>
        public static async Task Send(string action, string value)
        {
            if (BuildServerStatus.IsBuildServerDown)
                return;

            var key = await GetMetricsKey();
            Utils.Log($"File hash check (-42) {key}");
            var client = new Http.Client();
            client.Headers.Add((Consts.MetricsKeyHeader, key));
            await client.GetAsync($"{Consts.WabbajackBuildServerUri}metrics/{action}/{value}");
        }
    }
}