upd
This commit is contained in:
parent
fc52fc66f5
commit
d984dc4f7f
434
main.go
434
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
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user