package main import ( "net/http" "fmt" "github.com/gin-gonic/gin" "github.com/go-telegram-bot-api/telegram-bot-api" "github.com/retailcrm/api-client-go/v5" "github.com/retailcrm/mg-transport-api-client-go/v1" ) func connectHandler(c *gin.Context) { res := struct { Conn Connection Locale map[string]string }{ c.MustGet("account").(Connection), getLocale(), } 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(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("bot_already_created")}) return } bot, err := tgbotapi.NewBotAPI(b.Token) if err != nil { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("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 { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("error_creating_webhook")}) logger.Error(b.Token, err.Error()) return } if !wr.Ok { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("error_creating_webhook")}) logger.Error(b.Token, wr.ErrorCode, wr.Result) return } b.Name = bot.Self.FirstName ch := v1.Channel{ Type: "telegram", Events: []string{ "message_sent", "message_updated", "message_deleted", "message_read", }, } conn := getConnectionById(b.ConnectionID) var client = v1.New(conn.MGURL, conn.MGToken) data, status, err := client.ActivateTransportChannel(ch) if status != http.StatusCreated { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("error_activating_channel")}) logger.Error(conn.APIURL, status, err.Error(), data) return } b.Channel = data.ChannelID err = conn.createBot(b) if err != nil { 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(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("not_found_account")}) logger.Error(b.ID, "MGURL or MGToken is empty") return } var client = v1.New(conn.MGURL, conn.MGToken) data, status, err := client.DeactivateTransportChannel(getBotChannelByToken(b.Token)) if status > http.StatusOK { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("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]string }{ p, bots, getLocale(), } 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 { c.AbortWithStatusJSON(code, gin.H{"error": err.Error()}) return } err = conn.saveConnection() 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(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("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.StatusBadRequest { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": getLocalizedMessage("error_activity_mg")}) logger.Error(conn.APIURL, status, errr.ApiErr, data) return } conn.MGURL = data.Info["baseUrl"] conn.MGToken = data.Info["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 rec v5.ActivityCallback if err := c.ShouldBindJSON(&rec); err != nil { c.Error(err) return } conn := getConnection(rec.ClientId) if conn.ID == 0 { c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{ "success": false, "error": "Wrong data", }, ) return } conn.Active = rec.Activity.Active && !rec.Activity.Freeze if err := conn.setConnectionActivity(); err != nil { c.Error(err) return } c.JSON(http.StatusOK, gin.H{"success": true}) } 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, ), }, }, } }