mirror of
https://github.com/davidallendj/magellan.git
synced 2025-12-20 03:27:03 -07:00
101 lines
2.9 KiB
Go
101 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/OpenCHAMI/magellan/pkg/jaws"
|
|
"github.com/OpenCHAMI/magellan/pkg/pdu"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func transformToSMDFormat(inventory *pdu.PDUInventory) []map[string]any {
|
|
smdRecords := make([]map[string]any, 0)
|
|
|
|
rtsHostname := fmt.Sprintf("%s-rts:8083", inventory.Hostname)
|
|
pduBank := "B"
|
|
|
|
for _, outlet := range inventory.Outlets {
|
|
smdID := fmt.Sprintf("%sp1v%s", inventory.Hostname, outlet.ID)
|
|
odataID := fmt.Sprintf("/redfish/v1/PowerEquipment/RackPDUs/%s/Outlets/%s", pduBank, outlet.ID)
|
|
redfishURL := fmt.Sprintf("%s%s", rtsHostname, odataID)
|
|
powerControlTarget := fmt.Sprintf("%s/Actions/Outlet.PowerControl", odataID)
|
|
|
|
record := map[string]any{
|
|
"ID": smdID,
|
|
"Type": "CabinetPDUPowerConnector",
|
|
"RedfishType": "Outlet",
|
|
"RedfishSubtype": "Cx",
|
|
"OdataID": odataID,
|
|
"RedfishEndpointID": inventory.Hostname,
|
|
"Enabled": true,
|
|
"RedfishEndpointFQDN": rtsHostname,
|
|
"RedfishURL": redfishURL,
|
|
"ComponentEndpointType": "ComponentEndpointOutlet",
|
|
"RedfishOutletInfo": map[string]any{
|
|
"Name": outlet.Name,
|
|
"Actions": map[string]any{
|
|
"#Outlet.PowerControl": map[string]any{
|
|
"PowerState@Redfish.AllowableValues": []string{"On", "Off"},
|
|
"target": powerControlTarget,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
smdRecords = append(smdRecords, record)
|
|
}
|
|
return smdRecords
|
|
}
|
|
|
|
var pduCollectCmd = &cobra.Command{
|
|
Use: "collect [hosts...]",
|
|
Short: "Collect inventory from JAWS-based PDUs",
|
|
Long: `Connects to one or more PDUs with a JAWS interface to collect hardware inventory.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
if len(args) == 0 {
|
|
log.Error().Msg("no PDU hosts provided")
|
|
return
|
|
}
|
|
|
|
if username == "" || password == "" {
|
|
log.Error().Msg("--username and --password are required for PDU collection")
|
|
return
|
|
}
|
|
|
|
allSmdRecords := make([]map[string]any, 0)
|
|
|
|
for _, host := range args {
|
|
log.Info().Msgf("Collecting from PDU: %s", host)
|
|
config := jaws.CrawlerConfig{
|
|
URI: host,
|
|
Username: username,
|
|
Password: password,
|
|
Insecure: true,
|
|
}
|
|
|
|
inventory, err := jaws.CrawlPDU(config)
|
|
if err != nil {
|
|
log.Error().Err(err).Msgf("failed to crawl PDU %s", host)
|
|
continue
|
|
}
|
|
|
|
smdRecords := transformToSMDFormat(inventory)
|
|
|
|
allSmdRecords = append(allSmdRecords, smdRecords...)
|
|
}
|
|
|
|
output, err := json.MarshalIndent(allSmdRecords, "", " ")
|
|
if err != nil {
|
|
log.Error().Err(err).Msgf("failed to marshal SMD records to JSON")
|
|
}
|
|
fmt.Println(string(output))
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
PduCmd.AddCommand(pduCollectCmd)
|
|
|
|
pduCollectCmd.Flags().StringVarP(&username, "username", "u", "", "Set the PDU username")
|
|
pduCollectCmd.Flags().StringVarP(&password, "password", "p", "", "Set the PDU password")
|
|
}
|