Formatting changes

Azure · May 3, 2021 · c1e6263 · c1e6263
1 parent 4fc7f62
commit c1e6263
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 10 deletions.
diff --git a/src/Bicep.LangServer.UnitTests/Snippets/SnippetsProviderTests.cs b/src/Bicep.LangServer.UnitTests/Snippets/SnippetsProviderTests.cs
@@ -286,10 +286,10 @@ public void GetResourceBodyCompletionSnippets_WithNoStaticTemplate_ShouldReturnS
 	location: $3
 	name: $4
 	sku: {
-	name: $5
-	properties: {
-	loadBalancerType: $6
-	}
+		name: $5
+		properties: {
+			loadBalancerType: $6
+		}
 	}
 	$0
 }");

diff --git a/src/Bicep.LangServer/Snippets/SnippetsProvider.cs b/src/Bicep.LangServer/Snippets/SnippetsProvider.cs
@@ -228,7 +228,7 @@ public IEnumerable<Snippet> GetResourceBodyCompletionSnippets(TypeSymbol typeSym
 
                 foreach (KeyValuePair<string, TypeProperty> kvp in objectType.Properties.OrderBy(x => x.Key))
                 {
-                    string? snippetText = GetSnippetText(kvp.Value, ref index);
+                    string? snippetText = GetSnippetText(kvp.Value, indentLevel: 1, ref index);
 
                     if (snippetText is not null)
                     {
@@ -251,30 +251,33 @@ public IEnumerable<Snippet> GetResourceBodyCompletionSnippets(TypeSymbol typeSym
             return null;
         }
 
-        private string? GetSnippetText(TypeProperty typeProperty, ref int index)
+        private string? GetSnippetText(TypeProperty typeProperty, int indentLevel, ref int index)
         {
             if (typeProperty.Flags.HasFlag(TypePropertyFlags.Required))
             {
                 StringBuilder sb = new StringBuilder();
 
                 if (typeProperty.TypeReference.Type is ObjectType objectType)
                 {
-                    sb.AppendLine("\t" + typeProperty.Name + ": {");
+                    sb.AppendLine(GetIndentString(indentLevel) + typeProperty.Name + ": {");
+
+                    indentLevel++;
 
                     foreach (KeyValuePair<string, TypeProperty> kvp in objectType.Properties.OrderBy(x => x.Key))
                     {
-                        string? snippetText = GetSnippetText(kvp.Value, ref index);
+                        string? snippetText = GetSnippetText(kvp.Value, indentLevel, ref index);
                         if (snippetText is not null)
                         {
                             sb.Append(snippetText);
                         }
                     }
 
-                    sb.AppendLine("\t}");
+                    indentLevel--;
+                    sb.AppendLine(GetIndentString(indentLevel) + "}");
                 }
                 else
                 {
-                    sb.AppendLine("\t" + typeProperty.Name + ": $" + (index).ToString());
+                    sb.AppendLine(GetIndentString(indentLevel) + typeProperty.Name + ": $" + (index).ToString());
                     index++;
                 }
 
@@ -284,6 +287,11 @@ public IEnumerable<Snippet> GetResourceBodyCompletionSnippets(TypeSymbol typeSym
             return null;
         }
 
+        private string GetIndentString(int indentLevel)
+        {
+            return new string('\t', indentLevel);
+        }
+
         private Snippet GetEmptySnippet()
         {
             string label = "{}";