@servicetitan/navigation 9.2.1 → 10.0.1

package/src/utils/side-nav.ts

@@ -1,10 +1,11 @@
-import { isCounterPropsObject } from './counter-tag';
-import { HeaderNavigationItemData, HeaderNavigationItemSubmenu } from './navigation';
+import { CounterTagProps } from '../components/counter-tag';
+import { CounterTagData, CounterTagValue } from './counter-tag';
+import { NavigationSubmenuData } from './navigation';
 
 export function getSubmenuGroupTag(
-    submenu: HeaderNavigationItemSubmenu | undefined,
-    defaultTag: HeaderNavigationItemData['tag']
-): HeaderNavigationItemData['tag'] {
+    submenu: NavigationSubmenuData | undefined,
+    defaultTag: CounterTagProps | undefined
+): CounterTagProps | undefined {
     if (!submenu) {
         return defaultTag;
     }
@@ -13,9 +14,7 @@ export function getSubmenuGroupTag(
 
     for (const group of submenu.groups) {
         for (const link of group.links) {
-            const ltv: number | boolean | undefined = isCounterPropsObject(link.tag)
-                ? link.tag.value
-                : link.tag;
+            const ltv: number | boolean | undefined = link.counter;
 
             if (ltv) {
                 if (typeof ltv === 'number') {
@@ -30,5 +29,14 @@ export function getSubmenuGroupTag(
         }
     }
 
-    return tagValue ?? defaultTag;
+    return tagValue ? getCounterTag(tagValue) : defaultTag;
 }
+
+export const getCounterTag = (
+    counter: CounterTagValue | undefined,
+    tag?: CounterTagData | undefined
+): CounterTagProps | undefined => {
+    const value = counter ?? tag?.value;
+
+    return value ? { value, className: '' } : undefined;
+};