From 0444afe59b91c758c3f8cbf9c4c9b9444700593d Mon Sep 17 00:00:00 2001
From: Lucca Dukic <109136188+LuccaBitfly@users.noreply.github.com>
Date: Tue, 14 Jan 2025 16:59:26 +0100
Subject: [PATCH] fix: use free tier perks when dashboard user doesn't exist

---
 backend/pkg/api/data_access/data_access.go  | 3 +--
 backend/pkg/api/handlers/handler_service.go | 8 ++++++++
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/backend/pkg/api/data_access/data_access.go b/backend/pkg/api/data_access/data_access.go
index 165ec1b9b..af05c2950 100644
--- a/backend/pkg/api/data_access/data_access.go
+++ b/backend/pkg/api/data_access/data_access.go
@@ -14,7 +14,6 @@ import (
 	"github.com/gobitfly/beaconchain/pkg/commons/log"
 	"github.com/gobitfly/beaconchain/pkg/commons/types"
 	"github.com/jmoiron/sqlx"
-	"github.com/pkg/errors"
 )
 
 type DataAccessor interface {
@@ -202,4 +201,4 @@ func (d *DataAccessService) Close() {
 	}
 }
 
-var ErrNotFound = errors.New("not found")
+var ErrNotFound = db.ErrNotFound
diff --git a/backend/pkg/api/handlers/handler_service.go b/backend/pkg/api/handlers/handler_service.go
index 32a920bf9..18238dc46 100644
--- a/backend/pkg/api/handlers/handler_service.go
+++ b/backend/pkg/api/handlers/handler_service.go
@@ -207,6 +207,14 @@ func (h *HandlerService) getDashboardPremiumPerks(ctx context.Context, id types.
 	}
 	userInfo, err := h.daService.GetUserInfo(ctx, dashboardUser.UserId)
 	if err != nil {
+		if errors.Is(err, dataaccess.ErrNotFound) {
+			log.Warn("user not found for dashboard owner, returning free tier perks", log.Fields{"dashboard_id": id.Id, "user_id_of_dashboard": dashboardUser.UserId})
+			perk, err := h.daService.GetFreeTierPerks(ctx)
+			if err != nil {
+				return nil, err
+			}
+			return perk, nil
+		}
 		return nil, err
 	}