Add telemetry

Author: wangzelin007

src/azure-cli-core/azure/cli/core/commands/__init__.py

@@ -702,20 +702,32 @@ def _what_if(self, args):
         if '--what-if' in args:
             logger.debug("Entering what-if mode")
 
+            # Remove both --what-if and --export-bicep from args for processing
+            clean_args = [arg for arg in args if arg not in ['--what-if', '--export-bicep']]
+            command_parts = [arg for arg in clean_args if not arg.startswith('-') and arg != 'az']
+            command_name = ' '.join(command_parts) if command_parts else 'unknown'
+            safe_params = AzCliCommandInvoker._extract_parameter_names(args)
+            
+            # Set command details first so telemetry knows which command was attempted
+            telemetry.set_command_details(
+                command_name + ' --what-if',
+                self.data.get('output', 'json'),
+                safe_params
+            )
+            
             # Check if command is in whitelist
             if not self._is_command_supported_for_what_if(args):
                 error_msg = ("\"--what-if\" argument is not supported for this command.")
                 logger.error(error_msg)
+                telemetry.set_user_fault(summary='what-if-unsupported-command')
                 return CommandResultItem(None, exit_code=1, error=CLIError(error_msg))
 
             from azure.cli.core.what_if import show_what_if
+
+            # Check if --export-bicep is present
+            export_bicep = '--export-bicep' in args
+
             try:
-                # Check if --export-bicep is present
-                export_bicep = '--export-bicep' in args
-                
-                # Remove both --what-if and --export-bicep from args for processing
-                clean_args = [arg for arg in args if arg not in ['--what-if', '--export-bicep']]
-                
                 if export_bicep:
                     logger.debug("Export bicep mode enabled")
 
@@ -752,6 +764,8 @@ def _what_if(self, args):
                 # Default to 'json' if not already set
                 if 'output' not in self.cli_ctx.invocation.data or self.cli_ctx.invocation.data['output'] is None:
                     self.cli_ctx.invocation.data['output'] = 'json'
+                
+                telemetry.set_success(summary='what-if-completed')
 
                 # Return the formatted what-if output as the result
                 # Similar to the normal flow in execute() method
@@ -764,6 +778,8 @@ def _what_if(self, args):
             except (CLIError, ValueError, KeyError) as ex:
                 # If what-if service fails, still show an informative message
                 logger.error("What-if preview failed: %s", str(ex))
+                telemetry.set_exception(ex, fault_type='what-if-error', summary=str(ex)[:100])
+                telemetry.set_failure(summary='what-if-failed')
                 return CommandResultItem(None, exit_code=1,
                                          error=CLIError(f'What-if preview failed: {str(ex)}'))