mirror of
https://github.com/davidallendj/magellan.git
synced 2025-12-20 11:37:01 -07:00
108 lines
3.8 KiB
Go
108 lines
3.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
|
|
urlx "github.com/OpenCHAMI/magellan/internal/url"
|
|
"github.com/OpenCHAMI/magellan/pkg/crawler"
|
|
"github.com/OpenCHAMI/magellan/pkg/secrets"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// The `crawl` command walks a collection of Redfish endpoints to collect
|
|
// specfic inventory detail. This command only expects host names and does
|
|
// not require a scan to be performed beforehand.
|
|
var CrawlCmd = &cobra.Command{
|
|
Use: "crawl [uri]",
|
|
Example: ` magellan crawl https://bmc.example.com
|
|
magellan crawl https://bmc.example.com -i -u username -p password`,
|
|
Short: "Crawl a single BMC for inventory information",
|
|
Long: "Crawl a single BMC for inventory information with URI.\n\n NOTE: This command does not scan subnets, store scan information in cache, nor make a request to a specified host. It is used only to retrieve inventory data directly. Otherwise, use 'scan' and 'collect' instead.",
|
|
Args: func(cmd *cobra.Command, args []string) error {
|
|
// Validate that the only argument is a valid URI
|
|
var err error
|
|
if err := cobra.ExactArgs(1)(cmd, args); err != nil {
|
|
return err
|
|
}
|
|
args[0], err = urlx.Sanitize(args[0])
|
|
if err != nil {
|
|
return fmt.Errorf("failed to sanitize URI: %w", err)
|
|
}
|
|
return nil
|
|
},
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
uri = args[0]
|
|
store secrets.SecretStore
|
|
err error
|
|
)
|
|
// try and load credentials from local store first
|
|
store, err = secrets.OpenStore(secretsFile)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("failed to open local store...falling back to provided arguments")
|
|
// try and use the `username` and `password` arguments instead
|
|
store = secrets.NewStaticStore(username, password)
|
|
}
|
|
|
|
// found the store so try to load the creds
|
|
_, err = store.GetSecretByID(uri)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msgf("failed to get secrets for '%s'...", uri)
|
|
// if we have CLI flags set, then we want to override default stored creds
|
|
if username != "" && password != "" {
|
|
// finally, use the CLI arguments passed instead
|
|
log.Info().Msg("...using provided arguments for credentials")
|
|
store = secrets.NewStaticStore(username, password)
|
|
} else {
|
|
// try and get a default *stored* username/password
|
|
log.Info().Msg("...using default stored secrets for credentials")
|
|
secret, err := store.GetSecretByID(secrets.DEFAULT_KEY)
|
|
if err != nil {
|
|
// no default found, so use CLI arguments
|
|
log.Warn().Err(err).Msg("no default credentials found")
|
|
} else {
|
|
// found default values in local store so use them
|
|
var creds crawler.BMCUsernamePassword
|
|
err = json.Unmarshal([]byte(secret), &creds)
|
|
if err != nil {
|
|
log.Warn().Err(err).Msg("failed to unmarshal default store credentials")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
systems, err := crawler.CrawlBMCForSystems(crawler.CrawlerConfig{
|
|
URI: uri,
|
|
CredentialStore: store,
|
|
Insecure: insecure,
|
|
UseDefault: true,
|
|
})
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to crawl BMC")
|
|
}
|
|
// Marshal the inventory details to JSON
|
|
jsonData, err := json.MarshalIndent(systems, "", " ")
|
|
if err != nil {
|
|
log.Error().Err(err).Msg("failed to marshal JSON")
|
|
return
|
|
}
|
|
|
|
// Print the pretty JSON
|
|
fmt.Println(string(jsonData))
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
CrawlCmd.Flags().StringVarP(&username, "username", "u", "", "Set the username for the BMC")
|
|
CrawlCmd.Flags().StringVarP(&password, "password", "p", "", "Set the password for the BMC")
|
|
CrawlCmd.Flags().BoolVarP(&insecure, "insecure", "i", false, "Ignore SSL errors")
|
|
CrawlCmd.Flags().StringVarP(&secretsFile, "file", "f", "nodes.json", "set the secrets file with BMC credentials")
|
|
|
|
checkBindFlagError(viper.BindPFlag("crawl.insecure", CrawlCmd.Flags().Lookup("insecure")))
|
|
|
|
rootCmd.AddCommand(CrawlCmd)
|
|
}
|