From 05f1dfc6896d3affa7c494a1b955f230d836a4b7 Mon Sep 17 00:00:00 2001
From: Armand Philippot <git@armandphilippot.com>
Date: Fri, 27 Oct 2023 18:07:45 +0200
Subject: feat: replace next-themes with a custom ThemeProvider

To be honest, next-themes was working fine. However since I use a theme
provider for Prism code blocks, some code is duplicated between this
app and the library. So I prefer to use a custom Provider without the
options I don't need.
---
 src/utils/hooks/use-theme/index.ts           |  1 +
 src/utils/hooks/use-theme/use-theme.test.tsx | 82 ++++++++++++++++++++++++++++
 src/utils/hooks/use-theme/use-theme.ts       | 15 +++++
 3 files changed, 98 insertions(+)
 create mode 100644 src/utils/hooks/use-theme/index.ts
 create mode 100644 src/utils/hooks/use-theme/use-theme.test.tsx
 create mode 100644 src/utils/hooks/use-theme/use-theme.ts

(limited to 'src/utils/hooks/use-theme')

diff --git a/src/utils/hooks/use-theme/index.ts b/src/utils/hooks/use-theme/index.ts
new file mode 100644
index 0000000..4e8fc4a
--- /dev/null
+++ b/src/utils/hooks/use-theme/index.ts
@@ -0,0 +1 @@
+export * from './use-theme';
diff --git a/src/utils/hooks/use-theme/use-theme.test.tsx b/src/utils/hooks/use-theme/use-theme.test.tsx
new file mode 100644
index 0000000..feaabfa
--- /dev/null
+++ b/src/utils/hooks/use-theme/use-theme.test.tsx
@@ -0,0 +1,82 @@
+import { describe, expect, it } from '@jest/globals';
+import { act, renderHook } from '@testing-library/react';
+import type { FC, ReactNode } from 'react';
+import { ThemeProvider, type ThemeProviderProps } from '../../providers';
+import { useTheme } from './use-theme';
+
+const createWrapper = (
+  Wrapper: FC<ThemeProviderProps>,
+  config: ThemeProviderProps
+) =>
+  function CreatedWrapper({ children }: { children: ReactNode }) {
+    return <Wrapper {...config}>{children}</Wrapper>;
+  };
+
+describe('useTheme', () => {
+  it('should return the default value without provider and prevent update', () => {
+    const defaultTheme = 'system';
+    const { result } = renderHook(() => useTheme());
+
+    expect(result.current.theme).toBe(defaultTheme);
+
+    act(() => result.current.setTheme('dark'));
+
+    expect(result.current.theme).toBe(defaultTheme);
+  });
+
+  it('can update the value', () => {
+    const defaultTheme = 'dark';
+
+    const { result } = renderHook(() => useTheme(), {
+      wrapper: createWrapper(ThemeProvider, {
+        attribute: 'magnam',
+        defaultTheme,
+        storageKey: 'repellat',
+      }),
+    });
+
+    expect(result.current.theme).toBe(defaultTheme);
+
+    const newTheme = 'light';
+
+    act(() => result.current.setTheme(newTheme));
+
+    expect(result.current.theme).toBe(newTheme);
+  });
+
+  it('can toggle the theme from dark to light', () => {
+    const defaultTheme = 'dark';
+
+    const { result } = renderHook(() => useTheme(), {
+      wrapper: createWrapper(ThemeProvider, {
+        attribute: 'voluptatibus',
+        defaultTheme,
+        storageKey: 'qui',
+      }),
+    });
+
+    expect(result.current.theme).toBe(defaultTheme);
+
+    act(() => result.current.toggleTheme());
+
+    expect(result.current.theme).toBe('light');
+  });
+
+  it('can toggle the theme from light to dark', () => {
+    const defaultTheme = 'light';
+
+    const { result } = renderHook(() => useTheme(), {
+      wrapper: createWrapper(ThemeProvider, {
+        attribute: 'sed',
+        defaultTheme,
+        storageKey: 'ut',
+      }),
+    });
+
+    expect(result.current.theme).toBe(defaultTheme);
+
+    act(() => result.current.toggleTheme());
+
+    expect(result.current.theme).toBe('dark');
+  });
+});
diff --git a/src/utils/hooks/use-theme/use-theme.ts b/src/utils/hooks/use-theme/use-theme.ts
new file mode 100644
index 0000000..0605d8b
--- /dev/null
+++ b/src/utils/hooks/use-theme/use-theme.ts
@@ -0,0 +1,15 @@
+import { useCallback, useContext } from 'react';
+import { ThemeContext } from '../../providers';
+
+export const useTheme = () => {
+  const { resolvedTheme, theme, setTheme } = useContext(ThemeContext);
+
+  const toggleTheme = useCallback(() => {
+    setTheme(() => {
+      if (resolvedTheme === 'dark') return 'light';
+      return 'dark';
+    });
+  }, [resolvedTheme, setTheme]);
+
+  return { resolvedTheme, setTheme, theme, toggleTheme };
+};
-- 
cgit v1.2.3