Unique Attribute

westermo · May 29, 2024 · debf68d · debf68d
1 parent 0ab9731
commit debf68d
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 8 deletions.
diff --git a/YangParser/Generator/YangGenerator.cs b/YangParser/Generator/YangGenerator.cs
@@ -1,7 +1,6 @@
 using System;
 using System.Collections.Generic;
 using System.Collections.Immutable;
-using System.IO;
 using System.Linq;
 using Microsoft.CodeAnalysis;
 using Microsoft.CodeAnalysis.Text;
@@ -355,6 +354,11 @@ public class WhenAttribute(string xPath) : Attribute
                            public string XPath { get; } = xPath;
                        }
                        [AttributeUsage(AttributeTargets.All, AllowMultiple = true)]
+                       public class MustAttribute(string xPath) : Attribute
+                       {
+                           public string XPath { get; } = xPath;
+                       }
+                       [AttributeUsage(AttributeTargets.All, AllowMultiple = true)]
                        public class TargetAttribute(string xPath) : Attribute
                        {
                            public string XPath { get; } = xPath;
@@ -365,6 +369,11 @@ public class KeyAttribute(params string[] value) : Attribute
                            public string[] Value { get; } = value;
                        }
                        [AttributeUsage(AttributeTargets.All, AllowMultiple = false)]
+                       public class UniqueAttribute(params string[] value) : Attribute
+                       {
+                           public string[] Value { get; } = value;
+                       }
+                       [AttributeUsage(AttributeTargets.All, AllowMultiple = false)]
                        public class MinElementsAttribute(int value) : Attribute
                        {
                            public int Value { get; } = value;

diff --git a/YangParser/SemanticModel/Container.cs b/YangParser/SemanticModel/Container.cs
@@ -57,7 +57,7 @@ public override string ToCode()
     {
         var nodes = Children.Select(child => child.ToCode()).ToArray();
         var name = Argument;
-        if (Parent.Argument == Argument)
+        if (Parent!.Argument == Argument)
         {
             name = "sub-" + name;
         }

diff --git a/YangParser/SemanticModel/Must.cs b/YangParser/SemanticModel/Must.cs
@@ -10,14 +10,21 @@ public Must(YangStatement statement) : base(statement)
     {
         if (statement.Keyword != Keyword)
             throw new SemanticError($"Non-matching Keyword '{statement.Keyword}', expected {Keyword}", statement);
-
     }
 
     public const string Keyword = "must";
-    public override ChildRule[] PermittedChildren { get; } = [
+
+    public override ChildRule[] PermittedChildren { get; } =
+    [
         new ChildRule(Description.Keyword),
         new ChildRule(Reference.Keyword),
         new ChildRule(ErrorAppTag.Keyword),
         new ChildRule(ErrorMessage.Keyword)
-        ];
-}
+    ];
+
+    public override string ToCode()
+    {
+        Parent?.Attributes.Add($"Must(\"{SingleLine(Argument).Replace("\"", "\\\"")}\")");
+        return string.Empty;
+    }
+}
diff --git a/YangParser/SemanticModel/Unique.cs b/YangParser/SemanticModel/Unique.cs
@@ -10,12 +10,19 @@ public Unique(YangStatement statement) : base(statement)
     {
         if (statement.Keyword != Keyword)
             throw new SemanticError($"Non-matching Keyword '{statement.Keyword}', expected {Keyword}", statement);
-        
+
         ValidateChildren(statement);
-        Identifiers = Argument.Split(' ');
+        Identifiers = Argument.Split(' ', '\n', '\t');
     }
 
     public const string Keyword = "unique";
 
     public string[] Identifiers { get; }
+
+    public override string ToCode()
+    {
+        Parent?.Attributes.Add(
+            $"Unique({string.Join(",", Identifiers.Select(i => $"nameof({MakeName(SingleLine(i).Replace("\"", "\\\""))})"))})");
+        return string.Empty;
+    }
 }