Refactoring

3 files changed, 243 insertions, 0 deletions

diff --git a/plugins/plugin.go b/plugins/plugin.go
new file mode 100644
index 0000000..504c4af
--- /dev/null
+++ b/plugins/plugin.go
@@ -0,0 +1,7 @@
+package plugins
+
+import "context"
+
+type Plugin interface {
+	Run(ctx context.Context) error
+}
diff --git a/plugins/tg.go b/plugins/tg.go
new file mode 100644
index 0000000..ccd3a9b
--- /dev/null
+++ b/plugins/tg.go
@@ -0,0 +1,129 @@
+package plugins
+
+import (
+	"context"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"os"
+
+	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
+	"github.com/neonxp/tamtam"
+
+	"transport/lib"
+)
+
+type Telegram struct {
+	Routing []lib.Routing
+	API     *tgbotapi.BotAPI
+	Updates chan lib.Message
+	Bus     chan lib.Message
+}
+
+func NewTelegram(token string, updates chan lib.Message, bus chan lib.Message, routing []lib.Routing) *Telegram {
+	tgApi, err := tgbotapi.NewBotAPI(token)
+	if err != nil {
+		log.Panic(err)
+	}
+
+	return &Telegram{API: tgApi, Updates: updates, Bus: bus, Routing: routing}
+}
+
+func (t *Telegram) Run(ctx context.Context) error {
+
+	u := tgbotapi.NewUpdate(0)
+	u.Timeout = 60
+
+	updates, err := t.API.GetUpdatesChan(u)
+	if err != nil {
+		return err
+	}
+	for {
+		select {
+		case <-ctx.Done():
+			return nil
+		case upd := <-updates:
+			if upd.Message == nil { // ignore any non-Message Updates
+				continue
+			}
+			for _, r := range t.Routing {
+				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, _ := t.API.GetFileDirectURL(upd.Message.Sticker.Thumbnail.FileID)
+						images = append(images, s)
+					}
+					if upd.Message.Photo != nil && len(*upd.Message.Photo) > 0 {
+						p := (*upd.Message.Photo)[0]
+						s, _ := t.API.GetFileDirectURL(p.FileID)
+						images = append(images, s)
+					}
+					t.Bus <- lib.Message{
+						To:      r.TTID,
+						From:    from,
+						Text:    upd.Message.Text,
+						Images:  images,
+						Sticker: isSticker,
+					}
+				}
+			}
+		case msg := <-t.Updates:
+			att := make([]interface{}, 0)
+			for _, a := range msg.Images {
+				att = append(att, tamtam.Image{Url: a})
+			}
+			text := fmt.Sprintf("*%s*:\n%s", msg.From, msg.Text)
+			if len(msg.Text) > 0 {
+				res, err := t.API.Send(tgbotapi.MessageConfig{
+					BaseChat: tgbotapi.BaseChat{
+						ChatID:           msg.To,
+						ReplyToMessageID: 0,
+					},
+					Text:                  text,
+					DisableWebPagePreview: false,
+					ParseMode:             "markdown",
+				})
+				log.Printf("[TG] Send text: %#v %#v", res, err)
+			}
+			if len(msg.Images) > 0 {
+				for _, i := range msg.Images {
+					u, err := t.DownloadFile(i)
+					if err != nil {
+						log.Printf("[TG] Download image: %#v", err)
+						continue
+					}
+					m := tgbotapi.NewPhotoUpload(msg.To, u)
+					m.Caption = msg.Sticker
+					res, err := t.API.Send(m)
+					log.Printf("[TG] Send image: %#v %#v", res, err)
+				}
+			}
+		}
+	}
+}
+
+func (t *Telegram) DownloadFile(u string) (string, error) {
+	resp, err := http.Get(u)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	// Write the body to file
+
+	f, err := ioutil.TempFile(os.TempDir(), "tg*")
+	if err != nil {
+		return "", err
+	}
+	_, err = io.Copy(f, resp.Body)
+	defer f.Close()
+	return f.Name(), nil
+}
diff --git a/plugins/tt.go b/plugins/tt.go
new file mode 100644
index 0000000..89fa77c
--- /dev/null
+++ b/plugins/tt.go
@@ -0,0 +1,107 @@
+package plugins
+
+import (
+	"context"
+	"fmt"
+	"log"
+
+	"github.com/neonxp/tamtam"
+
+	"transport/lib"
+)
+
+type TamTam struct {
+	Routing []lib.Routing
+	API     *tamtam.Api
+	Updates chan lib.Message
+	Bus     chan lib.Message
+}
+
+func NewTamTam(token string, updates chan lib.Message, bus chan lib.Message, routing []lib.Routing) *TamTam {
+	return &TamTam{API: tamtam.New(token), Updates: updates, Bus: bus, Routing: routing}
+}
+
+func (t *TamTam) Run(ctx context.Context) error {
+	updates := make(chan interface{}, 1)
+	go func() {
+		if err := t.API.GetUpdatesLoop(ctx, updates); err != nil {
+			log.Printf("[TT] Error: %#v", err)
+		}
+	}()
+	for {
+		select {
+		case <-ctx.Done():
+			return nil
+		case upd := <-updates:
+			log.Printf("[TT] Received: %#v", upd)
+			switch upd := upd.(type) {
+			case tamtam.UpdateMessageCreated:
+				for _, r := range t.Routing {
+					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 = "СТИКЕР"
+							}
+						}
+						t.Bus <- lib.Message{
+							To:      r.TgID,
+							From:    from,
+							Text:    upd.Message.Body.Text,
+							Images:  images,
+							Sticker: isSticker,
+						}
+					}
+				}
+
+			default:
+				log.Printf("Unknown type: %#v", upd)
+			}
+		case msg := <-t.Updates:
+
+			text := fmt.Sprintf("*%s*:\n%s", msg.From, msg.Text)
+			if msg.Sticker != "" {
+				text = fmt.Sprintf("*%s*: [%s]", msg.From, msg.Sticker)
+			}
+			attachments := make([]interface{}, 0, len(msg.Images))
+			if len(msg.Images) > 0 {
+				toSent := []string{}
+				for _, i := range msg.Images {
+					ok := true
+					for _, j := range toSent {
+						if i == j {
+							ok = false
+						}
+					}
+					if ok {
+						toSent = append(toSent, i)
+					}
+				}
+				for _, i := range toSent {
+					attachments = append(attachments, tamtam.PhotoAttachment{
+						Type: "image",
+						Payload: tamtam.PhotoAttachmentPayload{
+							Url: &i,
+						},
+					})
+				}
+			}
+			if len(text) > 0 || len(attachments) > 0 {
+				res, err := t.API.SendMessage(msg.To, msg.To, &tamtam.NewMessageBody{
+					Text:        text,
+					Attachments: attachments,
+				})
+				log.Printf("[TT] Answer: %#v %#v", res, err)
+			}
+		}
+	}
+}