4fb0cdb847
All of the login methods need to do (effectively) the same thing once we have a token, so refactor this out into something we can reuse.
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
// mautrix-discord - A Matrix-Discord puppeting bridge.
|
|
// Copyright (C) 2024 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 connector
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"maunium.net/go/mautrix/bridgev2"
|
|
)
|
|
|
|
const LoginStepIDComplete = "fi.mau.discord.login.complete"
|
|
|
|
func (d *DiscordConnector) GetLoginFlows() []bridgev2.LoginFlow {
|
|
return []bridgev2.LoginFlow{
|
|
{
|
|
ID: LoginFlowIDRemoteAuth,
|
|
Name: "QR Code",
|
|
Description: "Scan a QR code with the Discord mobile app to log in.",
|
|
},
|
|
{
|
|
ID: LoginFlowIDBrowser,
|
|
Name: "Browser",
|
|
Description: "Log in to your Discord account in a web browser.",
|
|
},
|
|
{
|
|
ID: LoginFlowIDToken,
|
|
Name: "Token",
|
|
Description: "Provide a Discord user token to connect with.",
|
|
},
|
|
}
|
|
}
|
|
|
|
func (d *DiscordConnector) CreateLogin(ctx context.Context, user *bridgev2.User, flowID string) (bridgev2.LoginProcess, error) {
|
|
login := DiscordGenericLogin{
|
|
connector: d,
|
|
User: user,
|
|
}
|
|
|
|
switch flowID {
|
|
case LoginFlowIDToken:
|
|
return &DiscordTokenLogin{DiscordGenericLogin: &login}, nil
|
|
case LoginFlowIDRemoteAuth:
|
|
return &DiscordRemoteAuthLogin{DiscordGenericLogin: &login}, nil
|
|
case LoginFlowIDBrowser:
|
|
return &DiscordBrowserLogin{DiscordGenericLogin: &login}, nil
|
|
default:
|
|
return nil, fmt.Errorf("unknown discord login flow id")
|
|
}
|
|
}
|