Fixes to invoke-action

Author: chadiel

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Components/LongRunningOperationHelper.cs

@@ -76,7 +76,7 @@ internal LongRunningOperationHelper(string activityName, Func<ResourceManagerRes
         /// Waits for the operation to complete.
         /// </summary>
         /// <param name="operationResult">The operation result.</param>
-        internal JToken WaitOnOperation(OperationResult operationResult)
+        internal string WaitOnOperation(OperationResult operationResult)
         {
             // TODO: Re-factor this mess.
             this.ProgressTrackerObject.UpdateProgress("Starting", 0);
@@ -161,10 +161,11 @@ private TrackingOperationResult HandleOperationResponse(OperationResult operatio
         /// <param name="operationResult">The result of the operation.</param>
         private TrackingOperationResult HandleCreateOrUpdateResponse(OperationResult operationResult)
         {
-            var resource = operationResult.Value == null
-                ? null
-                : operationResult.Value
-                    .ToObject<Resource<InsensitiveDictionary<JToken>>>(JsonExtensions.JsonObjectTypeSerializer);
+            Resource<InsensitiveDictionary<JToken>> resource;
+            if (!operationResult.Value.TryConvertTo<Resource<InsensitiveDictionary<JToken>>>(out resource))
+            {
+                return null;
+            }
 
             if(resource == null && operationResult.HttpStatusCode == HttpStatusCode.Created)
             {
@@ -444,11 +445,12 @@ private string GetAzureAsyncOperationState(OperationResult operationResult)
             /// <param name="operationResult">The operation result.</param>
             private static string GetResourceState(OperationResult operationResult)
             {
-                var resource = operationResult.Value == null
-                    ? null
-                    : operationResult.Value
-                        .ToObject<Resource<InsensitiveDictionary<JToken>>>(JsonExtensions.JsonObjectTypeSerializer);
-
+                Resource<InsensitiveDictionary<JToken>> resource;
+                if (!operationResult.Value.TryConvertTo<Resource<InsensitiveDictionary<JToken>>>(out resource))
+                {
+                    return null;
+                }
+                
                 if (resource == null)
                 {
                     return null;

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Components/OperationResult.cs

@@ -56,6 +56,6 @@ public class OperationResult
         /// <summary>
         /// Gets or sets the value.
         /// </summary>
-        public JObject Value { get; set; }
+        public string Value { get; set; }
     }
 }

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Components/ResourceIdUtility.cs

@@ -40,7 +40,7 @@ public static string GetResourceId(string resourceId, string extensionResourceTy
                 resourceIdBuilder.Append(ResourceIdUtility.ProcessResourceTypeAndName(resourceType: extensionResourceType, resourceName: extensionResourceName));
             }
 
-            return resourceId.ToString();
+            return resourceIdBuilder.ToString();
         }
 
         /// <summary>

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Extensions/HttpMessageExtensions.cs

@@ -50,5 +50,36 @@ public static async Task<T> ReadContentAsJsonAsync<T>(this HttpResponseMessage m
                 }
             }
         }
+
+            {
+        /// <summary>
+        /// Reads the JSON content from the http response message.
+        /// </summary>
+        /// <typeparam name="T">The type of object contained in the JSON.</typeparam>
+        /// <param name="message">The response message to be read.</param>
+        /// <param name="rewindContentStream">Rewind content stream if set to true.</param>
+        /// <returns>An object of type T instantiated from the response message's body.</returns>
+        public static async Task<string> ReadContentAsStringAsync(this HttpResponseMessage message, bool rewindContentStream = false)
+        {
+            using (var stream = await message.Content
+                .ReadAsStreamAsync()
+                .ConfigureAwait(continueOnCapturedContext: false))
+            using (var streamReader = new StreamReader(stream))
+            {
+                var streamPosition = stream.Position;
+                try
+                {
+                    
+                    return streamReader.ReadToEnd();
+                }
+                finally
+                {
+                    if (stream.CanSeek && streamPosition != stream.Position && rewindContentStream)
+                    {
+                        stream.Seek(streamPosition, SeekOrigin.Begin);
+                    }
+                }
+            }
+        }
     }
 }

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Extensions/JsonExtensions.cs

@@ -147,6 +147,32 @@ public static bool CanConvertTo<TType>(this JToken jobject)
             return jobject.TryConvertTo(out ignored);
         }
 
