Support for calculation groups

anakic · Oct 30, 2024 · 5d45d0d · 5d45d0d
1 parent 3814945
commit 5d45d0d
Showing 1 changed file with 53 additions and 3 deletions.
diff --git a/Packer2.Library/DataModel/FolderDatabaseStore.cs b/Packer2.Library/DataModel/FolderDatabaseStore.cs
@@ -108,6 +108,50 @@ protected override string GetFileExtension(JToken elem)
             }
         }
 
+        class CalculationItemMapping : JsonPropertyZone
+        {
+            protected override string PayloadContainingObjectJsonPath => "";
+
+            protected override string GetPayloadProperty(JToken obj) => "expression";
+
+            protected override string ContainingFolder => "Items";
+
+            protected override string ElementsSelector => ".calculationGroup.calculationItems[*]";
+
+            protected override string GetFileName(JToken elem) => (string)elem["name"]!;
+
+            protected override string GetFileExtension(JToken elem) => "dax";
+        }
+
+        class CalculationGroupMapping : JsonElementZone
+        {
+            protected override string ElementsSelector => ".model.tables[*]";
+
+            protected override string ContainingFolder => "CalculationGroups";
+
+            protected override string GetSubfolderForElement(JToken elem)
+                => (string)elem["name"]!;
+
+            protected override string GetFileName(JToken elem) => "group";
+
+            protected override string GetFileExtension(JToken elem) => "json";
+
+            protected override IEnumerable<MappingZone> ChildMappings { get; }
+
+            protected override bool FilterSelectedElements(JToken element)
+            {
+                return element["calculationGroup"] != null;
+            }
+
+            public CalculationGroupMapping()
+            {
+                ChildMappings = new MappingZone[]
+                {
+                    new CalculationItemMapping()
+                };
+            }
+        }
+
         class TablesMapping : JsonElementZone
         {
             protected override string ElementsSelector => ".model.tables[*]";
@@ -123,13 +167,18 @@ protected override string GetSubfolderForElement(JToken elem)
 
             protected override IEnumerable<MappingZone> ChildMappings { get; }
 
+            protected override bool FilterSelectedElements(JToken element)
+            {
+                return element["calculationGroup"] == null;
+            }
+
             public TablesMapping()
             {
                 ChildMappings = new MappingZone[]
                 {
                     new PartitionExpressionsMapping(),
                     new ColumnExpressionsMapping(),
-                    new MeasureExpressionsMapping()
+                    new MeasureExpressionsMapping(),
                 };
             }
         }
@@ -140,8 +189,9 @@ public BimMappedRepository()
             {
                 Mappings = new MappingZone[]
                 {
-                new TablesMapping(),
-                new ModelExpressionsMapping()
+                    new TablesMapping(),
+                    new ModelExpressionsMapping(),
+                    new CalculationGroupMapping(),
                 };
             }