feat: map to Statsig CustomIDs (#210)

Signed-off-by: Jens Henneberg <[email protected]>

Author: jenshenneberg

.github/workflows/dotnet-format.yml

@@ -20,10 +20,10 @@ jobs:
     - name: Check out code
       uses: actions/checkout@v4
 
-    - name: Setup .NET Core 6.0
+    - name: Setup .NET
       uses: actions/setup-dotnet@v4
       with:
-        dotnet-version: 6.0.x
+        global-json-file: global.json
 
     - name: Install format tool
       run: dotnet tool install -g dotnet-format

src/OpenFeature.Contrib.Providers.Statsig/EvaluationContextExtensions.cs

@@ -1,4 +1,5 @@
-using OpenFeature.Model;
+using OpenFeature.Error;
+using OpenFeature.Model;
 using Statsig;
 
 namespace OpenFeature.Contrib.Providers.Statsig
@@ -14,6 +15,7 @@ internal static class EvaluationContextExtensions
         internal const string CONTEXT_LOCALE = "locale";
         internal const string CONTEXT_USER_AGENT = "userAgent";
         internal const string CONTEXT_PRIVATE_ATTRIBUTES = "privateAttributes";
+        internal const string CONTEXT_CUSTOM_IDS = "customIDs";
 
         public static StatsigUser AsStatsigUser(this EvaluationContext evaluationContext)
         {
@@ -49,7 +51,19 @@ public static StatsigUser AsStatsigUser(this EvaluationContext evaluationContext
                             var privateAttributes = item.Value.AsStructure;
                             foreach (var items in privateAttributes)
                             {
-                                user.AddPrivateAttribute(items.Key, items.Value);
+                                user.AddPrivateAttribute(items.Key, items.Value.AsObject);
+                            }
+                        }
+                        break;
+                    case CONTEXT_CUSTOM_IDS:
+                        if (item.Value.IsStructure)
+                        {
+                            var customIds = item.Value.AsStructure;
+                            foreach (var customId in customIds)
+                            {
+                                if (customId.Value.IsString)
+                                    user.AddCustomID(customId.Key, customId.Value.AsString);
+                                else throw new FeatureProviderException(Constant.ErrorType.TypeMismatch, "Only string values are supported for CustomIDs");
                             }
                         }
                         break;

src/OpenFeature.Contrib.Providers.Statsig/OpenFeature.Contrib.Providers.Statsig.csproj

@@ -4,7 +4,6 @@
 		<PackageId>OpenFeature.Contrib.Providers.Statsig</PackageId>
 		<VersionNumber>0.0.5</VersionNumber><!--x-release-please-version -->
 		<VersionPrefix>$(VersionNumber)</VersionPrefix>
-		<VersionSuffix>preview</VersionSuffix>
 		<AssemblyVersion>$(VersionNumber)</AssemblyVersion>
 		<FileVersion>$(VersionNumber)</FileVersion>
 		<Description>Statsig provider for .NET</Description>

test/OpenFeature.Contrib.Providers.Statsig.Test/EvaluationContextExtensionsTests.cs

@@ -1,6 +1,8 @@
 using AutoFixture.Xunit2;
+using OpenFeature.Error;
 using OpenFeature.Model;
 using System.Collections.Generic;
+using System.Linq;
 using Xunit;
 
 namespace OpenFeature.Contrib.Providers.Statsig.Test
@@ -80,7 +82,37 @@ public void AsStatsigUser_ShouldMapPrivateData(string key, string value)
             // Assert
             Assert.NotNull(statsigUser);
             Assert.True(statsigUser.PrivateAttributes.TryGetValue(key, out var mappedValue));
-            Assert.Equal(value, (mappedValue as Value)?.AsString);
+            Assert.Equal(value, mappedValue);
+        }
+
+        [Theory]
+        [AutoData]
+        public void AsStatsigUser_ShouldMapCustomIds(Dictionary<string, string> customIdKeyValues)
+        {
+
+            // Arrange
+            var customIdStructure = new Structure(customIdKeyValues.ToDictionary(kvp => kvp.Key, kvp => new Value(kvp.Value)));
+            var evaluationContext = EvaluationContext.Builder().Set(EvaluationContextExtensions.CONTEXT_CUSTOM_IDS, customIdStructure).Build();
+
+            // Act
+            var statsigUser = evaluationContext.AsStatsigUser();
+
+            // Assert
+            Assert.NotNull(statsigUser);
+            var result = statsigUser.CustomIDs.ToDictionary(kvp => kvp.Key, kvp => kvp.Value);
+            Assert.Equal(customIdKeyValues, result);
+        }
+
+        [Theory]
+        [AutoData]
+        public void AsStatsigUser_ShouldFailOnNonStringCustomId(Dictionary<string, int> customIdKeyValues)
+        {
+            // Arrange
+            var customIdStructure = new Structure(customIdKeyValues.ToDictionary(kvp => kvp.Key, kvp => new Value(kvp.Value)));
+            var evaluationContext = EvaluationContext.Builder().Set(EvaluationContextExtensions.CONTEXT_CUSTOM_IDS, customIdStructure).Build();
+
+            // Act and Assert
+            Assert.Throws<FeatureProviderException>(evaluationContext.AsStatsigUser);
         }
 
         [Fact]