Helper improvements

Author: morehawes

inc/Helpers/Waymark_Helper.php

@@ -525,13 +525,28 @@ static public function set_map_data_property($map_data, $key = false, $value = f
 
 	static public function add_map_link_to_description($map_id = null, $map_title = false, $map_data = null) {
 		$desc_append = '<div class="waymark-description-link">';
-		$desc_append .= esc_html__('Part of', 'waymark') . ' <b>';
-		if ($map_title) {
-			$desc_append .= '<a href="' . get_permalink($map_id) . '">' . $map_title . '</a>';
-		} else {
-			$desc_append .= '<a href="' . get_permalink($map_id) . '">' . esc_html__('Map', 'waymark') . '</a>';
+
+		// Title?
+		$map_title = ($map_title) ? $map_title : esc_html__('Map', 'waymark');
+
+		// Add link to Map
+		$desc_append .= '	<span class="waymark-map-link"><a href="' . get_permalink($map_id) . '">' . $map_title . '</a></span>';
+
+		// Is Map part of the waymark_collection taxonomy?
+		$collections = wp_get_post_terms($map_id, 'waymark_collection');
+
+		//If we have Collections
+		if (sizeof($collections)) {
+			//Sort by count ASC
+			usort($collections, function ($a, $b) {
+				return $a->count < $b->count;
+			});
+
+			// Add link to first
+			$desc_append .= '	<span class="waymark-collection-link"><a href="' . get_term_link($collections[0]) . '">' . $collections[0]->name . '</a></span>';
 		}
-		$desc_append .= '</b>.</div>';
+
+		$desc_append .= '</div>';
 
 		return self::set_map_data_property($map_data, 'description', $desc_append, true);
 	}