Refactoring

1 files changed, 10 insertions, 149 deletions

diff --git a/bot.go b/bot.go
index 0ad1620..d3d8a8e 100644
--- a/bot.go
+++ b/bot.go
@@ -1,171 +1,32 @@
 package main
 
 import (
-	"context"
-	"fmt"
 	"log"
 	"os"
-	"strings"
 
-	"github.com/go-telegram-bot-api/telegram-bot-api"
 	"github.com/neonxp/rutina"
-	"github.com/neonxp/tamtam"
-)
-
-type message struct {
-	From    string
-	To      int64
-	Text    string
-	Images  []string
-	Sticker string
-}
 
-type routing struct {
-	TgID int64
-	TTID int64
-}
+	"transport/lib"
+	"transport/plugins"
+)
 
 func main() {
 	tamtamToken := os.Getenv("TAMTAM")
 	telegramToken := os.Getenv("TELEGRAM")
 	r := rutina.New()
 
-	routes := []routing{
+	routes := []lib.Routing{
 		{TgID: -1001046507545, TTID: -69062921115159},
 	}
 
-	tt2tg := make(chan message, 1)
-	tg2tt := make(chan message, 1)
-
-	// region TamTam
-	tamtamApi := tamtam.New(tamtamToken)
-	ttch := make(chan interface{}, 1)
-	r.Go(func(ctx context.Context) error {
-		for {
-			select {
-			case <-ctx.Done():
-				return nil
-			case upd := <-ttch:
-				log.Printf("[TT] Received: %#v", upd)
-				switch upd := upd.(type) {
-				case tamtam.UpdateMessageCreated:
-					for _, r := range routes {
-						if r.TTID == upd.Message.Recipient.ChatId {
-							from := upd.Message.Sender.Name
-							if upd.Message.Sender.Username != "" {
-								from = fmt.Sprintf("%s (%s)", upd.Message.Sender.Name, upd.Message.Sender.Username)
-							}
-							isSticker := ""
-							images := make([]string, 0)
-							for _, a := range upd.Message.Body.Attachments {
-								switch a := a.(type) {
-								case *tamtam.PhotoAttachment:
-									images = append(images, a.Payload.Url)
-								case *tamtam.StickerAttachment:
-									images = append(images, a.Payload.Url)
-									isSticker = "STICKER"
-								}
-							}
-							tt2tg <- message{
-								To:      r.TgID,
-								From:    from,
-								Text:    upd.Message.Body.Text,
-								Images:  images,
-								Sticker: isSticker,
-							}
-						}
-					}
-
-				default:
-					log.Printf("Unknown type: %#v", upd)
-				}
-			case msg := <-tg2tt:
-
-				text := fmt.Sprintf("[TG] %s: %s", msg.From, msg.Text)
-				if msg.Sticker != "" {
-					text = fmt.Sprintf("[TG] %s: [%s]", msg.From, msg.Sticker)
-				}
-				if len(msg.Images) > 0 {
-					text += "[IMAGES] " + strings.Join(msg.Images, " ")
-				}
-				res, err := tamtamApi.SendMessage(msg.To, msg.To, &tamtam.NewMessageBody{
-					Text: text,
-				})
-				log.Printf("[TT] Answer: %#v %#v", res, err)
-			}
-		}
-	}, rutina.ShutdownIfDone, rutina.RestartIfFail)
-	r.Go(func(ctx context.Context) error {
-		return tamtamApi.GetUpdatesLoop(ctx, ttch)
-	}, rutina.ShutdownIfDone, rutina.RestartIfFail)
-	// endregion
-
-	// region Telegram
-	tgApi, err := tgbotapi.NewBotAPI(telegramToken)
-	if err != nil {
-		log.Panic(err)
-	}
+	tt2tg := make(chan lib.Message, 1)
+	tg2tt := make(chan lib.Message, 1)
 
-	u := tgbotapi.NewUpdate(0)
-	u.Timeout = 60
-	updates, err := tgApi.GetUpdatesChan(u)
-	r.Go(func(ctx context.Context) error {
-		for {
-			select {
-			case <-ctx.Done():
-				return nil
-			case upd := <-updates:
-				if upd.Message == nil { // ignore any non-Message Updates
-					continue
-				}
-				for _, r := range routes {
-					if r.TgID == upd.Message.Chat.ID {
-						from := fmt.Sprintf("%s %s", upd.Message.From.FirstName, upd.Message.From.LastName)
-						if upd.Message.From.UserName != "" {
-							from = fmt.Sprintf("%s %s (%s)", upd.Message.From.FirstName, upd.Message.From.LastName, upd.Message.From.UserName)
-						}
-						isSticker := ""
-						images := make([]string, 0)
-						if upd.Message.Sticker != nil {
-							isSticker = upd.Message.Sticker.Emoji
-							s, _ := tgApi.GetFileDirectURL(upd.Message.Sticker.Thumbnail.FileID)
-							images = append(images, s)
-						}
-						if upd.Message.Photo != nil {
-							for _, p := range *upd.Message.Photo {
-								s, _ := tgApi.GetFileDirectURL(p.FileID)
-								images = append(images, s)
-							}
-						}
-						tg2tt <- message{
-							To:      r.TTID,
-							From:    from,
-							Text:    upd.Message.Text,
-							Images:  images,
-							Sticker: isSticker,
-						}
-					}
-				}
-			case msg := <-tt2tg:
-				att := make([]interface{}, 0)
-				for _, a := range msg.Images {
-					att = append(att, tamtam.Image{Url: a})
-				}
-				text := fmt.Sprintf("[TT] %s: %s", msg.From, msg.Text)
-				if msg.Sticker != "" {
-					text = fmt.Sprintf("[TT] %s: [%s]", msg.From, msg.Sticker)
-				}
-				if len(msg.Images) > 0 {
-					text += "[IMAGES] " + strings.Join(msg.Images, " ")
-				}
-				m := tgbotapi.NewMessage(msg.To, text)
-				res, err := tgApi.Send(m)
-				log.Printf("[TG] Answer: %#v %#v", res, err)
-			}
-		}
-	}, rutina.ShutdownIfDone, rutina.RestartIfFail)
+	tg := plugins.NewTelegram(telegramToken, tt2tg, tg2tt, routes)
+	tt := plugins.NewTamTam(tamtamToken, tg2tt, tt2tg, routes)
 
-	// endregion
+	r.Go(tg.Run, rutina.ShutdownIfDone, rutina.RestartIfFail)
+	r.Go(tt.Run, rutina.ShutdownIfDone, rutina.RestartIfFail)
 
 	r.ListenOsSignals()
 	if err := r.Wait(); err != nil {