ssm_do_navigation

do-navigation.php

<?php
function ssm_do_navigation( $location = 'header' ) {

  $menu = '';
  $depth = '';
  $items_wrap = '';

  if ( $location == 'header' ) {

    $default_menu = get_field('default_primary_navigation', 'options');
    $branded_environments_menu = get_field('branded_environments_primary_navigation', 'options');
    $depth = 2;
    $items_wrap = '<ul class="medium-horizontal menu show-for-large dropdown" data-dropdown-menu>%3$s</ul>';

  } elseif ( $location == 'off-canvas' ) {

    $default_menu = get_field('default_off_canvas_navigation', 'options');
    $branded_environments_menu = get_field('branded_environments_primary_navigation', 'options');
    $depth = 1;
    $items_wrap = '<ul class="vertical menu">%3$s</ul>';

  } elseif ( $location == 'footer' ) {

    $default_menu = get_field('default_footer_navigation', 'options');
    $branded_environments_menu = get_field('branded_environments_footer_navigation', 'options');
    $depth = 2;
    $items_wrap = '<ul class="menu">%3$s</ul>';

  }

  if ( isset($_COOKIE['ref']) && $_COOKIE['ref'] == 'branded-env' || isset( $_GET['ref'] ) && $_GET['ref'] == 'branded-env' ) {

    $menu = $branded_environments_menu;

  } else {

    $menu = $default_menu;

  }

  return $nav = wp_nav_menu( array( 'menu' => $menu, 'container' => false, 'items_wrap' => $items_wrap, 'depth' => $depth, 'walker' => new Foundation_Nav_Walker ));

}