diff options
Diffstat (limited to 'cmd/gotosocial/action/admin/account/account.go')
| -rw-r--r-- | cmd/gotosocial/action/admin/account/account.go | 258 |
1 files changed, 258 insertions, 0 deletions
diff --git a/cmd/gotosocial/action/admin/account/account.go b/cmd/gotosocial/action/admin/account/account.go new file mode 100644 index 000000000..6b1029712 --- /dev/null +++ b/cmd/gotosocial/action/admin/account/account.go @@ -0,0 +1,258 @@ +/* + GoToSocial + Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org + + 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 <http://www.gnu.org/licenses/>. +*/ + +package account + +import ( + "context" + "errors" + "fmt" + "time" + + "github.com/spf13/viper" + "github.com/superseriousbusiness/gotosocial/cmd/gotosocial/action" + "github.com/superseriousbusiness/gotosocial/internal/config" + "github.com/superseriousbusiness/gotosocial/internal/db" + "github.com/superseriousbusiness/gotosocial/internal/db/bundb" + "github.com/superseriousbusiness/gotosocial/internal/gtsmodel" + "github.com/superseriousbusiness/gotosocial/internal/validate" + "golang.org/x/crypto/bcrypt" +) + +// Create creates a new account in the database using the provided flags. +var Create action.GTSAction = func(ctx context.Context) error { + dbConn, err := bundb.NewBunDBService(ctx) + if err != nil { + return fmt.Errorf("error creating dbservice: %s", err) + } + + username := viper.GetString(config.Keys.AdminAccountUsername) + if username == "" { + return errors.New("no username set") + } + if err := validate.Username(username); err != nil { + return err + } + + email := viper.GetString(config.Keys.AdminAccountEmail) + if email == "" { + return errors.New("no email set") + } + if err := validate.Email(email); err != nil { + return err + } + + password := viper.GetString(config.Keys.AdminAccountPassword) + if password == "" { + return errors.New("no password set") + } + if err := validate.NewPassword(password); err != nil { + return err + } + + _, err = dbConn.NewSignup(ctx, username, "", false, email, password, nil, "", "", false, false) + if err != nil { + return err + } + + return dbConn.Stop(ctx) +} + +// Confirm sets a user to Approved, sets Email to the current UnconfirmedEmail value, and sets ConfirmedAt to now. +var Confirm action.GTSAction = func(ctx context.Context) error { + dbConn, err := bundb.NewBunDBService(ctx) + if err != nil { + return fmt.Errorf("error creating dbservice: %s", err) + } + + username := viper.GetString(config.Keys.AdminAccountUsername) + if username == "" { + return errors.New("no username set") + } + if err := validate.Username(username); err != nil { + return err + } + + a, err := dbConn.GetLocalAccountByUsername(ctx, username) + if err != nil { + return err + } + + u := >smodel.User{} + if err := dbConn.GetWhere(ctx, []db.Where{{Key: "account_id", Value: a.ID}}, u); err != nil { + return err + } + + u.Approved = true + u.Email = u.UnconfirmedEmail + u.ConfirmedAt = time.Now() + if err := dbConn.UpdateByPrimaryKey(ctx, u); err != nil { + return err + } + + return dbConn.Stop(ctx) +} + +// Promote sets a user to admin. +var Promote action.GTSAction = func(ctx context.Context) error { + dbConn, err := bundb.NewBunDBService(ctx) + if err != nil { + return fmt.Errorf("error creating dbservice: %s", err) + } + + username := viper.GetString(config.Keys.AdminAccountUsername) + if username == "" { + return errors.New("no username set") + } + if err := validate.Username(username); err != nil { + return err + } + + a, err := dbConn.GetLocalAccountByUsername(ctx, username) + if err != nil { + return err + } + + u := >smodel.User{} + if err := dbConn.GetWhere(ctx, []db.Where{{Key: "account_id", Value: a.ID}}, u); err != nil { + return err + } + u.Admin = true + if err := dbConn.UpdateByPrimaryKey(ctx, u); err != nil { + return err + } + + return dbConn.Stop(ctx) +} + +// Demote sets admin on a user to false. +var Demote action.GTSAction = func(ctx context.Context) error { + dbConn, err := bundb.NewBunDBService(ctx) + if err != nil { + return fmt.Errorf("error creating dbservice: %s", err) + } + + username := viper.GetString(config.Keys.AdminAccountUsername) + if username == "" { + return errors.New("no username set") + } + if err := validate.Username(username); err != nil { + return err + } + + a, err := dbConn.GetLocalAccountByUsername(ctx, username) + if err != nil { + return err + } + + u := >smodel.User{} + if err := dbConn.GetWhere(ctx, []db.Where{{Key: "account_id", Value: a.ID}}, u); err != nil { + return err + } + u.Admin = false + if err := dbConn.UpdateByPrimaryKey(ctx, u); err != nil { + return err + } + + return dbConn.Stop(ctx) +} + +// Disable sets Disabled to true on a user. +var Disable action.GTSAction = func(ctx context.Context) error { + dbConn, err := bundb.NewBunDBService(ctx) + if err != nil { + return fmt.Errorf("error creating dbservice: %s", err) + } + + username := viper.GetString(config.Keys.AdminAccountUsername) + if username == "" { + return errors.New("no username set") + } + if err := validate.Username(username); err != nil { + return err + } + + a, err := dbConn.GetLocalAccountByUsername(ctx, username) + if err != nil { + return err + } + + u := >smodel.User{} + if err := dbConn.GetWhere(ctx, []db.Where{{Key: "account_id", Value: a.ID}}, u); err != nil { + return err + } + u.Disabled = true + if err := dbConn.UpdateByPrimaryKey(ctx, u); err != nil { + return err + } + + return dbConn.Stop(ctx) +} + +// Suspend suspends the target account, cleanly removing all of its media, followers, following, likes, statuses, etc. +var Suspend action.GTSAction = func(ctx context.Context) error { + // TODO + return nil +} + +// Password sets the password of target account. +var Password action.GTSAction = func(ctx context.Context) error { + dbConn, err := bundb.NewBunDBService(ctx) + if err != nil { + return fmt.Errorf("error creating dbservice: %s", err) + } + + username := viper.GetString(config.Keys.AdminAccountUsername) + if username == "" { + return errors.New("no username set") + } + if err := validate.Username(username); err != nil { + return err + } + + password := viper.GetString(config.Keys.AdminAccountPassword) + if password == "" { + return errors.New("no password set") + } + if err := validate.NewPassword(password); err != nil { + return err + } + + a, err := dbConn.GetLocalAccountByUsername(ctx, username) + if err != nil { + return err + } + + u := >smodel.User{} + if err := dbConn.GetWhere(ctx, []db.Where{{Key: "account_id", Value: a.ID}}, u); err != nil { + return err + } + + pw, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost) + if err != nil { + return fmt.Errorf("error hashing password: %s", err) + } + + u.EncryptedPassword = string(pw) + + if err := dbConn.UpdateByPrimaryKey(ctx, u); err != nil { + return err + } + + return nil +} |