1
0
mirror of synced 2025-02-24 16:03:14 +03:00
2020-03-27 15:11:00 +03:00

1179 lines
26 KiB
Go

package main
import (
"bytes"
"encoding/json"
"fmt"
"image/png"
"io"
"io/ioutil"
"net/http"
"strconv"
"strings"
"time"
"github.com/gin-gonic/gin"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
"github.com/h2non/filetype"
filetypes "github.com/h2non/filetype/matchers"
v5 "github.com/retailcrm/api-client-go/v5"
v1 "github.com/retailcrm/mg-transport-api-client-go/v1"
"golang.org/x/image/webp"
)
func connectHandler(c *gin.Context) {
res := struct {
Conn Connection
Locale map[string]interface{}
Year int
}{
c.MustGet("account").(Connection),
getLocale(),
time.Now().Year(),
}
c.HTML(http.StatusOK, "home", &res)
}
func addBotHandler(c *gin.Context) {
b := c.MustGet("bot").(Bot)
cl, err := getBotByToken(b.Token)
if err != nil {
c.Error(err)
return
}
if cl.ID != 0 {
c.AbortWithStatusJSON(BadRequest("bot_already_created"))
return
}
bot, err := tgbotapi.NewBotAPI(b.Token)
if err != nil {
c.AbortWithStatusJSON(BadRequest("incorrect_token"))
logger.Error(b.Token, err.Error())
return
}
bot.Debug = config.Debug
wr, err := bot.SetWebhook(tgbotapi.NewWebhook("https://" + config.HTTPServer.Host + "/telegram/" + bot.Token))
if err != nil || !wr.Ok {
c.AbortWithStatusJSON(BadRequest("error_creating_webhook"))
logger.Error(b.Token, err.Error(), wr)
return
}
b.Name = bot.Self.UserName
conn := getConnectionById(b.ConnectionID)
client := v1.New(conn.MGURL, conn.MGToken)
client.Debug = config.Debug
channelSettings := getChannelSettings()
if b.Name != "" {
channelSettings.Name = "@" + b.Name
}
data, status, err := client.ActivateTransportChannel(channelSettings)
if status != http.StatusCreated {
c.AbortWithStatusJSON(BadRequest("error_activating_channel"))
logger.Error(conn.APIURL, status, err.Error(), data)
return
}
b.Channel = data.ChannelID
b.Lang = "en"
hashSettings, err := getChannelSettingsHash()
if err != nil {
logger.Errorf("addBotHandler hashSettings apiURl: %s, err: %s", conn.APIURL, err.Error())
} else {
b.ChannelSettingsHash = hashSettings
}
err = conn.createBot(b)
if err != nil {
client.DeactivateTransportChannel(data.ChannelID)
c.Error(err)
return
}
c.JSON(http.StatusCreated, b)
}
func deleteBotHandler(c *gin.Context) {
b := c.MustGet("bot").(Bot)
conn := getConnectionById(b.ConnectionID)
if conn.MGURL == "" || conn.MGToken == "" {
c.AbortWithStatusJSON(BadRequest("not_found_account"))
return
}
var client = v1.New(conn.MGURL, conn.MGToken)
client.Debug = config.Debug
data, status, err := client.DeactivateTransportChannel(getBotChannelByToken(b.Token))
if status > http.StatusOK {
c.AbortWithStatusJSON(BadRequest("error_deactivating_channel"))
logger.Error(b.ID, status, err.Error(), data)
return
}
err = b.deleteBot()
if err != nil {
c.Error(err)
return
}
c.JSON(http.StatusOK, gin.H{})
}
func settingsHandler(c *gin.Context) {
uid := c.Param("uid")
p := getConnection(uid)
if p.ID == 0 {
c.Redirect(http.StatusFound, "/")
return
}
bots := p.getBotsByClientID()
res := struct {
Conn *Connection
Bots Bots
Locale map[string]interface{}
Year int
LangCode []string
}{
p,
bots,
getLocale(),
time.Now().Year(),
[]string{"en", "ru", "es"},
}
c.HTML(http.StatusOK, "form", &res)
}
func saveHandler(c *gin.Context) {
conn := c.MustGet("connection").(Connection)
_, err, code := getAPIClient(conn.APIURL, conn.APIKEY)
if err != nil {
if code == http.StatusInternalServerError {
c.Error(err)
} else {
c.AbortWithStatusJSON(code, gin.H{"error": err.Error()})
}
return
}
err = conn.saveConnectionByClientID()
if err != nil {
c.Error(err)
return
}
c.JSON(http.StatusOK, gin.H{"message": getLocalizedMessage("successful")})
}
func createHandler(c *gin.Context) {
conn := c.MustGet("connection").(Connection)
cl := getConnectionByURL(conn.APIURL)
if cl.ID != 0 {
c.AbortWithStatusJSON(BadRequest("connection_already_created"))
return
}
client, err, code := getAPIClient(conn.APIURL, conn.APIKEY)
if err != nil {
if code == http.StatusInternalServerError {
c.Error(err)
} else {
c.AbortWithStatusJSON(code, gin.H{"error": err.Error()})
}
return
}
conn.ClientID = GenerateToken()
data, status, errr := client.IntegrationModuleEdit(getIntegrationModule(conn.ClientID))
if errr.RuntimeErr != nil {
c.Error(errr.RuntimeErr)
return
}
if status == http.StatusPaymentRequired {
c.AbortWithStatusJSON(BadRequest("error_payment_mg"))
logger.Error(conn.APIURL, status, errr.ApiErr, data)
return
}
if status >= http.StatusBadRequest {
c.AbortWithStatusJSON(BadRequest("error_activity_mg"))
logger.Error(conn.APIURL, status, errr.ApiErr, data)
return
}
conn.MGURL = data.Info.MgTransportInfo.EndpointUrl
conn.MGToken = data.Info.MgTransportInfo.Token
conn.Active = true
err = conn.createConnection()
if err != nil {
c.Error(err)
return
}
c.JSON(
http.StatusCreated,
gin.H{
"url": "/settings/" + conn.ClientID,
"message": getLocalizedMessage("successful"),
},
)
}
func activityHandler(c *gin.Context) {
var (
activity v5.Activity
systemUrl = c.PostForm("systemUrl")
clientId = c.PostForm("clientId")
)
conn := getConnection(clientId)
if conn.ID == 0 {
c.AbortWithStatusJSON(http.StatusBadRequest,
gin.H{
"success": false,
"error": "Wrong data",
},
)
return
}
err := json.Unmarshal([]byte(c.PostForm("activity")), &activity)
if err != nil {
c.AbortWithStatusJSON(http.StatusBadRequest,
gin.H{
"success": false,
"error": "Wrong data",
},
)
return
}
conn.Active = activity.Active && !activity.Freeze
if systemUrl != "" {
conn.APIURL = systemUrl
}
hashSettings, err := getChannelSettingsHash()
if err != nil {
logger.Errorf("activityHandler hashSettings apiURl: %s, err: %s", conn.APIURL, err.Error())
} else {
updateBots(conn, hashSettings)
}
if err := conn.saveConnection(); err != nil {
c.Error(err)
return
}
c.JSON(http.StatusOK, gin.H{"success": true})
}
func setLangBotHandler(c *gin.Context) {
b := c.MustGet("bot").(Bot)
cl, err := getBotByToken(b.Token)
if err != nil {
c.Error(err)
return
}
cl.Lang = b.Lang
err = cl.save()
if err != nil {
c.Error(err)
return
}
c.JSON(http.StatusOK, gin.H{})
}
func getIntegrationModule(clientId string) v5.IntegrationModule {
return v5.IntegrationModule{
Code: config.TransportInfo.Code,
IntegrationCode: config.TransportInfo.Code,
Active: true,
Name: config.TransportInfo.Name,
ClientID: clientId,
Logo: fmt.Sprintf(
"https://%s%s",
config.HTTPServer.Host,
config.TransportInfo.LogoPath,
),
BaseURL: fmt.Sprintf(
"https://%s",
config.HTTPServer.Host,
),
AccountURL: fmt.Sprintf(
"https://%s/settings/%s",
config.HTTPServer.Host,
clientId,
),
Actions: map[string]string{"activity": "/actions/activity"},
Integrations: &v5.Integrations{
MgTransport: &v5.MgTransport{
WebhookUrl: fmt.Sprintf(
"https://%s/webhook/",
config.HTTPServer.Host,
),
},
},
}
}
func getChannelSettings(cid ...uint64) v1.Channel {
var channelID uint64
if len(cid) > 0 {
channelID = cid[0]
}
return v1.Channel{
ID: channelID,
Type: Type,
Settings: v1.ChannelSettings{
SpamAllowed: false,
Status: v1.Status{
Delivered: v1.ChannelFeatureSend,
Read: v1.ChannelFeatureNone,
},
Text: v1.ChannelSettingsText{
Creating: v1.ChannelFeatureBoth,
Editing: v1.ChannelFeatureBoth,
Quoting: v1.ChannelFeatureBoth,
Deleting: v1.ChannelFeatureReceive,
MaxCharsCount: MaxCharsCount,
},
Product: v1.Product{
Creating: v1.ChannelFeatureReceive,
Editing: v1.ChannelFeatureReceive,
},
Order: v1.Order{
Creating: v1.ChannelFeatureReceive,
Editing: v1.ChannelFeatureReceive,
},
File: v1.ChannelSettingsFilesBase{
Creating: v1.ChannelFeatureBoth,
Editing: v1.ChannelFeatureBoth,
Quoting: v1.ChannelFeatureBoth,
Deleting: v1.ChannelFeatureReceive,
Max: 1,
},
Image: v1.ChannelSettingsFilesBase{
Creating: v1.ChannelFeatureBoth,
Editing: v1.ChannelFeatureBoth,
Quoting: v1.ChannelFeatureBoth,
Deleting: v1.ChannelFeatureReceive,
Max: 10,
},
},
}
}
func updateChannelsSettings() {
hashSettings, err := getChannelSettingsHash()
if err != nil {
logger.Error(err.Error())
return
}
connections := getConnections()
if len(connections) > 0 {
for _, conn := range connections {
if !conn.Active {
logger.Infof(
"updateChannelsSettings connection %s deactivated",
conn.APIURL,
)
continue
}
updateBots(conn, hashSettings)
}
}
return
}
func updateBots(conn *Connection, hashSettings string) {
var channelIDs []uint64
bots := conn.getBotsByClientID()
if len(bots) > 0 {
client := v1.New(conn.MGURL, conn.MGToken)
client.Debug = config.Debug
for _, bot := range bots {
channelIDs = append(channelIDs, bot.Channel)
if bot.ChannelSettingsHash == hashSettings {
continue
}
channelSettings := getChannelSettings(bot.Channel)
if bot.Name != "" {
channelSettings.Name = "@" + bot.Name
}
data, status, err := client.UpdateTransportChannel(channelSettings)
if config.Debug {
logger.Infof(
"updateChannelsSettings apiURL: %s, ChannelID: %d, Data: %v, Status: %d, err: %v",
conn.APIURL, bot.Channel, data, status, err,
)
}
if err == nil {
bot.ChannelSettingsHash = hashSettings
err = bot.save()
if err != nil {
logger.Error(
"updateChannelsSettings bot.save apiURL: %s, bot.Channel: %d , err: %v",
conn.APIURL, bot.Channel, err,
)
}
}
}
deactivateChannels(client, channelIDs)
}
return
}
func deactivateChannels(client *v1.MgClient, channelIDs []uint64) {
channelListItems, status, err := client.TransportChannels(v1.Channels{Active: true})
if config.Debug {
logger.Debugf(
"TransportChannels ChannelListItems: %+v, Status: %d, err: %v",
channelListItems, status, err,
)
}
if len(channelListItems) > 0 {
for _, channel := range channelIDs {
for key, ch := range channelListItems {
if channel == ch.ID {
if len(channelListItems) == 1 {
channelListItems = channelListItems[:0]
break
}
channelListItems = append(channelListItems[:key], channelListItems[key+1:]...)
}
}
}
}
if len(channelListItems) > 0 {
for _, ch := range channelListItems {
channelListItems, status, err := client.DeactivateTransportChannel(ch.ID)
if config.Debug {
logger.Debugf(
"DeactivateTransportChannel ChannelListItems: %+v, Status: %d, err: %v",
channelListItems, status, err,
)
}
}
}
}
func telegramWebhookHandler(c *gin.Context) {
b := c.MustGet("bot").(Bot)
conn := getConnectionById(b.ConnectionID)
if !conn.Active {
c.AbortWithStatus(http.StatusOK)
return
}
var update tgbotapi.Update
if err := c.ShouldBindJSON(&update); err != nil {
c.Error(err)
return
}
if config.Debug {
logger.Debugf(
"mgWebhookHandler request:\nUpdateID: %v,\nMessage: %+v,\nEditedMessage: %+v",
update.UpdateID, update.Message, update.EditedMessage,
)
}
if update.Message != nil && shouldMessageBeIgnored(update.Message) {
logger.Infof("telegramWebhookHandler ignoring unprocessable message %+v", update.Message)
return
}
var client = v1.New(conn.MGURL, conn.MGToken)
client.Debug = config.Debug
if update.Message != nil {
nickname := update.Message.From.UserName
user := getUserByExternalID(update.Message.From.ID)
if update.Message.From.UserName == "" {
nickname = update.Message.From.FirstName
}
if user.Expired(config.UpdateInterval) || user.ID == 0 {
fileID, fileURL, err := GetFileIDAndURL(b.Token, update.Message.From.ID)
if err != nil {
c.Error(err)
return
}
if fileID != user.UserPhotoID && fileURL != "" {
picURL, err := UploadUserAvatar(fileURL)
if err != nil {
c.Error(err)
return
}
user.UserPhotoID = fileID
user.UserPhotoURL = picURL
}
if user.ExternalID == 0 {
user.ExternalID = update.Message.From.ID
}
err = user.save()
if err != nil {
c.Error(err)
return
}
}
lang := update.Message.From.LanguageCode
if len(update.Message.From.LanguageCode) > 2 {
lang = update.Message.From.LanguageCode[:2]
}
if config.Debug {
logger.Debugf("telegramWebhookHandler user %+v", user)
}
snd := v1.SendData{
Message: v1.Message{
ExternalID: strconv.Itoa(update.Message.MessageID),
Type: "text",
Text: update.Message.Text,
},
Originator: v1.OriginatorCustomer,
Customer: v1.Customer{
ExternalID: strconv.Itoa(update.Message.From.ID),
Nickname: nickname,
Firstname: update.Message.From.FirstName,
Avatar: user.UserPhotoURL,
Lastname: update.Message.From.LastName,
Language: lang,
},
Channel: b.Channel,
ExternalChatID: strconv.FormatInt(update.Message.Chat.ID, 10),
}
if update.Message.ReplyToMessage != nil {
snd.Quote = &v1.SendMessageRequestQuote{ExternalID: strconv.Itoa(update.Message.ReplyToMessage.MessageID)}
}
if snd.Message.Text == "" {
setLocale(update.Message.From.LanguageCode)
err := setAttachment(update.Message, client, &snd, b.Token)
if err != nil {
logger.Error(client.Token, err.Error())
c.AbortWithStatus(http.StatusBadRequest)
return
}
}
data, st, err := client.Messages(snd)
if err != nil {
logger.Error(b.Token, err.Error(), st, data)
if update.Message.ReplyToMessage != nil {
c.AbortWithStatus(http.StatusOK)
} else if st == http.StatusBadRequest && err.Error() == "Message with passed external_id already exists" {
logger.Errorf("Message with externalId '%s' is already exists - ignoring it", snd.Message.ExternalID)
c.JSON(http.StatusOK, gin.H{})
} else {
c.Error(err)
}
return
}
if config.Debug {
logger.Debugf("telegramWebhookHandler Type: SendMessage, Bot: %v, Message: %+v, Response: %+v", b.ID, snd, data)
}
}
if update.EditedMessage != nil {
if update.EditedMessage.Text == "" {
if getMessageID(update.EditedMessage) != "undefined" {
if config.Debug {
logger.Debug(b.Token, update.EditedMessage, "Only text messages can be updated")
}
c.JSON(http.StatusOK, gin.H{})
return
}
setLocale(update.EditedMessage.From.LanguageCode)
update.EditedMessage.Text = getLocalizedMessage(getMessageID(update.Message))
}
snd := v1.EditMessageRequest{
Message: v1.EditMessageRequestMessage{
ExternalID: strconv.Itoa(update.EditedMessage.MessageID),
Text: update.EditedMessage.Text,
},
Channel: b.Channel,
}
data, st, err := client.UpdateMessages(snd)
if err != nil {
logger.Error(b.Token, err.Error(), st, data)
c.Error(err)
return
}
if config.Debug {
logger.Debugf("telegramWebhookHandler Type: UpdateMessage, Bot: %v, Message: %v, Response: %v", b.ID, snd, data)
}
}
c.JSON(http.StatusOK, gin.H{})
}
func mgWebhookHandler(c *gin.Context) {
conn := c.MustGet("connection").(Connection)
var msg v1.WebhookRequest
if err := c.ShouldBindJSON(&msg); err != nil {
c.Error(err)
return
}
if config.Debug {
logger.Debugf("mgWebhookHandler request: %+v", msg)
}
uid, _ := strconv.Atoi(msg.Data.ExternalMessageID)
cid, _ := strconv.ParseInt(msg.Data.ExternalChatID, 10, 64)
b := getBot(conn.ID, msg.Data.ChannelID)
if b.ID == 0 {
c.AbortWithStatus(http.StatusBadRequest)
return
}
bot, err := tgbotapi.NewBotAPI(b.Token)
if err != nil {
logger.Error(b, err)
c.AbortWithStatus(http.StatusBadRequest)
return
}
bot.Debug = config.Debug
setLocale(b.Lang)
mgClient := v1.New(conn.MGURL, conn.MGToken)
switch msg.Type {
case "message_sent":
var mb string
var m tgbotapi.Chattable
switch msg.Data.Type {
case v1.MsgTypeProduct:
mb = fmt.Sprintf("*%s*\n", replaceMarkdownSymbols(msg.Data.Product.Name))
if msg.Data.Product.Cost != nil && msg.Data.Product.Cost.Value != 0 {
mb += fmt.Sprintf(
"\n%s: %s\n",
getLocalizedMessage("item_cost"),
getLocalizedTemplateMessage(
"cost_currency",
map[string]interface{}{
"Amount": msg.Data.Product.Cost.Value,
"Currency": currency[strings.ToLower(msg.Data.Product.Cost.Currency)],
},
),
)
}
if msg.Data.Product.Url != "" {
mb += replaceMarkdownSymbols(msg.Data.Product.Url)
} else {
mb += replaceMarkdownSymbols(msg.Data.Product.Img)
}
case v1.MsgTypeOrder:
mb = getOrderMessage(msg.Data.Order)
case v1.MsgTypeText:
mb = replaceMarkdownSymbols(msg.Data.Content)
case v1.MsgTypeImage:
m, err = photoMessage(msg.Data, mgClient, cid)
if err != nil {
logger.Errorf(
"GetFile request apiURL: %s, clientID: %s, err: %s",
conn.APIURL, conn.ClientID, err.Error(),
)
return
}
case v1.MsgTypeFile:
items := *msg.Data.Items
if len(items) > 0 {
m, err = documentMessage(items[0], mgClient, cid)
if err != nil {
logger.Errorf(
"GetFile request apiURL: %s, clientID: %s, err: %s",
conn.APIURL, conn.ClientID, err.Error(),
)
return
}
}
}
if mb != "" {
m, err = textMessage(cid, mb, msg.Data.QuoteExternalID)
if err != nil {
c.Error(err)
return
}
}
msgSend, err := bot.Send(m)
if err != nil {
logger.Error(err)
c.AbortWithStatus(http.StatusBadRequest)
return
}
if config.Debug {
logger.Debugf("mgWebhookHandler sent %+v", msgSend)
}
c.JSON(http.StatusOK, gin.H{"external_message_id": strconv.Itoa(msgSend.MessageID)})
case "message_updated":
msgSend, err := bot.Send(tgbotapi.NewEditMessageText(cid, uid, replaceMarkdownSymbols(msg.Data.Content)))
if err != nil {
logger.Error(err)
c.AbortWithStatus(http.StatusBadRequest)
return
}
if config.Debug {
logger.Debugf("mgWebhookHandler update %+v", msgSend)
}
c.AbortWithStatus(http.StatusOK)
case "message_deleted":
msgSend, err := bot.Send(tgbotapi.NewDeleteMessage(cid, uid))
if err != nil {
logger.Error(err)
c.AbortWithStatus(http.StatusBadRequest)
return
}
if config.Debug {
logger.Debugf("mgWebhookHandler delete %+v", msgSend)
}
c.JSON(http.StatusOK, gin.H{})
}
}
func getOrderMessage(dataOrder *v1.MessageDataOrder) string {
mb := "*" + getLocalizedMessage("order")
if dataOrder.Number != "" {
mb += " " + replaceMarkdownSymbols(dataOrder.Number)
}
if dataOrder.Date != "" {
mb += fmt.Sprintf(" (%s)", dataOrder.Date)
}
mb += "*\n"
if len(dataOrder.Items) > 0 {
mb += "\n"
for k, v := range dataOrder.Items {
mb += fmt.Sprintf(
"%d. %s",
k+1,
replaceMarkdownSymbols(v.Name),
)
if v.Quantity != nil {
if v.Quantity.Value != 0 {
mb += fmt.Sprintf(
" _%v_",
v.Quantity.Value,
)
}
}
if v.Price != nil {
if val, ok := currency[strings.ToLower(v.Price.Currency)]; ok {
mb += fmt.Sprintf(
" _x %s_\n",
getLocalizedTemplateMessage(
"cost_currency",
map[string]interface{}{
"Amount": v.Price.Value,
"Currency": val,
},
),
)
}
} else {
mb += "\n"
}
}
}
if dataOrder.Delivery != nil {
if dataOrder.Delivery.Name != "" {
mb += fmt.Sprintf(
"\n*%s:*\n%s",
getLocalizedMessage("delivery"),
replaceMarkdownSymbols(dataOrder.Delivery.Name),
)
}
if dataOrder.Delivery.Price != nil {
if val, ok := currency[strings.ToLower(dataOrder.Delivery.Price.Currency)]; ok && dataOrder.Delivery.Price.Value != 0 {
mb += fmt.Sprintf(
"; %s",
getLocalizedTemplateMessage(
"cost_currency",
map[string]interface{}{
"Amount": dataOrder.Delivery.Price.Value,
"Currency": val,
},
),
)
}
}
if dataOrder.Delivery.Address != "" {
mb += ";\n" + replaceMarkdownSymbols(dataOrder.Delivery.Address)
}
if dataOrder.Delivery.Comment != "" {
mb += ";\n" + replaceMarkdownSymbols(dataOrder.Delivery.Comment)
}
mb += "\n"
}
if len(dataOrder.Payments) > 0 {
mb += fmt.Sprintf(
"\n*%s:*\n",
getLocalizedMessage("payment"),
)
for _, v := range dataOrder.Payments {
mb += replaceMarkdownSymbols(v.Name)
if v.Amount != nil {
if val, ok := currency[strings.ToLower(v.Amount.Currency)]; ok && v.Amount.Value != 0 {
mb += fmt.Sprintf(
"; %s",
getLocalizedTemplateMessage(
"cost_currency",
map[string]interface{}{
"Amount": v.Amount.Value,
"Currency": val,
},
),
)
}
}
if v.Status != nil && v.Status.Name != "" {
mb += fmt.Sprintf(
" (%s)",
replaceMarkdownSymbols(v.Status.Name),
)
}
mb += "\n"
}
}
if dataOrder.Cost != nil {
if val, ok := currency[strings.ToLower(dataOrder.Cost.Currency)]; ok && dataOrder.Cost.Value != 0 {
mb += fmt.Sprintf(
"\n%s: %s",
getLocalizedMessage("order_total"),
getLocalizedTemplateMessage(
"cost_currency",
map[string]interface{}{
"Amount": dataOrder.Cost.Value,
"Currency": val,
},
),
)
}
}
return mb
}
func photoMessage(webhookData v1.WebhookData, mgClient *v1.MgClient, cid int64) (chattable tgbotapi.Chattable, err error) {
items := *webhookData.Items
if len(items) == 1 {
v := items
file, _, err := mgClient.GetFile(v[0].ID)
if err != nil {
return chattable, err
}
msg := tgbotapi.NewPhotoUpload(cid, nil)
msg.FileID = file.Url
msg.UseExisting = true
msg.Caption = webhookData.Content
chattable = msg
} else if len(items) > 1 {
var it []interface{}
for _, v := range items {
file, _, err := mgClient.GetFile(v.ID)
if err != nil {
logger.Errorf(
"GetFile request fileID: %s, err: %s",
v.ID, err.Error(),
)
continue
}
ip := tgbotapi.NewInputMediaPhoto(file.Url)
ip.Caption = webhookData.Content
it = append(it, ip)
}
chattable = tgbotapi.NewMediaGroup(cid, it)
}
return
}
func documentMessage(item v1.FileItem, mgClient *v1.MgClient, cid int64) (chattable tgbotapi.Chattable, err error) {
file, _, err := mgClient.GetFile(item.ID)
if err != nil {
return chattable, err
}
data, err := http.Get(file.Url)
if err != nil {
return chattable, err
}
tt := tgbotapi.FileReader{
Name: item.Caption,
Reader: data.Body,
Size: int64(item.Size),
}
chattable = tgbotapi.NewDocumentUpload(cid, tt)
return
}
func textMessage(cid int64, mb string, quoteExternalID string) (chattable tgbotapi.Chattable, err error) {
var qid int
m := tgbotapi.NewMessage(cid, mb)
if quoteExternalID != "" {
qid, err = strconv.Atoi(quoteExternalID)
if err != nil {
return
}
m.ReplyToMessageID = qid
}
m.ParseMode = "Markdown"
chattable = m
return
}
func setAttachment(attachments *tgbotapi.Message, client *v1.MgClient, snd *v1.SendData, botToken string) error {
var (
items []v1.Item
fileID string
)
t := getMessageID(attachments)
bot, err := tgbotapi.NewBotAPI(botToken)
if err != nil {
return err
}
caption := getLocalizedMessage(t)
switch t {
case "photo":
for _, v := range *attachments.Photo {
fileID = v.FileID
}
snd.Message.Type = v1.MsgTypeImage
snd.Message.Note = attachments.Caption
case "animation":
fileID = attachments.Animation.FileID
snd.Message.Type = v1.MsgTypeFile
caption += ".mp4"
case "document":
fileID = attachments.Document.FileID
snd.Message.Type = v1.MsgTypeFile
caption = attachments.Document.FileName
case "sticker":
fileID = attachments.Sticker.FileID
snd.Message.Type = v1.MsgTypeImage
case "voice":
fileID = attachments.Voice.FileID
snd.Message.Type = v1.MsgTypeAudio
default:
snd.Message.Text = getLocalizedMessage(t)
}
if fileID != "" {
file, err := getFileURL(fileID, bot)
if err != nil {
return err
}
item := v1.Item{}
fileUrl := fmt.Sprintf("https://api.telegram.org/file/bot%s/%s", botToken, file.FilePath)
switch {
case t == "sticker" || t == "voice":
item, _, err = getItemData(
client,
fileUrl,
caption,
)
if err != nil {
return err
}
case t == "animation":
item, _, err = getItemData(
client,
fileUrl,
caption,
)
if err != nil {
return err
}
item.Caption = item.ID + ".mp4"
default:
item, err = convertAndUploadImage(
client,
fileUrl,
)
if err != nil {
return err
}
}
items = append(items, item)
}
if len(items) > 0 {
snd.Message.Items = items
snd.Message.Text = attachments.Caption
}
return nil
}
func getItemData(client *v1.MgClient, url string, caption string) (v1.Item, int, error) {
item := v1.Item{}
data, st, err := client.UploadFileByURL(
v1.UploadFileByUrlRequest{
Url: url,
},
)
if err != nil {
return item, st, err
}
item.ID = data.ID
item.Caption = caption
return item, st, err
}
func getFileURL(fileID string, b *tgbotapi.BotAPI) (tgbotapi.File, error) {
return b.GetFile(tgbotapi.FileConfig{FileID: fileID})
}
func convertAndUploadImage(client *v1.MgClient, url string) (v1.Item, error) {
item := v1.Item{}
res, err := http.Get(url)
if err != nil {
return item, err
}
imgByte, err := ioutil.ReadAll(res.Body)
if err != nil {
return item, err
}
if kind, err := filetype.Match(imgByte); err != nil {
return item, err
} else if kind == filetypes.TypeWebp {
img, err := webp.Decode(res.Body)
if err != nil {
return item, err
}
pReader, pWriter := io.Pipe()
go func() {
defer pWriter.Close()
err = png.Encode(pWriter, img)
if err != nil {
logger.Info(item, err.Error())
}
}()
data, _, err := client.UploadFile(pReader)
if err != nil {
return item, err
}
item.ID = data.ID
} else {
data, _, err := client.UploadFile(bytes.NewReader(imgByte))
if err != nil {
return item, err
}
item.ID = data.ID
}
return item, nil
}