Чат-бот для объединения Telegram и Discord в ROX, в отличии от основого нет ВК и сайта

This commit is contained in:
Taz
2026-04-22 18:27:05 +02:00
parent 95d94c16b6
commit d3b54c682b
11 changed files with 20 additions and 298 deletions
+6 -121
View File
@@ -1,9 +1,7 @@
package main
import (
"encoding/json"
"log"
"net/http"
"os"
"strconv"
"strings"
@@ -14,13 +12,12 @@ func main() {
telegramToken := os.Getenv("TELEGRAM_TOKEN")
discordToken := os.Getenv("DISCORD_TOKEN")
vkToken := os.Getenv("VK_TOKEN")
if telegramToken == "" || discordToken == "" || vkToken == "" {
log.Fatal("Необходимо задать переменные окружения TELEGRAM_TOKEN, DISCORD_TOKEN, VK_TOKEN")
if telegramToken == "" || discordToken == "" {
log.Fatal("Необходимо задать переменные окружения TELEGRAM_TOKEN, DISCORD_TOKEN")
}
bridge, err := NewBridge(telegramToken, discordToken, vkToken)
bridge, err := NewBridge(telegramToken, discordToken)
if err != nil {
log.Fatalf("Ошибка инициализации Bridge: %v", err)
}
@@ -30,7 +27,6 @@ func main() {
// Получаем ID чатов/каналов из переменных окружения
telegramChatID := os.Getenv("TELEGRAM_CHAT_ID") // int64
discordChannelID := os.Getenv("DISCORD_CHANNEL_ID") // string
vkPeerID := os.Getenv("VK_PEER_ID") // int
forward := func(text string, from string, sender string, media []string) {
// Собираем итоговый текст с медиа URL, если есть
@@ -59,13 +55,6 @@ func main() {
_ = bridge.SendToDiscord(discordChannelID, "[TG] "+sender+": "+msg)
}
}
// Для VK и сайта оставляем ссылки
if vkPeerID != "" {
if id, err := strconv.Atoi(vkPeerID); err == nil {
_ = bridge.SendToVK(id, "[TG] "+sender+": "+msg)
}
}
_ = bridge.SendToSite("[TG] " + sender + ": " + msg)
case "discord":
// Для Telegram отправляем файлы, если есть, иначе текст
if telegramChatID != "" {
@@ -83,116 +72,12 @@ func main() {
}
}
}
// Для VK и сайта оставляем ссылки
if vkPeerID != "" {
if id, err := strconv.Atoi(vkPeerID); err == nil {
_ = bridge.SendToVK(id, "[DS] "+sender+": "+msg)
}
}
_ = bridge.SendToSite("[DS] " + sender + ": " + msg)
case "vk":
// Если когда-либо будут передаваться ссылки на медиа из VK, для TG/Discord можем тоже отправлять файлы
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, "[VK] "+sender+": "+text)
bridge.CleanupFiles(files)
} else {
_ = bridge.SendToTelegram(id, "[VK] "+sender+": "+msg)
}
} else {
_ = bridge.SendToTelegram(id, "[VK] "+sender+": "+msg)
}
}
}
if discordChannelID != "" {
if len(media) > 0 {
files, derr := bridge.DownloadFiles(media)
if derr == nil && len(files) > 0 {
_ = bridge.SendFilesToDiscord(discordChannelID, files, "[VK] "+sender+": "+text)
bridge.CleanupFiles(files)
} else {
_ = bridge.SendToDiscord(discordChannelID, "[VK] "+sender+": "+msg)
}
} else {
_ = bridge.SendToDiscord(discordChannelID, "[VK] "+sender+": "+msg)
}
}
// Для сайта оставляем ссылки
_ = bridge.SendToSite("[VK] " + sender + ": " + msg)
case "site":
// Из сайта пока не принимаем ссылки на файлы, отправляем как текст
if telegramChatID != "" {
if id, err := strconv.ParseInt(telegramChatID, 10, 64); err == nil {
_ = bridge.SendToTelegram(id, "[SITE] "+sender+": "+msg)
}
}
if discordChannelID != "" {
_ = bridge.SendToDiscord(discordChannelID, "[SITE] "+sender+": "+msg)
}
if vkPeerID != "" {
if id, err := strconv.Atoi(vkPeerID); err == nil {
_ = bridge.SendToVK(id, "[SITE] "+sender+": "+msg)
}
}
}
}
bridge.StartTelegramListener(func(text string, sender string, media []string) { forward(text, "telegram", sender, media) })
bridge.StartDiscordListener(func(text string, sender string, media []string) { forward(text, "discord", sender, media) })
bridge.StartVKListener(func(text string, sender string, media []string) { forward(text, "vk", sender, media) })
bridge.StartDiscordListener(discordChannelID, func(text string, sender string, media []string) { forward(text, "discord", sender, media) })
// webhook endpoint
http.HandleFunc("/webhook", func(w http.ResponseWriter, r *http.Request) {
if r.Method != http.MethodPost {
http.Error(w, "Method not allowed", http.StatusMethodNotAllowed)
return
}
apiKey := os.Getenv("WEBHOOK_API_KEY")
if apiKey == "" {
http.Error(w, "Webhook API key not set", http.StatusInternalServerError)
return
}
key := r.URL.Query().Get("key")
if key == "" {
// также можно принимать ключ в теле запроса
var req struct {
Key string `json:"key"`
Text string `json:"text"`
Sender string `json:"sender"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, "Bad request", http.StatusBadRequest)
return
}
if req.Key != apiKey {
http.Error(w, "Forbidden", http.StatusForbidden)
return
}
forward(req.Text, "site", req.Sender, nil)
w.WriteHeader(http.StatusOK)
w.Write([]byte("ok"))
return
}
if key != apiKey {
http.Error(w, "Forbidden", http.StatusForbidden)
return
}
var req struct {
Text string `json:"text"`
Sender string `json:"sender"`
}
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, "Bad request", http.StatusBadRequest)
return
}
forward(req.Text, "site", req.Sender, nil)
w.WriteHeader(http.StatusOK)
w.Write([]byte("ok"))
})
log.Println("Webhook listening on :8080 (/webhook)")
log.Fatal(http.ListenAndServe(":8080", nil))
log.Println("Chat Bridge запущен и работает")
select {} // Бесконечный цикл для поддержания работы программы
}