From dda7d2f598f23b61574609d327a8f680e25348c3 Mon Sep 17 00:00:00 2001
From: vasavk <vasavi.kanamarlapudi@verizon.com>
Date: Mon, 12 Feb 2024 13:27:11 +0530
Subject: [PATCH] story:ONEAPP-6315: show badge variants

---
 .../ButtonIconViewController.swift            | 21 ++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/VDSSample/ViewControllers/ButtonIconViewController.swift b/VDSSample/ViewControllers/ButtonIconViewController.swift
index 54feaac..423d0f5 100644
--- a/VDSSample/ViewControllers/ButtonIconViewController.swift
+++ b/VDSSample/ViewControllers/ButtonIconViewController.swift
@@ -58,6 +58,7 @@ class ButtonIconViewController: BaseViewController<ButtonIcon> {
     var disabledSwitch = Toggle()
     var selectableSwitch = Toggle()
     var badgeIndicatorSwitch = Toggle()
+    var variantOneSwitch = Toggle()
     
     lazy var badgeIndicatorExpandDirectionPickerSelectorView = {
         PickerSelectorView(title: "right",
@@ -74,7 +75,6 @@ class ButtonIconViewController: BaseViewController<ButtonIcon> {
     
     override func setupForm(){
         super.setupForm()
-//        addActionRow()
         addFormRow(label: "Disabled", view: disabledSwitch)
         addFormRow(label: "Surface", view: surfacePickerSelectorView)
         addFormRow(label: "Surface Type", view: surfaceTypePickerSelectorView)
@@ -91,12 +91,27 @@ class ButtonIconViewController: BaseViewController<ButtonIcon> {
         addFormRow(label: "Y Offset", view: centerY)
         addFormRow(label: "Show Badge Indicator", view: badgeIndicatorSwitch)
         addFormRow(label: "Expand Direction", view: badgeIndicatorExpandDirectionPickerSelectorView)
-                
+        addFormRow(label: "Badge Variants", view: variantOneSwitch)
+
+        variantOneSwitch.onChange = { [weak self] sender in
+            guard let self else { return }
+            if sender.isOn {
+                if badgeIndicatorSwitch.isOn {
+                    component.badgeIndicatorModel = ButtonIcon.BadgeIndicatorModel(kind: BadgeIndicator.Kind.simple, size: BadgeIndicator.Size.small)
+                } else {
+                    component.badgeIndicatorModel = nil
+                }
+            } else  {
+                component.badgeIndicatorModel = ButtonIcon.BadgeIndicatorModel(kind: BadgeIndicator.Kind.numbered, size: BadgeIndicator.Size.small, maximumDigits: BadgeIndicator.MaximumDigits.two, number: 999, trailingText: "New")
+            }
+        }
+        
         badgeIndicatorSwitch.onChange =  { [weak self] sender in
             guard let self else { return }
             self.component.showBadgeIndicator = sender.isOn
             self.setBadgeIndicatorModel()
             self.component.badgeIndicator.isHidden = !sender.isOn
+            variantOneSwitch.isOn = false
         }
         
         disabledSwitch.onChange =  { [weak self] sender in
@@ -148,7 +163,7 @@ class ButtonIconViewController: BaseViewController<ButtonIcon> {
     //sub models
     func setBadgeIndicatorModel() {
         if badgeIndicatorSwitch.isOn {
-            component.badgeIndicatorModel = ButtonIcon.BadgeIndicatorModel(kind: BadgeIndicator.Kind.numbered, size: BadgeIndicator.Size.small, maximumDigits: BadgeIndicator.MaximumDigits.three, number: 9999, trailingText: "New")
+            component.badgeIndicatorModel = ButtonIcon.BadgeIndicatorModel(kind: BadgeIndicator.Kind.numbered, size: BadgeIndicator.Size.small, maximumDigits: BadgeIndicator.MaximumDigits.two, number: 999)
         } else {
             component.badgeIndicatorModel = nil
         }