TheArchive/zitadel
1
0
Fork 0
mirror of https://github.com/zitadel/zitadel.git synced 2025-03-04 09:55:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: rename OTP to TOTP in v2 alpha user api

Browse Source 
This change renames the v2 user OTP registration endpoints and objects
to TOTP.
Also the v2 related code paths have been renamed to TOTP.

This change was discussed during the sprint review.
This commit is contained in:
Tim Möhlmann 2023-06-22 12:06:32 +02:00
parent df87907299
commit 56e33ce1a7
10 changed files with 89 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
internal/api/grpc/user/v2/otp.go
View File

@ -1,38 +0,0 @@
package user
import (
"context"
"github.com/zitadel/zitadel/internal/api/authz"
"github.com/zitadel/zitadel/internal/api/grpc/object/v2"
"github.com/zitadel/zitadel/internal/domain"
user "github.com/zitadel/zitadel/pkg/grpc/user/v2alpha"
)
func (s *Server) RegisterOTP(ctx context.Context, req *user.RegisterOTPRequest) (*user.RegisterOTPResponse, error) {
return otpDetailsToPb(
s.command.AddUserOTP(ctx, req.GetUserId(), authz.GetCtxData(ctx).ResourceOwner),
)
}
func otpDetailsToPb(otp *domain.OTPv2, err error) (*user.RegisterOTPResponse, error) {
if err != nil {
return nil, err
}
return &user.RegisterOTPResponse{
Details: object.DomainToDetailsPb(otp.ObjectDetails),
Uri: otp.URI,
Secret: otp.Secret,
}, nil
}
func (s *Server) VerifyOTPRegistration(ctx context.Context, req *user.VerifyOTPRegistrationRequest) (*user.VerifyOTPRegistrationResponse, error) {
objectDetails, err := s.command.CheckUserOTP(ctx, req.GetUserId(), req.GetCode(), authz.GetCtxData(ctx).ResourceOwner)
if err != nil {
return nil, err
}
return &user.VerifyOTPRegistrationResponse{
Details: object.DomainToDetailsPb(objectDetails),
}, nil
}

38
internal/api/grpc/user/v2/totp.go Normal file
View File

@ -0,0 +1,38 @@
package user
import (
"context"
"github.com/zitadel/zitadel/internal/api/authz"
"github.com/zitadel/zitadel/internal/api/grpc/object/v2"
"github.com/zitadel/zitadel/internal/domain"
user "github.com/zitadel/zitadel/pkg/grpc/user/v2alpha"
)
func (s *Server) RegisterTOTP(ctx context.Context, req *user.RegisterTOTPRequest) (*user.RegisterTOTPResponse, error) {
return totpDetailsToPb(
s.command.AddUserTOTP(ctx, req.GetUserId(), authz.GetCtxData(ctx).ResourceOwner),
)
}
func totpDetailsToPb(totp *domain.TOTP, err error) (*user.RegisterTOTPResponse, error) {
if err != nil {
return nil, err
}
return &user.RegisterTOTPResponse{
Details: object.DomainToDetailsPb(totp.ObjectDetails),
Uri: totp.URI,
Secret: totp.Secret,
}, nil
}
func (s *Server) VerifyTOTPRegistration(ctx context.Context, req *user.VerifyTOTPRegistrationRequest) (*user.VerifyTOTPRegistrationResponse, error) {
objectDetails, err := s.command.CheckUserTOTP(ctx, req.GetUserId(), req.GetCode(), authz.GetCtxData(ctx).ResourceOwner)
if err != nil {
return nil, err
}
return &user.VerifyTOTPRegistrationResponse{
Details: object.DomainToDetailsPb(objectDetails),
}, nil
}

34
internal/api/grpc/user/v2/otp_integration_test.go → internal/api/grpc/user/v2/totp_integration_test.go
View File

