Есть сайт написанный на Реакте, есть возможность смены темы с белой на чёрную с сохранением в LocalStorage, если оставить чёрную тему то при последующей загрузке страницы на долю секунды видна белая тема, как это можно исправить? Подскажите пожалуйста
LocalStorage:
function switchMode() {
if (document.documentElement.hasAttribute('data-theme')) {
document.documentElement.removeAttribute('data-theme');
localStorage.removeItem('data-theme');
} else {
document.documentElement.setAttribute('data-theme', 'dark');
localStorage.setItem('data-theme', 'dark');
}
}
window.onload = function () {
if (localStorage.getItem('data-theme') === 'dark') {
document.documentElement.setAttribute('data-theme', 'dark');
}
}