Creating a Separate nav Menu when logged in for wholesale

Store Design
, ,
1

I’ve created a separate wholesale section to the website. When users are tagged and login they will only see wholesale prices.

I have to edit the header.liquid file. Normally I replace
{% for link in linklists.main-menu.links %}
or may look like this
{% for link in linklists[section.settings.main_linklist].links %}

With the following code

{% assign menu_handle = ‘main-menu’ %}
{% if customer %}
{% if customer.tags contains ‘wholesale’ %}
{% assign menu_handle = ‘main-menu-wholesale’ %}
{% endif %}
{% endif %}
{% include ‘site-nav’, linklist: menu_handle %}.

Then normally I’d add this for mobile nav.
{% assign menu_handle = ‘main-menu’ %}
{% if customer %}
{% if customer.tags contains ‘wholesale’ %}
{% assign menu_handle = ‘main-menu-wholesale’ %}
{% endif %}
{% endif %}
{% for link in linklists[menu_handle].links %}

The problem is the header.liquid file doesn’t contain either line of code above for me to replace.

Client is using colorblock theme and I’m not sure what to do from here.

Any help?