File tree Expand file tree Collapse file tree 1 file changed +6
-14
lines changed Expand file tree Collapse file tree 1 file changed +6
-14
lines changed Original file line number Diff line number Diff line change @@ -221,10 +221,13 @@ export function useColorModeState(
221
221
// Add mode className
222
222
const customPropertiesEnabled = checkHasCustomPropertiesEnabled ( theme )
223
223
224
- const manualSetRef = React . useRef ( false )
225
224
const manuallySetMode = React . useCallback ( ( value : string | null ) => {
226
- manualSetRef . current = true
227
- setMode ( value )
225
+ setMode ( value || null )
226
+ if ( value ) {
227
+ storage . set ( value )
228
+ } else {
229
+ storage . clear ( )
230
+ }
228
231
} , [ ] )
229
232
230
233
// Set initial color mode in lazy
@@ -237,17 +240,6 @@ export function useColorModeState(
237
240
}
238
241
} , [ ] )
239
242
240
- // Store mode preference
241
- useIsomorphicLayoutEffect ( ( ) => {
242
- if ( manualSetRef . current ) {
243
- if ( mode ) {
244
- storage . set ( mode )
245
- } else {
246
- storage . clear ( )
247
- }
248
- }
249
- } , [ mode ] )
250
-
251
243
// Sync system mode
252
244
useIsomorphicLayoutEffect ( ( ) => {
253
245
const storedMode = storage . get ( )
You can’t perform that action at this time.
0 commit comments