From 3080e8e9961e7490ba16670b0677e9fe56f3117d Mon Sep 17 00:00:00 2001
From: Egor <egor.andrianovv@gmail.com>
Date: Sat, 31 Jan 2026 19:52:08 +0300
Subject: [PATCH 1/3] Update successNotification.ts

---
 src/store/successNotification.ts | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/store/successNotification.ts b/src/store/successNotification.ts
index eb04ab0..1d7961c 100644
--- a/src/store/successNotification.ts
+++ b/src/store/successNotification.ts
@@ -5,7 +5,8 @@ export type SuccessNotificationType =
   | 'balance_topup'
   | 'subscription_activated'
   | 'subscription_renewed'
-  | 'subscription_purchased';
+  | 'subscription_purchased'
+  | 'devices_purchased';
 
 export interface SuccessNotificationData {
   type: SuccessNotificationType;
@@ -21,6 +22,10 @@ export interface SuccessNotificationData {
   title?: string;
   /** Custom message override */
   message?: string;
+  /** Number of devices added */
+  devicesAdded?: number;
+  /** New total device limit */
+  newDeviceLimit?: number;
 }
 
 interface SuccessNotificationState {

From abe016c7b105c04d55dc13db3fa420b933aefa3a Mon Sep 17 00:00:00 2001
From: Egor <egor.andrianovv@gmail.com>
Date: Sat, 31 Jan 2026 19:53:23 +0300
Subject: [PATCH 2/3] Add files via upload

---
 src/components/SuccessNotificationModal.tsx | 51 ++++++++++++++++++++-
 src/components/WebSocketNotifications.tsx   | 15 ++++++
 2 files changed, 65 insertions(+), 1 deletion(-)

diff --git a/src/components/SuccessNotificationModal.tsx b/src/components/SuccessNotificationModal.tsx
index 8bd9ced..8249139 100644
--- a/src/components/SuccessNotificationModal.tsx
+++ b/src/components/SuccessNotificationModal.tsx
@@ -48,6 +48,16 @@ const RocketIcon = () => (
   </svg>
 );
 
+const DevicesIcon = () => (
+  <svg className="h-8 w-8" fill="none" viewBox="0 0 24 24" stroke="currentColor" strokeWidth={1.5}>
+    <path
+      strokeLinecap="round"
+      strokeLinejoin="round"
+      d="M10.5 1.5H8.25A2.25 2.25 0 006 3.75v16.5a2.25 2.25 0 002.25 2.25h7.5A2.25 2.25 0 0018 20.25V3.75a2.25 2.25 0 00-2.25-2.25H13.5m-3 0V3h3V1.5m-3 0h3m-3 18.75h3"
+    />
+  </svg>
+);
+
 const CloseIcon = () => (
   <svg className="h-5 w-5" fill="none" viewBox="0 0 24 24" stroke="currentColor" strokeWidth={2}>
     <path strokeLinecap="round" strokeLinejoin="round" d="M6 18L18 6M6 6l12 12" />
@@ -114,6 +124,7 @@ export default function SuccessNotificationModal() {
     data.type === 'subscription_activated' ||
     data.type === 'subscription_renewed' ||
     data.type === 'subscription_purchased';
+  const isDevicesPurchased = data.type === 'devices_purchased';
 
   // Format amount
   const formattedAmount = data.amountKopeks
@@ -158,6 +169,10 @@ export default function SuccessNotificationModal() {
       title = t('successNotification.subscriptionPurchased.title', 'Subscription purchased!');
       icon = <RocketIcon />;
       gradientClass = 'from-accent-500 to-purple-600';
+    } else if (data.type === 'devices_purchased') {
+      title = t('successNotification.devicesPurchased.title', 'Devices added!');
+      icon = <DevicesIcon />;
+      gradientClass = 'from-blue-500 to-cyan-600';
     }
   }
 
@@ -211,7 +226,31 @@ export default function SuccessNotificationModal() {
                   ? t('successNotification.amount', 'Amount')
                   : t('successNotification.price', 'Price')}
               </span>
-              <span className="text-lg font-bold text-success-400">+{formattedAmount}</span>
+              <span
+                className={`text-lg font-bold ${isDevicesPurchased ? 'text-dark-100' : 'text-success-400'}`}
+              >
+                {isDevicesPurchased ? '' : '+'}
+                {formattedAmount}
+              </span>
+            </div>
+          )}
+
+          {/* Devices info (for devices purchase) */}
+          {isDevicesPurchased && data.devicesAdded && (
+            <div className="flex items-center justify-between rounded-xl bg-dark-800/50 px-4 py-3">
+              <span className="text-dark-400">
+                {t('successNotification.devicesAdded', 'Devices added')}
+              </span>
+              <span className="text-lg font-bold text-blue-400">+{data.devicesAdded}</span>
+            </div>
+          )}
+
+          {isDevicesPurchased && data.newDeviceLimit && (
+            <div className="flex items-center justify-between rounded-xl bg-dark-800/50 px-4 py-3">
+              <span className="text-dark-400">
+                {t('successNotification.totalDevices', 'Total devices')}
+              </span>
+              <span className="font-semibold text-dark-100">{data.newDeviceLimit}</span>
             </div>
           )}
 
@@ -265,6 +304,16 @@ export default function SuccessNotificationModal() {
               </button>
             )}
 
