validate registry

deepfence · May 22, 2023 · fe97363 · fe97363
1 parent 2208879
commit fe97363
Show file tree

Hide file tree

Showing 8 changed files with 40 additions and 0 deletions.
diff --git a/deepfence_server/pkg/registry/acr/acr.go b/deepfence_server/pkg/registry/acr/acr.go
@@ -9,6 +9,7 @@ import (
 	"github.com/deepfence/ThreatMapper/deepfence_server/pkg/constants"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
 	"github.com/deepfence/golang_deepfence_sdk/utils/log"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(data []byte) (*RegistryACR, error) {
@@ -20,6 +21,10 @@ func New(data []byte) (*RegistryACR, error) {
 	return &r, nil
 }
 
+func (d *RegistryACR) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (d *RegistryACR) IsValidCredential() bool {
 	if d.NonSecret.AzureRegistryURL == "" {
 		return false

diff --git a/deepfence_server/pkg/registry/dockerprivate/dockerprivate.go b/deepfence_server/pkg/registry/dockerprivate/dockerprivate.go
@@ -8,6 +8,7 @@ import (
 	"github.com/deepfence/ThreatMapper/deepfence_server/model"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
 	"github.com/deepfence/golang_deepfence_sdk/utils/log"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(requestByte []byte) (*RegistryDockerPrivate, error) {
@@ -19,6 +20,10 @@ func New(requestByte []byte) (*RegistryDockerPrivate, error) {
 	return &r, nil
 }
 
+func (d *RegistryDockerPrivate) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (d *RegistryDockerPrivate) IsValidCredential() bool {
 	if d.NonSecret.DockerRegistryURL == "" {
 		return false

diff --git a/deepfence_server/pkg/registry/ecr/ecr.go b/deepfence_server/pkg/registry/ecr/ecr.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/deepfence/ThreatMapper/deepfence_server/model"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(requestByte []byte) (*RegistryECR, error) {
@@ -16,6 +17,10 @@ func New(requestByte []byte) (*RegistryECR, error) {
 	return &r, nil
 }
 
+func (d *RegistryECR) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (e *RegistryECR) IsValidCredential() bool {
 	return true
 }

diff --git a/deepfence_server/pkg/registry/gcr/gcr.go b/deepfence_server/pkg/registry/gcr/gcr.go
@@ -9,6 +9,7 @@ import (
 	"github.com/deepfence/ThreatMapper/deepfence_server/pkg/constants"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
 	"github.com/deepfence/golang_deepfence_sdk/utils/log"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(data []byte) (*RegistryGCR, error) {
@@ -20,6 +21,10 @@ func New(data []byte) (*RegistryGCR, error) {
 	return &r, nil
 }
 
+func (d *RegistryGCR) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (d *RegistryGCR) IsValidCredential() bool {
 	if d.NonSecret.RegistryURL == "" {
 		return false

diff --git a/deepfence_server/pkg/registry/gitlab/gitlab.go b/deepfence_server/pkg/registry/gitlab/gitlab.go
@@ -5,6 +5,7 @@ import (
 
 	"github.com/deepfence/ThreatMapper/deepfence_server/model"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(requestByte []byte) (*RegistryGitlab, error) {
@@ -16,6 +17,10 @@ func New(requestByte []byte) (*RegistryGitlab, error) {
 	return &r, nil
 }
 
+func (d *RegistryGitlab) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (e *RegistryGitlab) IsValidCredential() bool {
 	return true
 }

diff --git a/deepfence_server/pkg/registry/harbor/harbor.go b/deepfence_server/pkg/registry/harbor/harbor.go
@@ -8,6 +8,7 @@ import (
 	"github.com/deepfence/ThreatMapper/deepfence_server/model"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
 	"github.com/deepfence/golang_deepfence_sdk/utils/log"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(data []byte) (*RegistryHarbor, error) {
@@ -19,6 +20,10 @@ func New(data []byte) (*RegistryHarbor, error) {
 	return &r, nil
 }
 
+func (d *RegistryHarbor) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (d *RegistryHarbor) IsValidCredential() bool {
 	if d.NonSecret.HarborRegistryURL == "" {
 		return false

diff --git a/deepfence_server/pkg/registry/jfrog/jfrog.go b/deepfence_server/pkg/registry/jfrog/jfrog.go
@@ -8,6 +8,7 @@ import (
 	"github.com/deepfence/ThreatMapper/deepfence_server/model"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
 	"github.com/deepfence/golang_deepfence_sdk/utils/log"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(requestByte []byte) (*RegistryJfrog, error) {
@@ -19,6 +20,10 @@ func New(requestByte []byte) (*RegistryJfrog, error) {
 	return &r, nil
 }
 
+func (d *RegistryJfrog) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (d *RegistryJfrog) IsValidCredential() bool {
 	if d.NonSecret.JfrogRegistryURL == "" {
 		return false

diff --git a/deepfence_server/pkg/registry/quay/quay.go b/deepfence_server/pkg/registry/quay/quay.go
@@ -8,6 +8,7 @@ import (
 	"github.com/deepfence/ThreatMapper/deepfence_server/model"
 	"github.com/deepfence/golang_deepfence_sdk/utils/encryption"
 	"github.com/deepfence/golang_deepfence_sdk/utils/log"
+	"github.com/go-playground/validator/v10"
 )
 
 func New(data []byte) (*RegistryQuay, error) {
@@ -19,6 +20,10 @@ func New(data []byte) (*RegistryQuay, error) {
 	return &r, nil
 }
 
+func (d *RegistryQuay) ValidateFields(v *validator.Validate) error {
+	return v.Struct(d)
+}
+
 func (d *RegistryQuay) IsValidCredential() bool {
 	if d.NonSecret.QuayNamespace == "" {
 		return false