Auto tagging with ResourceTransformation (#33)

Signed-off-by: Piotr <[email protected]> Signed-off-by: Piotr <[email protected]> Co-authored-by: Piotr <[email protected]>
Avanade · Sep 13, 2022 · d7eb0eb · d7eb0eb
1 parent 91a8623
commit d7eb0eb
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 3 deletions.
diff --git a/samples/My.Hr/My.Hr.Infra.Tests/CoreExStackTests.cs b/samples/My.Hr/My.Hr.Infra.Tests/CoreExStackTests.cs
@@ -21,6 +21,28 @@ public async Task ResourceGroupHasNameTag()
         tags.Should().ContainKey("App");
     }
 
+    [Test]
+    public async Task AllResourcesHaveNameTag()
+    {
+        // unfortunately this doesn't test tags created by auto-tagging dove via ResourceTransformation
+        var (resources, outputs, dbOperationsMock) = await Testing.RunAsync();
+
+        var rs = resources.Select(async r =>
+        {
+            var tagsProp = r.GetType().GetProperty("Tags");
+
+            return tagsProp != null
+            ? (resource: r, tags: await tagsProp!.GetValue(r)!.GetValueAsync<System.Collections.Immutable.ImmutableDictionary<string, string>?>())
+            : (resource: r, tags: null);
+        });
+
+        var result = (await Task.WhenAll(rs)).Where(anyResource => anyResource.tags != null);
+
+        // Assert
+        result.Should().HaveCountGreaterThan(5);
+        result.Should().AllSatisfy(r => r.tags!.ContainsKey("App"), because: "All resources should be tagged");
+    }
+
     [Test]
     public async Task FunctionIsCreatedWithAUrl()
     {

diff --git a/samples/My.Hr/My.Hr.Infra/Program.cs b/samples/My.Hr/My.Hr.Infra/Program.cs
@@ -1,8 +1,29 @@
 // build CoreEx stack
 using My.Hr.Infra.Services;
+using Pulumi;
 
-return await Pulumi.Deployment.RunAsync(() =>
+return await Deployment.RunAsync(() =>
 {
-    // create ans use actual instance of DB Operations service
+    // create and use actual instance of DB Operations service
     return My.Hr.Infra.CoreExStack.ExecuteStackAsync(new DbOperations());
-}, null);
+}, new StackOptions
+{
+    // apply auto-tagging transformation
+    // https://gist.github.com/dbeattie71/1f8a1a9264ceb8161ad4c49de1ee3bb3
+    ResourceTransformations = new System.Collections.Generic.List<ResourceTransformation>{
+        (args) => {
+            var tagsProp = args.Args.GetType().GetProperty("Tags");
+
+            if(tagsProp?.GetValue(args.Args) is InputMap<string> tags)
+            {
+                Log.Debug("Adding tags to " + args.Resource.GetResourceName());
+
+                tags.Add("user:Stack", Deployment.Instance.StackName);
+                tags.Add("user:Project", Deployment.Instance.ProjectName);
+                tags.Add("App:Name", "CoreEx");
+            }
+
+            return new ResourceTransformationResult(args.Args, args.Options);
+        }
+    }
+});