fix: add default value for theme context

MarceloPrado · Apr 16, 2024 · e30d38e · e30d38e
1 parent d43810c
commit e30d38e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 12 deletions.
diff --git a/packages/flash-calendar/src/components/Calendar.stories.tsx b/packages/flash-calendar/src/components/Calendar.stories.tsx
@@ -162,6 +162,6 @@ export const DateRangePicker = (args: typeof KichenSink.args) => {
 
 export const ControlledColorScheme: StoryObj<typeof Calendar> = {
   args: {
-    colorSchemeToOverride: "dark",
+    colorSchemeToOverride: "light",
   },
 };
diff --git a/packages/flash-calendar/src/hooks/useCalendarTheme.ts b/packages/flash-calendar/src/hooks/useCalendarTheme.ts
@@ -6,17 +6,11 @@ export interface CalendarThemeContext {
   colorScheme?: ColorSchemeName;
 }
 
-export const calendarThemeContext = createContext<
-  CalendarThemeContext | undefined
->(undefined);
+const defaultThemeContext: CalendarThemeContext = { colorScheme: undefined };
 
-export const useCalendarThemeContext = () => {
-  const context = useContext(calendarThemeContext);
+export const calendarThemeContext =
+  createContext<CalendarThemeContext>(defaultThemeContext);
 
-  if (!context) {
-    throw new Error(
-      "useCalendarThemeContext must be called inside <calendarThemeContext.Provider>"
-    );
-  }
-  return context;
+export const useCalendarThemeContext = () => {
+  return useContext(calendarThemeContext);
 };