diff options
author | Igor Drozdov <idrozdov@gitlab.com> | 2020-12-10 14:40:31 +0000 |
---|---|---|
committer | Igor Drozdov <idrozdov@gitlab.com> | 2020-12-10 14:40:31 +0000 |
commit | 2fb14fb46bf4cffe8aff4a31794ae9d3588aac78 (patch) | |
tree | c4d4057fcdc5753086493c71b10f9ea5ee7b84d5 /internal/gitlabnet/twofactorverify/client.go | |
parent | 384f3036e3d9c501e29a7ce24ece1e887a14d53a (diff) | |
parent | 1293a33014c9cfc82b0bc1b9525987476b2aa857 (diff) | |
download | gitlab-shell-2fb14fb46bf4cffe8aff4a31794ae9d3588aac78.tar.gz |
Merge branch 'if-270554-2fa_verify_command' into 'main'
Add 2fa_verify command
See merge request gitlab-org/gitlab-shell!440
Diffstat (limited to 'internal/gitlabnet/twofactorverify/client.go')
-rw-r--r-- | internal/gitlabnet/twofactorverify/client.go | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/internal/gitlabnet/twofactorverify/client.go b/internal/gitlabnet/twofactorverify/client.go new file mode 100644 index 0000000..aab302b --- /dev/null +++ b/internal/gitlabnet/twofactorverify/client.go @@ -0,0 +1,90 @@ +package twofactorverify + +import ( + "context" + "errors" + "fmt" + "net/http" + + "gitlab.com/gitlab-org/gitlab-shell/client" + "gitlab.com/gitlab-org/gitlab-shell/internal/command/commandargs" + "gitlab.com/gitlab-org/gitlab-shell/internal/config" + "gitlab.com/gitlab-org/gitlab-shell/internal/gitlabnet" + "gitlab.com/gitlab-org/gitlab-shell/internal/gitlabnet/discover" +) + +type Client struct { + config *config.Config + client *client.GitlabNetClient +} + +type Response struct { + Success bool `json:"success"` + Message string `json:"message"` +} + +type RequestBody struct { + KeyId string `json:"key_id,omitempty"` + UserId int64 `json:"user_id,omitempty"` + OTPAttempt string `json:"otp_attempt"` +} + +func NewClient(config *config.Config) (*Client, error) { + client, err := gitlabnet.GetClient(config) + if err != nil { + return nil, fmt.Errorf("Error creating http client: %v", err) + } + + return &Client{config: config, client: client}, nil +} + +func (c *Client) VerifyOTP(ctx context.Context, args *commandargs.Shell, otp string) error { + requestBody, err := c.getRequestBody(ctx, args, otp) + if err != nil { + return err + } + + response, err := c.client.Post(ctx, "/two_factor_otp_check", requestBody) + if err != nil { + return err + } + defer response.Body.Close() + + return parse(response) +} + +func parse(hr *http.Response) error { + response := &Response{} + if err := gitlabnet.ParseJSON(hr, response); err != nil { + return err + } + + if !response.Success { + return errors.New(response.Message) + } + + return nil +} + +func (c *Client) getRequestBody(ctx context.Context, args *commandargs.Shell, otp string) (*RequestBody, error) { + client, err := discover.NewClient(c.config) + + if err != nil { + return nil, err + } + + var requestBody *RequestBody + if args.GitlabKeyId != "" { + requestBody = &RequestBody{KeyId: args.GitlabKeyId, OTPAttempt: otp} + } else { + userInfo, err := client.GetByCommandArgs(ctx, args) + + if err != nil { + return nil, err + } + + requestBody = &RequestBody{UserId: userInfo.UserId, OTPAttempt: otp} + } + + return requestBody, nil +} |