Switch startup, config, commands and db migrations to mautrix-go systems

main.go

@@ -1,43 +1,168 @@
+// mautrix-discord - A Matrix-Discord puppeting bridge.
+// Copyright (C) 2022 Tulir Asokan
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
 package main
 
 import (
-	"fmt"
-	"os"
+	_ "embed"
+	"sync"
 
-	"github.com/alecthomas/kong"
+	"go.mau.fi/mautrix-discord/database"
+	"maunium.net/go/mautrix/bridge"
+	"maunium.net/go/mautrix/bridge/commands"
+	"maunium.net/go/mautrix/id"
+	"maunium.net/go/mautrix/util/configupgrade"
 
 	"go.mau.fi/mautrix-discord/config"
-	"go.mau.fi/mautrix-discord/consts"
-	"go.mau.fi/mautrix-discord/globals"
-	"go.mau.fi/mautrix-discord/registration"
-	"go.mau.fi/mautrix-discord/run"
-	"go.mau.fi/mautrix-discord/version"
 )
 
-var cli struct {
-	globals.Globals
+// Information to find out exactly which commit the bridge was built from.
+// These are filled at build time with the -X linker flag.
+var (
+	Tag       = "unknown"
+	Commit    = "unknown"
+	BuildTime = "unknown"
+)
 
-	GenerateConfig       config.Cmd       `kong:"cmd,help='Generate the default configuration and exit.'"`
-	GenerateRegistration registration.Cmd `kong:"cmd,help='Generate the registration file for synapse and exit.'"`
-	Run                  run.Cmd          `kong:"cmd,help='Run the bridge.',default='1'"`
-	Version              version.Cmd      `kong:"cmd,help='Display the version and exit.'"`
+//go:embed example-config.yaml
+var ExampleConfig string
+
+type DiscordBridge struct {
+	bridge.Bridge
+
+	Config *config.Config
+	DB     *database.Database
+
+	provisioning *ProvisioningAPI
+
+	usersByMXID map[id.UserID]*User
+	usersByID   map[string]*User
+	usersLock   sync.Mutex
+
+	managementRooms     map[id.RoomID]*User
+	managementRoomsLock sync.Mutex
+
+	portalsByMXID map[id.RoomID]*Portal
+	portalsByID   map[database.PortalKey]*Portal
+	portalsLock   sync.Mutex
+
+	puppets             map[string]*Puppet
+	puppetsByCustomMXID map[id.UserID]*Puppet
+	puppetsLock         sync.Mutex
+}
+
+func (br *DiscordBridge) GetExampleConfig() string {
+	return ExampleConfig
+}
+
+func (br *DiscordBridge) GetConfigPtr() interface{} {
+	br.Config = &config.Config{
+		BaseConfig: &br.Bridge.Config,
+	}
+	br.Config.BaseConfig.Bridge = &br.Config.Bridge
+	return br.Config
+}
+
+func (br *DiscordBridge) Init() {
+	br.CommandProcessor = commands.NewProcessor(&br.Bridge)
+	br.RegisterCommands()
+
+	br.DB = database.New(br.Bridge.DB)
+}
+
+func (br *DiscordBridge) Start() {
+	if br.Config.Bridge.Provisioning.SharedSecret != "disable" {
+		br.provisioning = newProvisioningAPI(br)
+	}
+	go br.startUsers()
+}
+
+func (br *DiscordBridge) Stop() {
+	for _, user := range br.usersByMXID {
+		if user.Session == nil {
+			continue
+		}
+
+		br.Log.Debugln("Disconnecting", user.MXID)
+		user.Session.Close()
+	}
+}
+
+func (br *DiscordBridge) GetIPortal(mxid id.RoomID) bridge.Portal {
+	p := br.GetPortalByMXID(mxid)
+	if p == nil {
+		return nil
+	}
+	return p
+}
+
+func (br *DiscordBridge) GetIUser(mxid id.UserID, create bool) bridge.User {
+	p := br.GetUserByMXID(mxid)
+	if p == nil {
+		return nil
+	}
+	return p
+}
+
+func (br *DiscordBridge) IsGhost(mxid id.UserID) bool {
+	_, isGhost := br.ParsePuppetMXID(mxid)
+	return isGhost
+}
+
+func (br *DiscordBridge) GetIGhost(mxid id.UserID) bridge.Ghost {
+	p := br.GetPuppetByMXID(mxid)
+	if p == nil {
+		return nil
+	}
+	return p
+}
+
+func (br *DiscordBridge) CreatePrivatePortal(id id.RoomID, user bridge.User, ghost bridge.Ghost) {
+	//TODO implement
 }
 
 func main() {
-	ctx := kong.Parse(
-		&cli,
-		kong.Name(consts.Name),
-		kong.Description(consts.Description),
-		kong.UsageOnError(),
-		kong.ConfigureHelp(kong.HelpOptions{
-			Compact: true,
-			Summary: true,
-		}),
-	)
+	br := &DiscordBridge{
+		usersByMXID: make(map[id.UserID]*User),
+		usersByID:   make(map[string]*User),
 
-	err := ctx.Run(&cli.Globals)
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "error: %s\n", err)
-		os.Exit(1)
+		managementRooms: make(map[id.RoomID]*User),
+
+		portalsByMXID: make(map[id.RoomID]*Portal),
+		portalsByID:   make(map[database.PortalKey]*Portal),
+
+		puppets:             make(map[string]*Puppet),
+		puppetsByCustomMXID: make(map[id.UserID]*Puppet),
 	}
+	br.Bridge = bridge.Bridge{
+		Name:         "mautrix-discord",
+		URL:          "https://github.com/mautrix/discord",
+		Description:  "A Matrix-Discord puppeting bridge.",
+		Version:      "0.1.0",
+		ProtocolName: "Discord",
+
+		ConfigUpgrader: &configupgrade.StructUpgrader{
+			SimpleUpgrader: configupgrade.SimpleUpgrader(config.DoUpgrade),
+			Blocks:         config.SpacedBlocks,
+			Base:           ExampleConfig,
+		},
+
+		Child: br,
+	}
+	br.InitVersion(Tag, Commit, BuildTime)
+
+	br.Main()
 }