Add -Force, use DataStore functions to write file

Author: Aman Orazaev

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Dataplane/Commands/Export-AzureAsInstanceLog.cs

@@ -14,12 +14,14 @@
 
 using System;
 using System.IO;
+using System.Globalization;
 using System.Linq;
 using System.Management.Automation;
 using System.Management.Instrumentation;
 using System.Net.Http;
 using System.Net.Http.Headers;
 using System.Threading.Tasks;
+using Microsoft.Azure.Commands.Common.Authentication;
 using Microsoft.Azure.Commands.AnalysisServices.Dataplane.Models;
 using Microsoft.Azure.Commands.AnalysisServices.Dataplane.Properties;
 using Microsoft.Azure.Commands.Common.Authentication.Abstractions;
@@ -30,6 +32,7 @@
 
 namespace Microsoft.Azure.Commands.AnalysisServices.Dataplane
 {
+
     /// <summary>
     /// Cmdlet to export an Analysis Services server log to file
     /// </summary>
@@ -48,6 +51,10 @@ public class ExportAzureAnalysisServerLog : AzurePSCmdlet
         [ValidateNotNullOrEmpty]
         public string OutputPath { get; set; }
 
+        [Parameter(Mandatory = false, HelpMessage = "Overwrite file if exists")]
+        [ValidateNotNullOrEmpty]
+        public SwitchParameter Force { get; set; }
+
         public IAsAzureHttpClient AsAzureHttpClient { get; private set; }
 
         public ITokenCacheItemProvider TokenCacheItemProvider { get; private set; }
@@ -159,10 +166,15 @@ public override void ExecuteCmdlet()
                 using (HttpResponseMessage message = AsAzureHttpClient.CallGetAsync(
                     resolvedUriBuilder.Uri,
                     logfileEndpoint,
-                    accessToken).Result)
+                    accessToken).ConfigureAwait(false).GetAwaiter().GetResult())
                 {
                     message.EnsureSuccessStatusCode();
-                    File.WriteAllText(this.OutputPath, message.Content.ReadAsStringAsync().Result);
+                    string actionWarning = string.Format(CultureInfo.CurrentCulture, Resources.ExportingLogOverwriteWarning, this.OutputPath);
+                    if (AzureSession.Instance.DataStore.FileExists(this.OutputPath) && !this.Force.IsPresent && !ShouldContinue(actionWarning, Resources.Confirm))
+                    {
+                        return;
+                    }
+                    AzureSession.Instance.DataStore.WriteFile(this.OutputPath, message.Content.ReadAsStringAsync().Result);
                 }
             }
         }

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Dataplane/Properties/Resources.Designer.cs

@@ -132,6 +132,12 @@
   <data name="ExportingLogFromAnalysisServicesServer" xml:space="preserve">
     <value>Export log from Analysis Services server</value>
   </data>
+  <data name="ExportingLogOverwriteWarning" xml:space="preserve">
+    <value>Writing log to file {0}</value>
+  </data>
+  <data name="Confirm" xml:space="preserve">
+    <value>Confirm</value>
+  </data>
   <data name="UnknownEnvironment" xml:space="preserve">
     <value>Unknown Analysis Services environment name specified.</value>
   </data>

src/ResourceManager/AnalysisServices/Commands.AnalysisServices.Dataplane/Properties/Resources.resx

@@ -12,7 +12,7 @@ Exports a log from an instance of Analysis Services server in the currently logg
 ## SYNTAX
 
 ```
-Export-AzureAnalysisServicesInstanceLog [-Instance] <String> [-OutputPath] <String> [-WhatIf]
+Export-AzureAnalysisServicesInstanceLog [-Instance] <String> [-OutputPath] <String> [-WhatIf] [-Force]
 ```
 
 ## DESCRIPTION
@@ -60,6 +60,21 @@ Accept pipeline input: False
 Accept wildcard characters: False
 ```
 
+### -Force
+Overwrite file if exists without asking
+
+```yaml
+Type: SwitchParameter
+Parameter Sets: (All)
+Aliases:
+
+Required: False
+Position: Named
+Default value: None
+Accept pipeline input: False
+Accept wildcard characters: False
+```
+
 ## INPUTS
 
 ## OUTPUTS