diff --git a/static/js/menu.js b/static/js/menu.js index 12bb40f..6c1f350 100644 --- a/static/js/menu.js +++ b/static/js/menu.js @@ -2,13 +2,10 @@ var toggle = document.getElementById('toggle'); var menu = document.getElementById('menu'); toggle.addEventListener('click', function () { - if (menu.classList.contains('main-nav__list--active')) { - this.setAttribute('aria-expanded', 'false'); - this.classList.remove('main-nav__btn--active'); - menu.classList.remove('main-nav__list--active'); - } else { - menu.classList.add('main-nav__list--active'); - this.classList.add('main-nav__btn--active'); - this.setAttribute('aria-expanded', 'true'); - } + menu.classList.toggle('main-nav__list--active'); + this.classList.toggle('main-nav__btn--active'); + this.setAttribute( + 'aria-expanded', + this.getAttribute('aria-expanded') === 'true' ? 'false' : 'true' + ); });