burble.dn42/bird-lg-go
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Simultaneous query for telegram bot

Browse Source
This commit is contained in:
Lan Tian 2020-04-09 13:43:53 +08:00
parent abd0eda45b
commit 77873d6835
No known key found for this signature in database
GPG Key ID: 27F31700E751EC22
1 changed files with 37 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
frontend/telegram_bot.go
View File

@ -36,6 +36,20 @@ func telegramIsCommand(message string, command string) bool {
return b return b
} }
func telegramDefaultPostProcess(s string) string {
return strings.TrimSpace(s)
}
func telegramBatchRequestFormat(servers []string, endpoint string, command string, postProcess func(string) string) string {
results := batchRequest(servers, endpoint, command)
result := ""
for i, r := range results {
result += servers[i] + "\n"
result += postProcess(r) + "\n\n"
}
return result
}
func webHandlerTelegramBot(w http.ResponseWriter, r *http.Request) { func webHandlerTelegramBot(w http.ResponseWriter, r *http.Request) {
// Parse only needed fields of incoming JSON body // Parse only needed fields of incoming JSON body
var err error var err error
@ -52,9 +66,11 @@ func webHandlerTelegramBot(w http.ResponseWriter, r *http.Request) {
} }
// Select only one server based on webhook URL // Select only one server based on webhook URL
server := r.URL.Path[len("/telegram/"):] var servers []string
if len(server) == 0 { if len(r.URL.Path[len("/telegram/"):]) == 0 {
server = setting.servers[0] servers = setting.servers
} else {
servers = strings.Split(r.URL.Path[len("/telegram/"):], "+")
} }
// Parse target // Parse target
@ -68,29 +84,33 @@ func webHandlerTelegramBot(w http.ResponseWriter, r *http.Request) {
// - traceroute // - traceroute
if telegramIsCommand(request.Message.Text, "trace") || telegramIsCommand(request.Message.Text, "trace4") { if telegramIsCommand(request.Message.Text, "trace") || telegramIsCommand(request.Message.Text, "trace4") {
commandResult = strings.Join(batchRequest([]string{server}, "traceroute", target), "") commandResult = telegramBatchRequestFormat(servers, "traceroute", target, telegramDefaultPostProcess)
} else if telegramIsCommand(request.Message.Text, "trace6") { } else if telegramIsCommand(request.Message.Text, "trace6") {
commandResult = strings.Join(batchRequest([]string{server}, "traceroute6", target), "") commandResult = telegramBatchRequestFormat(servers, "traceroute6", target, telegramDefaultPostProcess)
} else if telegramIsCommand(request.Message.Text, "route") || telegramIsCommand(request.Message.Text, "route4") { } else if telegramIsCommand(request.Message.Text, "route") || telegramIsCommand(request.Message.Text, "route4") {
commandResult = strings.Join(batchRequest([]string{server}, "bird", "show route for "+target+" primary"), "") commandResult = telegramBatchRequestFormat(servers, "bird", "show route for "+target+" primary", telegramDefaultPostProcess)
} else if telegramIsCommand(request.Message.Text, "route6") { } else if telegramIsCommand(request.Message.Text, "route6") {
commandResult = strings.Join(batchRequest([]string{server}, "bird6", "show route for "+target+" primary"), "") commandResult = telegramBatchRequestFormat(servers, "bird6", "show route for "+target+" primary", telegramDefaultPostProcess)
} else if telegramIsCommand(request.Message.Text, "path") || telegramIsCommand(request.Message.Text, "path4") { } else if telegramIsCommand(request.Message.Text, "path") || telegramIsCommand(request.Message.Text, "path4") {
tempResult := strings.Join(batchRequest([]string{server}, "bird", "show route for "+target+" all primary"), "") commandResult = telegramBatchRequestFormat(servers, "bird", "show route for "+target+" all primary", func(result string) string {
for _, s := range strings.Split(tempResult, "\n") { for _, s := range strings.Split(result, "\n") {
if strings.Contains(s, "BGP.as_path: ") { if strings.Contains(s, "BGP.as_path: ") {
commandResult = strings.Split(s, ":")[1] return strings.TrimSpace(strings.Split(s, ":")[1])
} }
} }
return ""
})
} else if telegramIsCommand(request.Message.Text, "path6") { } else if telegramIsCommand(request.Message.Text, "path6") {
tempResult := strings.Join(batchRequest([]string{server}, "bird6", "show route for "+target+" all primary"), "") commandResult = telegramBatchRequestFormat(servers, "bird6", "show route for "+target+" all primary", func(result string) string {
for _, s := range strings.Split(tempResult, "\n") { for _, s := range strings.Split(result, "\n") {
if strings.Contains(s, "BGP.as_path: ") { if strings.Contains(s, "BGP.as_path: ") {
commandResult = strings.Split(s, ":")[1] return strings.TrimSpace(strings.Split(s, ":")[1])
} }
} }
return ""
})
} else if telegramIsCommand(request.Message.Text, "whois") { } else if telegramIsCommand(request.Message.Text, "whois") {
tempResult := whois(target) tempResult := whois(target)