feat: Prompt for privileged user if host collectors present in spec (#1513)

* feat: Prompt for privileged user if host collectors present

* Prompt preflight checks that have host collectors

* Show cursor before prompting
This commit is contained in:
Evans Mungai
2024-03-28 11:51:19 +00:00
committed by GitHub
parent 76c52d2b93
commit e6aff48f1b
3 changed files with 67 additions and 5 deletions

View File

@@ -106,6 +106,23 @@ func runTroubleshoot(v *viper.Viper, args []string) error {
})
}
if interactive {
c := color.New()
c.Println(fmt.Sprintf("\r%s\r", cursor.ClearEntireLine()))
}
if interactive {
if len(mainBundle.Spec.HostCollectors) > 0 && !util.IsRunningAsRoot() {
msg := "Some host collectors may require elevated privileges to run.\nDo you want to exit and rerun the command as a privileged user?"
fmt.Print(cursor.Show())
if util.PromptYesNo(msg) {
fmt.Println("Exiting...")
return nil
}
fmt.Print(cursor.Hide())
}
}
var wg sync.WaitGroup
collectorCB := func(c chan interface{}, msg string) { c <- msg }
progressChan := make(chan interface{})
@@ -174,11 +191,6 @@ func runTroubleshoot(v *viper.Viper, args []string) error {
nonInteractiveOutput := analysisOutput{}
if interactive {
c := color.New()
c.Println(fmt.Sprintf("\r%s\r", cursor.ClearEntireLine()))
}
response, err := supportbundle.CollectSupportBundleFromSpec(&mainBundle.Spec, additionalRedactors, createOpts)
if err != nil {
return errors.Wrap(err, "failed to run collect and analyze process")