CheckIfInteractive

Author: huangpf

src/Common/Commands.Common/AzurePSCmdlet.cs

@@ -225,6 +225,28 @@ protected void SaveDataCollectionProfile()
             WriteWarning(string.Format(Resources.DataCollectionSaveFileInformation, fileFullPath));
         }
 
+        protected bool CheckIfInteractive()
+        {
+            bool interactive = true;
+            try
+            {
+                var test = this.Host.UI.RawUI.CursorSize;
+            }
+            catch (HostException ex)
+            {
+                if (ex.Message.StartsWith("A command that prompts the user failed"))
+                {
+                    interactive = false;
+                }
+                else
+                {
+                    throw ex;
+                }
+            }
+
+            return interactive;
+        }
+
         /// <summary>
         /// Prompt for the current data collection profile
         /// </summary>
@@ -234,7 +256,7 @@ protected void PromptForDataCollectionProfileIfNotExists()
             // Initialize it from the environment variable or profile file.
             InitializeDataCollectionProfile();
 
-            if (GeneralUtilities.CheckIfInteractive() && !_dataCollectionProfile.EnableAzureDataCollection.HasValue)
+            if (CheckIfInteractive() && !_dataCollectionProfile.EnableAzureDataCollection.HasValue)
             {
                 WriteWarning(Resources.DataCollectionPrompt);
 
@@ -259,7 +281,7 @@ protected void PromptForDataCollectionProfileIfNotExists()
                 bool enabled = false;
                 if (this.Host.UI.RawUI.KeyAvailable)
                 {
-                    KeyInfo keyInfo = this.Host.UI.RawUI.ReadKey(ReadKeyOptions.NoEcho);
+                    KeyInfo keyInfo = this.Host.UI.RawUI.ReadKey(ReadKeyOptions.NoEcho | ReadKeyOptions.AllowCtrlC | ReadKeyOptions.IncludeKeyDown);
                     enabled = (keyInfo.Character == 'Y' || keyInfo.Character == 'y');
                 }
 

src/Common/Commands.Common/GeneralUtilities.cs

@@ -430,14 +430,5 @@ public static void EnsureDefaultProfileDirectoryExists()
                 AzureSession.DataStore.CreateDirectory(AzureSession.ProfileDirectory);
             }
         }
-
-        /// <summary>
-        /// Check if the PS program is run in interactive mode.
-        /// </summary>
-        public static bool CheckIfInteractive()
-        {
-            var arguments = Environment.GetCommandLineArgs();
-            return arguments.Any(s => !string.Equals(s, "-noninteractive"));
-        }
     }
 }