From 045a05247034a4be359920a6734a4fb6943921cb Mon Sep 17 00:00:00 2001
From: soykje <jean-paul.lescouzeres@adevinta.com>
Date: Tue, 5 Dec 2023 16:58:45 +0100
Subject: [PATCH] feat(progress-tracker): add disabled state management

---
 .../src/ProgressTrackerStep.styles.ts         | 23 +++++++----
 .../src/ProgressTrackerStep.tsx               | 39 ++++++++++++++-----
 2 files changed, 46 insertions(+), 16 deletions(-)

diff --git a/packages/components/progress-tracker/src/ProgressTrackerStep.styles.ts b/packages/components/progress-tracker/src/ProgressTrackerStep.styles.ts
index edc24199b..3482f072c 100644
--- a/packages/components/progress-tracker/src/ProgressTrackerStep.styles.ts
+++ b/packages/components/progress-tracker/src/ProgressTrackerStep.styles.ts
@@ -1,6 +1,6 @@
 import { cva, type VariantProps } from 'class-variance-authority'
 
-export const stepWrapperVariant = cva(
+export const stepItemVariant = cva(
   [
     'relative inline-flex flex-auto first:grow-0 justify-center',
     // Progress Track
@@ -36,6 +36,7 @@ export const stepWrapperVariant = cva(
   }
 )
 
+// eslint-disable-next-line tailwindcss/no-custom-classname
 export const stepButtonVariant = cva(
   [
     'block mx-sm mb-auto',
@@ -46,17 +47,24 @@ export const stepButtonVariant = cva(
     'before:flex before:items-center before:justify-center before:rounded-full',
     'before:border-sm before:text-on-basic-container',
     'before:text-body-2 before:font-bold',
-    'hover:before:bg-basic/dim-5',
   ],
   {
     variants: {
       size: {
-        sm: ['before:w-[16px] before:h-[16px]', 'before:content-[""]'],
-        md: 'before:w-[24px] before:h-[24px]',
-        lg: 'before:w-[32px] before:h-[32px]',
+        sm: ['before:w-sz-16 before:h-sz-16', 'before:content-["_"]'],
+        md: 'before:w-sz-24 before:h-sz-24',
+        lg: 'before:w-sz-32 before:h-sz-32',
+      },
+      complete: {
+        true: [
+          'before:content-[url(data:image/svg+xml;base64,PHN2ZyB2aWV3Qm94PScwIDAgMjQgMjQnIGhlaWdodD0nMTYnIHdpZHRoPScxNicgeG1sbnM9J2h0dHA6Ly93d3cudzMub3JnLzIwMDAvc3ZnJyBmaWxsPSdjdXJyZW50Q29sb3InIHN0cm9rZT0nbm9uZSc+PHBhdGggZD0nbTguOTIsMTkuMDhjLS4xOCwwLS4zNi0uMDMtLjUzLS4xcy0uMzMtLjE3LS40Ny0uMzFsLTUuNDktNS4zNGMtLjI4LS4yOC0uNDItLjYxLS40Mi0xcy4xNC0uNzMuNDItMWMuMjgtLjI4LjYyLS40MSwxLjAyLS40MXMuNzQuMTQsMS4wNS40MWw0LjQzLDQuMywxMC42Mi0xMC4yOWMuMjgtLjI4LjYyLS40MiwxLjAyLS40My4zOSwwLC43My4xMywxLjAyLjQzLjI4LjI4LjQyLjYxLjQyLDFzLS4xNC43My0uNDIsMWwtMTEuNjUsMTEuMzJjLS4xNC4xNC0uMy4yNC0uNDcuMzEtLjE3LjA3LS4zNS4xLS41My4xWic+PC9wYXRoPjwvc3ZnPg==)]',
+          'before:inline-flex before:leading-none',
+          'hover:before:bg-basic/dim-5',
+        ],
+        false: '',
       },
       active: {
-        true: 'before:bg-basic-container',
+        true: ['before:bg-basic-container', 'hover:before:bg-basic/dim-5'],
         false: '',
       },
       disabled: {
@@ -65,8 +73,9 @@ export const stepButtonVariant = cva(
       },
     },
     defaultVariants: {
-      active: false,
       disabled: false,
+      active: false,
+      complete: false,
       size: 'lg',
     },
   }
diff --git a/packages/components/progress-tracker/src/ProgressTrackerStep.tsx b/packages/components/progress-tracker/src/ProgressTrackerStep.tsx
index 8bd073e05..fc0886832 100644
--- a/packages/components/progress-tracker/src/ProgressTrackerStep.tsx
+++ b/packages/components/progress-tracker/src/ProgressTrackerStep.tsx
@@ -1,14 +1,15 @@
 import { type ComponentPropsWithoutRef, forwardRef, useEffect, useId, useState } from 'react'
 
 import { useProgressTrackerContext } from './ProgressTrackerContext'
-import { stepButtonVariant, stepWrapperVariant } from './ProgressTrackerStep.styles'
+import { stepButtonVariant, stepItemVariant } from './ProgressTrackerStep.styles'
 
 export type ProgressTrackerStepProps = ComponentPropsWithoutRef<'li'> & {
   label?: string
+  disabled?: boolean
 }
 
 export const ProgressTrackerStep = forwardRef<HTMLLIElement, ProgressTrackerStepProps>(
-  ({ label, className, ...rest }, ref) => {
+  ({ label, disabled = false, className, ...rest }, ref) => {
     const {
       stepIndex: activeStepIndex,
       steps,
@@ -20,21 +21,41 @@ export const ProgressTrackerStep = forwardRef<HTMLLIElement, ProgressTrackerStep
     const ID = useId()
     const stepId = `step-${ID}`
 
-    const [isActive, setIsActive] = useState(false)
+    const [progressState, setProgressState] = useState<
+      'active' | 'complete' | 'disabled' | undefined
+    >(() => (disabled ? 'disabled' : undefined))
 
     useEffect(() => setSteps(steps => steps.add(stepId)), [])
 
-    useEffect(
-      () => setIsActive([...steps].indexOf(stepId) === activeStepIndex),
-      [activeStepIndex, steps, stepId]
-    )
+    useEffect(() => {
+      if ([...steps].indexOf(stepId) === activeStepIndex) {
+        setProgressState('active')
+      } else if ([...steps].indexOf(stepId) < activeStepIndex) {
+        setProgressState('complete')
+      } else {
+        setProgressState(undefined)
+      }
+    }, [activeStepIndex, steps, stepId])
 
     return (
-      <li id={stepId} ref={ref} {...rest} className={stepWrapperVariant({ size })}>
+      <li
+        id={stepId}
+        ref={ref}
+        data-state={progressState}
+        className={stepItemVariant({ size })}
+        {...rest}
+      >
         <button
           type="button"
           onClick={() => onStepClick(stepId)}
-          className={stepButtonVariant({ active: isActive, size, className })}
+          disabled={progressState === 'disabled'}
+          className={stepButtonVariant({
+            complete: progressState === 'complete',
+            active: progressState === 'active',
+            disabled: progressState === 'disabled',
+            size,
+            className,
+          })}
         >
           {label && <span className="block text-body-2 font-bold text-on-surface">{label}</span>}
         </button>