+        /// <summary>
+        /// Checks if a conversion from the supplied <see cref="JToken"/> to a <typeparamref name="TType"/> can be made.
+        /// </summary>
+        /// <typeparam name="TType">The type to convert to.</typeparam>
+        /// <param name="str">The string.</param>
+        /// <param name="result">The result.</param>
+        public static bool TryConvertTo<TType>(this string str, out TType result)
+        {
+            JToken tmp = null;
+            try
+            {
+                tmp = str.ToJToken();
+            }
+            catch (FormatException)
+            {
+            }
+            catch (ArgumentException)
+            {
+            }
+            catch (JsonException)
+            {
+            }
+
+            return tmp.TryConvertTo<TType>(out result);
+        }
+
         /// <summary>
         /// Checks if a conversion from the supplied <see cref="JToken"/> to a <typeparamref name="TType"/> can be made.
         /// </summary>

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Implementation/InvokeAzureResourceActionCmdlet.cs

@@ -80,10 +80,10 @@ protected override void OnProcessRecord()
                             odataQuery: this.ODataQuery);
 
                     var activity = string.Format("POST {0}", managementUri.PathAndQuery);
-                    var result = this.GetLongRunningOperationTracker(activityName: activity, isResourceCreateOrUpdate: false)
+                    var resultString = this.GetLongRunningOperationTracker(activityName: activity, isResourceCreateOrUpdate: false)
                         .WaitOnOperation(operationResult: operationResult);
 
-                    this.WriteObject(result, ResourceObjectFormat.New);
+                    this.TryConvertAndWriteObject(resultString);
                 });
         }
 

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Implementation/MoveAzureResourceCmdlet.cs

@@ -173,7 +173,7 @@ private void RunCmdlet()
                                 isResourceCreateOrUpdate: false)
                             .WaitOnOperation(operationResult: operationResult);
 
-                        this.WriteObject(result, ResourceObjectFormat.New);
+                        this.TryConvertAndWriteObject(result, ResourceObjectFormat.New);
                     }
                     else
                     {

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Implementation/NewAzureResourceCmdlet.cs

@@ -129,7 +129,7 @@ protected override void OnProcessRecord()
                     var result = this.GetLongRunningOperationTracker(activityName: activity, isResourceCreateOrUpdate: true)
                         .WaitOnOperation(operationResult: operationResult);
 
-                    this.WriteObject(result, this.OutputObjectFormat.Value);
+                    this.TryConvertAndWriteObject(result, this.OutputObjectFormat.Value);
                 });
         }
 

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Implementation/RemoveAzureResourceCmdlet.cs

@@ -60,7 +60,7 @@ protected override void OnProcessRecord()
                     var result = this.GetLongRunningOperationTracker(activityName: activity, isResourceCreateOrUpdate: false)
                         .WaitOnOperation(operationResult: operationResult);
 
-                    this.WriteObject(result, ResourceObjectFormat.New);
+                    this.TryConvertAndWriteObject(result, ResourceObjectFormat.New);
                 });
         }
     }

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Implementation/ResourceManagerCmdletBase.cs

@@ -259,6 +259,24 @@ public ResourceManagerRestRestClient GetResourcesClient()
                         headerValues: AzureSession.ClientFactory.UserAgents));
         }
 
+        /// <summary>
+        /// Writes the object
+        /// </summary>
+        /// <param name="resultString">The result as a string</param>
+        /// <param name="objectFormat">The <see cref="ResourceObjectFormat"/></param>
+        protected void TryConvertAndWriteObject(string resultString, ResourceObjectFormat objectFormat)
+        {
+            JToken resultJToken;
+            if (resultString.TryConvertTo<JToken>(out resultJToken))
+            {
+                this.WriteObject(resultJToken, objectFormat);
+            }
+            else
+            {
+                this.WriteObject(resultString);
+            }
+        }
+
         /// <summary>
         /// Writes a <see cref="JToken"/> object as a <see cref="PSObject"/>.
         /// </summary>

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/Implementation/SetAzureResourceCmdlet.cs

@@ -124,7 +124,7 @@ protected override void OnProcessRecord()
                     var result = this.GetLongRunningOperationTracker(activityName: activity, isResourceCreateOrUpdate: true)
                         .WaitOnOperation(operationResult: operationResult);
 
-                    this.WriteObject(result, this.OutputObjectFormat.Value);
+                    this.TryConvertAndWriteObject(result, this.OutputObjectFormat.Value);
                 });
         }
 

src/ResourceManager/ResourceManager/Commands.ResourceManager/Cmdlets/RestClients/ResourceManagerRestClientBase.cs

@@ -293,7 +293,7 @@ private async Task<OperationResult> GetOperationResult(HttpResponseMessage respo
             var operationResult = new OperationResult
             {
                 Value = await response
-                    .ReadContentAsJsonAsync<JObject>()
+                    .ReadContentAsStringAsync()
                     .ConfigureAwait(continueOnCapturedContext: false),
             };