From d984dc4f7fc5f79dc978dc40970c6e1a53f5dc78 Mon Sep 17 00:00:00 2001 From: shoopea Date: Fri, 3 May 2019 12:08:15 +0800 Subject: [PATCH] upd --- main.go | 434 ++++++++++++++++++++++++++++---------------------------- 1 file changed, 215 insertions(+), 219 deletions(-) diff --git a/main.go b/main.go index 187c983..0f325f3 100644 --- a/main.go +++ b/main.go @@ -1,219 +1,215 @@ -package main - -import ( - "fmt" - "log" - "time" - "encoding/json" - - "github.com/Arman92/go-tdlib" - "github.com/streadway/amqp" -) - -type MsgDirection int - -const ( - Incoming MsgDirection = 1 - Outgoing MsgDirection = 2 -) - -type ChatWarsMessage struct { - UserID64 int64 `json:"user_id"` - Direction MsgDirection `json:"direction"` - MsgText string `json:"msg"` - MsgID64 int64 `json:"msg_id"` -} - -const user_chtwrsbot = 408101137 - -var ownUserID64 = int64(0) -var ownUserID32 = int32(0) - - -func main() { -// msgMutex = &sync.Mutex{} - tdlib.SetLogVerbosityLevel(1) - tdlib.SetFilePath("./errors.txt") - - // Create new instance of client - client := tdlib.NewClient(tdlib.Config{ - APIID: "187786", - APIHash: "e782045df67ba48e441ccb105da8fc85", - SystemLanguageCode: "en", - DeviceModel: "ChatWarsClient", - SystemVersion: "0.1", - ApplicationVersion: "0.1", - UseMessageDatabase: true, - UseFileDatabase: true, - UseChatInfoDatabase: true, - UseTestDataCenter: false, - DatabaseDirectory: "./tdlib-db", - FileDirectory: "./tdlib-files", - IgnoreFileNames: false, - }) - - for { - currentState, _ := client.Authorize() - if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateWaitPhoneNumberType { - fmt.Print("Enter phone: ") - var number string - fmt.Scanln(&number) - _, err := client.SendPhoneNumber(number) - if err != nil { - fmt.Printf("Error sending phone number: %v", err) - } - } else if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateWaitCodeType { - fmt.Print("Enter code: ") - var code string - fmt.Scanln(&code) - _, err := client.SendAuthCode(code) - if err != nil { - fmt.Printf("Error sending auth code : %v", err) - } - } else if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateWaitPasswordType { - fmt.Print("Enter Password: ") - var password string - fmt.Scanln(&password) - _, err := client.SendAuthPassword(password) - if err != nil { - fmt.Printf("Error sending auth password: %v", err) - } - } else if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateReadyType { - fmt.Println("Authorization Ready! Let's rock") - break - } - } - - ownUserID32 = OwnUserID(client) - ownUserID64 = int64(OwnUserID(client)) - - go ListenCW(client) - - fmt.Println("Started !") - - // Main loop - for { - time.Sleep(30 * time.Second) - } - -} - -func ListenCW(c *tdlib.Client) { - eventFilter := func(msg *tdlib.TdMessage) bool { - updateMsg := (*msg).(*tdlib.UpdateNewMessage) - chatID := updateMsg.Message.ChatID - if (chatID == user_chtwrsbot) { - return true - } - return false - } - - conn, err := amqp.Dial("amqp://shoopea:UmDd5g4WRa2MzqOHsG2T@localhost:5672/chatwars_shoopea") - if err != nil { - log.Fatal(err) - } - defer conn.Close() - - ch, err := conn.Channel() - if err != nil { - log.Fatal(err) - } - defer ch.Close() - - q, err := ch.QueueDeclare( - "msg", // name - false, // durable - false, // delete when unused - false, // exclusive - false, // no-wait - nil, // arguments - ) - if err != nil { - log.Fatal(err) - } - - receiver := c.AddEventReceiver(&tdlib.UpdateNewMessage{}, eventFilter, 100) - for newMsg := range receiver.Chan { - updateMsg := (newMsg).(*tdlib.UpdateNewMessage) - senderUserID := updateMsg.Message.SenderUserID - mType := updateMsg.Message.Content.GetMessageContentEnum() - if mType == "messageText" { - user, err := c.GetUser(senderUserID) - if err != nil { - fmt.Println("ListenCW:", err.Error()) - continue - } - - txt := updateMsg.Message.Content.(*tdlib.MessageText).Text.Text - t := time.Now() - - m := ChatWarsMessage{ - UserID64 : ownUserID64, - MsgText : string(txt), - MsgID64 : updateMsg.Message.ID, - } - if senderUserID == ownUserID32 { - m.Direction = Outgoing - } else { - m.Direction = Incoming - } - - b, err := json.Marshal(m) - - _ = ch.Publish( - "", // exchange - q.Name, // routing key - false, // mandatory - false, // immediate - amqp.Publishing{ - ContentType: "application/json", - Body: []byte(b), - }) - - - - fmt.Printf("[%d-%02d-%02d %02d:%02d:%02d-00:00]", t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute(), t.Second()) - fmt.Println(" === CHATWARS (", user.Username, ") =====================================================================") - fmt.Println(txt, "\n") - fmt.Println("************ DETAILS ************") - fmt.Println("ID : ", updateMsg.Message.ID) - fmt.Println("SenderUserID : ", updateMsg.Message.SenderUserID) - fmt.Println("ChatID : ", updateMsg.Message.ChatID) - fmt.Println("SendingState : ", updateMsg.Message.SendingState) - fmt.Println("IsOutgoing : ", updateMsg.Message.IsOutgoing) - fmt.Println("CanBeEdited : ", updateMsg.Message.CanBeEdited) - fmt.Println("CanBeForwarded : ", updateMsg.Message.CanBeForwarded) - fmt.Println("IsChannelPost : ", updateMsg.Message.IsChannelPost) - fmt.Println("ContainsUnreadMention : ", updateMsg.Message.ContainsUnreadMention) - fmt.Println("ForwardInfo : ", updateMsg.Message.ForwardInfo) - fmt.Println("ReplyToMessageID : ", updateMsg.Message.ReplyToMessageID) - fmt.Println("ViaBotUserID : ", updateMsg.Message.ViaBotUserID) - fmt.Println("================================================================================================================") - - } - } -} - -func ForwardMsg(c *tdlib.Client, msgID int64, fromChatID int64, toChatID int64) (int64) { - msgIDs := make([]int64, 1) - msgIDs[0] = msgID - msgs, _ := c.ForwardMessages(toChatID, fromChatID, msgIDs, false, false, false) - if msgs != nil { - return msgs.Messages[0].ID - } else { - return 0 - } -} - -func DeleteMsg(c *tdlib.Client, chatID int64, msgID int64) { - msgIDs := make([]int64, 1) - msgIDs[0] = msgID - c.DeleteMessages(chatID, msgIDs, false) - fmt.Println("Deleting message ", msgID) -} - -func OwnUserID(c *tdlib.Client) (int32) { - user, _ := c.GetMe() - return user.ID; -} - +package main + +import ( + "encoding/json" + "fmt" + "log" + "time" + + "github.com/Arman92/go-tdlib" + "github.com/streadway/amqp" +) + +type MsgDirection int + +const ( + Incoming MsgDirection = 1 + Outgoing MsgDirection = 2 +) + +type ChatWarsMessage struct { + MsgID64 int64 `json:"msg_id"` + UserID64 int64 `json:"user_id"` + Direction MsgDirection `json:"direction"` + MsgText string `json:"msg"` +} + +const user_chtwrsbot = 408101137 + +var ownUserID64 = int64(0) +var ownUserID32 = int32(0) + +func main() { + // msgMutex = &sync.Mutex{} + tdlib.SetLogVerbosityLevel(1) + tdlib.SetFilePath("./errors.txt") + + // Create new instance of client + client := tdlib.NewClient(tdlib.Config{ + APIID: "187786", + APIHash: "e782045df67ba48e441ccb105da8fc85", + SystemLanguageCode: "en", + DeviceModel: "ChatWarsClient", + SystemVersion: "0.1", + ApplicationVersion: "0.1", + UseMessageDatabase: true, + UseFileDatabase: true, + UseChatInfoDatabase: true, + UseTestDataCenter: false, + DatabaseDirectory: "./tdlib-db", + FileDirectory: "./tdlib-files", + IgnoreFileNames: false, + }) + + for { + currentState, _ := client.Authorize() + if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateWaitPhoneNumberType { + fmt.Print("Enter phone: ") + var number string + fmt.Scanln(&number) + _, err := client.SendPhoneNumber(number) + if err != nil { + fmt.Printf("Error sending phone number: %v", err) + } + } else if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateWaitCodeType { + fmt.Print("Enter code: ") + var code string + fmt.Scanln(&code) + _, err := client.SendAuthCode(code) + if err != nil { + fmt.Printf("Error sending auth code : %v", err) + } + } else if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateWaitPasswordType { + fmt.Print("Enter Password: ") + var password string + fmt.Scanln(&password) + _, err := client.SendAuthPassword(password) + if err != nil { + fmt.Printf("Error sending auth password: %v", err) + } + } else if currentState.GetAuthorizationStateEnum() == tdlib.AuthorizationStateReadyType { + fmt.Println("Authorization Ready! Let's rock") + break + } + } + + ownUserID32 = OwnUserID(client) + ownUserID64 = int64(OwnUserID(client)) + + go ListenCW(client) + + fmt.Println("Started !") + + // Main loop + for { + time.Sleep(30 * time.Second) + } + +} + +func ListenCW(c *tdlib.Client) { + eventFilter := func(msg *tdlib.TdMessage) bool { + updateMsg := (*msg).(*tdlib.UpdateNewMessage) + chatID := updateMsg.Message.ChatID + if chatID == user_chtwrsbot { + return true + } + return false + } + + conn, err := amqp.Dial("amqp://shoopea:UmDd5g4WRa2MzqOHsG2T@localhost:5672/chatwars") + if err != nil { + log.Fatal(err) + } + defer conn.Close() + + ch, err := conn.Channel() + if err != nil { + log.Fatal(err) + } + defer ch.Close() + + q, err := ch.QueueDeclare( + "msg", // name + false, // durable + false, // delete when unused + false, // exclusive + false, // no-wait + nil, // arguments + ) + if err != nil { + log.Fatal(err) + } + + receiver := c.AddEventReceiver(&tdlib.UpdateNewMessage{}, eventFilter, 100) + for newMsg := range receiver.Chan { + updateMsg := (newMsg).(*tdlib.UpdateNewMessage) + senderUserID := updateMsg.Message.SenderUserID + mType := updateMsg.Message.Content.GetMessageContentEnum() + if mType == "messageText" { + user, err := c.GetUser(senderUserID) + if err != nil { + fmt.Println("ListenCW:", err.Error()) + continue + } + + txt := updateMsg.Message.Content.(*tdlib.MessageText).Text.Text + t := time.Now() + + m := ChatWarsMessage{ + UserID64: ownUserID64, + MsgText: string(txt), + MsgID64: updateMsg.Message.ID, + } + if senderUserID == ownUserID32 { + m.Direction = Outgoing + } else { + m.Direction = Incoming + } + + b, err := json.Marshal(m) + + _ = ch.Publish( + "", // exchange + q.Name, // routing key + false, // mandatory + false, // immediate + amqp.Publishing{ + ContentType: "application/json", + Body: []byte(b), + }) + + fmt.Printf("[%d-%02d-%02d %02d:%02d:%02d-00:00]", t.Year(), t.Month(), t.Day(), t.Hour(), t.Minute(), t.Second()) + fmt.Println(" === CHATWARS (", user.Username, ") =====================================================================") + fmt.Println(txt, "\n") + fmt.Println("************ DETAILS ************") + fmt.Println("ID : ", updateMsg.Message.ID) + fmt.Println("SenderUserID : ", updateMsg.Message.SenderUserID) + fmt.Println("ChatID : ", updateMsg.Message.ChatID) + fmt.Println("SendingState : ", updateMsg.Message.SendingState) + fmt.Println("IsOutgoing : ", updateMsg.Message.IsOutgoing) + fmt.Println("CanBeEdited : ", updateMsg.Message.CanBeEdited) + fmt.Println("CanBeForwarded : ", updateMsg.Message.CanBeForwarded) + fmt.Println("IsChannelPost : ", updateMsg.Message.IsChannelPost) + fmt.Println("ContainsUnreadMention : ", updateMsg.Message.ContainsUnreadMention) + fmt.Println("ForwardInfo : ", updateMsg.Message.ForwardInfo) + fmt.Println("ReplyToMessageID : ", updateMsg.Message.ReplyToMessageID) + fmt.Println("ViaBotUserID : ", updateMsg.Message.ViaBotUserID) + fmt.Println("================================================================================================================") + + } + } +} + +func ForwardMsg(c *tdlib.Client, msgID int64, fromChatID int64, toChatID int64) int64 { + msgIDs := make([]int64, 1) + msgIDs[0] = msgID + msgs, _ := c.ForwardMessages(toChatID, fromChatID, msgIDs, false, false, false) + if msgs != nil { + return msgs.Messages[0].ID + } else { + return 0 + } +} + +func DeleteMsg(c *tdlib.Client, chatID int64, msgID int64) { + msgIDs := make([]int64, 1) + msgIDs[0] = msgID + c.DeleteMessages(chatID, msgIDs, false) + fmt.Println("Deleting message ", msgID) +} + +func OwnUserID(c *tdlib.Client) int32 { + user, _ := c.GetMe() + return user.ID +}