79 lines
2.5 KiB
Go
79 lines
2.5 KiB
Go
|
package cmd
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/json"
|
||
|
"fmt"
|
||
|
"net/http"
|
||
|
"pixelridgesoftworks.com/BackGo/pkg/backup"
|
||
|
)
|
||
|
|
||
|
// InterfaceCLI manages commands related to node interface operations
|
||
|
type InterfaceCLI struct {
|
||
|
NodeAddress string
|
||
|
DBManager *backup.DBManager // Assuming DBManager is part of the backup package
|
||
|
}
|
||
|
|
||
|
// NewInterfaceCLI creates a new InterfaceCLI instance
|
||
|
func NewInterfaceCLI(nodeAddress string, dbManager *backup.DBManager) *InterfaceCLI {
|
||
|
return &InterfaceCLI{
|
||
|
NodeAddress: nodeAddress,
|
||
|
DBManager: dbManager,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// TriggerCommand sends a trigger command to a node
|
||
|
func (cli *InterfaceCLI) TriggerCommand(action, value string) error {
|
||
|
return cli.sendCommandToNode("trigger", action, value)
|
||
|
}
|
||
|
|
||
|
// GetInfo sends a get_info command to a node
|
||
|
func (cli *InterfaceCLI) GetInfo(value string) error {
|
||
|
return cli.sendCommandToNode("get_info", cli.NodeAddress, value)
|
||
|
}
|
||
|
|
||
|
// AddNode handles adding a new node on the master node
|
||
|
func (cli *InterfaceCLI) AddNode(hostname, token string) error {
|
||
|
// Verify the hostname resolves to an expected IP before adding
|
||
|
// This might involve additional logic to determine what the expected IP is
|
||
|
expectedIP := "" // Placeholder for expected IP determination logic
|
||
|
if !cli.DBManager.VerifyHostname(hostname, expectedIP) {
|
||
|
return fmt.Errorf("hostname %s does not resolve to the expected IP %s", hostname, expectedIP)
|
||
|
}
|
||
|
|
||
|
// Create a token for the new node
|
||
|
if err := cli.DBManager.CreateToken(hostname, token); err != nil {
|
||
|
return fmt.Errorf("failed to create token for hostname %s: %v", hostname, err)
|
||
|
}
|
||
|
|
||
|
fmt.Printf("Node %s added successfully with token %s\n", hostname, token)
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// GenCert generates a certificate for a node
|
||
|
func (cli *InterfaceCLI) GenCert(nodeName string) error {
|
||
|
return cli.sendCommandToNode("gencert", nodeName, "")
|
||
|
}
|
||
|
|
||
|
// sendCommandToNode sends a command to the specified node
|
||
|
func (cli *InterfaceCLI) sendCommandToNode(action, node, value string) error {
|
||
|
payload := backup.CommandPayload{
|
||
|
Action: action,
|
||
|
Node: node,
|
||
|
Value: value,
|
||
|
}
|
||
|
payloadBytes, err := json.Marshal(payload)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("error marshaling command payload: %v", err)
|
||
|
}
|
||
|
|
||
|
resp, err := http.Post(fmt.Sprintf("http://%s/api/command", cli.NodeAddress), "application/json", bytes.NewReader(payloadBytes))
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("error sending command to node %s: %v", node, err)
|
||
|
}
|
||
|
defer resp.Body.Close()
|
||
|
|
||
|
fmt.Printf("Command '%s' sent to node %s successfully\n", action, node)
|
||
|
return nil
|
||
|
}
|