package db import ( "context" "os" "emperror.dev/errors" "github.com/georgysavva/scany/v2/pgxscan" "github.com/jackc/pgx/v5" "golang.org/x/oauth2" ) type FediverseApp struct { ID int64 // Instance is the instance's base API url, excluding schema Instance string ClientID string ClientSecret string InstanceType string } func (f FediverseApp) ClientConfig() *oauth2.Config { if f.MastodonCompatible() { return &oauth2.Config{ ClientID: f.ClientID, ClientSecret: f.ClientSecret, Endpoint: oauth2.Endpoint{ AuthURL: "https://" + f.Instance + "/oauth/authorize", TokenURL: "https://" + f.Instance + "/oauth/token", AuthStyle: oauth2.AuthStyleInParams, }, Scopes: []string{"read:accounts"}, RedirectURL: os.Getenv("BASE_URL") + "/auth/login/mastodon/" + f.Instance, } } return &oauth2.Config{ ClientID: f.ClientID, ClientSecret: f.ClientSecret, Endpoint: oauth2.Endpoint{ AuthURL: "https://" + f.Instance + "/auth", TokenURL: "https://" + f.Instance + "/api/auth/session/oauth", AuthStyle: oauth2.AuthStyleInHeader, }, Scopes: []string{"read:account"}, RedirectURL: os.Getenv("BASE_URL") + "/auth/login/misskey/" + f.Instance, } } func (f FediverseApp) MastodonCompatible() bool { return f.InstanceType == "mastodon" || f.InstanceType == "pleroma" || f.InstanceType == "akkoma" || f.InstanceType == "incestoma" || f.InstanceType == "pixelfed" || f.InstanceType == "gotosocial" } func (f FediverseApp) Misskey() bool { return f.InstanceType == "misskey" || f.InstanceType == "foundkey" || f.InstanceType == "calckey" || f.InstanceType == "firefish" } const ErrNoInstanceApp = errors.Sentinel("instance doesn't have an app") func (db *DB) FediverseApp(ctx context.Context, instance string) (fa FediverseApp, err error) { sql, args, err := sq.Select("*").From("fediverse_apps").Where("instance = ?", instance).ToSql() if err != nil { return fa, errors.Wrap(err, "building sql") } err = pgxscan.Get(ctx, db, &fa, sql, args...) if err != nil { if errors.Cause(err) == pgx.ErrNoRows { return fa, ErrNoInstanceApp } return fa, errors.Wrap(err, "executing query") } return fa, nil } func (db *DB) FediverseAppByID(ctx context.Context, id int64) (fa FediverseApp, err error) { sql, args, err := sq.Select("*").From("fediverse_apps").Where("id = ?", id).ToSql() if err != nil { return fa, errors.Wrap(err, "building sql") } err = pgxscan.Get(ctx, db, &fa, sql, args...) if err != nil { return fa, errors.Wrap(err, "executing query") } return fa, nil } func (db *DB) CreateFediverseApp(ctx context.Context, instance, instanceType, clientID, clientSecret string) (fa FediverseApp, err error) { sql, args, err := sq.Insert("fediverse_apps"). Columns("instance", "instance_type", "client_id", "client_secret"). Values(instance, instanceType, clientID, clientSecret). Suffix("RETURNING *").ToSql() if err != nil { return fa, errors.Wrap(err, "building query") } err = pgxscan.Get(ctx, db, &fa, sql, args...) if err != nil { return fa, errors.Wrap(err, "executing query") } return fa, nil }