[CODE-621]: Move registration func out of controller struct

This commit is contained in:
Akhilesh Pandey 2023-07-26 09:38:59 +05:30
parent f18f142401
commit 71eab8e042

View File

@ -11,25 +11,23 @@ import (
"github.com/harness/gitness/types" "github.com/harness/gitness/types"
) )
type registrationCheck func(ctx context.Context) (bool, error)
type Controller struct { type Controller struct {
principalStore store.PrincipalStore principalStore store.PrincipalStore
config *types.Config config *types.Config
IsUserRegistrationAllowed registrationCheck
} }
func NewController(principalStore store.PrincipalStore, config *types.Config) *Controller { func NewController(principalStore store.PrincipalStore, config *types.Config) *Controller {
return &Controller{ return &Controller{
principalStore: principalStore, principalStore: principalStore,
config: config, config: config,
IsUserRegistrationAllowed: func(ctx context.Context) (bool, error) {
usrCount, err := principalStore.CountUsers(ctx)
if err != nil {
return false, err
}
return usrCount == 0 || config.AllowSignUp, nil
},
} }
} }
func (c *Controller) IsUserRegistrationAllowed(ctx context.Context) (bool, error) {
usrCount, err := c.principalStore.CountUsers(ctx)
if err != nil {
return false, err
}
return usrCount == 0 || c.config.AllowSignUp, nil
}