Refinement implemented

westermo · May 30, 2024 · 72355e1 · 72355e1
1 parent 99db906
commit 72355e1
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/Compiler.Tests/ParsingTests.cs b/Compiler.Tests/ParsingTests.cs
@@ -75,7 +75,11 @@ import three {
                                             prefix b;
                                         }
                                         grouping A {
-                                            uses b:A;
+                                            uses b:A { 
+                                                refine target {
+                                                    default 80;
+                                                }
+                                            }
                                             container C {
                                                 uses C;
                                             }
@@ -134,6 +138,7 @@ leaf module-set-id {
                                                   }
                                                   leaf test { type operator; }
                                               }
+                                              leaf target { type int32; }
                                           }
                                       }
                                       """;
@@ -175,6 +180,7 @@ public void GroupingTest()
             }
             Assert.IsNotType<Uses>(statement);
         }
+        output.WriteLine(compilationUnit.ToCode());
 
         Log.Clear();
     }

diff --git a/YangParser/SemanticModel/Grouping.cs b/YangParser/SemanticModel/Grouping.cs
@@ -117,6 +117,24 @@ public IStatement[] WithUse(Uses use)
         }
 
         Parent.Replace(copy, []);
+        foreach (var refinement in use.Children.OfType<Refine>())
+        {
+            var path = refinement.Argument.Split('/');
+            var current = copy;
+            foreach (var element in path)
+            {
+                var prefix = element.Prefix(out var name);
+                var origin = current;
+                current = origin.Children.FirstOrDefault(c => c.Argument == name);
+                if (current is null)
+                {
+                    Log.Write($"Could not find part '{name}' of path {refinement.Argument} in source {origin}");
+                    break; //Target not present, nothing to refine.
+                }
+            }
+
+            current?.Insert(refinement.Children);
+        }
 
         return copy.Children;
     }