From 1962b49971cf1602ed5ee20b33193e10022cf8cb Mon Sep 17 00:00:00 2001 From: Patrick Bajao Date: Thu, 8 Aug 2019 15:29:33 +0800 Subject: Implement AuthorizedPrincipals command Build this command when `Executable` name is `gitlab-shell-authorized-principals-check`. Feature flag is the same name. --- go/internal/command/command.go | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'go/internal/command/command.go') diff --git a/go/internal/command/command.go b/go/internal/command/command.go index d6b96f1..071cd1d 100644 --- a/go/internal/command/command.go +++ b/go/internal/command/command.go @@ -2,6 +2,7 @@ package command import ( "gitlab.com/gitlab-org/gitlab-shell/go/internal/command/authorizedkeys" + "gitlab.com/gitlab-org/gitlab-shell/go/internal/command/authorizedprincipals" "gitlab.com/gitlab-org/gitlab-shell/go/internal/command/commandargs" "gitlab.com/gitlab-org/gitlab-shell/go/internal/command/discover" "gitlab.com/gitlab-org/gitlab-shell/go/internal/command/fallback" @@ -38,6 +39,8 @@ func buildCommand(e *executable.Executable, args commandargs.CommandArgs, config return buildShellCommand(args.(*commandargs.Shell), config, readWriter) case executable.AuthorizedKeysCheck: return buildAuthorizedKeysCommand(args.(*commandargs.AuthorizedKeys), config, readWriter) + case executable.AuthorizedPrincipalsCheck: + return buildAuthorizedPrincipalsCommand(args.(*commandargs.AuthorizedPrincipals), config, readWriter) } return nil @@ -73,3 +76,11 @@ func buildAuthorizedKeysCommand(args *commandargs.AuthorizedKeys, config *config return &authorizedkeys.Command{Config: config, Args: args, ReadWriter: readWriter} } + +func buildAuthorizedPrincipalsCommand(args *commandargs.AuthorizedPrincipals, config *config.Config, readWriter *readwriter.ReadWriter) Command { + if !config.FeatureEnabled(executable.AuthorizedPrincipalsCheck) { + return nil + } + + return &authorizedprincipals.Command{Config: config, Args: args, ReadWriter: readWriter} +} -- cgit v1.2.1