Index: apachesolr_search.module
===================================================================
RCS file: /cvs/drupal-contrib/contributions/modules/apachesolr/apachesolr_search.module,v
retrieving revision 1.1.2.6.2.90
diff -u -p -r1.1.2.6.2.90 apachesolr_search.module
--- apachesolr_search.module	6 May 2009 01:15:35 -0000	1.1.2.6.2.90
+++ apachesolr_search.module	15 May 2009 16:24:33 -0000
@@ -444,51 +444,91 @@ function apachesolr_search_block($op = '
         $initial_limits = variable_get('apachesolr_facet_query_initial_limits', array());
         $limit_default = variable_get('apachesolr_facet_query_initial_limit_default', 10);
 
+        $reflect_hierarchy = variable_get('apachesolr_facet_reflect_hierarchy', array());
+
         // Handle taxonomy vocabulary facets
         if ((strpos($delta, 'im_vid_') === 0) && module_exists('taxonomy')) {
 
           if (is_object($response->facet_counts->facet_fields->$delta)) {
-            $contains_active = FALSE;
-            $terms = array();
+            $vid = substr($delta, 7);
+            if (is_numeric($vid)) {
+              $vocab = taxonomy_vocabulary_load($vid);
+              $terms = array();
+
+              foreach ($response->facet_counts->facet_fields->$delta as $tid => $count) {
+                if ($tid == '_empty_') {
+                  // TODO - for now we don't handle facet missing.
+                  continue;
+                }
+                $unclick_link = '';
+                unset($active);
 
-            foreach ($response->facet_counts->facet_fields->$delta as $tid => $count) {
-              if ($tid == '_empty_') {
-                // TODO - for now we don't handle facet missing.
-                continue;
-              }
-              $unclick_link = '';
-              unset($active);
-              $term = taxonomy_get_term($tid);
-              $new_query = clone $query;
-              if ($active = $query->has_filter('tid', $tid)) {
-                $contains_active = TRUE;
-                $new_query->remove_filter('tid', $term->tid);
-                $path = $new_query->get_path();
-                $querystring = $new_query->get_url_querystring();
-                $unclick_link = theme('apachesolr_unclick_link', $path, $querystring);
-              }
-              else {
-                $new_query->add_filter('tid', $term->tid);
-                $path = $new_query->get_path();
-                $querystring = $new_query->get_url_querystring();
+                $parents = array();
+                if ($reflect_hierarchy['apachesolr_search'][$delta]) {
+                  $parents = taxonomy_get_parents_all($tid);
+                  if (!empty($parents[1]) && !$query->has_filter('tid', $parents[1]->tid)) {
+                    continue;
+                  }
+
+                }
+                else if ($term = taxonomy_get_term($tid)) {
+                  $parents[] = $term;
+                }
+
+                $new_query = clone $query;
+
+                $active = $query->has_filter('tid', $tid);
+                if ($active) {
+                  $unclick_link = apachesolr_get_unclick_link($query, 'tid', $tid, $reflect_hierarchy['apachesolr_search'][$delta] ? $vid : FALSE);
+                }
+                else {
+                  $new_query->add_filter('tid', $tid);
+                  $path = $new_query->get_path();
+                  $querystring = $new_query->get_url_querystring();
+                }
+
+                // if numdocs == 1 and !active, don't add.
+                if ($response->response->numFound > 1 || $active) {
+                  $sort_key = '';
+                  $term = NULL;
+                  foreach (array_reverse($parents) as $term) {
+                    $count = $response->facet_counts->facet_fields->$delta->{$term->tid};
+                    $countsort = $count == 0 ? '' : 1 / $count;
+                    $countsort = $query->has_filter('tid', $term->tid) ? $countsort : 1 + $countsort;
+                    $sort_key .= ($sort_key ? '|' : '') . $countsort . ':' . $term->name; 
+                  }
+                  $terms[$term->vid][$sort_key] = theme('apachesolr_facet_item', $term->name, $count, $path, $querystring, $active, $unclick_link, $response->response->numFound, array());
+                }
               }
-              $countsort = $count == 0 ? '' : 1 / $count;
-              // if numdocs == 1 and !active, don't add.
-              if ($response->response->numFound > 1 || $active) {
-                $terms[$term->vid][$active ? $countsort . $term->name : 1 + $countsort . $term->name] = theme('apachesolr_facet_item', $term->name, $count, $path, $querystring, $active, $unclick_link, $response->response->numFound);
+
+              // Process all terms into an item list
+              if (is_array($terms) && isset($terms[$vid]) && is_array($terms[$vid])) {
+                ksort($terms[$vid]);
+                
+                // Create array of nested items
+                unset($items);
+                foreach ($terms[$vid] as $key => $value) {
+                  // Sample $key = "0.017:Fruit|0.017:Apples|0.017:Red apples"
+                  $parts = explode("|", $key);
+                  $current_part = array_pop($parts);
+                  $previous_parts = $parts;
+                  if ($previous_parts) {
+                    $keys = implode("']['children']['", $previous_parts);
+                    $eval = "\$items['{$keys}']['children']['{$current_part}'] = array('data' => \$value);";
+                  } else {
+                    $eval = "\$items['{$current_part}'] = array('data' => \$value);";
+                  }
+                  eval($eval);
+                }                
+                
+                $limit = isset($initial_limits['apachesolr_search'][$delta]) ? $initial_limits['apachesolr_search'][$delta] : $limit_default;
+                return array(
+                  'subject' => t('Filter by @name', array('@name' => $vocab->name)),
+                  'content' => theme('apachesolr_facet_list', $items, $limit),
+                );
               }
             }
           }
-          $vid = substr($delta, 7);
-          $vocab = taxonomy_vocabulary_load($vid);
-          if (is_numeric($vid) && is_array($terms) && isset($terms[$vid]) && is_array($terms[$vid])) {
-            ksort($terms[$vid]);
-            $limit = isset($initial_limits['apachesolr_search'][$delta]) ? $initial_limits['apachesolr_search'][$delta] : $limit_default;
-            return array(
-              'subject' => t('Filter by @name', array('@name' => $vocab->name)),
-              'content' => theme('apachesolr_facet_list', $terms[$vid], $limit),
-            );
-          }
           return;
         }
 
@@ -503,11 +543,14 @@ function apachesolr_search_block($op = '
             $links[] = apachesolr_l($query->get_query_basic(), $path, $options);
             foreach($fields as $field) {
               if ($field['#name']) {
-                $new_query = clone $query;
-                $new_query->remove_filter($field['#name'], $field['#value']);
-                $path = $new_query->get_path();
-                $querystring = $new_query->get_url_querystring();
-                $unclick_link = theme('apachesolr_unclick_link', $path, $querystring);
+                $vid = FALSE;
+                if (module_exists('taxonomy') && 'tid' == $field['#name']) {
+                  $term = taxonomy_get_term($field['#value']);
+                  if ($reflect_hierarchy['apachesolr_search']['im_vid_'.$term->vid]) {
+                    $vid = $term->vid;
+                  }
+                }
+                $unclick_link = apachesolr_get_unclick_link($query, $field['#name'], $field['#value'], $vid);
                 if (! $fielddisplay = theme("apachesolr_breadcrumb_". $field['#name'], $field['#value'])) {
                   $fielddisplay = $field['#value'];
                 }
@@ -557,6 +600,30 @@ function apachesolr_search_block($op = '
   }
 }
 
+function apachesolr_get_unclick_link(&$query, $field_name, $field_value, $vid = FALSE) {
+  static $unclick_links = array();
+
+  if (empty($unclick_links[$field_name][$field_value])) {
+    $new_query = clone $query;
+
+    $new_query->remove_filter($field_name, $field_value);
+
+    if ($vid && 'tid' == $field_name) {
+      $childs = taxonomy_get_tree($vid, $field_value);
+      foreach ($childs as $child) {
+        $new_query->remove_filter('tid', $child->tid);
+      }
+    }
+
+    $path = 'search/' . arg(1) . '/' . $new_query->get_query_basic();
+    $querystring = $new_query->get_url_querystring();
+
+    $unclick_links[$field_name][$field_value] = theme('apachesolr_unclick_link', $path, $querystring);
+  }
+
+  return $unclick_links[$field_name][$field_value];
+}
+
 /**
  * Callback function for the 'Filter by book' facet block.
  */
@@ -761,3 +828,5 @@ function theme_apachesolr_currentsearch(
 function theme_apachesolr_search_snippets($doc, $snippets) {
   return implode(' ... ', $snippets);
 }
+
+     
\ No newline at end of file