github/drone
5
0
Fork 0
mirror of https://github.com/harness/drone.git synced 2025-05-08 13:51:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Akhilesh Pandey 2023-07-26 09:38:59 +05:30
parent f18f142401
commit 71eab8e042
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
internal/api/controller/system/controller.go
View File

@ -11,25 +11,23 @@ import (
"github.com/harness/gitness/types"
)
type registrationCheck func(ctx context.Context) (bool, error)
type Controller struct {
principalStore store.PrincipalStore
config *types.Config
IsUserRegistrationAllowed registrationCheck
principalStore store.PrincipalStore
config *types.Config
}
func NewController(principalStore store.PrincipalStore, config *types.Config) *Controller {
return &Controller{
principalStore: principalStore,
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
}