2017-05-22 16:49:32 +01:00
|
|
|
package accounts
|
2017-05-19 10:27:03 +01:00
|
|
|
|
|
|
|
import (
|
2017-09-18 14:15:27 +01:00
|
|
|
"context"
|
2017-09-01 10:13:10 +01:00
|
|
|
"errors"
|
2020-02-13 17:27:33 +00:00
|
|
|
"net/url"
|
2017-05-30 17:51:40 +01:00
|
|
|
|
2017-05-23 17:43:05 +01:00
|
|
|
"github.com/matrix-org/dendrite/clientapi/auth/authtypes"
|
2020-02-13 17:27:33 +00:00
|
|
|
"github.com/matrix-org/dendrite/clientapi/auth/storage/accounts/postgres"
|
|
|
|
"github.com/matrix-org/dendrite/clientapi/auth/storage/accounts/sqlite3"
|
2017-07-17 18:10:56 +01:00
|
|
|
"github.com/matrix-org/dendrite/common"
|
2017-05-19 10:27:03 +01:00
|
|
|
"github.com/matrix-org/gomatrixserverlib"
|
|
|
|
)
|
|
|
|
|
2020-02-13 17:27:33 +00:00
|
|
|
type Database interface {
|
|
|
|
common.PartitionStorer
|
|
|
|
GetAccountByPassword(ctx context.Context, localpart, plaintextPassword string) (*authtypes.Account, error)
|
|
|
|
GetProfileByLocalpart(ctx context.Context, localpart string) (*authtypes.Profile, error)
|
|
|
|
SetAvatarURL(ctx context.Context, localpart string, avatarURL string) error
|
|
|
|
SetDisplayName(ctx context.Context, localpart string, displayName string) error
|
|
|
|
CreateAccount(ctx context.Context, localpart, plaintextPassword, appserviceID string) (*authtypes.Account, error)
|
|
|
|
UpdateMemberships(ctx context.Context, eventsToAdd []gomatrixserverlib.Event, idsToRemove []string) error
|
|
|
|
GetMembershipInRoomByLocalpart(ctx context.Context, localpart, roomID string) (authtypes.Membership, error)
|
|
|
|
GetMembershipsByLocalpart(ctx context.Context, localpart string) (memberships []authtypes.Membership, err error)
|
|
|
|
SaveAccountData(ctx context.Context, localpart, roomID, dataType, content string) error
|
|
|
|
GetAccountData(ctx context.Context, localpart string) (global []gomatrixserverlib.ClientEvent, rooms map[string][]gomatrixserverlib.ClientEvent, err error)
|
|
|
|
GetAccountDataByType(ctx context.Context, localpart, roomID, dataType string) (data *gomatrixserverlib.ClientEvent, err error)
|
|
|
|
GetNewNumericLocalpart(ctx context.Context) (int64, error)
|
|
|
|
SaveThreePIDAssociation(ctx context.Context, threepid, localpart, medium string) (err error)
|
|
|
|
RemoveThreePIDAssociation(ctx context.Context, threepid string, medium string) (err error)
|
|
|
|
GetLocalpartForThreePID(ctx context.Context, threepid string, medium string) (localpart string, err error)
|
|
|
|
GetThreePIDsForLocalpart(ctx context.Context, localpart string) (threepids []authtypes.ThreePID, err error)
|
|
|
|
GetFilter(ctx context.Context, localpart string, filterID string) (*gomatrixserverlib.Filter, error)
|
|
|
|
PutFilter(ctx context.Context, localpart string, filter *gomatrixserverlib.Filter) (string, error)
|
|
|
|
CheckAccountAvailability(ctx context.Context, localpart string) (bool, error)
|
|
|
|
GetAccountByLocalpart(ctx context.Context, localpart string) (*authtypes.Account, error)
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewDatabase(dataSourceName string, serverName gomatrixserverlib.ServerName) (Database, error) {
|
|
|
|
uri, err := url.Parse(dataSourceName)
|
2017-05-19 10:27:03 +01:00
|
|
|
if err != nil {
|
2020-02-13 17:27:33 +00:00
|
|
|
return postgres.NewDatabase(dataSourceName, serverName)
|
2017-05-19 10:27:03 +01:00
|
|
|
}
|
2020-02-13 17:27:33 +00:00
|
|
|
switch uri.Scheme {
|
|
|
|
case "postgres":
|
|
|
|
return postgres.NewDatabase(dataSourceName, serverName)
|
|
|
|
case "file":
|
|
|
|
return sqlite3.NewDatabase(dataSourceName, serverName)
|
|
|
|
default:
|
|
|
|
return postgres.NewDatabase(dataSourceName, serverName)
|
2017-05-19 10:27:03 +01:00
|
|
|
}
|
|
|
|
}
|
2017-09-01 10:13:10 +01:00
|
|
|
|
|
|
|
// Err3PIDInUse is the error returned when trying to save an association involving
|
|
|
|
// a third-party identifier which is already associated to a local user.
|
|
|
|
var Err3PIDInUse = errors.New("This third-party identifier is already in use")
|