diff options
author | Jason Mojica <jason@jasonmojica.net> | 2018-10-13 08:35:00 -0400 |
---|---|---|
committer | George Goldberg <george@gberg.me> | 2018-10-13 13:35:00 +0100 |
commit | f025e749a903ad7ee621761d6c1a6aa28592ede3 (patch) | |
tree | 5969714efd86392625f6156c417f04c141ccb8e1 /cmd/mattermost/commands/command.go | |
parent | 908ed5555f8a3d37cd057035b2792d66c8b7838a (diff) | |
download | chat-f025e749a903ad7ee621761d6c1a6aa28592ede3.tar.gz chat-f025e749a903ad7ee621761d6c1a6aa28592ede3.tar.bz2 chat-f025e749a903ad7ee621761d6c1a6aa28592ede3.zip |
MM-12355: Add CLI command "command create" (#9569)
* Add create command
* Create CreateCommand Tests
* Change method flag description
* Change nil test assertion
Diffstat (limited to 'cmd/mattermost/commands/command.go')
-rw-r--r-- | cmd/mattermost/commands/command.go | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/cmd/mattermost/commands/command.go b/cmd/mattermost/commands/command.go index aee2eb290..63f18bed6 100644 --- a/cmd/mattermost/commands/command.go +++ b/cmd/mattermost/commands/command.go @@ -5,8 +5,10 @@ package commands import ( "errors" + "strings" "fmt" + "github.com/mattermost/mattermost-server/app" "github.com/mattermost/mattermost-server/model" "github.com/spf13/cobra" @@ -17,6 +19,15 @@ var CommandCmd = &cobra.Command{ Short: "Management of slash commands", } +var CommandCreateCmd = &cobra.Command{ + Use: "create [team]", + Short: "Create a custom slash command", + Long: `Create a custom slash command for the specified team.`, + Args: cobra.MinimumNArgs(1), + Example: ` command create myteam --title MyCommand --description "My Command Description" --trigger-word mycommand --url http://localhost:8000/my-slash-handler --creator myusername --response-username my-bot-username --icon http://localhost:8000/my-slash-handler-bot-icon.png --autocomplete --post`, + RunE: createCommandCmdF, +} + var CommandMoveCmd = &cobra.Command{ Use: "move", Short: "Move a slash command to a different team", @@ -34,13 +45,92 @@ var CommandListCmd = &cobra.Command{ } func init() { + CommandCreateCmd.Flags().String("title", "", "Command Title") + CommandCreateCmd.Flags().String("description", "", "Command Description") + CommandCreateCmd.Flags().String("trigger-word", "", "Command Trigger Word (required)") + CommandCreateCmd.MarkFlagRequired("trigger-word") + CommandCreateCmd.Flags().String("url", "", "Command Callback URL (required)") + CommandCreateCmd.MarkFlagRequired("url") + CommandCreateCmd.Flags().String("creator", "", "Command Creator's Username (required)") + CommandCreateCmd.MarkFlagRequired("creator") + CommandCreateCmd.Flags().String("response-username", "", "Command Response Username") + CommandCreateCmd.Flags().String("icon", "", "Command Icon URL") + CommandCreateCmd.Flags().Bool("autocomplete", false, "Show Command in autocomplete list") + CommandCreateCmd.Flags().String("autocompleteDesc", "", "Short Command Description for autocomplete list") + CommandCreateCmd.Flags().String("autocompleteHint", "", "Command Arguments displayed as help in autocomplete list") + CommandCreateCmd.Flags().Bool("post", false, "Use POST method for Callback URL") + CommandCmd.AddCommand( + CommandCreateCmd, CommandMoveCmd, CommandListCmd, ) RootCmd.AddCommand(CommandCmd) } +func createCommandCmdF(command *cobra.Command, args []string) error { + a, err := InitDBCommandContextCobra(command) + if err != nil { + return err + } + defer a.Shutdown() + + team := getTeamFromTeamArg(a, args[0]) + if team == nil { + return errors.New("unable to find team '" + args[0] + "'") + } + + title, _ := command.Flags().GetString("title") + description, _ := command.Flags().GetString("description") + trigger, _ := command.Flags().GetString("trigger-word") + + if strings.HasPrefix(trigger, "/") { + return errors.New("a trigger word cannot begin with a /") + } + if strings.Contains(trigger, " ") { + return errors.New("a trigger word must not contain spaces") + } + + url, _ := command.Flags().GetString("url") + creator, _ := command.Flags().GetString("creator") + user := getUserFromUserArg(a, creator) + if user == nil { + return errors.New("unable to find user '" + creator + "'") + } + responseUsername, _ := command.Flags().GetString("response-username") + icon, _ := command.Flags().GetString("icon") + autocomplete, _ := command.Flags().GetBool("autocomplete") + autocompleteDesc, _ := command.Flags().GetString("autocompleteDesc") + autocompleteHint, _ := command.Flags().GetString("autocompleteHint") + post, errp := command.Flags().GetBool("post") + method := "P" + if errp != nil || post == false { + method = "G" + } + + newCommand := &model.Command{ + CreatorId: user.Id, + TeamId: team.Id, + Trigger: trigger, + Method: method, + Username: responseUsername, + IconURL: icon, + AutoComplete: autocomplete, + AutoCompleteDesc: autocompleteDesc, + AutoCompleteHint: autocompleteHint, + DisplayName: title, + Description: description, + URL: url, + } + + if _, err := a.CreateCommand(newCommand); err != nil { + return errors.New("unable to create command '" + newCommand.Trigger + "'. " + err.Error()) + } + CommandPrettyPrintln("created command '" + newCommand.Trigger + "'") + + return nil +} + func moveCommandCmdF(command *cobra.Command, args []string) error { a, err := InitDBCommandContextCobra(command) if err != nil { |