Initial commit
This commit is contained in:
James
commit
1a79f1c87b
|
|
@ -0,0 +1,5 @@
|
|||
# Binaries
|
||||
*.exe
|
||||
node_modules/
|
||||
.venv/
|
||||
__pycache__/
|
||||
|
|
@ -0,0 +1,28 @@
|
|||
module message-bridge
|
||||
|
||||
go 1.25.6
|
||||
|
||||
require (
|
||||
filippo.io/edwards25519 v1.1.0 // indirect
|
||||
github.com/beeper/argo-go v1.1.2 // indirect
|
||||
github.com/coder/websocket v1.8.14 // indirect
|
||||
github.com/elliotchance/orderedmap/v3 v3.1.0 // indirect
|
||||
github.com/google/uuid v1.6.0 // indirect
|
||||
github.com/gorilla/mux v1.8.1 // indirect
|
||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/mattn/go-sqlite3 v1.14.33 // indirect
|
||||
github.com/petermattis/goid v0.0.0-20260113132338-7c7de50cc741 // indirect
|
||||
github.com/rs/zerolog v1.34.0 // indirect
|
||||
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e // indirect
|
||||
github.com/vektah/gqlparser/v2 v2.5.27 // indirect
|
||||
go.mau.fi/libsignal v0.2.1 // indirect
|
||||
go.mau.fi/util v0.9.5 // indirect
|
||||
go.mau.fi/whatsmeow v0.0.0-20260129212019-7787ab952245 // indirect
|
||||
golang.org/x/crypto v0.47.0 // indirect
|
||||
golang.org/x/exp v0.0.0-20260112195511-716be5621a96 // indirect
|
||||
golang.org/x/net v0.49.0 // indirect
|
||||
golang.org/x/sys v0.40.0 // indirect
|
||||
golang.org/x/text v0.33.0 // indirect
|
||||
google.golang.org/protobuf v1.36.11 // indirect
|
||||
)
|
||||
|
|
@ -0,0 +1,54 @@
|
|||
filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
|
||||
filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
|
||||
github.com/beeper/argo-go v1.1.2 h1:UQI2G8F+NLfGTOmTUI0254pGKx/HUU/etbUGTJv91Fs=
|
||||
github.com/beeper/argo-go v1.1.2/go.mod h1:M+LJAnyowKVQ6Rdj6XYGEn+qcVFkb3R/MUpqkGR0hM4=
|
||||
github.com/coder/websocket v1.8.14 h1:9L0p0iKiNOibykf283eHkKUHHrpG7f65OE3BhhO7v9g=
|
||||
github.com/coder/websocket v1.8.14/go.mod h1:NX3SzP+inril6yawo5CQXx8+fk145lPDC6pumgx0mVg=
|
||||
github.com/coreos/go-systemd/v22 v22.5.0/go.mod h1:Y58oyj3AT4RCenI/lSvhwexgC+NSVTIJ3seZv2GcEnc=
|
||||
github.com/elliotchance/orderedmap/v3 v3.1.0 h1:j4DJ5ObEmMBt/lcwIecKcoRxIQUEnw0L804lXYDt/pg=
|
||||
github.com/elliotchance/orderedmap/v3 v3.1.0/go.mod h1:G+Hc2RwaZvJMcS4JpGCOyViCnGeKf0bTYCGTO4uhjSo=
|
||||
github.com/godbus/dbus/v5 v5.0.4/go.mod h1:xhWf0FNVPg57R7Z0UbKHbJfkEywrmjJnf7w5xrFpKfA=
|
||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/gorilla/mux v1.8.1 h1:TuBL49tXwgrFYWhqrNgrUNEY92u81SPhu7sTdzQEiWY=
|
||||
github.com/gorilla/mux v1.8.1/go.mod h1:AKf9I4AEqPTmMytcMc0KkNouC66V3BtZ4qD5fmWSiMQ=
|
||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
||||
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
||||
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/mattn/go-sqlite3 v1.14.33 h1:A5blZ5ulQo2AtayQ9/limgHEkFreKj1Dv226a1K73s0=
|
||||
github.com/mattn/go-sqlite3 v1.14.33/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=
|
||||
github.com/petermattis/goid v0.0.0-20260113132338-7c7de50cc741 h1:KPpdlQLZcHfTMQRi6bFQ7ogNO0ltFT4PmtwTLW4W+14=
|
||||
github.com/petermattis/goid v0.0.0-20260113132338-7c7de50cc741/go.mod h1:pxMtw7cyUw6B2bRH0ZBANSPg+AoSud1I1iyJHI69jH4=
|
||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||
github.com/rs/xid v1.6.0/go.mod h1:7XoLgs4eV+QndskICGsho+ADou8ySMSjJKDIan90Nz0=
|
||||
github.com/rs/zerolog v1.34.0 h1:k43nTLIwcTVQAncfCw4KZ2VY6ukYoZaBPNOE8txlOeY=
|
||||
github.com/rs/zerolog v1.34.0/go.mod h1:bJsvje4Z08ROH4Nhs5iH600c3IkWhwp44iRc54W6wYQ=
|
||||
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e h1:MRM5ITcdelLK2j1vwZ3Je0FKVCfqOLp5zO6trqMLYs0=
|
||||
github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e/go.mod h1:XV66xRDqSt+GTGFMVlhk3ULuV0y9ZmzeVGR4mloJI3M=
|
||||
github.com/vektah/gqlparser/v2 v2.5.27 h1:RHPD3JOplpk5mP5JGX8RKZkt2/Vwj/PZv0HxTdwFp0s=
|
||||
github.com/vektah/gqlparser/v2 v2.5.27/go.mod h1:D1/VCZtV3LPnQrcPBeR/q5jkSQIPti0uYCP/RI0gIeo=
|
||||
go.mau.fi/libsignal v0.2.1 h1:vRZG4EzTn70XY6Oh/pVKrQGuMHBkAWlGRC22/85m9L0=
|
||||
go.mau.fi/libsignal v0.2.1/go.mod h1:iVvjrHyfQqWajOUaMEsIfo3IqgVMrhWcPiiEzk7NgoU=
|
||||
go.mau.fi/util v0.9.5 h1:7AoWPCIZJGv4jvtFEuCe3GhAbI7uF9ckIooaXvwlIR4=
|
||||
go.mau.fi/util v0.9.5/go.mod h1:g1uvZ03VQhtTt2BgaRGVytS/Zj67NV0YNIECch0sQCQ=
|
||||
go.mau.fi/whatsmeow v0.0.0-20260129212019-7787ab952245 h1:Pdrwc7vLH6DrWa2Tk19pBTwlUfV0vJLU6V9xNZ2UwGE=
|
||||
go.mau.fi/whatsmeow v0.0.0-20260129212019-7787ab952245/go.mod h1:jDLOQLLiYXcm4vMB6vtPcBLU387sRY+P3vOElxX8srA=
|
||||
golang.org/x/crypto v0.47.0 h1:V6e3FRj+n4dbpw86FJ8Fv7XVOql7TEwpHapKoMJ/GO8=
|
||||
golang.org/x/crypto v0.47.0/go.mod h1:ff3Y9VzzKbwSSEzWqJsJVBnWmRwRSHt/6Op5n9bQc4A=
|
||||
golang.org/x/exp v0.0.0-20260112195511-716be5621a96 h1:Z/6YuSHTLOHfNFdb8zVZomZr7cqNgTJvA8+Qz75D8gU=
|
||||
golang.org/x/exp v0.0.0-20260112195511-716be5621a96/go.mod h1:nzimsREAkjBCIEFtHiYkrJyT+2uy9YZJB7H1k68CXZU=
|
||||
golang.org/x/net v0.49.0 h1:eeHFmOGUTtaaPSGNmjBKpbng9MulQsJURQUAfUwY++o=
|
||||
golang.org/x/net v0.49.0/go.mod h1:/ysNB2EvaqvesRkuLAyjI1ycPZlQHM3q01F02UY/MV8=
|
||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.12.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.40.0 h1:DBZZqJ2Rkml6QMQsZywtnjnnGvHza6BTfYFWY9kjEWQ=
|
||||
golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||
golang.org/x/text v0.33.0 h1:B3njUFyqtHDUI5jMn1YIr5B0IE2U0qck04r6d4KPAxE=
|
||||
golang.org/x/text v0.33.0/go.mod h1:LuMebE6+rBincTi9+xWTY8TztLzKHc/9C1uBCG27+q8=
|
||||
google.golang.org/protobuf v1.36.11 h1:fV6ZwhNocDyBLK0dj+fg8ektcVegBBuEolpbTQyBNVE=
|
||||
google.golang.org/protobuf v1.36.11/go.mod h1:HTf+CrKn2C3g5S8VImy6tdcUvCska2kB7j23XfzDpco=
|
||||
|
|
@ -0,0 +1,559 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"os/exec"
|
||||
"os/signal"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sync"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/gorilla/mux"
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
"github.com/skip2/go-qrcode"
|
||||
"go.mau.fi/whatsmeow"
|
||||
"go.mau.fi/whatsmeow/proto/waE2E"
|
||||
"go.mau.fi/whatsmeow/store/sqlstore"
|
||||
"go.mau.fi/whatsmeow/types"
|
||||
"go.mau.fi/whatsmeow/types/events"
|
||||
waLog "go.mau.fi/whatsmeow/util/log"
|
||||
)
|
||||
|
||||
type Message struct {
|
||||
ID string `json:"id"`
|
||||
Platform string `json:"platform"`
|
||||
From string `json:"from"`
|
||||
FromName string `json:"from_name,omitempty"`
|
||||
To string `json:"to,omitempty"`
|
||||
Body string `json:"body"`
|
||||
Timestamp time.Time `json:"timestamp"`
|
||||
IsGroup bool `json:"is_group"`
|
||||
GroupName string `json:"group_name,omitempty"`
|
||||
MediaType string `json:"media_type,omitempty"`
|
||||
MediaURL string `json:"media_url,omitempty"`
|
||||
MediaPath string `json:"media_path,omitempty"`
|
||||
HasMedia bool `json:"has_media"`
|
||||
Transcription string `json:"transcription,omitempty"`
|
||||
}
|
||||
|
||||
type MessageBridge struct {
|
||||
waClient *whatsmeow.Client
|
||||
messages []Message
|
||||
msgMutex sync.RWMutex
|
||||
dataDir string
|
||||
webhookURL string
|
||||
webhookToken string
|
||||
qrChan chan string
|
||||
connected bool
|
||||
webhooksSent map[string]bool // Track sent webhook message IDs
|
||||
webhooksMutex sync.Mutex
|
||||
}
|
||||
|
||||
var bridge *MessageBridge
|
||||
|
||||
func NewMessageBridge(dataDir string) *MessageBridge {
|
||||
return &MessageBridge{
|
||||
dataDir: dataDir,
|
||||
messages: make([]Message, 0),
|
||||
qrChan: make(chan string, 1),
|
||||
webhookURL: os.Getenv("WEBHOOK_URL"),
|
||||
webhookToken: os.Getenv("WEBHOOK_TOKEN"),
|
||||
webhooksSent: make(map[string]bool),
|
||||
}
|
||||
}
|
||||
|
||||
func (b *MessageBridge) InitWhatsApp() error {
|
||||
ctx := context.Background()
|
||||
dbPath := filepath.Join(b.dataDir, "whatsapp.db")
|
||||
container, err := sqlstore.New(ctx, "sqlite3", "file:"+dbPath+"?_foreign_keys=on", waLog.Noop)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to create store: %w", err)
|
||||
}
|
||||
|
||||
deviceStore, err := container.GetFirstDevice(ctx)
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to get device: %w", err)
|
||||
}
|
||||
|
||||
b.waClient = whatsmeow.NewClient(deviceStore, waLog.Noop)
|
||||
b.waClient.AddEventHandler(b.handleWhatsAppEvent)
|
||||
|
||||
if b.waClient.Store.ID == nil {
|
||||
// Need to pair
|
||||
qrChan, _ := b.waClient.GetQRChannel(ctx)
|
||||
err = b.waClient.Connect()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to connect: %w", err)
|
||||
}
|
||||
|
||||
go func() {
|
||||
for evt := range qrChan {
|
||||
if evt.Event == "code" {
|
||||
select {
|
||||
case b.qrChan <- evt.Code:
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
}()
|
||||
} else {
|
||||
err = b.waClient.Connect()
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to connect: %w", err)
|
||||
}
|
||||
b.connected = true
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (b *MessageBridge) handleWhatsAppEvent(evt interface{}) {
|
||||
switch v := evt.(type) {
|
||||
case *events.Connected:
|
||||
b.connected = true
|
||||
log.Println("WhatsApp connected")
|
||||
|
||||
case *events.Message:
|
||||
msg := b.parseWhatsAppMessage(v)
|
||||
if msg != nil {
|
||||
b.addMessage(*msg)
|
||||
b.sendWebhook(*msg)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (b *MessageBridge) parseWhatsAppMessage(evt *events.Message) *Message {
|
||||
info := evt.Info
|
||||
|
||||
msg := &Message{
|
||||
ID: info.ID,
|
||||
Platform: "whatsapp",
|
||||
From: info.Sender.User,
|
||||
Timestamp: info.Timestamp,
|
||||
IsGroup: info.IsGroup,
|
||||
}
|
||||
|
||||
// Get sender name
|
||||
if info.PushName != "" {
|
||||
msg.FromName = info.PushName
|
||||
}
|
||||
|
||||
// Get group info
|
||||
if info.IsGroup {
|
||||
msg.To = info.Chat.User
|
||||
// Try to get group name from cache
|
||||
ctx := context.Background()
|
||||
groupInfo, err := b.waClient.GetGroupInfo(ctx, info.Chat)
|
||||
if err == nil {
|
||||
msg.GroupName = groupInfo.Name
|
||||
}
|
||||
}
|
||||
|
||||
// Parse message content
|
||||
if evt.Message != nil {
|
||||
if evt.Message.GetConversation() != "" {
|
||||
msg.Body = evt.Message.GetConversation()
|
||||
} else if evt.Message.GetExtendedTextMessage() != nil {
|
||||
msg.Body = evt.Message.GetExtendedTextMessage().GetText()
|
||||
} else if img := evt.Message.GetImageMessage(); img != nil {
|
||||
msg.HasMedia = true
|
||||
msg.MediaType = "image"
|
||||
msg.Body = img.GetCaption()
|
||||
b.downloadMedia(evt, msg)
|
||||
} else if vid := evt.Message.GetVideoMessage(); vid != nil {
|
||||
msg.HasMedia = true
|
||||
msg.MediaType = "video"
|
||||
msg.Body = vid.GetCaption()
|
||||
b.downloadMedia(evt, msg)
|
||||
} else if audio := evt.Message.GetAudioMessage(); audio != nil {
|
||||
msg.HasMedia = true
|
||||
if audio.GetPTT() {
|
||||
msg.MediaType = "voice"
|
||||
} else {
|
||||
msg.MediaType = "audio"
|
||||
}
|
||||
b.downloadMedia(evt, msg)
|
||||
} else if doc := evt.Message.GetDocumentMessage(); doc != nil {
|
||||
msg.HasMedia = true
|
||||
msg.MediaType = "document"
|
||||
msg.Body = doc.GetFileName()
|
||||
b.downloadMedia(evt, msg)
|
||||
}
|
||||
}
|
||||
|
||||
return msg
|
||||
}
|
||||
|
||||
func (b *MessageBridge) downloadMedia(evt *events.Message, msg *Message) {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
|
||||
defer cancel()
|
||||
|
||||
// Get the downloadable message
|
||||
var downloadable whatsmeow.DownloadableMessage
|
||||
var mimeType string
|
||||
if img := evt.Message.GetImageMessage(); img != nil {
|
||||
downloadable = img
|
||||
mimeType = img.GetMimetype()
|
||||
} else if vid := evt.Message.GetVideoMessage(); vid != nil {
|
||||
downloadable = vid
|
||||
mimeType = vid.GetMimetype()
|
||||
} else if audio := evt.Message.GetAudioMessage(); audio != nil {
|
||||
downloadable = audio
|
||||
mimeType = audio.GetMimetype()
|
||||
} else if doc := evt.Message.GetDocumentMessage(); doc != nil {
|
||||
downloadable = doc
|
||||
mimeType = doc.GetMimetype()
|
||||
}
|
||||
|
||||
if downloadable == nil {
|
||||
return
|
||||
}
|
||||
|
||||
data, err := b.waClient.Download(ctx, downloadable)
|
||||
if err != nil {
|
||||
log.Printf("Failed to download media: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
// Save to file
|
||||
mediaDir := filepath.Join(b.dataDir, "media")
|
||||
os.MkdirAll(mediaDir, 0755)
|
||||
|
||||
ext := getExtFromMime(mimeType, msg.MediaType, msg.Body)
|
||||
filename := fmt.Sprintf("%s_%s%s", msg.Platform, msg.ID, ext)
|
||||
filePath := filepath.Join(mediaDir, filename)
|
||||
|
||||
if err := os.WriteFile(filePath, data, 0644); err != nil {
|
||||
log.Printf("Failed to save media: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
msg.MediaURL = "/media/" + filename
|
||||
msg.MediaPath = filePath
|
||||
log.Printf("Saved media: %s (%d bytes, %s)", filePath, len(data), mimeType)
|
||||
|
||||
// Transcribe voice messages
|
||||
if msg.MediaType == "voice" {
|
||||
b.transcribeAudio(msg)
|
||||
}
|
||||
}
|
||||
|
||||
// transcribeAudio calls the transcription script for voice messages
|
||||
func (b *MessageBridge) transcribeAudio(msg *Message) {
|
||||
transcriptScript := os.Getenv("TRANSCRIBE_SCRIPT")
|
||||
if transcriptScript == "" {
|
||||
transcriptScript = "/home/johan/clawd/scripts/transcribe-audio.sh"
|
||||
}
|
||||
|
||||
// Check if script exists and GROQ_API_KEY is set
|
||||
if _, err := os.Stat(transcriptScript); os.IsNotExist(err) {
|
||||
log.Printf("Transcription script not found: %s", transcriptScript)
|
||||
return
|
||||
}
|
||||
if os.Getenv("GROQ_API_KEY") == "" {
|
||||
log.Printf("GROQ_API_KEY not set, skipping transcription")
|
||||
return
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
|
||||
cmd := exec.CommandContext(ctx, transcriptScript, msg.MediaPath)
|
||||
var stdout, stderr bytes.Buffer
|
||||
cmd.Stdout = &stdout
|
||||
cmd.Stderr = &stderr
|
||||
|
||||
if err := cmd.Run(); err != nil {
|
||||
log.Printf("Transcription failed: %v (stderr: %s)", err, stderr.String())
|
||||
return
|
||||
}
|
||||
|
||||
transcription := strings.TrimSpace(stdout.String())
|
||||
if transcription != "" {
|
||||
msg.Transcription = transcription
|
||||
log.Printf("Transcribed voice message: %s", truncate(transcription, 100))
|
||||
}
|
||||
}
|
||||
|
||||
func getExtFromMime(mimeType, mediaType, body string) string {
|
||||
// Try mime type first
|
||||
switch {
|
||||
case strings.HasPrefix(mimeType, "image/jpeg"):
|
||||
return ".jpg"
|
||||
case strings.HasPrefix(mimeType, "image/png"):
|
||||
return ".png"
|
||||
case strings.HasPrefix(mimeType, "image/webp"):
|
||||
return ".webp"
|
||||
case strings.HasPrefix(mimeType, "image/gif"):
|
||||
return ".gif"
|
||||
case strings.HasPrefix(mimeType, "video/mp4"):
|
||||
return ".mp4"
|
||||
case strings.HasPrefix(mimeType, "audio/ogg"):
|
||||
return ".ogg"
|
||||
case strings.HasPrefix(mimeType, "audio/mpeg"):
|
||||
return ".mp3"
|
||||
case strings.HasPrefix(mimeType, "audio/mp4"):
|
||||
return ".m4a"
|
||||
case strings.HasPrefix(mimeType, "application/pdf"):
|
||||
return ".pdf"
|
||||
}
|
||||
|
||||
// Fallback to media type
|
||||
switch mediaType {
|
||||
case "image":
|
||||
return ".jpg"
|
||||
case "video":
|
||||
return ".mp4"
|
||||
case "audio", "voice":
|
||||
return ".ogg"
|
||||
case "document":
|
||||
if body != "" {
|
||||
if ext := filepath.Ext(body); ext != "" {
|
||||
return ext
|
||||
}
|
||||
}
|
||||
return ".bin"
|
||||
}
|
||||
return ".bin"
|
||||
}
|
||||
|
||||
func (b *MessageBridge) addMessage(msg Message) {
|
||||
b.msgMutex.Lock()
|
||||
defer b.msgMutex.Unlock()
|
||||
|
||||
b.messages = append(b.messages, msg)
|
||||
|
||||
// Keep last 1000 messages in memory
|
||||
if len(b.messages) > 1000 {
|
||||
b.messages = b.messages[len(b.messages)-1000:]
|
||||
}
|
||||
|
||||
log.Printf("[%s] %s (%s): %s", msg.Platform, msg.FromName, msg.From, truncate(msg.Body, 50))
|
||||
}
|
||||
|
||||
func (b *MessageBridge) sendWebhook(msg Message) {
|
||||
if b.webhookURL == "" {
|
||||
return
|
||||
}
|
||||
|
||||
// Check if we've already sent a webhook for this message
|
||||
b.webhooksMutex.Lock()
|
||||
if b.webhooksSent[msg.ID] {
|
||||
b.webhooksMutex.Unlock()
|
||||
return
|
||||
}
|
||||
b.webhooksSent[msg.ID] = true
|
||||
// Prune old entries to prevent memory leak (keep last 1000)
|
||||
if len(b.webhooksSent) > 1000 {
|
||||
b.webhooksSent = make(map[string]bool)
|
||||
b.webhooksSent[msg.ID] = true
|
||||
}
|
||||
b.webhooksMutex.Unlock()
|
||||
|
||||
payload, _ := json.Marshal(map[string]interface{}{
|
||||
"event": "new_message",
|
||||
"message": msg,
|
||||
})
|
||||
|
||||
go func() {
|
||||
req, err := http.NewRequest("POST", b.webhookURL, strings.NewReader(string(payload)))
|
||||
if err != nil {
|
||||
log.Printf("Webhook request error: %v", err)
|
||||
return
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
if b.webhookToken != "" {
|
||||
req.Header.Set("Authorization", "Bearer "+b.webhookToken)
|
||||
}
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
log.Printf("Webhook send error: %v", err)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode >= 400 {
|
||||
log.Printf("Webhook error response: %d", resp.StatusCode)
|
||||
} else {
|
||||
log.Printf("Webhook sent: %s from %s", msg.Platform, msg.From)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
func truncate(s string, max int) string {
|
||||
if len(s) <= max {
|
||||
return s
|
||||
}
|
||||
return s[:max] + "..."
|
||||
}
|
||||
|
||||
// HTTP Handlers
|
||||
func handleStatus(w http.ResponseWriter, r *http.Request) {
|
||||
status := map[string]interface{}{
|
||||
"whatsapp": map[string]interface{}{
|
||||
"connected": bridge.connected,
|
||||
"needs_qr": bridge.waClient != nil && bridge.waClient.Store.ID == nil,
|
||||
},
|
||||
"message_count": len(bridge.messages),
|
||||
}
|
||||
json.NewEncoder(w).Encode(status)
|
||||
}
|
||||
|
||||
func handleQR(w http.ResponseWriter, r *http.Request) {
|
||||
format := r.URL.Query().Get("format")
|
||||
|
||||
select {
|
||||
case code := <-bridge.qrChan:
|
||||
// Put it back for other requests
|
||||
select {
|
||||
case bridge.qrChan <- code:
|
||||
default:
|
||||
}
|
||||
|
||||
if format == "png" {
|
||||
png, err := qrcode.Encode(code, qrcode.Medium, 256)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), 500)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "image/png")
|
||||
w.Write(png)
|
||||
} else if format == "base64" {
|
||||
png, err := qrcode.Encode(code, qrcode.Medium, 256)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), 500)
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(map[string]string{
|
||||
"qr_base64": base64.StdEncoding.EncodeToString(png),
|
||||
"qr_raw": code,
|
||||
})
|
||||
} else {
|
||||
// Terminal QR
|
||||
qr, _ := qrcode.New(code, qrcode.Medium)
|
||||
w.Header().Set("Content-Type", "text/plain")
|
||||
w.Write([]byte(qr.ToSmallString(false)))
|
||||
}
|
||||
default:
|
||||
if bridge.connected {
|
||||
json.NewEncoder(w).Encode(map[string]string{"status": "already_connected"})
|
||||
} else {
|
||||
json.NewEncoder(w).Encode(map[string]string{"status": "no_qr_available"})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func handleMessages(w http.ResponseWriter, r *http.Request) {
|
||||
bridge.msgMutex.RLock()
|
||||
defer bridge.msgMutex.RUnlock()
|
||||
|
||||
platform := r.URL.Query().Get("platform")
|
||||
limit := 50
|
||||
|
||||
var filtered []Message
|
||||
for i := len(bridge.messages) - 1; i >= 0 && len(filtered) < limit; i-- {
|
||||
msg := bridge.messages[i]
|
||||
if platform == "" || msg.Platform == platform {
|
||||
filtered = append(filtered, msg)
|
||||
}
|
||||
}
|
||||
|
||||
json.NewEncoder(w).Encode(filtered)
|
||||
}
|
||||
|
||||
func handleMedia(w http.ResponseWriter, r *http.Request) {
|
||||
vars := mux.Vars(r)
|
||||
filename := vars["filename"]
|
||||
|
||||
mediaPath := filepath.Join(bridge.dataDir, "media", filename)
|
||||
http.ServeFile(w, r, mediaPath)
|
||||
}
|
||||
|
||||
func handleSend(w http.ResponseWriter, r *http.Request) {
|
||||
var req struct {
|
||||
Platform string `json:"platform"`
|
||||
To string `json:"to"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
|
||||
http.Error(w, err.Error(), 400)
|
||||
return
|
||||
}
|
||||
|
||||
if req.Platform == "whatsapp" && bridge.waClient != nil && bridge.connected {
|
||||
jid := types.NewJID(req.To, types.DefaultUserServer)
|
||||
_, err := bridge.waClient.SendMessage(context.Background(), jid, &waE2E.Message{
|
||||
Conversation: &req.Message,
|
||||
})
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), 500)
|
||||
return
|
||||
}
|
||||
json.NewEncoder(w).Encode(map[string]string{"status": "sent"})
|
||||
} else {
|
||||
http.Error(w, "platform not connected", 400)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
dataDir := os.Getenv("DATA_DIR")
|
||||
if dataDir == "" {
|
||||
dataDir = filepath.Join(os.Getenv("HOME"), ".message-bridge")
|
||||
}
|
||||
os.MkdirAll(dataDir, 0755)
|
||||
|
||||
port := os.Getenv("PORT")
|
||||
if port == "" {
|
||||
port = "8030"
|
||||
}
|
||||
|
||||
bridge = NewMessageBridge(dataDir)
|
||||
|
||||
// Initialize WhatsApp
|
||||
log.Println("Initializing WhatsApp...")
|
||||
if err := bridge.InitWhatsApp(); err != nil {
|
||||
log.Printf("WhatsApp init error: %v", err)
|
||||
}
|
||||
|
||||
// HTTP server
|
||||
r := mux.NewRouter()
|
||||
r.HandleFunc("/status", handleStatus).Methods("GET")
|
||||
r.HandleFunc("/qr", handleQR).Methods("GET")
|
||||
r.HandleFunc("/messages", handleMessages).Methods("GET")
|
||||
r.HandleFunc("/media/{filename}", handleMedia).Methods("GET")
|
||||
r.HandleFunc("/send", handleSend).Methods("POST")
|
||||
|
||||
srv := &http.Server{
|
||||
Addr: ":" + port,
|
||||
Handler: r,
|
||||
}
|
||||
|
||||
go func() {
|
||||
log.Printf("Message Bridge listening on :%s", port)
|
||||
if err := srv.ListenAndServe(); err != http.ErrServerClosed {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}()
|
||||
|
||||
// Wait for shutdown
|
||||
c := make(chan os.Signal, 1)
|
||||
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
|
||||
<-c
|
||||
|
||||
log.Println("Shutting down...")
|
||||
if bridge.waClient != nil {
|
||||
bridge.waClient.Disconnect()
|
||||
}
|
||||
srv.Shutdown(context.Background())
|
||||
}
|
||||
Binary file not shown.
Loading…
Reference in New Issue