134 lines
2.4 KiB
Go
134 lines
2.4 KiB
Go
package bridge
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/alecthomas/kong"
|
|
|
|
"maunium.net/go/mautrix/appservice"
|
|
"maunium.net/go/mautrix/event"
|
|
"maunium.net/go/mautrix/format"
|
|
"maunium.net/go/mautrix/id"
|
|
|
|
"gitlab.com/beeper/discord/consts"
|
|
"gitlab.com/beeper/discord/remoteauth"
|
|
"gitlab.com/beeper/discord/version"
|
|
)
|
|
|
|
type globals struct {
|
|
context *kong.Context
|
|
|
|
bridge *Bridge
|
|
bot *appservice.IntentAPI
|
|
portal *Portal
|
|
handler *commandHandler
|
|
roomID id.RoomID
|
|
user *User
|
|
replyTo id.EventID
|
|
}
|
|
|
|
func (g *globals) reply(msg string) {
|
|
content := format.RenderMarkdown(msg, true, false)
|
|
content.MsgType = event.MsgNotice
|
|
intent := g.bot
|
|
|
|
if g.portal != nil && g.portal.IsPrivateChat() {
|
|
intent = g.portal.MainIntent()
|
|
}
|
|
|
|
_, err := intent.SendMessageEvent(g.roomID, event.EventMessage, content)
|
|
if err != nil {
|
|
g.handler.log.Warnfln("Failed to reply to command from %q: %v", g.user.MXID, err)
|
|
}
|
|
}
|
|
|
|
type commands struct {
|
|
globals
|
|
|
|
Help helpCmd `kong:"cmd,help='Displays this message.'"`
|
|
Login loginCmd `kong:"cmd,help='Log in to Discord.'"`
|
|
Version versionCmd `kong:"cmd,help='Displays the version of the bridge.'"`
|
|
}
|
|
|
|
type helpCmd struct {
|
|
Command []string `kong:"arg,optional,help='The command to get help on.'"`
|
|
}
|
|
|
|
func (c *helpCmd) Run(g *globals) error {
|
|
ctx, err := kong.Trace(g.context.Kong, c.Command)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if ctx.Error != nil {
|
|
return err
|
|
}
|
|
|
|
err = ctx.PrintUsage(true)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fmt.Fprintln(g.context.Stdout)
|
|
|
|
return nil
|
|
}
|
|
|
|
type versionCmd struct{}
|
|
|
|
func (c *versionCmd) Run(g *globals) error {
|
|
fmt.Fprintln(g.context.Stdout, consts.Name, version.String)
|
|
|
|
return nil
|
|
}
|
|
|
|
type loginCmd struct{}
|
|
|
|
func (l *loginCmd) Run(g *globals) error {
|
|
client, err := remoteauth.New()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
qrChan := make(chan string)
|
|
doneChan := make(chan struct{})
|
|
|
|
go func() {
|
|
code := <-qrChan
|
|
|
|
_, err := g.user.sendQRCode(g.bot, g.roomID, code)
|
|
if err != nil {
|
|
fmt.Fprintln(g.context.Stdout, "failed to generate the qrcode")
|
|
|
|
return
|
|
}
|
|
}()
|
|
|
|
ctx := context.Background()
|
|
|
|
if err := client.Dial(ctx, qrChan, doneChan); err != nil {
|
|
close(qrChan)
|
|
close(doneChan)
|
|
|
|
return err
|
|
}
|
|
|
|
<-doneChan
|
|
|
|
user, err := client.Result()
|
|
if err != nil {
|
|
fmt.Println(g.context.Stdout, "failed to log in")
|
|
|
|
return err
|
|
}
|
|
|
|
if err := g.user.Login(user.Token); err != nil {
|
|
fmt.Println(g.context.Stdout, "failed to login", err)
|
|
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|