Added kmb and refactored query.Parse
This commit is contained in:
55
main.go
55
main.go
@@ -1,9 +1,11 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
mtrbus "github.com/tgckpg/golifehk/datasources/mtr/bus"
|
||||
kmb "github.com/tgckpg/golifehk/datasources/kmb"
|
||||
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
||||
)
|
||||
|
||||
@@ -29,13 +31,58 @@ func main() {
|
||||
|
||||
log.Printf( "[%s] %s", update.Message.From.UserName, update.Message.Text )
|
||||
|
||||
result := mtrbus.Query( "zh", update.Message.Text )
|
||||
mesg, err := mtrbus.Query( "zh-Hant", update.Message.Text ).Message()
|
||||
|
||||
if update.Message.Chat.IsGroup() {
|
||||
if err == nil {
|
||||
msg := tgbotapi.NewMessage( update.Message.Chat.ID, mesg )
|
||||
msg.ParseMode = "MarkdownV2"
|
||||
msg.ReplyToMessageID = update.Message.MessageID
|
||||
|
||||
bot.Send( msg )
|
||||
}
|
||||
|
||||
} else {
|
||||
|
||||
var msg tgbotapi.MessageConfig
|
||||
if err == nil {
|
||||
msg = tgbotapi.NewMessage( update.Message.Chat.ID, mesg )
|
||||
msg.ParseMode = "MarkdownV2"
|
||||
} else {
|
||||
msg = tgbotapi.NewMessage( update.Message.Chat.ID, fmt.Sprintf( "%s", err ) )
|
||||
}
|
||||
|
||||
if result.Error == nil || !update.Message.Chat.IsGroup() {
|
||||
msg := tgbotapi.NewMessage( update.Message.Chat.ID, result.Message() )
|
||||
msg.ParseMode = "MarkdownV2"
|
||||
msg.ReplyToMessageID = update.Message.MessageID
|
||||
bot.Send( msg )
|
||||
}
|
||||
|
||||
mesg, err2 := kmb.Query( "zh-Hant", update.Message.Text ).Message()
|
||||
|
||||
if err != nil && err2 != nil {
|
||||
continue
|
||||
}
|
||||
|
||||
err = err2
|
||||
if update.Message.Chat.IsGroup() {
|
||||
if err == nil {
|
||||
msg := tgbotapi.NewMessage( update.Message.Chat.ID, mesg )
|
||||
msg.ParseMode = "MarkdownV2"
|
||||
msg.ReplyToMessageID = update.Message.MessageID
|
||||
|
||||
bot.Send( msg )
|
||||
}
|
||||
|
||||
} else {
|
||||
|
||||
var msg tgbotapi.MessageConfig
|
||||
if err == nil {
|
||||
msg = tgbotapi.NewMessage( update.Message.Chat.ID, mesg )
|
||||
msg.ParseMode = "MarkdownV2"
|
||||
} else {
|
||||
msg = tgbotapi.NewMessage( update.Message.Chat.ID, fmt.Sprintf( "%s", err ) )
|
||||
}
|
||||
|
||||
msg.ReplyToMessageID = update.Message.MessageID
|
||||
bot.Send( msg )
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user