@ -13,24 +13,24 @@ import (
user "github.com/zitadel/zitadel/pkg/grpc/user/v2alpha" user "github.com/zitadel/zitadel/pkg/grpc/user/v2alpha"
) )
func TestServer_RegisterOTP(t *testing.T) { func TestServer_RegisterTOTP(t *testing.T) {
// userID := Tester.CreateHumanUser(CTX).GetUserId() // userID := Tester.CreateHumanUser(CTX).GetUserId()
type args struct { type args struct {
ctx context.Context ctx context.Context
req *user.RegisterOTPRequest req *user.RegisterTOTPRequest
} }
tests := []struct { tests := []struct {
name string name string
args args args args
want *user.RegisterOTPResponse want *user.RegisterTOTPResponse
wantErr bool wantErr bool
}{ }{
{ {
name: "missing user id", name: "missing user id",
args: args{ args: args{
ctx: CTX, ctx: CTX,
req: &user.RegisterOTPRequest{}, req: &user.RegisterTOTPRequest{},
}, },
wantErr: true, wantErr: true,
}, },
@ -38,7 +38,7 @@ func TestServer_RegisterOTP(t *testing.T) {
name: "user mismatch", name: "user mismatch",
args: args{ args: args{
ctx: CTX, ctx: CTX,
req: &user.RegisterOTPRequest{ req: &user.RegisterTOTPRequest{
UserId: "wrong", UserId: "wrong",
}, },
}, },
@ -50,11 +50,11 @@ func TestServer_RegisterOTP(t *testing.T) {
name: "human user", name: "human user",
args: args{ args: args{
ctx: CTX, ctx: CTX,
req: &user.RegisterOTPRequest{ req: &user.RegisterTOTPRequest{
UserId: userID, UserId: userID,
}, },
}, },
want: &user.RegisterOTPResponse{ want: &user.RegisterTOTPResponse{
Details: &object.Details{ Details: &object.Details{
ResourceOwner: Tester.Organisation.ID, ResourceOwner: Tester.Organisation.ID,
}, },
@ -64,7 +64,7 @@ func TestServer_RegisterOTP(t *testing.T) {
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
got, err := Client.RegisterOTP(tt.args.ctx, tt.args.req) got, err := Client.RegisterTOTP(tt.args.ctx, tt.args.req)
if tt.wantErr { if tt.wantErr {
require.Error(t, err) require.Error(t, err)
return return
@ -78,11 +78,11 @@ func TestServer_RegisterOTP(t *testing.T) {
} }
} }
func TestServer_VerifyOTPRegistration(t *testing.T) { func TestServer_VerifyTOTPRegistration(t *testing.T) {
userID := Tester.CreateHumanUser(CTX).GetUserId() userID := Tester.CreateHumanUser(CTX).GetUserId()
/* TODO: after we are able to obtain a Bearer token for a human user /* TODO: after we are able to obtain a Bearer token for a human user
reg, err := Client.RegisterOTP(CTX, &user.RegisterOTPRequest{ reg, err := Client.RegisterTOTP(CTX, &user.RegisterTOTPRequest{
UserId: userID, UserId: userID,
}) })
require.NoError(t, err) require.NoError(t, err)
@ -92,19 +92,19 @@ func TestServer_VerifyOTPRegistration(t *testing.T) {
type args struct { type args struct {
ctx context.Context ctx context.Context
req *user.VerifyOTPRegistrationRequest req *user.VerifyTOTPRegistrationRequest
} }
tests := []struct { tests := []struct {
name string name string
args args args args
want *user.VerifyOTPRegistrationResponse want *user.VerifyTOTPRegistrationResponse
wantErr bool wantErr bool
}{ }{
{ {
name: "user mismatch", name: "user mismatch",
args: args{ args: args{
ctx: CTX, ctx: CTX,
req: &user.VerifyOTPRegistrationRequest{ req: &user.VerifyTOTPRegistrationRequest{
UserId: "wrong", UserId: "wrong",
}, },
}, },
@ -114,7 +114,7 @@ func TestServer_VerifyOTPRegistration(t *testing.T) {
name: "wrong code", name: "wrong code",
args: args{ args: args{
ctx: CTX, ctx: CTX,
req: &user.VerifyOTPRegistrationRequest{ req: &user.VerifyTOTPRegistrationRequest{
UserId: userID, UserId: userID,
Code: "123", Code: "123",
}, },
@ -127,12 +127,12 @@ func TestServer_VerifyOTPRegistration(t *testing.T) {
name: "success", name: "success",
args: args{ args: args{
ctx: CTX, ctx: CTX,
req: &user.VerifyOTPRegistrationRequest{ req: &user.VerifyTOTPRegistrationRequest{
UserId: userID, UserId: userID,
Code: code, Code: code,
}, },
}, },
want: &user.VerifyOTPRegistrationResponse{ want: &user.VerifyTOTPRegistrationResponse{
Details: &object.Details{ Details: &object.Details{
ResourceOwner: Tester.Organisation.ResourceOwner, ResourceOwner: Tester.Organisation.ResourceOwner,
}, },
@ -142,7 +142,7 @@ func TestServer_VerifyOTPRegistration(t *testing.T) {
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
got, err := Client.VerifyOTPRegistration(tt.args.ctx, tt.args.req) got, err := Client.VerifyTOTPRegistration(tt.args.ctx, tt.args.req)
if tt.wantErr { if tt.wantErr {
require.Error(t, err) require.Error(t, err)
return return

14
internal/api/grpc/user/v2/otp_test.go → internal/api/grpc/user/v2/totp_test.go
View File

@ -14,15 +14,15 @@ import (
user "github.com/zitadel/zitadel/pkg/grpc/user/v2alpha" user "github.com/zitadel/zitadel/pkg/grpc/user/v2alpha"
) )
func Test_otpDetailsToPb(t *testing.T) { func Test_totpDetailsToPb(t *testing.T) {
type args struct { type args struct {
otp *domain.OTPv2 otp *domain.TOTP
err error err error
} }
tests := []struct { tests := []struct {
name string name string
args args args args
want *user.RegisterOTPResponse want *user.RegisterTOTPResponse
wantErr error wantErr error
}{ }{
{ {
@ -35,7 +35,7 @@ func Test_otpDetailsToPb(t *testing.T) {
{ {
name: "success", name: "success",
args: args{ args: args{
otp: &domain.OTPv2{ otp: &domain.TOTP{
ObjectDetails: &domain.ObjectDetails{ ObjectDetails: &domain.ObjectDetails{
Sequence: 123, Sequence: 123,
EventDate: time.Unix(456, 789), EventDate: time.Unix(456, 789),
@ -45,7 +45,7 @@ func Test_otpDetailsToPb(t *testing.T) {
URI: "URI", URI: "URI",
}, },
}, },
want: &user.RegisterOTPResponse{ want: &user.RegisterTOTPResponse{
Details: &object.Details{ Details: &object.Details{
Sequence: 123, Sequence: 123,
ChangeDate: &timestamppb.Timestamp{ ChangeDate: &timestamppb.Timestamp{
@ -61,10 +61,10 @@ func Test_otpDetailsToPb(t *testing.T) {
} }
for _, tt := range tests { for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) { t.Run(tt.name, func(t *testing.T) {
got, err := otpDetailsToPb(tt.args.otp, tt.args.err) got, err := totpDetailsToPb(tt.args.otp, tt.args.err)
require.ErrorIs(t, err, tt.wantErr) require.ErrorIs(t, err, tt.wantErr)
if !proto.Equal(tt.want, got) { if !proto.Equal(tt.want, got) {
t.Errorf("RegisterOTPResponse =\n%v\nwant\n%v", got, tt.want) t.Errorf("RegisterTOTPResponse =\n%v\nwant\n%v", got, tt.want)
} }
}) })
} }

8
internal/command/user_human_otp.go
View File

@ -45,7 +45,7 @@ func (c *Commands) AddHumanOTP(ctx context.Context, userID, resourceowner string
if userID == "" { if userID == "" {
return nil, caos_errs.ThrowInvalidArgument(nil, "COMMAND-5M0sd", "Errors.User.UserIDMissing") return nil, caos_errs.ThrowInvalidArgument(nil, "COMMAND-5M0sd", "Errors.User.UserIDMissing")
} }
prep, err := c.createHumanOTP(ctx, userID, resourceowner) prep, err := c.createHumanTOTP(ctx, userID, resourceowner)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -62,14 +62,14 @@ func (c *Commands) AddHumanOTP(ctx context.Context, userID, resourceowner string
}, nil }, nil
} }
type preparedOTP struct { type preparedTOTP struct {
wm *HumanOTPWriteModel wm *HumanOTPWriteModel
userAgg *eventstore.Aggregate userAgg *eventstore.Aggregate
key *otp.Key key *otp.Key
cmds []eventstore.Command cmds []eventstore.Command
} }
func (c *Commands) createHumanOTP(ctx context.Context, userID, resourceOwner string) (*preparedOTP, error) { func (c *Commands) createHumanTOTP(ctx context.Context, userID, resourceOwner string) (*preparedTOTP, error) {
human, err := c.getHuman(ctx, userID, resourceOwner) human, err := c.getHuman(ctx, userID, resourceOwner)
if err != nil { if err != nil {
logging.Log("COMMAND-DAqe1").WithError(err).WithField("traceID", tracing.TraceIDFromCtx(ctx)).Debug("unable to get human for loginname") logging.Log("COMMAND-DAqe1").WithError(err).WithField("traceID", tracing.TraceIDFromCtx(ctx)).Debug("unable to get human for loginname")
@ -107,7 +107,7 @@ func (c *Commands) createHumanOTP(ctx context.Context, userID, resourceOwner str
if err != nil { if err != nil {
return nil, err return nil, err
} }
return &preparedOTP{ return &preparedTOTP{
wm: otpWriteModel, wm: otpWriteModel,
userAgg: userAgg, userAgg: userAgg,
key: key, key: key,

2
internal/command/user_human_otp_test.go
View File

@ -512,7 +512,7 @@ func TestCommands_createHumanOTP(t *testing.T) {
}, },
}, },
} }
got, err := c.createHumanOTP(tt.args.ctx, tt.args.userID, tt.args.resourceOwner) got, err := c.createHumanTOTP(tt.args.ctx, tt.args.userID, tt.args.resourceOwner)
require.ErrorIs(t, err, tt.wantErr) require.ErrorIs(t, err, tt.wantErr)
if tt.want { if tt.want {
require.NotNil(t, got) require.NotNil(t, got)

8
internal/command/user_v2_otp.go → internal/command/user_v2_totp.go
View File

@ -7,25 +7,25 @@ import (
"github.com/zitadel/zitadel/internal/domain" "github.com/zitadel/zitadel/internal/domain"
) )
func (c *Commands) AddUserOTP(ctx context.Context, userID, resourceowner string) (*domain.OTPv2, error) { func (c *Commands) AddUserTOTP(ctx context.Context, userID, resourceowner string) (*domain.TOTP, error) {
if err := authz.UserIDInCTX(ctx, userID); err != nil { if err := authz.UserIDInCTX(ctx, userID); err != nil {
return nil, err return nil, err
} }
prep, err := c.createHumanOTP(ctx, userID, resourceowner) prep, err := c.createHumanTOTP(ctx, userID, resourceowner)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if err = c.pushAppendAndReduce(ctx, prep.wm, prep.cmds...); err != nil { if err = c.pushAppendAndReduce(ctx, prep.wm, prep.cmds...); err != nil {
return nil, err return nil, err
} }
return &domain.OTPv2{ return &domain.TOTP{
ObjectDetails: writeModelToObjectDetails(&prep.wm.WriteModel), ObjectDetails: writeModelToObjectDetails(&prep.wm.WriteModel),
Secret: prep.key.Secret(), Secret: prep.key.Secret(),
URI: prep.key.URL(), URI: prep.key.URL(),
}, nil }, nil
} }
func (c *Commands) CheckUserOTP(ctx context.Context, userID, code, resourceOwner string) (*domain.ObjectDetails, error) { func (c *Commands) CheckUserTOTP(ctx context.Context, userID, code, resourceOwner string) (*domain.ObjectDetails, error) {
if err := authz.UserIDInCTX(ctx, userID); err != nil { if err := authz.UserIDInCTX(ctx, userID); err != nil {
return nil, err return nil, err
} }

8
internal/command/user_v2_otp_test.go → internal/command/user_v2_totp_test.go
View File

@ -21,7 +21,7 @@ import (
"github.com/zitadel/zitadel/internal/repository/user" "github.com/zitadel/zitadel/internal/repository/user"
) )
func TestCommands_AddUserOTP(t *testing.T) { func TestCommands_AddUserTOTP(t *testing.T) {
ctx := authz.NewMockContext("inst1", "org1", "user1") ctx := authz.NewMockContext("inst1", "org1", "user1")
userAgg := &user.NewAggregate("user1", "org1").Aggregate userAgg := &user.NewAggregate("user1", "org1").Aggregate
@ -175,7 +175,7 @@ func TestCommands_AddUserOTP(t *testing.T) {
}, },
}, },
} }
got, err := c.AddUserOTP(ctx, tt.args.userID, tt.args.resourceowner) got, err := c.AddUserTOTP(ctx, tt.args.userID, tt.args.resourceowner)
require.ErrorIs(t, err, tt.wantErr) require.ErrorIs(t, err, tt.wantErr)
if tt.want { if tt.want {
require.NotNil(t, got) require.NotNil(t, got)
@ -187,7 +187,7 @@ func TestCommands_AddUserOTP(t *testing.T) {
} }
} }
func TestCommands_CheckUserOTP(t *testing.T) { func TestCommands_CheckUserTOTP(t *testing.T) {
ctx := authz.NewMockContext("inst1", "org1", "user1") ctx := authz.NewMockContext("inst1", "org1", "user1")
cryptoAlg := crypto.CreateMockEncryptionAlg(gomock.NewController(t)) cryptoAlg := crypto.CreateMockEncryptionAlg(gomock.NewController(t))
@ -252,7 +252,7 @@ func TestCommands_CheckUserOTP(t *testing.T) {
}, },
}, },
} }
got, err := c.CheckUserOTP(ctx, tt.args.userID, tt.args.code, tt.args.resourceOwner) got, err := c.CheckUserTOTP(ctx, tt.args.userID, tt.args.code, tt.args.resourceOwner)
require.ErrorIs(t, err, tt.wantErr) require.ErrorIs(t, err, tt.wantErr)
if tt.want { if tt.want {
require.NotNil(t, got) require.NotNil(t, got)

2
internal/domain/human_otp.go
View File

@ -17,7 +17,7 @@ type OTP struct {
State MFAState State MFAState
} }
type OTPv2 struct { type TOTP struct {
*ObjectDetails *ObjectDetails
Secret string Secret string

26
proto/zitadel/user/v2alpha/user_service.proto
View File

@ -271,9 +271,9 @@ service UserService {
}; };
} }
rpc RegisterOTP (RegisterOTPRequest) returns (RegisterOTPResponse) { rpc RegisterTOTP (RegisterTOTPRequest) returns (RegisterTOTPResponse) {
option (google.api.http) = { option (google.api.http) = {
post: "/v2alpha/users/{user_id}/otp" post: "/v2alpha/users/{user_id}/totp"
body: "*" body: "*"
}; };
@ -283,8 +283,8 @@ service UserService {
} }
}; };
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = { option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
summary: "Start the registration of an OTP generator for a user"; summary: "Start the registration of a TOTP generator for a user";
description: "Start the registration of a OTP generator for a user, as a response a secret returned, which is used to initialize a TOTP app or device." description: "Start the registration of a TOTP generator for a user, as a response a secret returned, which is used to initialize a TOTP app or device."
responses: { responses: {
key: "200" key: "200"
value: { value: {
@ -294,9 +294,9 @@ service UserService {
}; };
} }
rpc VerifyOTPRegistration (VerifyOTPRegistrationRequest) returns (VerifyOTPRegistrationResponse) { rpc VerifyTOTPRegistration (VerifyTOTPRegistrationRequest) returns (VerifyTOTPRegistrationResponse) {
option (google.api.http) = { option (google.api.http) = {
post: "/v2alpha/users/{user_id}/otp/_verify" // Why underscore here?? post: "/v2alpha/users/{user_id}/totp/_verify"
body: "*" body: "*"
}; };
@ -306,8 +306,8 @@ service UserService {
} }
}; };
option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = { option (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_operation) = {
summary: "Verify a u2f token for a user"; summary: "Verify a TOTP generator for a user";
description: "Verify the OTP registration with a generated code." description: "Verify the TOTP registration with a generated code."
responses: { responses: {
key: "200" key: "200"
value: { value: {
@ -719,7 +719,7 @@ message VerifyU2FRegistrationResponse{
zitadel.object.v2alpha.Details details = 1; zitadel.object.v2alpha.Details details = 1;
} }
message RegisterOTPRequest { message RegisterTOTPRequest {
string user_id = 1 [ string user_id = 1 [
(validate.rules).string = {min_len: 1, max_len: 200}, (validate.rules).string = {min_len: 1, max_len: 200},
(google.api.field_behavior) = REQUIRED, (google.api.field_behavior) = REQUIRED,
@ -731,7 +731,7 @@ message RegisterOTPRequest {
]; ];
} }
message RegisterOTPResponse { message RegisterTOTPResponse {
zitadel.object.v2alpha.Details details = 1; zitadel.object.v2alpha.Details details = 1;
string uri = 2 [ string uri = 2 [
(grpc.gateway.protoc_gen_openapiv2.options.openapiv2_field) = { (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_field) = {
@ -745,7 +745,7 @@ message RegisterOTPResponse {
]; ];
} }
message VerifyOTPRegistrationRequest { message VerifyTOTPRegistrationRequest {
string user_id = 1 [ string user_id = 1 [
(validate.rules).string = {min_len: 1, max_len: 200}, (validate.rules).string = {min_len: 1, max_len: 200},
(google.api.field_behavior) = REQUIRED, (google.api.field_behavior) = REQUIRED,
@ -759,13 +759,13 @@ message VerifyOTPRegistrationRequest {
(validate.rules).string = {min_len: 1, max_len: 200}, (validate.rules).string = {min_len: 1, max_len: 200},
(google.api.field_behavior) = REQUIRED, (google.api.field_behavior) = REQUIRED,
(grpc.gateway.protoc_gen_openapiv2.options.openapiv2_field) = { (grpc.gateway.protoc_gen_openapiv2.options.openapiv2_field) = {
description: "Code generated by OTP app or device" description: "Code generated by TOTP app or device"
example: "\"123456\""; example: "\"123456\"";
} }
]; ];
} }
message VerifyOTPRegistrationResponse { message VerifyTOTPRegistrationResponse {
zitadel.object.v2alpha.Details details = 1; zitadel.object.v2alpha.Details details = 1;
} }