Public Access
152 lines
4.6 KiB
Go
152 lines
4.6 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func main() {
|
|
log.Println("Go Chat Bridge запущен.")
|
|
|
|
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")
|
|
}
|
|
|
|
bridge, err := NewBridge(telegramToken, discordToken, vkToken)
|
|
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
|
|
vkPeerID := os.Getenv("VK_PEER_ID") // int
|
|
|
|
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":
|
|
if discordChannelID != "" {
|
|
_ = bridge.SendToDiscord(discordChannelID, "[TG] "+sender+": "+msg)
|
|
}
|
|
if vkPeerID != "" {
|
|
if id, err := strconv.Atoi(vkPeerID); err == nil {
|
|
_ = bridge.SendToVK(id, "[TG] "+sender+": "+msg)
|
|
}
|
|
}
|
|
_ = bridge.SendToSite("[TG] " + sender + ": " + msg)
|
|
case "discord":
|
|
if telegramChatID != "" {
|
|
if id, err := strconv.ParseInt(telegramChatID, 10, 64); err == nil {
|
|
_ = bridge.SendToTelegram(id, "[DS] "+sender+": "+msg)
|
|
}
|
|
}
|
|
if vkPeerID != "" {
|
|
if id, err := strconv.Atoi(vkPeerID); err == nil {
|
|
_ = bridge.SendToVK(id, "[DS] "+sender+": "+msg)
|
|
}
|
|
}
|
|
_ = bridge.SendToSite("[DS] " + sender + ": " + msg)
|
|
case "vk":
|
|
if telegramChatID != "" {
|
|
if id, err := strconv.ParseInt(telegramChatID, 10, 64); err == nil {
|
|
_ = bridge.SendToTelegram(id, "[VK] "+sender+": "+msg)
|
|
}
|
|
}
|
|
if discordChannelID != "" {
|
|
_ = 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) })
|
|
|
|
// 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))
|
|
}
|