Check StringList Type

Author: huangpf

src/ServiceManagement/Compute/Commands.ServiceManagement.Preview/Automation/Create-ParameterTree.ps1

@@ -35,6 +35,7 @@ function New-ParameterTreeNode
     $node | Add-Member -Type NoteProperty -Name Parent -Value $Parent;
     $node | Add-Member -Type NoteProperty -Name IsListItem -Value $false;
     $node | Add-Member -Type NoteProperty -Name AllStrings -Value $false;
+    $node | Add-Member -Type NoteProperty -Name OneStringList -Value $false;
     $node | Add-Member -Type NoteProperty -Name Properties -Value @();
     $node | Add-Member -Type NoteProperty -Name SubNodes -Value @();
 
@@ -72,6 +73,10 @@ function Create-ParameterTreeImpl
         {
             $treeNode.AllStrings = $true;
         }
+        elseif (Contains-OnlyStringList $TypeInfo)
+        {
+            $treeNode.OneStringList = $true;
+        }
 
         $padding = ($Depth.ToString() + (' ' * (4 * ($Depth + 1))));
         if ($Depth -gt 0)
@@ -83,6 +88,10 @@ function Create-ParameterTreeImpl
         {
             $annotation = " *";
         }
+        elseif ($treeNode.OneStringList)
+        {
+            $annotation = " ^";
+        }
 
         Write-Verbose ($padding + "[ Node ] " + $treeNode.Name + $annotation);
         Write-Verbose ($padding + "[Parent] " + $Parent.Name);

src/ServiceManagement/Compute/Commands.ServiceManagement.Preview/Automation/Generate-FunctionCommand.ps1

@@ -56,6 +56,7 @@ function Generate-CliFunctionCommandImpl
     $methodParamNameList = @();
     $methodParamTypeDict = @{};
     $allStringFieldCheck = @{};
+    $oneStringListCheck = @{};
 
     # 3. CLI Code
     # 3.1 Types
@@ -69,6 +70,8 @@ function Generate-CliFunctionCommandImpl
             $methodParamTypeDict.Add($paramItem.Name, $paramType);
             $allStringFields = Contains-OnlyStringFields $paramType;
             $allStringFieldCheck.Add($paramItem.Name, $allStringFields);
+            $oneStringList = Contains-OnlyStringList $paramType;
+            $oneStringListCheck.Add($paramItem.Name, $oneStringList);
 
             if ($paramType.Namespace -like $ModelNameSpace)
             {

src/ServiceManagement/Compute/Commands.ServiceManagement.Preview/Automation/ParameterTypeHelper.ps1

@@ -35,4 +35,36 @@ function Contains-OnlyStringFields
     }
 
     return $result;
-}
+}
+
+function Contains-OnlyStringList
+{
+    param(
+        [Parameter(Mandatory = $True)]
+        [System.Type]$parameterType
+    )
+
+    if ($parameterType -eq $null)
+    {
+        return $false;
+    }
+
+    if ($parameterType.GetProperties().Count -ne 1)
+    {
+        return $false;
+    }
+    else
+    {
+        [System.Reflection.PropertyInfo]$propInfoItem = ($parameterType.GetProperties())[0];
+        if ($propInfoItem.PropertyType.FullName.StartsWith("System.Collections.Generic.IList"))
+        {
+            [System.Type]$itemType = $propInfoItem.PropertyType.GenericTypeArguments[0];
+            if ($itemType.IsEquivalentTo([string]))
+            {
+                return $true;
+            }
+        }
+
+        return $false;
+    }
+}