This commit is contained in:
@@ -0,0 +1,84 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
|
||||
"maunium.net/go/mautrix/appservice"
|
||||
"maunium.net/go/mautrix/event"
|
||||
)
|
||||
|
||||
func main() {
|
||||
configPath := flag.String("config", "config.yaml", "Path to configuration file")
|
||||
dbPath := flag.String("database", "ntfy-bridge.db", "Path to SQLite database file")
|
||||
flag.Parse()
|
||||
|
||||
cfg, err := LoadConfig(*configPath)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to load config: %v", err)
|
||||
}
|
||||
|
||||
db, err := NewDatabase(*dbPath)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to open database: %v", err)
|
||||
}
|
||||
defer db.Close()
|
||||
|
||||
reg := &appservice.Registration{
|
||||
ID: cfg.Appservice.ID,
|
||||
AppToken: cfg.Appservice.ASToken,
|
||||
ServerToken: cfg.Appservice.HSToken,
|
||||
SenderLocalpart: cfg.Appservice.BotLocalpart,
|
||||
}
|
||||
|
||||
as, err := appservice.CreateFull(appservice.CreateOpts{
|
||||
Registration: reg,
|
||||
HomeserverDomain: cfg.Homeserver.Domain,
|
||||
HomeserverURL: cfg.Homeserver.URL,
|
||||
HostConfig: appservice.HostConfig{Hostname: cfg.Appservice.Address, Port: cfg.Appservice.Port},
|
||||
})
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to create appservice: %v", err)
|
||||
}
|
||||
|
||||
ghostPrefix := fmt.Sprintf("@%s_", cfg.Appservice.BotLocalpart)
|
||||
|
||||
handler := &MatrixHandler{
|
||||
AS: as,
|
||||
DB: db,
|
||||
Ntfy: NewNtfyClient(cfg.Bridge.NtfyURL),
|
||||
Config: &cfg.Bridge,
|
||||
HSDomain: cfg.Homeserver.Domain,
|
||||
GhostPrefix: ghostPrefix,
|
||||
}
|
||||
|
||||
ep := appservice.NewEventProcessor(as)
|
||||
ep.On(event.EventMessage, handler.HandleEvent)
|
||||
ep.On(event.StateMember, handler.HandleEvent)
|
||||
|
||||
bridges, err := db.GetAllBridges()
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to load bridges: %v", err)
|
||||
}
|
||||
|
||||
botIntent := as.BotIntent()
|
||||
for topic, roomID := range bridges {
|
||||
log.Printf("[startup] Re-establishing ntfy listener for topic %q -> room %s", topic, roomID)
|
||||
handler.Ntfy.StartNtfyListener(topic, roomID, botIntent, as.BotMXID())
|
||||
}
|
||||
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
ep.Start(ctx)
|
||||
|
||||
log.Printf("[main] Starting appservice on %s:%d", cfg.Appservice.Address, cfg.Appservice.Port)
|
||||
log.Printf("[main] ntfy base URL: %s", cfg.Bridge.NtfyURL)
|
||||
log.Printf("[main] Loaded %d existing bridge(s)", len(bridges))
|
||||
as.Start()
|
||||
|
||||
_ = os.Stdout
|
||||
}
|
||||
Reference in New Issue
Block a user