Initial commit

internal/telegram/exported.go

@@ -0,0 +1,31 @@
+// Fantasy World Zookeeper Bot
+// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
+
+package telegram
+
+import (
+	"github.com/rs/zerolog"
+	"gitlab.com/toby3d/telegram"
+	"lab.wtfteam.pro/fat0troll/fw_zookeeper/context"
+)
+
+var (
+	c   *context.Context
+	log zerolog.Logger
+)
+
+// Telegram is a struch which handles Telegram instance handling functions
+type Telegram struct {
+	bot *telegram.Bot
+}
+
+// New initializes package
+func New(cc *context.Context) {
+	c = cc
+	log = c.Logger.With().Str("domain", "telegram").Int("version", 1).Logger()
+	t := &Telegram{}
+
+	log.Info().Msg("Starting Telegram instance")
+
+	t.StartBot()
+}

internal/telegram/telegram.go

@@ -0,0 +1,97 @@
+// Fantasy World Zookeeper Bot
+// Copyright (c) 2018 Vladimir "fat0troll" Hodakov
+
+package telegram
+
+import (
+	"net"
+
+	http "github.com/valyala/fasthttp"
+	"gitlab.com/toby3d/telegram"
+	"golang.org/x/net/proxy"
+	"lab.wtfteam.pro/fat0troll/fw_zookeeper/internal/router"
+)
+
+func (t *Telegram) proxyDialer(addr string) (net.Conn, error) {
+	log.Debug().Msgf("Proxy used: %s", c.Config.Telegram.Proxy.Address)
+	proxyAuth := proxy.Auth{}
+	if c.Config.Telegram.Proxy.Username != "" {
+		proxyAuth.User = c.Config.Telegram.Proxy.Username
+		proxyAuth.Password = c.Config.Telegram.Proxy.Password
+	}
+	var dialProxy proxy.Dialer
+	var err error
+	dialProxy, err = proxy.SOCKS5("tcp", c.Config.Telegram.Proxy.Address, &proxyAuth, proxy.Direct)
+	if err != nil {
+		log.Fatal().Err(err).Msg("Failed to dial proxy")
+	}
+
+	return dialProxy.Dial("tcp", addr)
+}
+
+// Bot returns Telegram instance
+func (t *Telegram) Bot() *telegram.Bot {
+	return t.bot
+}
+
+// StartBot starts connection with Telegram
+func (t *Telegram) StartBot() {
+	// Any errors here considered fatal, because main purpose of this app is Telegram interactions
+	var err error
+	var updates telegram.UpdatesChannel
+	if c.Config.Telegram.Proxy.Enabled {
+		t.bot = new(telegram.Bot)
+		client := new(http.Client)
+		client.Dial = t.proxyDialer
+		t.bot.SetClient(client)
+		t.bot.AccessToken = c.Config.Telegram.Token
+		t.bot.User, err = t.bot.GetMe()
+	} else {
+		t.bot, err = telegram.New(c.Config.Telegram.Token)
+	}
+	if err != nil {
+		log.Fatal().Err(err)
+	}
+
+	if c.Config.Telegram.Webhook.Enabled {
+		var url *http.URI
+		url.Parse(nil, []byte(c.Config.Telegram.Webhook.Domain))
+		if len(url.Host()) == 0 {
+			log.Fatal().Msg("Can't parse webhook URL: got empty host")
+		}
+		log.Info().Msg("Trying to set webhook: " + url.String() + t.bot.AccessToken)
+
+		webhook := telegram.NewWebhook(url.String()+t.bot.AccessToken, nil)
+		webhook.MaxConnections = 40
+
+		updates = t.bot.NewWebhookChannel(url, webhook, "", "", c.Config.Telegram.Webhook.Listen)
+	} else {
+		log.Warn().Msg("Using long-polling for updates (not recommended)")
+		var info *telegram.WebhookInfo
+		info, err = t.bot.GetWebhookInfo()
+		if err != nil {
+			log.Fatal().Err(err)
+		}
+		if info != nil && info.URL != "" {
+			log.Info().Msg("Deleting old webhook...")
+			_, err := t.bot.DeleteWebhook()
+			if err != nil {
+				log.Fatal().Err(err)
+			}
+		}
+		updatesParams := telegram.GetUpdatesParameters{
+			Offset:  0,
+			Limit:   100,
+			Timeout: 60,
+		}
+		updates = t.bot.NewLongPollingChannel(&updatesParams)
+	}
+
+	log.Info().Msg("Connection with Telegram established")
+
+	for update := range updates {
+		log.Debug().Msgf("%+v", update)
+		go router.Requests.Respond(update)
+	}
+
+}