mirror of
https://github.com/davidallendj/magellan.git
synced 2025-12-20 03:27:03 -07:00
110 lines
4.1 KiB
Go
110 lines
4.1 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os/user"
|
|
|
|
magellan "github.com/OpenCHAMI/magellan/internal"
|
|
"github.com/OpenCHAMI/magellan/internal/api/smd"
|
|
"github.com/OpenCHAMI/magellan/internal/db/sqlite"
|
|
"github.com/OpenCHAMI/magellan/internal/log"
|
|
"github.com/cznic/mathutil"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
var (
|
|
forceUpdate bool
|
|
)
|
|
|
|
// The `collect` command fetches data from a collection of BMC nodes.
|
|
// This command should be ran after the `scan` to find available hosts
|
|
// on a subnet.
|
|
var collectCmd = &cobra.Command{
|
|
Use: "collect",
|
|
Short: "Collect system information by interrogating BMC node",
|
|
Long: "Send request(s) to a collection of hosts running Redfish services found stored from the 'scan' in cache.\n" +
|
|
"See the 'scan' command on how to perform a scan.\n\n" +
|
|
"Examples:\n" +
|
|
" magellan collect --cache ./assets.db --output ./logs --timeout 30 --cacert cecert.pem\n" +
|
|
" magellan collect --host smd.example.com --port 27779 --username username --password password",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
// make application logger
|
|
l := log.NewLogger(logrus.New(), logrus.DebugLevel)
|
|
|
|
// get probe states stored in db from scan
|
|
probeStates, err := sqlite.GetProbeResults(cachePath)
|
|
if err != nil {
|
|
l.Log.Errorf("failed toget states: %v", err)
|
|
}
|
|
|
|
// try to load access token either from env var, file, or config if var not set
|
|
if accessToken == "" {
|
|
var err error
|
|
accessToken, err = LoadAccessToken()
|
|
if err != nil {
|
|
l.Log.Errorf("failed to load access token: %v", err)
|
|
}
|
|
}
|
|
|
|
if verbose {
|
|
fmt.Printf("access token: %v\n", accessToken)
|
|
}
|
|
|
|
//
|
|
if concurrency <= 0 {
|
|
concurrency = mathutil.Clamp(len(probeStates), 1, 255)
|
|
}
|
|
q := &magellan.QueryParams{
|
|
Username: username,
|
|
Password: password,
|
|
Protocol: protocol,
|
|
Timeout: timeout,
|
|
Concurrency: concurrency,
|
|
Verbose: verbose,
|
|
CaCertPath: cacertPath,
|
|
OutputPath: outputPath,
|
|
ForceUpdate: forceUpdate,
|
|
AccessToken: accessToken,
|
|
}
|
|
err = magellan.CollectAll(&probeStates, l, q)
|
|
if err != nil {
|
|
l.Log.Errorf("failed to collect data: %v", err)
|
|
}
|
|
|
|
// add necessary headers for final request (like token)
|
|
headers := make(map[string]string)
|
|
if q.AccessToken != "" {
|
|
headers["Authorization"] = "Bearer " + q.AccessToken
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
currentUser, _ = user.Current()
|
|
collectCmd.PersistentFlags().StringVar(&smd.Host, "host", smd.Host, "set the host to the SMD API")
|
|
collectCmd.PersistentFlags().IntVarP(&smd.Port, "port", "p", smd.Port, "set the port to the SMD API")
|
|
collectCmd.PersistentFlags().StringVar(&username, "username", "", "set the BMC user")
|
|
collectCmd.PersistentFlags().StringVar(&password, "password", "", "set the BMC password")
|
|
collectCmd.PersistentFlags().StringVar(&protocol, "protocol", "https", "set the protocol used to query")
|
|
collectCmd.PersistentFlags().StringVarP(&outputPath, "output", "o", fmt.Sprintf("/tmp/%smagellan/data/", currentUser.Username+"/"), "set the path to store collection data")
|
|
collectCmd.PersistentFlags().BoolVar(&forceUpdate, "force-update", false, "set flag to force update data sent to SMD")
|
|
collectCmd.PersistentFlags().StringVar(&cacertPath, "cacert", "", "path to CA cert. (defaults to system CAs)")
|
|
|
|
// set flags to only be used together
|
|
collectCmd.MarkFlagsRequiredTogether("username", "password")
|
|
|
|
viper.BindPFlag("collect.driver", collectCmd.Flags().Lookup("driver"))
|
|
viper.BindPFlag("collect.host", collectCmd.Flags().Lookup("host"))
|
|
viper.BindPFlag("collect.port", collectCmd.Flags().Lookup("port"))
|
|
viper.BindPFlag("collect.username", collectCmd.Flags().Lookup("username"))
|
|
viper.BindPFlag("collect.password", collectCmd.Flags().Lookup("password"))
|
|
viper.BindPFlag("collect.protocol", collectCmd.Flags().Lookup("protocol"))
|
|
viper.BindPFlag("collect.output", collectCmd.Flags().Lookup("output"))
|
|
viper.BindPFlag("collect.force-update", collectCmd.Flags().Lookup("force-update"))
|
|
viper.BindPFlag("collect.cacert", collectCmd.Flags().Lookup("secure-tls"))
|
|
viper.BindPFlags(collectCmd.Flags())
|
|
|
|
rootCmd.AddCommand(collectCmd)
|
|
}
|