Hogy kell ezt a jQuery kódot függvénnyé alakítani?

Sziasztok.

Hogy kell ezt a jQuery kódot függvénnyé alakítani?

$('.submenu').click( function(){

if($(this).parent().hasClass('active')) {

$(this).parent().removeClass('active');

}

else {

$(this).parent().addClass('active');

}

});

$('#menu-button').click( function(){

if($('#menu').hasClass('active')) {

$('#menu').removeClass('active');

}

else {

$('#menu').addClass('active');

}

})

Én így csináltam:

function snOnClickClass(onClick, className, parent = false) {

$(onClick).click( function(){

parent = (!parent) ? $(this).parent() : $(parent);

if(parent.hasClass(className)) {

parent.removeClass(className);

}

else {

parent.addClass(className);

}

});

}

snOnClickClass('.submenu', 'active');

snOnClickClass('#menu-button', 'active', '#menu');

De nem úgy működik, ahogy akarom. Mindig csak az kap active class-t, amelyikre legelőször kattintottam. Miért? Vagy van valami normálisabb megoldás, amivel "szebbé" alakíthatom a fenti kódot?