summaryrefslogtreecommitdiff
path: root/internal/gitlabnet/twofactorverify/client.go
diff options
context:
space:
mode:
authorIgor Drozdov <idrozdov@gitlab.com>2020-12-10 14:40:31 +0000
committerIgor Drozdov <idrozdov@gitlab.com>2020-12-10 14:40:31 +0000
commit2fb14fb46bf4cffe8aff4a31794ae9d3588aac78 (patch)
treec4d4057fcdc5753086493c71b10f9ea5ee7b84d5 /internal/gitlabnet/twofactorverify/client.go
parent384f3036e3d9c501e29a7ce24ece1e887a14d53a (diff)
parent1293a33014c9cfc82b0bc1b9525987476b2aa857 (diff)
downloadgitlab-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.go90
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
+}