func SendDockerCommand(cmd tree_docker.DockerCmd, node string, target []string, tag []string, group []string) {
var err tree_lib.TreeError
if !tree_api.API_INIT(node) {
fmt.Println("Unable to init api client")
fmt.Println("Exiting ...")
return
}
err.From = tree_lib.FROM_SEND_DOCKER_COMMAND
var (
api_cmd = tree_api.Command{}
wait = make(chan bool)
)
api_cmd.ID = tree_lib.RandomString(20)
api_cmd.Data, err.Err = ffjson.Marshal(cmd)
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
api_cmd.CommandType = tree_api.COMMAND_CONTAINER
tree_event.ON(tree_event.ON_CHILD_CONNECTED, func(ev *tree_event.Event) {
path := &tree_graph.Path{From: node, Nodes: target, Groups: group, Tags: tag}
tree_api.SendCommand(&api_cmd, path, func(e *tree_event.Event, c tree_api.Command) bool {
fmt.Println(string(c.Data))
fmt.Println(c.Ended)
if c.Ended {
return false
}
return true
})
wait <- true
})
<-wait
}
func SendEventTriggerCommand(cmd *cobra.Command, args []string) {
var (
event_name string
err tree_lib.TreeError
node string
targets []string
target_groups []string
target_tags []string
)
err.From = tree_lib.FROM_SEND_COMMAND
node, err.Err = cmd.Flags().GetString("node")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
targets, err.Err = cmd.Flags().GetStringSlice("target")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
target_groups, err.Err = cmd.Flags().GetStringSlice("group")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
target_tags, err.Err = cmd.Flags().GetStringSlice("tag")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
event_name, err.Err = cmd.Flags().GetString("event")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
if !tree_api.API_INIT(node) {
fmt.Println("Unable to init api client")
fmt.Println("Exiting ...")
return
}
var (
api_cmd = tree_api.Command{}
wait_to_end = make(chan bool)
)
api_cmd.Data = []byte(event_name)
api_cmd.ID = tree_lib.RandomString(20)
api_cmd.CommandType = tree_api.COMMAND_TRIGGER_EVENT
tree_event.ON(tree_event.ON_CHILD_CONNECTED, func(ev *tree_event.Event) {
path := &tree_graph.Path{From: node, Nodes: targets, Tags: target_tags, Groups: target_groups}
tree_api.SendCommand(&api_cmd, path, func(e *tree_event.Event, c tree_api.Command) bool {
fmt.Println(string(c.Data))
fmt.Println(c.Ended)
// TODO: End coming faster than other messages FIX !!!!
if c.Ended {
return false
}
return true
})
wait_to_end <- true
})
<-wait_to_end
}
func ListInfos(cmd *cobra.Command, args []string) {
var (
err tree_lib.TreeError
node string
targets []string
groups []string
tags []string
info tree_api.Info
)
err.From = tree_lib.FROM_LIST_INFOS
node, err.Err = cmd.Flags().GetString("node")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
targets, err.Err = cmd.Flags().GetStringSlice("target")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
groups, err.Err = cmd.Flags().GetStringSlice("group")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
tags, err.Err = cmd.Flags().GetStringSlice("tag")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
if !tree_api.API_INIT(node) {
fmt.Println("Unable to init api client")
fmt.Println("Exiting ...")
return
}
var (
api_cmd = tree_api.Command{}
wait = make(chan bool)
)
info.Group = groups
info.Target = targets
info.Tag = tags
api_cmd.Data, err.Err = ffjson.Marshal(info)
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
api_cmd.ID = tree_lib.RandomString(20)
api_cmd.CommandType = tree_api.COMMAND_LIST
tree_event.ON(tree_event.ON_CHILD_CONNECTED, func(ev *tree_event.Event) {
path := &tree_graph.Path{From: node, Nodes: []string{node}}
tree_api.SendCommand(&api_cmd, path, func(e *tree_event.Event, c tree_api.Command) bool {
var (
err tree_lib.TreeError
info = make(map[string]node_info.NodeInfo)
)
err.Err = ffjson.Unmarshal(c.Data, &info)
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return false
}
for _, a := range info {
fmt.Println("name: ", a.Name, ", Adress: ", a.TreeIp, ":", a.TreePort, ", Value: ", a.Value)
fmt.Println("groups:", a.Groups)
fmt.Println("tags:", a.Tags)
fmt.Println("childs", a.Childs)
}
return false
})
wait <- true
})
<-wait
}
func SendAddEventHandlerCommand(cmd *cobra.Command, args []string) {
var (
event_name string
err tree_lib.TreeError
handler custom_event.Handler
node string
targets []string
target_groups []string
target_tags []string
)
err.From = tree_lib.FROM_SEND_COMMAND
node, err.Err = cmd.Flags().GetString("node")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
targets, err.Err = cmd.Flags().GetStringSlice("target")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
target_groups, err.Err = cmd.Flags().GetStringSlice("group")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
target_tags, err.Err = cmd.Flags().GetStringSlice("tag")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
event_name, err.Err = cmd.Flags().GetString("event")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
handler.Handler, err.Err = cmd.Flags().GetString("handler")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
handler.IsFile, err.Err = cmd.Flags().GetBool("file")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
handler.ExecUser, err.Err = cmd.Flags().GetString("user")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
handler.Password, err.Err = cmd.Flags().GetString("pass")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
var (
api_cmd = tree_api.Command{}
wait_to_end = make(chan bool)
)
api_cmd.Data, err.Err = ffjson.Marshal(map[string]interface{}{
"name": event_name,
"handlers": []custom_event.Handler{handler},
})
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
if !tree_api.API_INIT(node) {
fmt.Println("Unable to init api client")
fmt.Println("Exiting ...")
return
}
api_cmd.ID = tree_lib.RandomString(20)
api_cmd.CommandType = tree_api.COMMAND_ADD_CUSTOM_EVENT
tree_event.ON(tree_event.ON_CHILD_CONNECTED, func(ev *tree_event.Event) {
path := &tree_graph.Path{From: node, Nodes: targets, Tags: target_tags, Groups: target_groups}
tree_api.SendCommand(&api_cmd, path, func(e *tree_event.Event, c tree_api.Command) bool {
fmt.Println(string(c.Data))
fmt.Println(c.Ended)
// TODO: End coming faster than other messages FIX !!!!
if c.Ended {
return false
}
return true
})
wait_to_end <- true
})
<-wait_to_end
}
func UpdateInfo(cmd *cobra.Command, args []string) {
var (
node string
err tree_lib.TreeError
target string
ip string
port int
add_to_group string
delete_from_group string
add_to_tag string
delete_from_tag string
delete_child string
add_child string
info node_info.NodeInfo
)
err.From = tree_lib.FROM_UPDATE_INFO
node, err.Err = cmd.Flags().GetString("node")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
target, err.Err = cmd.Flags().GetString("target")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
ip, err.Err = cmd.Flags().GetString("ip")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
port, err.Err = cmd.Flags().GetInt("port")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
add_to_group, err.Err = cmd.Flags().GetString("add_group")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
delete_from_group, err.Err = cmd.Flags().GetString("delete_group")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
add_to_tag, err.Err = cmd.Flags().GetString("add_tag")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
delete_from_tag, err.Err = cmd.Flags().GetString("delete_tag")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
add_child, err.Err = cmd.Flags().GetString("add_child")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
delete_child, err.Err = cmd.Flags().GetString("delete_child")
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
if !tree_api.API_INIT(node) {
fmt.Println("Unable to init api client")
fmt.Println("Exiting ...")
return
}
info.Name = target
info.TreeIp = ip
info.TreePort = port
info.Childs = append(info.Childs, add_child, delete_child)
info.Groups = append(info.Groups, add_to_group, delete_from_group)
info.Tags = append(info.Tags, add_to_tag, delete_from_tag)
var (
api_cmd = tree_api.Command{}
wait = make(chan bool)
)
api_cmd.ID = tree_lib.RandomString(20)
api_cmd.Data, err.Err = ffjson.Marshal(info)
if !err.IsNull() {
tree_log.Error(err.From, err.Error())
return
}
api_cmd.CommandType = tree_api.COMMAND_UPDATE
tree_event.ON(tree_event.ON_CHILD_CONNECTED, func(ev *tree_event.Event) {
path := &tree_graph.Path{From: node, Nodes: []string{node}}
tree_api.SendCommand(&api_cmd, path, func(e *tree_event.Event, c tree_api.Command) bool {
return false
})
wait <- true
})
<-wait
}