Skip to content

Commit

Permalink
mmctl: add listing oauth apps (mattermost#25497)
Browse files Browse the repository at this point in the history
* mmctl: add listing oauth apps

* Fix tests

* Update mmctl docs

* Fix tests

* Update server/cmd/mmctl/commands/oauth.go

Co-authored-by: Ibrahim Serdar Acikgoz <[email protected]>

* Update server/cmd/mmctl/commands/oauth.go

Co-authored-by: Ibrahim Serdar Acikgoz <[email protected]>

* Add paging

* Update docs

---------

Co-authored-by: Ibrahim Serdar Acikgoz <[email protected]>
  • Loading branch information
jwilander and isacikgoz authored Dec 5, 2023
1 parent df6c8c5 commit e0a36cf
Show file tree
Hide file tree
Showing 7 changed files with 333 additions and 0 deletions.
1 change: 1 addition & 0 deletions server/cmd/mmctl/client/client.go
Original file line number Diff line number Diff line change
Expand Up @@ -149,4 +149,5 @@ type Client interface {
GeneratePresignedURL(ctx context.Context, name string) (*model.PresignURLResponse, *model.Response, error)
ResetSamlAuthDataToEmail(ctx context.Context, includeDeleted bool, dryRun bool, userIDs []string) (int64, *model.Response, error)
GenerateSupportPacket(ctx context.Context) ([]byte, *model.Response, error)
GetOAuthApps(ctx context.Context, page, perPage int) ([]*model.OAuthApp, *model.Response, error)
}
83 changes: 83 additions & 0 deletions server/cmd/mmctl/commands/oauth.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,83 @@
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package commands

import (
"context"
"fmt"

"github.com/mattermost/mattermost/server/public/model"

"github.com/mattermost/mattermost/server/v8/cmd/mmctl/client"
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"

"github.com/pkg/errors"
"github.com/spf13/cobra"
)

var OAuthCmd = &cobra.Command{
Use: "oauth",
Short: "Management of OAuth2 apps",
}

var ListOAuthAppsCmd = &cobra.Command{
Use: "list",
Short: "List OAuth2 apps",
Long: "list all OAuth2 apps",
Example: " oauth list",
RunE: withClient(listOAuthAppsCmdF),
Args: cobra.NoArgs,
}

func listOAuthAppsCmdF(c client.Client, command *cobra.Command, args []string) error {
page, err := command.Flags().GetInt("page")
if err != nil {
return err
}
perPage, err := command.Flags().GetInt("per-page")
if err != nil {
return err
}

apps, _, err := c.GetOAuthApps(context.Background(), page, perPage)
if err != nil {
return errors.Wrap(err, "Failed to fetch oauth2 apps")
}

userIds := make([]string, len(apps))
for i := range apps {
userIds[i] = apps[i].CreatorId
}

users, _, err := c.GetUsersByIds(context.Background(), userIds)
if err != nil {
return errors.Wrap(err, "Failed to fetch users for oauth2 apps")
}

usersByID := map[string]*model.User{}
for _, user := range users {
usersByID[user.Id] = user
}

for _, app := range apps {
ownerName := app.CreatorId
if owner, ok := usersByID[app.CreatorId]; ok {
ownerName = owner.Username
}
printer.PrintT(fmt.Sprintf("{{.Id}}: {{.Name}} (Created by %s)", ownerName), app)
}

return nil
}

func init() {
ListOAuthAppsCmd.Flags().Int("page", 0, "Page number to fetch for the list of OAuth2 apps")
ListOAuthAppsCmd.Flags().Int("per-page", 200, "Number of OAuth2 apps to be fetched")

OAuthCmd.AddCommand(
ListOAuthAppsCmd,
)

RootCmd.AddCommand(OAuthCmd)
}
138 changes: 138 additions & 0 deletions server/cmd/mmctl/commands/oauth_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,138 @@
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package commands

import (
"context"
"strconv"

"github.com/mattermost/mattermost/server/public/model"
"github.com/pkg/errors"

"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"

"github.com/spf13/cobra"
)

func (s *MmctlUnitTestSuite) TestListOAuthAppsCmd() {
oauthAppID := "oauthAppID"
oauthAppName := "oauthAppName"
userID := "userID"

cmd := &cobra.Command{}
cmd.Flags().Int("page", 0, "")
cmd.Flags().Int("per-page", 200, "")

s.Run("Listing oauth apps", func() {
printer.Clean()

mockOAuthApp := model.OAuthApp{
Id: oauthAppID,
Name: oauthAppName,
CreatorId: userID,
}
mockUser := model.User{Id: mockOAuthApp.CreatorId, Username: "mockuser"}

s.client.
EXPECT().
GetOAuthApps(context.Background(), 0, 200).
Return([]*model.OAuthApp{&mockOAuthApp}, &model.Response{}, nil).
Times(1)

s.client.
EXPECT().
GetUsersByIds(context.Background(), []string{mockOAuthApp.CreatorId}).
Return([]*model.User{&mockUser}, &model.Response{}, nil).
Times(1)

err := listOAuthAppsCmdF(s.client, cmd, []string{})
s.Require().Nil(err)
s.Len(printer.GetLines(), 1)
s.Len(printer.GetErrorLines(), 0)
s.Require().Equal(&mockOAuthApp, printer.GetLines()[0])
})

s.Run("Listing oauth apps with paging", func() {
printer.Clean()

mockOAuthApp := model.OAuthApp{
Id: oauthAppID,
Name: oauthAppName,
CreatorId: userID,
}
mockUser := model.User{Id: mockOAuthApp.CreatorId, Username: "mockuser"}

pageCmd := &cobra.Command{}
pageCmd.Flags().Int("page", 0, "")
pageCmd.Flags().Int("per-page", 200, "")

page := 1
perPage := 2
_ = pageCmd.Flags().Set("page", strconv.Itoa(page))
_ = pageCmd.Flags().Set("per-page", strconv.Itoa(perPage))

s.client.
EXPECT().
GetOAuthApps(context.Background(), 1, 2).
Return([]*model.OAuthApp{&mockOAuthApp}, &model.Response{}, nil).
Times(1)

s.client.
EXPECT().
GetUsersByIds(context.Background(), []string{mockOAuthApp.CreatorId}).
Return([]*model.User{&mockUser}, &model.Response{}, nil).
Times(1)

err := listOAuthAppsCmdF(s.client, pageCmd, []string{})
s.Require().Nil(err)
s.Len(printer.GetLines(), 1)
s.Len(printer.GetErrorLines(), 0)
s.Require().Equal(&mockOAuthApp, printer.GetLines()[0])
})

s.Run("Unable to list oauth apps", func() {
printer.Clean()

mockError := errors.New("mock error")

s.client.
EXPECT().
GetOAuthApps(context.Background(), 0, 200).
Return(nil, &model.Response{}, mockError).
Times(1)

err := listOAuthAppsCmdF(s.client, cmd, []string{})
s.Require().NotNil(err)
s.Len(printer.GetLines(), 0)
s.EqualError(err, "Failed to fetch oauth2 apps: mock error")
})

s.Run("Unable to get users for oauth apps", func() {
printer.Clean()

mockOAuthApp := model.OAuthApp{
Id: oauthAppID,
Name: oauthAppName,
CreatorId: userID,
}
mockError := errors.New("mock error")

s.client.
EXPECT().
GetOAuthApps(context.Background(), 0, 200).
Return([]*model.OAuthApp{&mockOAuthApp}, &model.Response{}, nil).
Times(1)

s.client.
EXPECT().
GetUsersByIds(context.Background(), []string{mockOAuthApp.CreatorId}).
Return(nil, &model.Response{}, mockError).
Times(1)

err := listOAuthAppsCmdF(s.client, cmd, []string{})
s.Require().NotNil(err)
s.Len(printer.GetLines(), 0)
s.EqualError(err, "Failed to fetch users for oauth2 apps: mock error")
})
}
1 change: 1 addition & 0 deletions server/cmd/mmctl/docs/mmctl.rst
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ SEE ALSO
* `mmctl ldap <mmctl_ldap.rst>`_ - LDAP related utilities
* `mmctl license <mmctl_license.rst>`_ - Licensing commands
* `mmctl logs <mmctl_logs.rst>`_ - Display logs in a human-readable format
* `mmctl oauth <mmctl_oauth.rst>`_ - Management of OAuth2 apps
* `mmctl permissions <mmctl_permissions.rst>`_ - Management of permissions
* `mmctl plugin <mmctl_plugin.rst>`_ - Management of plugins
* `mmctl post <mmctl_post.rst>`_ - Management of posts
Expand Down
41 changes: 41 additions & 0 deletions server/cmd/mmctl/docs/mmctl_oauth.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
.. _mmctl_oauth:

mmctl oauth
-----------

Management of OAuth2 apps

Synopsis
~~~~~~~~


Management of OAuth2 apps

Options
~~~~~~~

::

-h, --help help for oauth

Options inherited from parent commands
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

::

--config string path to the configuration file (default "$XDG_CONFIG_HOME/mmctl/config")
--disable-pager disables paged output
--insecure-sha1-intermediate allows to use insecure TLS protocols, such as SHA-1
--insecure-tls-version allows to use TLS versions 1.0 and 1.1
--json the output format will be in json format
--local allows communicating with the server through a unix socket
--quiet prevent mmctl to generate output for the commands
--strict will only run commands if the mmctl version matches the server one
--suppress-warnings disables printing warning messages

SEE ALSO
~~~~~~~~

* `mmctl <mmctl.rst>`_ - Remote client for the Open Source, self-hosted Slack-alternative
* `mmctl oauth list <mmctl_oauth_list.rst>`_ - List OAuth2 apps

53 changes: 53 additions & 0 deletions server/cmd/mmctl/docs/mmctl_oauth_list.rst
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
.. _mmctl_oauth_list:

mmctl oauth list
----------------

List OAuth2 apps

Synopsis
~~~~~~~~


list all OAuth2 apps

::

mmctl oauth list [flags]

Examples
~~~~~~~~

::

oauth list

Options
~~~~~~~

::

-h, --help help for list
--page int Page number to fetch for the list of OAuth2 apps
--per-page int Number of OAuth2 apps to be fetched (default 200)

Options inherited from parent commands
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

::

--config string path to the configuration file (default "$XDG_CONFIG_HOME/mmctl/config")
--disable-pager disables paged output
--insecure-sha1-intermediate allows to use insecure TLS protocols, such as SHA-1
--insecure-tls-version allows to use TLS versions 1.0 and 1.1
--json the output format will be in json format
--local allows communicating with the server through a unix socket
--quiet prevent mmctl to generate output for the commands
--strict will only run commands if the mmctl version matches the server one
--suppress-warnings disables printing warning messages

SEE ALSO
~~~~~~~~

* `mmctl oauth <mmctl_oauth.rst>`_ - Management of OAuth2 apps

16 changes: 16 additions & 0 deletions server/cmd/mmctl/mocks/client_mock.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

0 comments on commit e0a36cf

Please sign in to comment.