init

.gitignore

@@ -10,3 +10,6 @@
 
 # Output of the go coverage tool, specifically when used with LiteIDE
 *.out
+go.sum
+config.yaml
+Crisp_Telegram_bot

go.mod

@@ -0,0 +1,14 @@
+module github.com/tonyzzzzzz/Crisp_Telegram_bot
+
+go 1.12
+
+require (
+	github.com/crisp-im/go-crisp-api v3.3.2+incompatible
+	github.com/go-redis/redis v6.15.5+incompatible
+	github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible
+	github.com/gorilla/websocket v1.4.1 // indirect
+	github.com/graarh/golang-socketio v0.0.0-20170510162725-2c44953b9b5f // indirect
+	github.com/jinzhu/gorm v1.9.11
+	github.com/spf13/viper v1.4.0
+	github.com/technoweenie/multipartstreamer v1.0.1 // indirect
+)

main.go

@@ -0,0 +1,201 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"strconv"
+	"time"
+
+	"github.com/crisp-im/go-crisp-api/crisp"
+	"github.com/go-redis/redis"
+	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
+	"github.com/spf13/viper"
+	"github.com/tonyzzzzzz/Crisp_Telegram_bot/utils"
+)
+
+var bot *tgbotapi.BotAPI
+var client *crisp.Client
+var redisClient *redis.Client
+var config *viper.Viper
+
+// CrispMessageInfo stores the original message
+type CrispMessageInfo struct {
+	WebsiteID string
+	SessionID string
+}
+
+// MarshalBinary serializes CrispMessageInfo into binary
+func (s *CrispMessageInfo) MarshalBinary() ([]byte, error) {
+	return json.Marshal(s)
+}
+
+// UnmarshalBinary deserializes CrispMessageInfo into struct
+func (s *CrispMessageInfo) UnmarshalBinary(data []byte) error {
+	return json.Unmarshal(data, s)
+}
+
+func contains(s []interface{}, e int64) bool {
+	for _, a := range s {
+		if a == e {
+			return true
+		}
+	}
+	return false
+}
+
+func replyToUser(update *tgbotapi.Update) {
+	if update.Message.ReplyToMessage == nil {
+		msg := tgbotapi.NewMessage(update.Message.Chat.ID, "请回复一个消息")
+		bot.Send(msg)
+		return
+	}
+
+	res, err := redisClient.Get(strconv.Itoa(update.Message.ReplyToMessage.MessageID)).Result()
+
+	if err != nil {
+		msg := tgbotapi.NewMessage(update.Message.Chat.ID, "ERROR: "+err.Error())
+		bot.Send(msg)
+		return
+	}
+
+	var msgInfo CrispMessageInfo
+	err = json.Unmarshal([]byte(res), &msgInfo)
+
+	if err := json.Unmarshal([]byte(res), &msgInfo); err != nil {
+		msg := tgbotapi.NewMessage(update.Message.Chat.ID, "ERROR: "+err.Error())
+		bot.Send(msg)
+		return
+	}
+
+	if update.Message.Text != "" {
+		client.Website.SendTextMessageInConversation(msgInfo.WebsiteID, msgInfo.SessionID, crisp.ConversationTextMessageNew{
+			Type:    "text",
+			From:    "operator",
+			Origin:  "chat",
+			Content: update.Message.Text,
+		})
+	}
+
+	msg := tgbotapi.NewMessage(update.Message.Chat.ID, "回复成功！")
+	bot.Send(msg)
+}
+
+func sendMsgToAdmins(text string, WebsiteID string, SessionID string) {
+	for _, id := range config.Get("admins").([]interface{}) {
+		msg := tgbotapi.NewMessage(id.(int64), text)
+		msg.ParseMode = "Markdown"
+		sent, _ := bot.Send(msg)
+
+		redisClient.Set(strconv.Itoa(sent.MessageID), &CrispMessageInfo{
+			WebsiteID,
+			SessionID,
+		}, 12*time.Hour)
+	}
+}
+
+func init() {
+	config = utils.GetConfig()
+
+	log.Printf("Initializing Redis...")
+
+	redisClient = redis.NewClient(&redis.Options{
+		Addr:     config.GetString("redis.host"),
+		Password: config.GetString("redis.password"),
+		DB:       config.GetInt("redis.db"),
+	})
+
+	var err error
+
+	_, err = redisClient.Ping().Result()
+	if err != nil {
+		log.Panic(err)
+	}
+
+	log.Printf("Initializing Bot...")
+
+	bot, err = tgbotapi.NewBotAPI(config.GetString("telegram.key"))
+
+	if err != nil {
+		log.Panic(err)
+	}
+
+	bot.Debug = config.GetBool("debug")
+	bot.RemoveWebhook()
+
+	log.Printf("Authorized on account %s", bot.Self.UserName)
+
+	log.Printf("Initializing Crisp Listner")
+	client = crisp.New()
+	// Set authentication parameters
+	client.Authenticate(config.GetString("crisp.identifier"), config.GetString("crisp.key"))
+
+	// Connect to realtime events backend and listen (only to 'message:send' namespace)
+	client.Events.Listen(
+		[]string{
+			"message:send",
+		},
+
+		func(reg *crisp.EventsRegister) {
+			// Socket is connected: now listening for events
+
+			// Notice: if the realtime socket breaks at any point, this function will be called again upon reconnect (to re-bind events)
+			// Thus, ensure you only use this to register handlers
+
+			// Register handler on 'message:send/text' namespace
+			reg.On("message:send/text", func(evt crisp.EventsReceiveTextMessage) {
+				text := fmt.Sprintf(`*%s(%s): *%s`, *evt.User.Nickname, *evt.User.UserID, *evt.Content)
+				sendMsgToAdmins(text, *evt.WebsiteID, *evt.SessionID)
+			})
+
+			// Register handler on 'message:send/file' namespace
+			reg.On("message:send/file", func(evt crisp.EventsReceiveFileMessage) {
+				text := fmt.Sprintf(`*%s(%s): *[File](%s)`, *evt.User.Nickname, *evt.User.UserID, evt.Content.URL)
+				sendMsgToAdmins(text, *evt.WebsiteID, *evt.SessionID)
+			})
+
+			// Register handler on 'message:send/animation' namespace
+			reg.On("message:send/animation", func(evt crisp.EventsReceiveAnimationMessage) {
+				text := fmt.Sprintf(`*%s(%s): *[Animation](%s)`, *evt.User.Nickname, *evt.User.UserID, evt.Content.URL)
+				sendMsgToAdmins(text, *evt.WebsiteID, *evt.SessionID)
+			})
+		},
+
+		func() {
+			log.Printf("Crisp listener disconnected, reconnecting...")
+		},
+
+		func() {
+			log.Fatal("Crisp listener error, check your API key or internet connection?")
+		},
+	)
+}
+
+func main() {
+	var updates tgbotapi.UpdatesChannel
+
+	log.Print("Start pooling")
+	u := tgbotapi.NewUpdate(0)
+	u.Timeout = 60
+
+	updates, _ = bot.GetUpdatesChan(u)
+
+	for update := range updates {
+		if update.Message == nil {
+			continue
+		}
+
+		log.Printf("%s %s: %s", update.Message.From.FirstName, update.Message.From.LastName, update.Message.Text)
+
+		switch update.Message.Command() {
+		case "start":
+			msg := tgbotapi.NewMessage(update.Message.Chat.ID, "Blinkload Telegram 客服助手")
+			msg.ParseMode = "Markdown"
+			bot.Send(msg)
+		}
+
+		if contains(config.Get("admins").([]interface{}), int64(update.Message.From.ID)) {
+			replyToUser(&update)
+		}
+	}
+}

utils/config.go

@@ -0,0 +1,38 @@
+package utils
+
+import (
+	"log"
+	"strings"
+
+	"github.com/spf13/viper"
+)
+
+// GetConfig returns the global config
+func GetConfig() *viper.Viper {
+	c := viper.New()
+	c.SetConfigType("yaml")
+	c.SetConfigName("config")
+	c.AddConfigPath(".")
+	c.AutomaticEnv()
+
+	c.SetDefault("debug", true)
+	c.SetDefault("admins", []interface{}{})
+
+	c.SetDefault("redis.host", "localhost:6379")
+	c.SetDefault("redis.db", 0)
+	c.SetDefault("redis.password", "")
+
+	c.SetDefault("crisp.identifier", "")
+	c.SetDefault("crisp.key", "")
+
+	c.SetDefault("telegram.key", "")
+
+	replacer := strings.NewReplacer(".", "_")
+	c.SetEnvKeyReplacer(replacer)
+
+	if err := c.ReadInConfig(); err != nil {
+		log.Fatal(err.Error())
+	}
+
+	return c
+}