Check user is not nil before getting Provider Data

rarimo · Apr 5, 2024 · c8faaf4 · c8faaf4
1 parent 5517b33
commit c8faaf4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/internal/service/api/handlers/get_provider_by_identity_id.go b/internal/service/api/handlers/get_provider_by_identity_id.go
@@ -21,7 +21,7 @@ func GetProviderByIdentityId(w http.ResponseWriter, r *http.Request) {
 
 	provider, err := KYCService(r).GetProviderByIdentityId(req)
 	switch {
-	case errors.Is(err, identityproviders.ErrProviderNotFound):
+	case errors.Is(err, identityproviders.ErrProviderNotFound), errors.Is(err, identityproviders.ErrUserNotFound):
 		Log(r).WithField("reason", err).Debug("Not found")
 		ape.RenderErr(w, problems.NotFound())
 		return

diff --git a/internal/service/core/handlers.go b/internal/service/core/handlers.go
@@ -155,6 +155,10 @@ func (k *kycService) GetProviderByIdentityId(req *requests.GetProviderByIdentity
 		return "", errors.Wrap(err, "failed to get user from db with provided identityID")
 	}
 
+	if user == nil {
+		return "", identityproviders.ErrUserNotFound
+	}
+
 	domainData := unstopdom.Domain{}
 	if err := json.Unmarshal(user.ProviderData, &domainData); err != nil {
 		return "", errors.Wrap(err, "failed to unmarshal json")

diff --git a/internal/service/core/identity_providers/models.go b/internal/service/core/identity_providers/models.go
@@ -13,6 +13,7 @@ var (
 
 	// Not found errors
 	ErrProviderNotFound = errors.New("provider not found")
+	ErrUserNotFound     = errors.New("user not found")
 )
 
 type IdentityProviderName string