sailpoint-cli/cmd/va/troubleshoot.go

package va

import (
	"errors"
	"fmt"
	"sync"
	"time"

	"os"
	"path"

	"github.com/fatih/color"
	"github.com/sailpoint-oss/sailpoint-cli/internal/terminal"
	"github.com/sailpoint-oss/sailpoint-cli/internal/va"
	"github.com/spf13/cobra"
	"github.com/vbauerster/mpb/v8"
)

func NewTroubleshootCmd(term terminal.Terminal) *cobra.Command {
	var output string
	cmd := &cobra.Command{
		Use:     "troubleshoot",
		Short:   "Perform troubleshooting operations against a virtual appliance",
		Long:    "\nPerform troubleshooting operations against a virtual appliance\n\n",
		Example: "sail va troubleshoot 10.10.10.10",
		Args:    cobra.MinimumNArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {

			if output == "" {
				output, _ = os.Getwd()
			}

			var credentials []string
			for credential := 0; credential < len(args); credential++ {
				password, _ := term.PromptPassword(fmt.Sprintf("Enter Password for %v:", args[credential]))
				credentials = append(credentials, password)
			}

			for index, endpoint := range args {
				outputDir := path.Join(output, endpoint)

				if _, err := os.Stat(outputDir); errors.Is(err, os.ErrNotExist) {
					err := os.MkdirAll(outputDir, 0700)
					if err != nil {
						return err
					}
				}

				password := credentials[index]

				orgErr := va.RunVACmdLive(endpoint, password, TroubleshootingScript)
				if orgErr != nil {
					return orgErr
				}

				var wg sync.WaitGroup
				p := mpb.New(mpb.WithWidth(60),
					mpb.PopCompletedMode(),
					mpb.WithRefreshRate(180*time.Millisecond),
					mpb.WithWaitGroup(&wg))

				err := va.CollectVAFiles(endpoint, password, outputDir, []string{"/home/sailpoint/stuntlog.txt"}, p)
				if err != nil {
					return err
				}

				color.Green("stuntlog copied to %s", outputDir)
			}

			return nil

		}}

	cmd.Flags().StringP("endpoint", "e", "", "Host to troubleshoot")
	cmd.Flags().StringVarP(&output, "output", "o", "", "Path to save the log file")

	return cmd

}