From 0884f89431cd26bcc9674b3b7ab628b090f5881e Mon Sep 17 00:00:00 2001 From: tobi <31960611+tsmethurst@users.noreply.github.com> Date: Tue, 7 Dec 2021 13:31:39 +0100 Subject: Implement Cobra CLI tooling, Viper config tooling (#336) * start pulling out + replacing urfave and config * replace many many instances of config * move more stuff => viper * properly remove urfave * move some flags to root command * add testrig commands to root * alias config file keys * start adding cli parsing tests * reorder viper init * remove config path alias * fmt * change config file keys to non-nested * we're more or less in business now * tidy up the common func * go fmt * get tests passing again * add note about the cliparsing tests * reorganize * update docs with changes * structure cmd dir better * rename + move some files around * fix dangling comma --- cmd/gotosocial/admin.go | 169 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 169 insertions(+) create mode 100644 cmd/gotosocial/admin.go (limited to 'cmd/gotosocial/admin.go') diff --git a/cmd/gotosocial/admin.go b/cmd/gotosocial/admin.go new file mode 100644 index 000000000..9ee4461ea --- /dev/null +++ b/cmd/gotosocial/admin.go @@ -0,0 +1,169 @@ +/* + 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 . +*/ + +package main + +import ( + "github.com/spf13/cobra" + "github.com/superseriousbusiness/gotosocial/cmd/gotosocial/action/admin/account" + "github.com/superseriousbusiness/gotosocial/cmd/gotosocial/action/admin/trans" + "github.com/superseriousbusiness/gotosocial/cmd/gotosocial/flag" + "github.com/superseriousbusiness/gotosocial/internal/config" +) + +func adminCommands() *cobra.Command { + adminCmd := &cobra.Command{ + Use: "admin", + Short: "gotosocial admin-related tasks", + } + + /* + ADMIN ACCOUNT COMMANDS + */ + + adminAccountCmd := &cobra.Command{ + Use: "account", + Short: "admin commands related to accounts", + } + flag.AdminAccount(adminAccountCmd, config.Defaults) + + adminAccountCreateCmd := &cobra.Command{ + Use: "create", + Short: "create a new account", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Create) + }, + } + flag.AdminAccountCreate(adminAccountCreateCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountCreateCmd) + + adminAccountConfirmCmd := &cobra.Command{ + Use: "confirm", + Short: "confirm an existing account manually, thereby skipping email confirmation", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Confirm) + }, + } + flag.AdminAccount(adminAccountConfirmCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountConfirmCmd) + + adminAccountPromoteCmd := &cobra.Command{ + Use: "promote", + Short: "promote an account to admin", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Promote) + }, + } + flag.AdminAccount(adminAccountPromoteCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountPromoteCmd) + + adminAccountDemoteCmd := &cobra.Command{ + Use: "demote", + Short: "demote an account from admin to normal user", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Demote) + }, + } + flag.AdminAccount(adminAccountDemoteCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountDemoteCmd) + + adminAccountDisableCmd := &cobra.Command{ + Use: "disable", + Short: "prevent an account from signing in or posting etc, but don't delete anything", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Disable) + }, + } + flag.AdminAccount(adminAccountDisableCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountDisableCmd) + + adminAccountSuspendCmd := &cobra.Command{ + Use: "suspend", + Short: "completely remove an account and all of its posts, media, etc", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Suspend) + }, + } + flag.AdminAccount(adminAccountSuspendCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountSuspendCmd) + + adminAccountPasswordCmd := &cobra.Command{ + Use: "password", + Short: "set a new password for the given account", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), account.Password) + }, + } + flag.AdminAccountPassword(adminAccountPasswordCmd, config.Defaults) + adminAccountCmd.AddCommand(adminAccountPasswordCmd) + + adminCmd.AddCommand(adminAccountCmd) + + /* + ADMIN IMPORT/EXPORT COMMANDS + */ + + adminExportCmd := &cobra.Command{ + Use: "export", + Short: "export data from the database to file at the given path", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), trans.Export) + }, + } + flag.AdminTrans(adminExportCmd, config.Defaults) + adminCmd.AddCommand(adminExportCmd) + + adminImportCmd := &cobra.Command{ + Use: "import", + Short: "import data from a file into the database", + PreRunE: func(cmd *cobra.Command, args []string) error { + return preRun(cmd) + }, + RunE: func(cmd *cobra.Command, args []string) error { + return run(cmd.Context(), trans.Import) + }, + } + flag.AdminTrans(adminImportCmd, config.Defaults) + adminCmd.AddCommand(adminImportCmd) + + return adminCmd +} -- cgit v1.2.3