From b96b50c01f93275ecc7ec273b63b22fb3a22e171 Mon Sep 17 00:00:00 2001
From: vladp <vlad.popovtsevv@gmail.com>
Date: Mon, 9 Feb 2026 11:28:23 +0700
Subject: [PATCH] add referal only for b2c

---
 src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java b/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java
index 867ce06..6f69ac2 100644
--- a/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java
+++ b/src/main/java/ru/soune/nocopy/handler/RegRequestHandler.java
@@ -101,7 +101,9 @@ public class RegRequestHandler implements RequestHandler {
         regAnswer.setVerified(true);
         regAnswer.setActive(true);
 
-        referralService.onRegister(authToken.getUser().getId(), regRequest.getReferralLink());
+        if (regRequest.getAccountType().equals("b2c")) {
+            referralService.onRegister(authToken.getUser().getId(), regRequest.getReferralLink());
+        }
 
         return new BaseResponse(request.getMsgId(), MessageCode.SUCCESS.getCode(),
                 MessageCode.SUCCESS.getDescription(), regAnswer);