+            {isDevicesPurchased && (
+              <button
+                onClick={handleGoToSubscription}
+                className="flex w-full items-center justify-center gap-2 rounded-xl bg-gradient-to-r from-blue-500 to-cyan-600 py-3.5 font-bold text-white shadow-lg shadow-blue-500/25 transition-all hover:from-blue-400 hover:to-cyan-500 active:from-blue-600 active:to-cyan-700"
+              >
+                <DevicesIcon />
+                <span>{t('successNotification.goToSubscription', 'Go to Subscription')}</span>
+              </button>
+            )}
+
             <button
               onClick={handleClose}
               className="w-full rounded-xl bg-dark-800 py-3 font-semibold text-dark-300 transition-colors hover:bg-dark-700 hover:text-dark-100"
diff --git a/src/components/WebSocketNotifications.tsx b/src/components/WebSocketNotifications.tsx
index 5c1034c..17626e5 100644
--- a/src/components/WebSocketNotifications.tsx
+++ b/src/components/WebSocketNotifications.tsx
@@ -170,6 +170,21 @@ export default function WebSocketNotifications() {
         return;
       }
 
+      if (type === 'subscription.devices_purchased') {
+        // Show prominent success modal for device purchase
+        showSuccessModal({
+          type: 'devices_purchased',
+          amountKopeks: message.amount_kopeks,
+          devicesAdded: message.devices_added,
+          newDeviceLimit: message.new_device_limit,
+        });
+        queryClient.invalidateQueries({ queryKey: ['subscription'] });
+        queryClient.invalidateQueries({ queryKey: ['balance'] });
+        queryClient.invalidateQueries({ queryKey: ['transactions'] });
+        refreshUser();
+        return;
+      }
+
       // Autopay events
       if (type === 'autopay.success') {
         const amount = message.amount_rubles ?? (message.amount_kopeks ?? 0) / 100;

From dea0562a0dbf85aa33a52d82ad5648d6bca12366 Mon Sep 17 00:00:00 2001
From: Egor <egor.andrianovv@gmail.com>
Date: Sat, 31 Jan 2026 19:53:49 +0300
Subject: [PATCH 3/3] Update useWebSocket.ts

---
 src/hooks/useWebSocket.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/hooks/useWebSocket.ts b/src/hooks/useWebSocket.ts
index 5d52fa8..6a719c5 100644
--- a/src/hooks/useWebSocket.ts
+++ b/src/hooks/useWebSocket.ts
@@ -23,6 +23,9 @@ export interface WSMessage {
   new_expires_at?: string;
   tariff_name?: string;
   days_left?: number;
+  // Device purchase events
+  devices_added?: number;
+  new_device_limit?: number;
   // Autopay events
   required_kopeks?: number;
   required_rubles?: number;