94104102d2
It works, but nothing is persisted yet as the user object needs some tweaking especially the database side.
135 lines
2.5 KiB
Go
135 lines
2.5 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.Printfln(g.context.Stdout, "failed to log in")
|
|
|
|
return err
|
|
}
|
|
|
|
g.user.User.ID = user.UserID
|
|
g.user.User.Discriminator = user.Discriminator
|
|
g.user.User.Username = user.Username
|
|
|
|
g.handler.log.Warnln("users:", user)
|
|
g.handler.log.Warnln("err:", err)
|
|
|
|
return nil
|
|
}
|