diff --git a/cmd/pdu-collect.go b/cmd/pdu-collect.go index 6b4f1e8..0eb6753 100644 --- a/cmd/pdu-collect.go +++ b/cmd/pdu-collect.go @@ -10,6 +10,8 @@ import ( "github.com/spf13/cobra" ) +var mock bool + func transformToSMDFormat(inventory *pdu.PDUInventory) []map[string]any { smdRecords := make([]map[string]any, 0) @@ -53,6 +55,51 @@ var pduCollectCmd = &cobra.Command{ 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 mock { + log.Info().Msg("Running in --mock mode. Generating hardcoded PDU payload to standard output.") + + type PDUInventoryForSMD struct { + Model string `json:"Model"` + SerialNumber string `json:"SerialNumber"` + FirmwareVersion string `json:"FirmwareVersion"` + Outlets []any `json:"Outlets"` + } + type PayloadForSMD struct { + ID string `json:"ID"` + Type string `json:"Type"` + FQDN string `json:"FQDN"` + Hostname string `json:"Hostname"` + Enabled bool `json:"Enabled"` + RediscoverOnUpdate bool `json:"RediscoverOnUpdate"` + PDUInventory PDUInventoryForSMD `json:"PDUInventory"` + } + + mockPayload := PayloadForSMD{ + ID: "x9999m0", + Type: "CabinetPDUController", + FQDN: "x9999m0-rts.mock:8083", + Hostname: "x9999m0-rts.mock:8083", + Enabled: true, + RediscoverOnUpdate: false, + PDUInventory: PDUInventoryForSMD{ + Model: "MOCK-PRO2", + SerialNumber: "MOCK-SN-12345", + FirmwareVersion: "v9.9z", + Outlets: []any{ + map[string]string{"id": "ZA01", "name": "Mock_Server_01", "state": "On", "socket_type": "Cx"}, + map[string]string{"id": "ZA02", "name": "Mock_Server_02", "state": "Off", "socket_type": "Cx"}, + }, + }, + } + + jsonData, err := json.MarshalIndent(mockPayload, "", " ") + if err != nil { + log.Fatal().Err(err).Msg("Failed to marshal mock payload") + } + + fmt.Println(string(jsonData)) + return + } if len(args) == 0 { log.Error().Msg("no PDU hosts provided") return @@ -98,4 +145,5 @@ func init() { pduCollectCmd.Flags().StringVarP(&username, "username", "u", "", "Set the PDU username") pduCollectCmd.Flags().StringVarP(&password, "password", "p", "", "Set the PDU password") + pduCollectCmd.Flags().BoolVar(&mock, "mock", false, "Run in mock mode, sending hardcoded data to SMD") }