Public Access
Чат-бот для объединения Telegram и Discord в ROX, в отличии от основого нет ВК и сайта
This commit is contained in:
@@ -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 {} // Бесконечный цикл для поддержания работы программы
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user