From 59c1aca734f399a4a4e0ad60fcccda7666a7b068 Mon Sep 17 00:00:00 2001
From: Clara Castillo <claracastillochale@gmail.com>
Date: Mon, 3 Jun 2024 10:43:07 +0200
Subject: [PATCH] fix: allow to render custom icons for toasts

fixes #10
---
 libs/ngx-sonner/src/lib/toaster.component.ts  | 22 +++++++++++++------
 .../src/tests/toaster-test.component.ts       | 19 +++++++++++++++-
 libs/ngx-sonner/src/tests/toaster.spec.ts     | 12 ++++++++++
 3 files changed, 45 insertions(+), 8 deletions(-)

diff --git a/libs/ngx-sonner/src/lib/toaster.component.ts b/libs/ngx-sonner/src/lib/toaster.component.ts
index 781f757..fb036d1 100644
--- a/libs/ngx-sonner/src/lib/toaster.component.ts
+++ b/libs/ngx-sonner/src/lib/toaster.component.ts
@@ -79,13 +79,21 @@ import { Position, Theme, ToasterProps } from './types';
                 [classes]="toastOptions().classes ?? {}"
                 [duration]="toastOptions().duration ?? duration()"
                 [unstyled]="toastOptions().unstyled ?? false">
-                <ngx-sonner-loader
-                  [isVisible]="toast.type === 'loading'"
-                  loading-icon />
-                <ngx-sonner-icon type="success" success-icon />
-                <ngx-sonner-icon type="error" error-icon />
-                <ngx-sonner-icon type="warning" warning-icon />
-                <ngx-sonner-icon type="info" info-icon />
+                <ng-content select="[loading-icon]" loading-icon>
+                  <ngx-sonner-loader [isVisible]="toast.type === 'loading'" />
+                </ng-content>
+                <ng-content select="[success-icon]" success-icon>
+                  <ngx-sonner-icon type="success" />
+                </ng-content>
+                <ng-content select="[error-icon]" error-icon>
+                  <ngx-sonner-icon type="error" />
+                </ng-content>
+                <ng-content select="[warning-icon]" warning-icon>
+                  <ngx-sonner-icon type="warning" />
+                </ng-content>
+                <ng-content select="[info-icon]" info-icon>
+                  <ngx-sonner-icon type="info" />
+                </ng-content>
               </ngx-sonner-toast>
             }
           </ol>
diff --git a/libs/ngx-sonner/src/tests/toaster-test.component.ts b/libs/ngx-sonner/src/tests/toaster-test.component.ts
index 5a6b1d4..ebdcda3 100644
--- a/libs/ngx-sonner/src/tests/toaster-test.component.ts
+++ b/libs/ngx-sonner/src/tests/toaster-test.component.ts
@@ -18,7 +18,24 @@ export type ToastTestInputs = {
     <ngx-sonner-toaster
       [dir]="dir()"
       [theme]="theme()"
-      [closeButton]="closeButton()" />
+      [closeButton]="closeButton()">
+      <svg
+        success-icon
+        xmlns="http://www.w3.org/2000/svg"
+        width="24"
+        height="24"
+        viewBox="0 0 24 24"
+        fill="none"
+        stroke="currentColor"
+        stroke-width="2"
+        stroke-linecap="round"
+        stroke-linejoin="round"
+        class="lucide lucide-badge-check">
+        <path
+          d="M3.85 8.62a4 4 0 0 1 4.78-4.77 4 4 0 0 1 6.74 0 4 4 0 0 1 4.78 4.78 4 4 0 0 1 0 6.74 4 4 0 0 1-4.77 4.78 4 4 0 0 1-6.75 0 4 4 0 0 1-4.78-4.77 4 4 0 0 1 0-6.76Z" />
+        <path d="m9 12 2 2 4-4" />
+      </svg>
+    </ngx-sonner-toaster>
     <button data-testid="trigger" (click)="onClick()">Trigger</button>
   `,
   changeDetection: ChangeDetectionStrategy.OnPush,
diff --git a/libs/ngx-sonner/src/tests/toaster.spec.ts b/libs/ngx-sonner/src/tests/toaster.spec.ts
index effac62..5cb7efb 100644
--- a/libs/ngx-sonner/src/tests/toaster.spec.ts
+++ b/libs/ngx-sonner/src/tests/toaster.spec.ts
@@ -284,4 +284,16 @@ describe('Toaster', () => {
     const closeButton = container.querySelector('[data-close-button]');
     expect(closeButton).not.toBeNull();
   });
+
+  it('should render the custom icon when provided', async () => {
+    const { user, trigger, container } = await setup({
+      callback: toast => toast.success('Hello world'),
+    });
+    await user.click(trigger);
+    const icon = container.querySelector('[data-icon]');
+    expect(icon).not.toBeNull();
+    expect(icon).toContainHTML(
+      '<svg success-icon xmlns="http://www.w3.org/2000/svg" width="24" height="24" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="lucide lucide-badge-check"><path d="M3.85 8.62a4 4 0 0 1 4.78-4.77 4 4 0 0 1 6.74 0 4 4 0 0 1 4.78 4.78 4 4 0 0 1 0 6.74 4 4 0 0 1-4.77 4.78 4 4 0 0 1-6.75 0 4 4 0 0 1-4.78-4.77 4 4 0 0 1 0-6.76Z" /><path d="m9 12 2 2 4-4" /></svg>'
+    );
+  });
 });