fetchtelegram.go

package main

import (
	"fmt"
	"time"

	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
)

var tgbot *tgbotapi.BotAPI

func setupTelegram() {
	var settings globalSettings
	database.First(&settings)
	if settings.APIKey != "" {
		var err error
		tgbot, err = tgbotapi.NewBotAPI(settings.APIKey)
		output(func(print printer) {
			if err != nil {
				print("unable to connect to telegram")
			} else {
				print(fmt.Sprintf("Telegram Identity: %s", tgbot.Self.UserName))
			}
		})
		u := tgbotapi.NewUpdate(settings.APIOffset)
		u.Timeout = 60
		var botupdates tgbotapi.UpdatesChannel
		botupdates, err = tgbot.GetUpdatesChan(u)
		output(func(print printer) {
			if err != nil {
				print("unable to establish updates connection")
			}
		})
		go telegramFetcher(botupdates)
	} else {
		output(func(print printer) {
			print("No Telegram API key set. SEt one and restart in order to start operations.")
		})
	}
}

func telegramFetcher(updates tgbotapi.UpdatesChannel) {
	for update := range updates {
		var settings globalSettings
		database.First(&settings)
		settings.APIOffset = update.UpdateID + 1
		database.Save(&settings)

		incomingmessage := update.Message
		userID := incomingmessage.From.ID

		storeduser := chat{TelegramID: userID}
		database.Where(&storeduser).First(&storeduser)
		if database.NewRecord(storeduser) {
			//this is a new record, so we have to fill in the details.
			storeduser.TelegramFirstName = incomingmessage.From.FirstName
			storeduser.TelegramLastName = incomingmessage.From.LastName
			storeduser.TelegramUserName = incomingmessage.From.UserName
			if storeduser.TelegramUserName == "" {
				storeduser.TelegramUserName = storeduser.TelegramFirstName
			}
			if storeduser.TelegramUserName == "" {
				storeduser.TelegramUserName = storeduser.TelegramLastName
			}
			storeduser.TelegramChatID = incomingmessage.Chat.ID
			database.Create(&storeduser)
		}

		storedmessage := message{Inbound: true, Text: incomingmessage.Text, Date: int(time.Now().Unix()), ChatID: storeduser.ID, Read: false}
		database.Create(&storedmessage)
		if activeChatID == storeduser.ID {
			output(func(print printer) { print(storedmessage.toString()) })
		} else {
			updateSidebar()
		}
		automationqueue <- automationitem{Chat: storeduser, Message: storedmessage}
		var chats []chat
		database.Where(&chat{Admin: true}).Find(&chats)
		for _, admin := range chats {
			if admin.ID != storeduser.ID {
				admin.sendMessage(fmt.Sprintf("[I] <%s> %s", storeduser.TelegramUserName, incomingmessage.Text))
			}
		}
	}
}