diff --git a/bot/command.go b/bot/command.go index e2fc10f..2bc3b27 100644 --- a/bot/command.go +++ b/bot/command.go @@ -2,340 +2,45 @@ package bot import ( "fmt" - - timeago "github.com/ararog/timeago" - log "github.com/sirupsen/logrus" ) -type commandFunc func(func(string), string, []string) +type commandFunc func(from string, params []string) string -// list help -func (b *Bot) help(answer func(string), from string, params []string) { - msg := fmt.Sprintf("Hi %s there are the following commands:\n", from) - for _, cmd := range b.commands { - msg = fmt.Sprintf("%s - .%s\n", msg, cmd) - } - answer(msg) +type Command struct { + Name string + Description string + Commands []*Command + Action commandFunc } -// add a chat to send log to a chat -func (b *Bot) addSend(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD IPAddress/Hostname\n or\n CMD IPAddress/Hostname to") - return - } - host := params[0] - to := from - if len(params) > 1 { - to = params[1] - } - - h := b.db.GetHost(host) - if h == nil { - h = b.db.NewHost(host) - } - if h == nil { - answer(fmt.Sprintf("could not create host %s", host)) - return - } - n, ok := b.db.NotifiesByAddress[to] - if !ok { - n = b.db.NewNotify(to) - } - if n == nil { - answer(fmt.Sprintf("could not create notify %s in list of %s", to, host)) - return - } - h.AddNotify(n) - - answer(fmt.Sprintf("added %s in list of %s", to, host)) -} - -//add a chat to send log to a chat -func (b *Bot) delSend(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD IPAddress/Hostname\n or\n CMD IPAddress/Hostname to") - return - } - host := params[0] - to := from - if len(params) > 1 { - to = params[1] - } - - if h := b.db.GetHost(host); h != nil { - h.DeleteNotify(to) - answer(fmt.Sprintf("removed %s in list of %s", to, host)) - } else { - answer("not found host") - } - -} - -// list all hostname with the chat where it send to -func (b *Bot) listSend(answer func(string), from string, params []string) { - msg := "sending:\n" - all := false - of := from - if len(params) > 0 { - if params[0] == "all" { - all = true - } else { - of = params[0] +func (c Command) Run(from string, args []string) string { + if len(args) > 0 { + cmdName := args[0] + if cmdName == "help" { + return c.Help() } - } - for _, host := range b.db.Hosts { - toList := "" - show := all - for _, to := range host.Notifies { - if all { - toList = fmt.Sprintf("%s , %s", toList, to) - } else if to == of { - show = true + if len(c.Commands) > 0 { + for _, cmd := range c.Commands { + if cmd.Name == cmdName { + return cmd.Run(from, args[1:]) + } } - } - if !show { - continue - } - if len(toList) > 3 { - toList = toList[3:] - } - if host.Name != "" { - msg = fmt.Sprintf("%s%s (%s): %s\n", msg, host.Address, host.Name, toList) - } else { - msg = fmt.Sprintf("%s%s: %s\n", msg, host.Address, toList) + return fmt.Sprintf("command %s not found\n%s", cmdName, c.Help()) } } - - answer(msg) + if c.Action != nil { + return c.Action(from, args) + } + return c.Help() } -// add hostname -func (b *Bot) addHostname(answer func(string), from string, params []string) { - if len(params) < 2 { - answer("invalid: CMD IPAddress/Hostname NewHostname") - return - } - addr := params[0] - name := params[1] - - h := b.db.GetHost(addr) - if h == nil { - h = b.db.NewHost(addr) - } - b.db.ChangeHostname(h, name) - - answer(fmt.Sprintf("set for %s the hostname %s", addr, name)) -} - -func (b *Bot) delHostname(answer func(string), from string, params []string) { - if len(params) < 2 { - answer("invalid: CMD IPAddress/Hostname") - return - } - addr := params[0] - h := b.db.GetHost(addr) - if h != nil { - b.db.DeleteHost(h) - if h.Name != "" { - answer(fmt.Sprintf("remove host %s with hostname %s", h.Address, h.Name)) - } else { - answer(fmt.Sprintf("remove host %s", h.Address)) +func (c Command) Help() string { + if len(c.Commands) > 0 { + msg := fmt.Sprintf("%s\n-------------------", c.Description) + for _, cmd := range c.Commands { + msg = fmt.Sprintf("%s\n%s: %s", msg, cmd.Name, cmd.Description) } - } else { - answer("could not found host") + return msg } -} - -// list all host with his ip -func (b *Bot) listHostname(answer func(string), from string, params []string) { - msg := "hostnames:\n" - for _, host := range b.db.Hosts { - if host.Lastseen.Year() > 1 { - got, _ := timeago.TimeAgoFromNowWithTime(host.Lastseen) - msg = fmt.Sprintf("%s%s - %s (%s)\n", msg, host.Address, host.Name, got) - } else { - msg = fmt.Sprintf("%s%s - %s\n", msg, host.Address, host.Name) - } - } - answer(msg) -} - -// set a filter by max -func (b *Bot) listMaxfilter(answer func(string), from string, params []string) { - msg := "filters: " - if len(params) > 0 && params[0] == "all" { - msg = fmt.Sprintf("%s\n", msg) - for _, n := range b.db.Notifies { - msg = fmt.Sprintf("%s%s - %s\n", msg, n.Address(), n.MaxPrioIn.String()) - } - } else { - of := from - if len(params) > 0 { - of = params[0] - } - if notify, ok := b.db.NotifiesByAddress[of]; ok { - msg = fmt.Sprintf("%s %s is %s", msg, of, notify.MaxPrioIn.String()) - } - } - answer(msg) -} - -// set a filter to a max -func (b *Bot) setMaxfilter(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD Priority\n or\n CMD Channel Priority") - return - } - to := from - var max log.Level - var err error - - if len(params) > 1 { - to = params[0] - max, err = log.ParseLevel(params[1]) - } else { - max, err = log.ParseLevel(params[0]) - } - if err != nil { - answer("invalid priority: CMD Priority\n or\n CMD Channel Priority") - return - } - n, ok := b.db.NotifiesByAddress[to] - if !ok { - n = b.db.NewNotify(to) - } - - n.MaxPrioIn = max - - answer(fmt.Sprintf("set filter for %s to %s", to, max.String())) -} - -// list of regex filter -func (b *Bot) listRegex(answer func(string), from string, params []string) { - msg := "regexs:\n" - if len(params) > 0 && params[0] == "all" { - for _, n := range b.db.Notifies { - msg = fmt.Sprintf("%s%s\n-------------\n", msg, n.Address()) - for expression := range n.RegexIn { - msg = fmt.Sprintf("%s - %s\n", msg, expression) - } - } - } else { - of := from - if len(params) > 0 { - of = params[0] - } - if n, ok := b.db.NotifiesByAddress[of]; ok { - msg = fmt.Sprintf("%s%s\n-------------\n", msg, of) - for expression := range n.RegexIn { - msg = fmt.Sprintf("%s - %s\n", msg, expression) - } - } - } - answer(msg) -} - -// add a regex filter -func (b *Bot) addRegex(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD regex\n or\n CMD channel regex") - return - } - of := from - regex := params[0] - if len(params) > 1 { - of = params[0] - regex = params[1] - } - - n := b.db.NotifiesByAddress[of] - if err := n.AddRegex(regex); err == nil { - answer(fmt.Sprintf("add regex for \"%s\" to %s", of, regex)) - } else { - answer(fmt.Sprintf("\"%s\" is no valid regex expression: %s", regex, err)) - } -} - -// del a regex filter -func (b *Bot) delRegex(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD regex\n or\n CMD channel regex") - return - } - of := from - regex := params[0] - if len(params) > 1 { - of = params[0] - regex = params[1] - } - n := b.db.NotifiesByAddress[of] - delete(n.RegexIn, regex) - b.listRegex(answer, of, []string{}) -} - -// list of regex replace -func (b *Bot) listRegexReplace(answer func(string), from string, params []string) { - msg := "replaces:\n" - if len(params) > 0 && params[0] == "all" { - for _, n := range b.db.Notifies { - msg = fmt.Sprintf("%s%s\n-------------\n", msg, n.Address()) - for expression, value := range n.RegexReplace { - msg = fmt.Sprintf("%s - \"%s\" : \"%s\"\n", msg, expression, value) - } - } - } else { - of := from - if len(params) > 0 { - of = params[0] - } - if n, ok := b.db.NotifiesByAddress[of]; ok { - msg = fmt.Sprintf("%s%s\n-------------\n", msg, of) - for expression, value := range n.RegexReplace { - msg = fmt.Sprintf("%s - \"%s\" : \"%s\"\n", msg, expression, value) - } - } - } - answer(msg) -} - -// add a regex replace -func (b *Bot) addRegexReplace(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD regex replace\n or\n CMD channel regex replace") - return - } - of := from - regex := params[0] - value := params[1] - if len(params) > 2 { - of = params[0] - regex = params[1] - value = params[2] - } - - n := b.db.NotifiesByAddress[of] - if err := n.AddRegexReplace(regex, value); err == nil { - answer(fmt.Sprintf("add replace in \"%s\" for \"%s\" to \"%s\"", of, regex, value)) - } else { - answer(fmt.Sprintf("\"%s\" to \"%s\" is no valid regex replace expression: %s", regex, value, err)) - } -} - -// del a regex replace -func (b *Bot) delRegexReplace(answer func(string), from string, params []string) { - if len(params) < 1 { - answer("invalid: CMD regex\n or\n CMD channel regex") - return - } - of := from - regex := params[0] - if len(params) > 1 { - of = params[0] - regex = params[1] - } - n := b.db.NotifiesByAddress[of] - - delete(n.RegexReplace, regex) - b.listRegexReplace(answer, of, []string{}) + return c.Description } diff --git a/bot/filter.go b/bot/filter.go new file mode 100644 index 0000000..ede3450 --- /dev/null +++ b/bot/filter.go @@ -0,0 +1,97 @@ +package bot + +import ( + "fmt" + + "dev.sum7.eu/genofire/logmania/database" +) + +func NewFilter(db *database.DB) *Command { + return &Command{ + Name: "filter", + Description: "list and configurate regex filter for channel by message content", + Commands: []*Command{ + &Command{ + Name: "add", + Description: "add regex filter for channel: [channel] regex", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: [channel] regex" + } + of := from + regex := params[0] + if len(params) > 1 { + of = params[0] + regex = params[1] + } + + n := db.NotifiesByAddress[of] + if err := n.AddRegex(regex); err != nil { + return fmt.Sprintf("\"%s\" is no valid regex expression: %s", regex, err) + } + return fmt.Sprintf("add regex for \"%s\" to %s", of, regex) + }, + }, + &Command{ + Name: "del", + Description: "del regex filter for channel: [channel] regex", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: [channel] regex" + } + of := from + regex := params[0] + if len(params) > 1 { + of = params[0] + regex = params[1] + } + n := db.NotifiesByAddress[of] + delete(n.RegexIn, regex) + return "deleted" + }, + }, + &Command{ + Name: "all", + Description: "list of all channels", + Action: func(from string, params []string) string { + msg := "filter:\n" + for _, n := range db.Notifies { + msg = fmt.Sprintf("%s%s\n-------------\n", msg, n.Address()) + for expression := range n.RegexIn { + msg = fmt.Sprintf("%s - %s\n", msg, expression) + } + } + return msg + }, + }, + &Command{ + Name: "channel", + Description: "list of given channel: channel", + Action: func(from string, params []string) string { + msg := "filter:\n" + if len(params) != 1 { + return "invalid: no channel given" + } + of := params[0] + if n, ok := db.NotifiesByAddress[of]; ok { + msg = fmt.Sprintf("%s%s\n-------------\n", msg, of) + for expression := range n.RegexIn { + msg = fmt.Sprintf("%s - %s\n", msg, expression) + } + } + return msg + }, + }, + }, + Action: func(from string, params []string) string { + msg := "filter:\n" + if n, ok := db.NotifiesByAddress[from]; ok { + msg = fmt.Sprintf("%s%s\n-------------\n", msg, from) + for expression := range n.RegexIn { + msg = fmt.Sprintf("%s - %s\n", msg, expression) + } + } + return msg + }, + } +} diff --git a/bot/hostname.go b/bot/hostname.go new file mode 100644 index 0000000..66fbd3d --- /dev/null +++ b/bot/hostname.go @@ -0,0 +1,67 @@ +package bot + +import ( + "fmt" + + "dev.sum7.eu/genofire/logmania/database" + timeago "github.com/ararog/timeago" +) + +func NewHostname(db *database.DB) *Command { + return &Command{ + Name: "hostname", + Description: "alternative short (host)names for long IP-Addresses or URLs (and time of last recieved input)", + Commands: []*Command{ + &Command{ + Name: "set", + Description: "set or replace a hostname: IPAddress/Hostname NewHostname", + Action: func(from string, params []string) string { + if len(params) != 2 { + return "invalid: IPAddress/Hostname NewHostname" + } + addr := params[0] + name := params[1] + + h := db.GetHost(addr) + if h == nil { + h = db.NewHost(addr) + } + db.ChangeHostname(h, name) + + return fmt.Sprintf("set for %s the hostname %s", addr, name) + }, + }, + &Command{ + Name: "del", + Description: "delete a hostname: IPAddress/Hostname", + Action: func(from string, params []string) string { + if len(params) != 1 { + return "invalid: IPAddress/Hostname" + } + addr := params[0] + h := db.GetHost(addr) + if h != nil { + db.DeleteHost(h) + if h.Name != "" { + return fmt.Sprintf("remove host %s with hostname %s", h.Address, h.Name) + } + return fmt.Sprintf("remove host %s", h.Address) + } + return "could not found host" + }, + }, + }, + Action: func(from string, params []string) string { + msg := "hostnames:\n" + for _, host := range db.Hosts { + if host.Lastseen.Year() > 1 { + got, _ := timeago.TimeAgoFromNowWithTime(host.Lastseen) + msg = fmt.Sprintf("%s%s - %s (%s)\n", msg, host.Address, host.Name, got) + } else { + msg = fmt.Sprintf("%s%s - %s\n", msg, host.Address, host.Name) + } + } + return msg + }, + } +} diff --git a/bot/main.go b/bot/main.go index 47e162c..f18ecb8 100644 --- a/bot/main.go +++ b/bot/main.go @@ -1,54 +1,36 @@ package bot import ( - "fmt" - "strings" + "github.com/mattn/go-shellwords" "dev.sum7.eu/genofire/logmania/database" ) type Bot struct { - db *database.DB - commandsMap map[string]commandFunc - commands []string + Command } func NewBot(db *database.DB) *Bot { - b := &Bot{ - db: db, - } - b.commandsMap = map[string]commandFunc{ - "help": b.help, - "send-add": b.addSend, - "send-list": b.listSend, - "send-del": b.delSend, - "hostname-set": b.addHostname, - "hostname-list": b.listHostname, - "hostname-del": b.delHostname, - "filter-set": b.setMaxfilter, - "filter-list": b.listMaxfilter, - "regex-add": b.addRegex, - "regex-list": b.listRegex, - "regex-del": b.delRegex, - "replace-add": b.addRegexReplace, - "replace-list": b.listRegexReplace, - "replace-del": b.delRegexReplace, - } - for k := range b.commandsMap { - b.commands = append(b.commands, k) - } - return b + return &Bot{Command{ + Description: "logmania bot, to configurate live all settings", + Commands: []*Command{ + NewFilter(db), + NewHostname(db), + NewPriority(db), + NewReplace(db), + NewSend(db), + }, + }} } -func (b *Bot) Handle(answer func(string), from, msg string) { - msgParts := strings.Split(msg, " ") - if len(msgParts[0]) <= 0 || msgParts[0][0] != '.' { - return +func (b *Bot) Handle(from, msg string) string { + msgParts, err := shellwords.Parse(msg) + if err != nil { + return "" } - cmdName := msgParts[0][1:] - if cmd, ok := b.commandsMap[cmdName]; ok { - cmd(answer, from, msgParts[1:]) - } else { - answer(fmt.Sprintf("not found command: !%s", cmdName)) + if len(msgParts) <= 0 || msgParts[0][0] != '.' { + return "" } + msgParts[0] = msgParts[0][1:] + return b.Run(from, msgParts) } diff --git a/bot/priority.go b/bot/priority.go new file mode 100644 index 0000000..68b52c8 --- /dev/null +++ b/bot/priority.go @@ -0,0 +1,81 @@ +package bot + +import ( + "fmt" + + log "github.com/sirupsen/logrus" + + "dev.sum7.eu/genofire/logmania/database" +) + +func NewPriority(db *database.DB) *Command { + return &Command{ + Name: "priority", + Description: "list and configurate priority in channel", + Commands: []*Command{ + &Command{ + Name: "set", + Description: "set max priority of channel: [channel] Priority", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: [channel] Priority" + } + to := from + var max log.Level + var err error + + if len(params) > 1 { + to = params[0] + max, err = log.ParseLevel(params[1]) + } else { + max, err = log.ParseLevel(params[0]) + } + if err != nil { + return "invalid: [Channel] Priority" + } + n, ok := db.NotifiesByAddress[to] + if !ok { + n = db.NewNotify(to) + } + + n.MaxPrioIn = max + + return fmt.Sprintf("set filter for %s to %s", to, max.String()) + }, + }, + &Command{ + Name: "all", + Description: "list of all channels", + Action: func(from string, params []string) string { + msg := "priority: \n" + for _, n := range db.Notifies { + msg = fmt.Sprintf("%s%s - %s\n", msg, n.Address(), n.MaxPrioIn.String()) + } + return msg + }, + }, + &Command{ + Name: "channel", + Description: "list of given channel: channel", + Action: func(from string, params []string) string { + if len(params) != 1 { + return "invalid: no channel given" + } + of := params[0] + msg := "priority: \n" + if notify, ok := db.NotifiesByAddress[of]; ok { + msg = fmt.Sprintf("%s %s is %s", msg, of, notify.MaxPrioIn.String()) + } + return msg + }, + }, + }, + Action: func(from string, params []string) string { + msg := "priority: \n" + if notify, ok := db.NotifiesByAddress[from]; ok { + msg = fmt.Sprintf("%s %s is %s", msg, from, notify.MaxPrioIn.String()) + } + return msg + }, + } +} diff --git a/bot/replace.go b/bot/replace.go new file mode 100644 index 0000000..45df105 --- /dev/null +++ b/bot/replace.go @@ -0,0 +1,101 @@ +package bot + +import ( + "fmt" + + "dev.sum7.eu/genofire/logmania/database" +) + +func NewReplace(db *database.DB) *Command { + return &Command{ + Name: "replace", + Description: "list and configurate replace content of message for channel", + Commands: []*Command{ + &Command{ + Name: "add", + Description: "add regex replace for channel: [channel] regex replace", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: [channel] regex replace" + } + of := from + regex := params[0] + value := params[1] + if len(params) > 2 { + of = params[0] + regex = params[1] + value = params[2] + } + + n := db.NotifiesByAddress[of] + if err := n.AddRegexReplace(regex, value); err != nil { + return fmt.Sprintf("\"%s\" to \"%s\" is no valid regex replace expression: %s", regex, value, err) + } + return fmt.Sprintf("add replace in \"%s\" for \"%s\" to \"%s\"", of, regex, value) + }, + }, + &Command{ + Name: "del", + Description: "del regex replace for channel: [channel] regex replace", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: [channel] regex replace" + } + of := from + regex := params[0] + if len(params) > 1 { + of = params[0] + regex = params[1] + } + n := db.NotifiesByAddress[of] + + delete(n.RegexReplace, regex) + return "deleted" + }, + }, + + &Command{ + Name: "all", + Description: "list of all channels", + Action: func(from string, params []string) string { + msg := "replaces:\n" + for _, n := range db.Notifies { + msg = fmt.Sprintf("%s%s\n-------------\n", msg, n.Address()) + for expression, value := range n.RegexReplace { + msg = fmt.Sprintf("%s - \"%s\" : \"%s\"\n", msg, expression, value) + } + } + return msg + }, + }, + &Command{ + Name: "channel", + Description: "list of given channel: channel", + Action: func(from string, params []string) string { + if len(params) != 1 { + return "invalid: no channel given" + } + of := params[0] + msg := "replaces:\n" + if n, ok := db.NotifiesByAddress[of]; ok { + msg = fmt.Sprintf("%s%s\n-------------\n", msg, of) + for expression, value := range n.RegexReplace { + msg = fmt.Sprintf("%s - \"%s\" : \"%s\"\n", msg, expression, value) + } + } + return msg + }, + }, + }, + Action: func(from string, params []string) string { + msg := "replaces:\n" + if n, ok := db.NotifiesByAddress[from]; ok { + msg = fmt.Sprintf("%s%s\n-------------\n", msg, from) + for expression, value := range n.RegexReplace { + msg = fmt.Sprintf("%s - \"%s\" : \"%s\"\n", msg, expression, value) + } + } + return msg + }, + } +} diff --git a/bot/send.go b/bot/send.go new file mode 100644 index 0000000..381f0b8 --- /dev/null +++ b/bot/send.go @@ -0,0 +1,142 @@ +package bot + +import ( + "fmt" + + "dev.sum7.eu/genofire/logmania/database" +) + +func NewSend(db *database.DB) *Command { + return &Command{ + Name: "send", + Description: "list and configurate destination for hostnames", + Commands: []*Command{ + &Command{ + Name: "add", + Description: "add a destination for host with: IPAddress/Hostname [to]", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: IPAddress/Hostname [to]" + } + host := params[0] + to := from + if len(params) > 1 { + to = params[1] + } + + h := db.GetHost(host) + if h == nil { + h = db.NewHost(host) + } + if h == nil { + return fmt.Sprintf("could not create host %s", host) + } + n, ok := db.NotifiesByAddress[to] + if !ok { + n = db.NewNotify(to) + } + if n == nil { + return fmt.Sprintf("could not create notify %s in list of %s", to, host) + } + h.AddNotify(n) + + return fmt.Sprintf("added %s in list of %s", to, host) + }, + }, + &Command{ + Name: "del", + Description: "del a destination for host with: IPAddress/Hostname [to]", + Action: func(from string, params []string) string { + if len(params) < 1 { + return "invalid: IPAddress/Hostname [to]" + } + host := params[0] + to := from + if len(params) > 1 { + to = params[1] + } + + if h := db.GetHost(host); h != nil { + h.DeleteNotify(to) + return fmt.Sprintf("removed %s in list of %s", to, host) + } + return "not found host" + }, + }, + &Command{ + Name: "all", + Description: "list of all hosts with there channels", + Action: func(from string, params []string) string { + msg := "sending:\n" + for _, host := range db.Hosts { + toList := "" + for _, to := range host.Notifies { + toList = fmt.Sprintf("%s , %s", toList, to) + } + if len(toList) > 3 { + toList = toList[3:] + } + if host.Name != "" { + msg = fmt.Sprintf("%s%s (%s): %s\n", msg, host.Address, host.Name, toList) + } else { + msg = fmt.Sprintf("%s%s: %s\n", msg, host.Address, toList) + } + } + return msg + }, + }, + &Command{ + Name: "channel", + Description: "list all host of given channel: channel", + Action: func(from string, params []string) string { + if len(params) != 1 { + return "invalid: no channel given" + } + + of := params[0] + msg := "sending:\n" + + for _, host := range db.Hosts { + show := false + for _, to := range host.Notifies { + if to == of { + show = true + break + } + } + if !show { + continue + } + if host.Name != "" { + msg = fmt.Sprintf("%s%s (%s)\n", msg, host.Address, host.Name) + } else { + msg = fmt.Sprintf("%s%s\n", msg, host.Address) + } + } + return msg + }, + }, + }, + Action: func(from string, params []string) string { + msg := "sending:\n" + for _, host := range db.Hosts { + show := false + for _, to := range host.Notifies { + if to == from { + show = true + break + } + } + if !show { + continue + } + if host.Name != "" { + msg = fmt.Sprintf("%s%s (%s)\n", msg, host.Address, host.Name) + } else { + msg = fmt.Sprintf("%s%s\n", msg, host.Address) + } + } + return msg + }, + } +} diff --git a/output/websocket/main.go b/output/websocket/main.go index 8236fcd..7a8f18a 100644 --- a/output/websocket/main.go +++ b/output/websocket/main.go @@ -47,9 +47,10 @@ func Init(configInterface interface{}, db *database.DB, bot *bot.Bot) output.Out logger.Warnf("receive unknown websocket message: %s", msg.Subject) continue } - bot.Handle(func(answer string) { + answer := bot.Handle("", msg.Body.(string)) + if answer != "" { msg.Answer("bot", answer) - }, "", msg.Body.(string)) + } } }() diff --git a/output/xmpp/main.go b/output/xmpp/main.go index eb39ee8..4282b7a 100644 --- a/output/xmpp/main.go +++ b/output/xmpp/main.go @@ -119,20 +119,22 @@ func Init(configInterface interface{}, db *database.DB, bot *bot.Bot) output.Out from = proto + ":" + from } - bot.Handle(func(answer string) { - to := msg.From - if msg.Type == xmpp.MessageTypeGroupchat && !to.IsBare() { - to = to.Bare() - } - err := client.Send(&xmpp.MessageClient{ - Type: msg.Type, - To: to, - Body: answer, - }) - if err != nil { - logger.Error("xmpp to ", msg.From.String(), " error:", err) - } - }, from, msg.Body) + answer := bot.Handle(from, msg.Body) + if answer == "" { + continue + } + to := msg.From + if msg.Type == xmpp.MessageTypeGroupchat && !to.IsBare() { + to = to.Bare() + } + err := client.Send(&xmpp.MessageClient{ + Type: msg.Type, + To: to, + Body: answer, + }) + if err != nil { + logger.Error("xmpp to ", msg.From.String(), " error:", err) + } } } }()