Files
turn-bridge/main.go
T

84 lines
2.8 KiB
Go

package main
import (
"log"
"os"
"strconv"
"strings"
)
func main() {
log.Println("Go Chat Bridge запущен.")
telegramToken := os.Getenv("TELEGRAM_TOKEN")
discordToken := os.Getenv("DISCORD_TOKEN")
if telegramToken == "" || discordToken == "" {
log.Fatal("Необходимо задать переменные окружения TELEGRAM_TOKEN, DISCORD_TOKEN")
}
bridge, err := NewBridge(telegramToken, discordToken)
if err != nil {
log.Fatalf("Ошибка инициализации Bridge: %v", err)
}
log.Println("Bridge успешно инициализирован:", bridge)
// Получаем ID чатов/каналов из переменных окружения
telegramChatID := os.Getenv("TELEGRAM_CHAT_ID") // int64
discordChannelID := os.Getenv("DISCORD_CHANNEL_ID") // string
forward := func(text string, from string, sender string, media []string) {
// Собираем итоговый текст с медиа URL, если есть
msg := text
if len(media) > 0 {
if msg != "" {
msg += "\n"
}
msg += strings.Join(media, "\n")
}
log.Printf("[Bridge] Пересылка сообщения от %s (%s): %s", from, sender, msg)
switch from {
case "telegram":
// Для Discord отправляем файлы, если есть, иначе текст
if discordChannelID != "" {
if len(media) > 0 {
files, derr := bridge.DownloadFiles(media)
if derr == nil && len(files) > 0 {
_ = bridge.SendFilesToDiscord(discordChannelID, files, "[TG] "+sender+": "+text)
bridge.CleanupFiles(files)
} else {
_ = bridge.SendToDiscord(discordChannelID, "[TG] "+sender+": "+msg)
}
} else {
_ = bridge.SendToDiscord(discordChannelID, "[TG] "+sender+": "+msg)
}
}
case "discord":
// Для Telegram отправляем файлы, если есть, иначе текст
if telegramChatID != "" {
if id, err := strconv.ParseInt(telegramChatID, 10, 64); err == nil {
if len(media) > 0 {
files, derr := bridge.DownloadFiles(media)
if derr == nil && len(files) > 0 {
_ = bridge.SendFilesToTelegram(id, files, "[DS] "+sender+": "+text)
bridge.CleanupFiles(files)
} else {
_ = bridge.SendToTelegram(id, "[DS] "+sender+": "+msg)
}
} else {
_ = bridge.SendToTelegram(id, "[DS] "+sender+": "+msg)
}
}
}
}
}
bridge.StartTelegramListener(func(text string, sender string, media []string) { forward(text, "telegram", sender, media) })
bridge.StartDiscordListener(discordChannelID, func(text string, sender string, media []string) { forward(text, "discord", sender, media) })
log.Println("Chat Bridge запущен и работает")
select {} // Бесконечный цикл для поддержания работы программы
}