Update

Author: huangpf

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

@@ -22,9 +22,12 @@ param(
 
     [Parameter(Mandatory = $true)]
     [string]$ModelClassNameSpace,
+    
+    [Parameter(Mandatory = $true)]
+    [string]$FileOutputFolder,
 
     [Parameter(Mandatory = $false)]
-    [string]$FileOutputFolder = $null
+    [System.Reflection.MethodInfo]$FriendMethodInfo
 )
 
 . "$PSScriptRoot\Import-StringFunction.ps1";

src/ServiceManagement/Compute/Commands.ServiceManagement.Preview/Automation/Import-TypeFunction.ps1

@@ -408,9 +408,7 @@ function Get-VerbTermNameAndSuffix
         [string]$MethodName
     )
 
-    $verb = $MethodName;
-    $suffix = $null;
-
+    $found = $false;
     foreach ($key in $common_verb_mapping.Keys)
     {
         if ($MethodName.StartsWith($key))
@@ -426,11 +424,17 @@ function Get-VerbTermNameAndSuffix
             {
                 $suffix += "WithDeallocation";
             }
-
+            $found = $true;
             break;
         }
     }
 
+    if (-not $found)
+    {
+        $verb = "Invoke";
+        $suffix = $MethodName;
+    }
+
     Write-Output $verb;
     Write-Output $suffix;
 }
@@ -698,3 +702,44 @@ function Get-ProperTypeName
 
     return $typeStr;
 }
+
+# Check if 2 methods have the same parameter list
+function CheckIf-SameParameterList
+{
+    param
+    (
+        [Parameter(Mandatory = $true)]
+        [System.Reflection.MethodInfo]$methodInfo,
+        
+        [Parameter(Mandatory = $true)]
+        [System.Reflection.MethodInfo]$friendMethodInfo
+    )
+
+    if ($methodInfo -eq $null -or $friendMethodInfo -eq $null)
+    {
+        return $false;
+    }
+
+    $myParams = $methodInfo.GetParameters();
+    $friendParams = $friendMethodInfo.GetParameters();
+    if ($myParams.Count -ne $friendParams.Count)
+    {
+        return $false;
+    }
+
+    for ($i = 0; $i -lt $myParams.Count -and $i -lt $friendParams.Count; $i++)
+    {
+        [System.Reflection.ParameterInfo]$paramInfo = $myParams[$i];
+        [System.Reflection.ParameterInfo]$friendInfo = $friendParams[$i];
+        if ($paramInfo.Name -ne $friendInfo.Name)
+        {
+            return $false;
+        }
+        elseif (-not $paramInfo.ParameterType.IsEquivalentTo($friendInfo.ParameterType))
+        {
+            return $false;
+        }
+    }
+
+    return $true;
+}

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

@@ -1172,6 +1172,7 @@ else
 
         $qualified_methods = @();
         $total_method_count = 0;
+        $friendMethodDict = @{};
         foreach ($mtItem in $methods)
         {
             [System.Reflection.MethodInfo]$methodInfo = $mtItem;
@@ -1190,6 +1191,23 @@ else
 
             $qualified_methods += $mtItem;
             $total_method_count++;
+
+            # Handle Friend Methods
+            if ($mtItem.Name -eq 'Deallocate' -and (-not $friendMethodDict.ContainsKey($mtItem.Name)))
+            {
+                $methods2 = Get-OperationMethods $operation_type;
+                foreach ($friendMethodInfo in $methods2)
+                {
+                    if ($friendMethodInfo.Name -eq 'PowerOff')
+                    {
+                        if (CheckIf-SameParameterList $methodInfo $friendMethodInfo)
+                        {
+                            $friendMethodDict.Add($mtItem.Name, $friendMethodInfo);
+                            break;
+                        }
+                    }
+                }
+            }
         }
 
         $method_count = 0;
@@ -1218,10 +1236,12 @@ else
             }
             Write-Verbose $SEC_LINE;
 
+            $friendMethodInfo = $friendMethodDict[$methodInfo.Name];
             $outputs = (. $PSScriptRoot\Generate-FunctionCommand.ps1 -OperationName $opShortName `
                                                                      -MethodInfo $methodInfo `
                                                                      -ModelClassNameSpace $client_model_namespace `
-                                                                     -FileOutputFolder $opOutFolder);
+                                                                     -FileOutputFolder $opOutFolder `
+                                                                     -FriendMethodInfo $friendMethodInfo);
 
             if ($outputs.Count -ne $null)
             {