func init() {
maker := func() client.Crudder { return &client.Deployment{} }
singularName := "deployment"
deployments := makeCommandTree(singularName, maker)
deployments.AddCommand(&cobra.Command{
Use: "bind [id] to [roleId]",
Short: "Bind a deployment to a role",
Run: func(c *cobra.Command, args []string) {
if len(args) != 3 || args[1] != "to" {
log.Fatalf("%v requires 2 arguments seperated by \"to\"", c.UseLine())
}
obj := &client.Deployment{}
if client.Fetch(obj, args[0]) != nil {
log.Fatalf("Failed to fetch %v\n", singularName)
}
role := &client.Role{}
if client.Fetch(role, args[2]) != nil {
log.Fatalf("Failed to fetch role\n")
}
nr := &client.DeploymentRole{}
nr.RoleID = role.ID
nr.DeploymentID = obj.ID
if err := client.BaseCreate(nr); err != nil {
log.Fatalf("Failed to create deploymentrole for deployment:%v role:%v\n", args[0], args[2])
}
fmt.Println(prettyJSON(nr))
},
})
deployments.AddCommand(&cobra.Command{
Use: "parent [id]",
Short: "Get the parent deployment of this deployment",
Run: func(c *cobra.Command, args []string) {
if len(args) != 1 {
log.Fatalf("%v requires 1 argument\n", c.UseLine())
}
obj := &client.Deployment{}
if client.Fetch(obj, args[0]) != nil {
log.Fatalf("Failed to fetch %v\n", singularName)
}
if !obj.ParentID.Valid {
log.Fatalf("System deployment does not have a parent")
return
}
res, err := obj.Parent()
if err != nil {
log.Fatalf("Failed to get parent of deployment %v\n", args[0])
}
fmt.Println(prettyJSON(res))
},
})
app.AddCommand(deployments)
}
func init() {
maker := func() client.Crudder { return &client.Node{} }
singularName := "node"
nodes := makeCommandTree(singularName, maker)
nodes.AddCommand(&cobra.Command{
Use: "addresses [id] on [networkId]",
Short: "Fetch all addresses allocated to node [id] on a given network",
Run: func(c *cobra.Command, args []string) {
if len(args) != 3 {
log.Fatalf("%v requires 2 arguments", c.UseLine())
}
node := &client.Node{}
net := &client.Network{}
if err := client.Fetch(node, args[0]); err != nil {
log.Fatalln("Failed to fetch node:", err)
}
if err := client.Fetch(net, args[2]); err != nil {
log.Fatalln("Failed to fetch network:", err)
}
allocations := []*client.NetworkAllocation{}
matcher := make(map[string]interface{})
matcher["network_id"] = net.ID
matcher["node_id"] = node.ID
if err := client.Match("network_allocations", matcher, &allocations); err != nil {
log.Fatalln("Failed to fetch allocations:", err)
}
addresses := make([]string, len(allocations))
for i := range allocations {
addresses[i] = allocations[i].Address
}
type result struct {
Node string `json:"node"`
Network string `json:"network"`
Category string `json:"category"`
Addresses []string `json:"addresses"`
}
res := result{Node: node.Name, Network: net.Name, Category: net.Category, Addresses: addresses}
fmt.Println(prettyJSON(res))
},
})
nodes.AddCommand(&cobra.Command{
Use: "bind [id] to [roleId]",
Short: "Bind a node to a role",
Run: func(c *cobra.Command, args []string) {
if len(args) != 3 || args[1] != "to" {
log.Fatalf("%v requires 2 arguments seperated by \"to\"", c.UseLine())
}
obj := &client.Node{}
role := &client.Role{}
if err := client.Fetch(obj, args[0]); err != nil {
log.Fatalf("Failed to fetch Node: %v\n", err.Error())
}
if err := client.Fetch(role, args[2]); err != nil {
log.Fatalf("Failed to fetch Role: %v\n", err.Error())
}
nr := &client.NodeRole{}
if err := client.Init(nr); err != nil {
log.Fatalf("Failed to initialize NodeRole with defaults\n%v\n", err)
}
nr.RoleID = role.ID
nr.NodeID = obj.ID
nr.DeploymentID = obj.DeploymentID
if err := client.BaseCreate(nr); err != nil {
log.Fatalf("Failed to create noderole for node:%v role:%v\n", args[0], args[2])
}
fmt.Println(prettyJSON(nr))
},
})
nodes.AddCommand(&cobra.Command{
Use: "import [json]",
Short: "Import a node definition as a JSON blob. May include a mac and an IP hint.",
Run: func(c *cobra.Command, args []string) {
if len(args) != 1 {
log.Fatalf("%v requires 1 argument\n", c.UseLine())
}
obj := &client.Node{}
if err := client.Import(obj, []byte(args[0])); err != nil {
log.Fatalf("Failed to create node\n%v\n", err)
}
fmt.Println(prettyJSON(obj))
},
})
nodes.AddCommand(&cobra.Command{
Use: "move [id] to [deployment]",
Short: "Atomically move a node from one deployment to another",
Run: func(c *cobra.Command, args []string) {
if len(args) != 3 {
log.Fatalf("%v requires 2 argument\n", c.UseLine())
}
obj := &client.Node{}
if client.Fetch(obj, args[0]) != nil {
log.Fatalf("Failed to fetch %v\n", singularName)
}
depl := &client.Deployment{}
if client.Fetch(depl, args[2]) != nil {
log.Fatalf("Failed fetch deployment")
}
if obj.Move(depl) != nil {
log.Fatalf("Failed to move node %v to deployment %v\n", args[0], args[2])
}
fmt.Println(prettyJSON(obj))
},
})
nodes.AddCommand(&cobra.Command{
Use: "poweractions [id]",
Short: "Get the power actions for this node",
Run: func(c *cobra.Command, args []string) {
if len(args) != 1 {
log.Fatalf("%v requires 1 argument\n", c.UseLine())
}
obj := &client.Node{}
if client.SetId(obj, args[0]) != nil {
log.Fatalf("Failed to parse ID %v for an %v\n", args[0], singularName)
}
res, err := obj.PowerActions()
if err != nil {
log.Fatalf("Failed to get power actions for node %v\n", args[0])
}
fmt.Println(prettyJSON(res))
},
})
nodes.AddCommand(&cobra.Command{
Use: "power [id] [action]",
Short: "Perform a power control action for this node",
Run: func(c *cobra.Command, args []string) {
if len(args) != 2 {
log.Fatalf("%v requires 2 arguments\n", c.UseLine())
}
obj := &client.Node{}
if client.SetId(obj, args[0]) != nil {
log.Fatalf("Failed to parse ID %v for an %v\n", args[0], singularName)
}
err := obj.Power(args[1])
if err != nil {
log.Fatalf("Failed perform power action %v\n", args[1])
}
},
})
nodes.AddCommand(&cobra.Command{
Use: "activebootenv [id]",
Short: "Get the boot environment that the provisioner has configured for the node.",
Run: func(c *cobra.Command, args []string) {
if len(args) != 1 {
log.Fatalf("%v requires 1 argument\n", c.UseLine())
}
obj := &client.Node{}
if client.SetId(obj, args[0]) != nil {
log.Fatalf("Failed to parse ID %v for an %v\n", args[0], singularName)
}
fmt.Println(obj.ActiveBootstate())
},
})
app.AddCommand(nodes)
}
func init() {
maker := func() client.Crudder { return &client.Network{} }
network := makeCommandTree("network", maker)
cmds := []*cobra.Command{
&cobra.Command{
Use: "import [json]",
Short: "Import a network + optional ranges and routers into Crowbar",
Run: func(c *cobra.Command, args []string) {
if len(args) != 1 {
log.Fatalf("%v requires 1 argument\n", c.UseLine())
}
obj := &client.Network{}
if err := client.Import(obj, []byte(args[0])); err != nil {
log.Fatalf("Unable to import network: %v\n", err)
}
fmt.Println(prettyJSON(obj))
},
},
&cobra.Command{
Use: "add [id] to [nodeId]",
Short: "Add a Network to a Node. This will implicitly allocate an address from the default host ranges in the network.",
Run: func(c *cobra.Command, args []string) {
if len(args) != 3 {
log.Fatalf("%v requires 2 arguments\n", c.UseLine())
}
node := &client.Node{}
network := &client.Network{}
if err := client.Fetch(node, args[2]); err != nil {
log.Fatalf("Unable to fetch node\n%v\n", err)
}
if err := client.Fetch(network, args[0]); err != nil {
log.Fatalf("Unable to fetch network\n%v\n", err)
}
ranges, err := network.AutoRanges(node)
if err != nil {
log.Fatalln("Failed to get auto ranges for network", err)
}
res := make([]*client.NetworkAllocation, len(ranges))
unwind := true
defer func() {
if unwind {
for _, allocation := range res {
if allocation != nil {
client.Destroy(allocation)
}
}
}
}()
for i, netRange := range ranges {
alloc := &client.NetworkAllocation{}
alloc.NodeID = null.IntFrom(node.ID)
alloc.NetworkID = null.IntFrom(network.ID)
alloc.NetworkRangeID = null.IntFrom(netRange.ID)
if err := client.BaseCreate(alloc); err != nil {
log.Fatalf("Unable to create new NetworkAllocation: %v", err)
}
res[i] = alloc
}
unwind = false
fmt.Println(prettyJSON(res))
},
},
&cobra.Command{
Use: "alloc [id] on [nodeId] from [rangeId] hint [address]",
Short: "Add a Network to a Node with an address allocated from a specific Range.",
Run: func(c *cobra.Command, args []string) {
if len(args) != 5 && len(args) != 7 {
log.Fatalf("%v requires 3 or 4 arguments", c.UseLine())
}
node := &client.Node{}
if err := client.Fetch(node, args[0]); err != nil {
log.Fatalln("Unable to fetch node from Crowbar\n%v\n", err)
}
netRange := &client.NetworkRange{}
if err := client.SetId(netRange, args[4]); err != nil {
vals := map[string]interface{}{}
vals["name"] = args[4]
network := &client.Network{}
if err := client.Fetch(network, args[2]); err != nil {
log.Fatalln("Unable to fetch network from Crowbar\n%v\n", err)
}
vals["network_id"] = network.ID
netRanges := []*client.NetworkRange{}
if err := client.Match(netRange.ApiName(), vals, &netRanges); err != nil {
log.Fatalf("Unable to fetch ranges matching %#v\n%v\n", netRange, err)
}
if len(netRanges) != 1 {
log.Fatalln("Supplied network range information is ambiguous")
}
netRange = netRanges[0]
} else {
if err := client.Read(netRange); err != nil {
log.Fatalln("Unable to fetch network range\n%v\n", err)
}
}
alloc := &client.NetworkAllocation{}
alloc.NetworkID = null.IntFrom(netRange.NetworkID)
alloc.NetworkRangeID = null.IntFrom(netRange.ID)
alloc.NodeID = null.IntFrom(node.ID)
if len(args) == 7 {
alloc.Address = args[6]
}
if err := client.BaseCreate(alloc); err != nil {
log.Fatalf("Unable to add node to network\n%v\n", err)
}
fmt.Println(prettyJSON(alloc))
},
},
}
network.AddCommand(cmds...)
app.AddCommand(network)
maker = func() client.Crudder { return &client.NetworkRange{} }
networkrange := makeCommandTree("networkrange", maker)
app.AddCommand(networkrange)
maker = func() client.Crudder { return &client.NetworkAllocation{} }
app.AddCommand(makeCommandTree("networkallocation", maker))
maker = func() client.Crudder { return &client.NetworkRouter{} }
app.AddCommand(makeCommandTree("networkrouter", maker))
}