Skip to content

Commit

Permalink
implement GET /commands (#5865)
Browse files Browse the repository at this point in the history
  • Loading branch information
cpanato authored and jwilander committed Apr 3, 2017
1 parent 67a8770 commit 88b8df3
Show file tree
Hide file tree
Showing 5 changed files with 168 additions and 2 deletions.
2 changes: 1 addition & 1 deletion api/command.go
Expand Up @@ -34,7 +34,7 @@ func InitCommand() {
}

func listCommands(c *Context, w http.ResponseWriter, r *http.Request) {
commands, err := app.ListCommands(c.TeamId, c.T)
commands, err := app.ListAutocompleteCommands(c.TeamId, c.T)
if err != nil {
c.Err = err
return
Expand Down
47 changes: 47 additions & 0 deletions api4/command.go
Expand Up @@ -5,6 +5,7 @@ package api4

import (
"net/http"
"strconv"

l4g "github.com/alecthomas/log4go"
"github.com/mattermost/platform/app"
Expand All @@ -16,6 +17,7 @@ func InitCommand() {
l4g.Debug(utils.T("api.command.init.debug"))

BaseRoutes.Commands.Handle("", ApiSessionRequired(createCommand)).Methods("POST")
BaseRoutes.Commands.Handle("", ApiSessionRequired(listCommands)).Methods("GET")
}

func createCommand(c *Context, w http.ResponseWriter, r *http.Request) {
Expand Down Expand Up @@ -44,3 +46,48 @@ func createCommand(c *Context, w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusCreated)
w.Write([]byte(rcmd.ToJson()))
}

func listCommands(c *Context, w http.ResponseWriter, r *http.Request) {
customOnly, failConv := strconv.ParseBool(r.URL.Query().Get("custom_only"))
if failConv != nil {
customOnly = false
}

teamId := r.URL.Query().Get("team_id")

if len(teamId) == 0 {
c.SetInvalidParam("team_id")
return
}

commands := []*model.Command{}
err := &model.AppError{}
if customOnly {
if !app.SessionHasPermissionToTeam(c.Session, teamId, model.PERMISSION_MANAGE_SLASH_COMMANDS) {
c.SetPermissionError(model.PERMISSION_MANAGE_SLASH_COMMANDS)
return
}
commands, err = app.ListTeamCommands(teamId)
if err != nil {
c.Err = err
return
}
} else {
//User with no permission should see only system commands
if !app.SessionHasPermissionToTeam(c.Session, teamId, model.PERMISSION_MANAGE_SLASH_COMMANDS) {
commands, err = app.ListAutocompleteCommands(teamId, c.T)
if err != nil {
c.Err = err
return
}
} else {
commands, err = app.ListAllCommands(teamId, c.T)
if err != nil {
c.Err = err
return
}
}
}

w.Write([]byte(model.CommandListToJson(commands)))
}
77 changes: 77 additions & 0 deletions api4/command_test.go
Expand Up @@ -59,3 +59,80 @@ func TestCreateCommand(t *testing.T) {
CheckNotImplementedStatus(t, resp)
CheckErrorMessage(t, resp, "api.command.disabled.app_error")
}

func TestListCommands(t *testing.T) {
th := Setup().InitBasic().InitSystemAdmin()
defer TearDown()
Client := th.Client

newCmd := &model.Command{
CreatorId: th.BasicUser.Id,
TeamId: th.BasicTeam.Id,
URL: "http://nowhere.com",
Method: model.COMMAND_METHOD_POST,
Trigger: "custom_command"}

_, resp := th.SystemAdminClient.CreateCommand(newCmd)
CheckNoError(t, resp)

t.Run("ListSystemAndCustomCommands", func(t *testing.T) {
listCommands, resp := th.SystemAdminClient.ListCommands(th.BasicTeam.Id, false)
CheckNoError(t, resp)

foundEcho := false
foundCustom := false
for _, command := range listCommands {
if command.Trigger == "echo" {
foundEcho = true
}
if command.Trigger == "custom_command" {
foundCustom = true
}
}
if !foundEcho {
t.Fatal("Couldn't find echo command")
}
if !foundCustom {
t.Fatal("Should list the custom command")
}
})

t.Run("ListCustomOnlyCommands", func(t *testing.T) {
listCommands, resp := th.SystemAdminClient.ListCommands(th.BasicTeam.Id, true)
CheckNoError(t, resp)

if len(listCommands) > 1 {
t.Fatal("Should list just one custom command")
}
if listCommands[0].Trigger != "custom_command" {
t.Fatal("Wrong custom command trigger")
}
})

t.Run("UserWithNoPermissionForCustomCommands", func(t *testing.T) {
_, resp := Client.ListCommands(th.BasicTeam.Id, true)
CheckForbiddenStatus(t, resp)
})

t.Run("RegularUserCanListOnlySystemCommands", func(t *testing.T) {
listCommands, resp := Client.ListCommands(th.BasicTeam.Id, false)
CheckNoError(t, resp)

foundEcho := false
foundCustom := false
for _, command := range listCommands {
if command.Trigger == "echo" {
foundEcho = true
}
if command.Trigger == "custom_command" {
foundCustom = true
}
}
if !foundEcho {
t.Fatal("Couldn't find echo command")
}
if foundCustom {
t.Fatal("Should not list the custom command")
}
})
}
33 changes: 32 additions & 1 deletion app/command.go
Expand Up @@ -61,7 +61,8 @@ func CreateCommandPost(post *model.Post, teamId string, response *model.CommandR
return post, nil
}

func ListCommands(teamId string, T goi18n.TranslateFunc) ([]*model.Command, *model.AppError) {
// previous ListCommands now ListAutocompleteCommands
func ListAutocompleteCommands(teamId string, T goi18n.TranslateFunc) ([]*model.Command, *model.AppError) {
commands := make([]*model.Command, 0, 32)
seen := make(map[string]bool)
for _, value := range commandProviders {
Expand Down Expand Up @@ -103,6 +104,36 @@ func ListTeamCommands(teamId string) ([]*model.Command, *model.AppError) {
}
}

func ListAllCommands(teamId string, T goi18n.TranslateFunc) ([]*model.Command, *model.AppError) {
commands := make([]*model.Command, 0, 32)
seen := make(map[string]bool)
for _, value := range commandProviders {
cpy := *value.GetCommand(T)
if cpy.AutoComplete && !seen[cpy.Id] {
cpy.Sanitize()
seen[cpy.Trigger] = true
commands = append(commands, &cpy)
}
}

if *utils.Cfg.ServiceSettings.EnableCommands {
if result := <-Srv.Store.Command().GetByTeam(teamId); result.Err != nil {
return nil, result.Err
} else {
teamCmds := result.Data.([]*model.Command)
for _, cmd := range teamCmds {
if !seen[cmd.Id] {
cmd.Sanitize()
seen[cmd.Trigger] = true
commands = append(commands, cmd)
}
}
}
}

return commands, nil
}

func ExecuteCommand(args *model.CommandArgs) (*model.CommandResponse, *model.AppError) {
parts := strings.Split(args.Command, " ")
trigger := parts[0][1:]
Expand Down
11 changes: 11 additions & 0 deletions model/client4.go
Expand Up @@ -2019,6 +2019,17 @@ func (c *Client4) CreateCommand(cmd *Command) (*Command, *Response) {
}
}

// ListCommands will retrieve a list of commands available in the team.
func (c *Client4) ListCommands(teamId string, customOnly bool) ([]*Command, *Response) {
query := fmt.Sprintf("?team_id=%v&custom_only=%v", teamId, customOnly)
if r, err := c.DoApiGet(c.GetCommandsRoute()+query, ""); err != nil {
return nil, &Response{StatusCode: r.StatusCode, Error: err}
} else {
defer closeBody(r)
return CommandListFromJson(r.Body), BuildResponse(r)
}
}

// Status Section

// GetUserStatus returns a user based on the provided user id string.
Expand Down

0 comments on commit 88b8df3

Please sign in to comment.