drupal/includes/tablesort.inc

<?php
// $Id$

function tablesort_init($header) {
  $ts = tablesort_get_order($header);
  $ts['sort'] = tablesort_get_sort($header);
  $ts['query_string'] = tablesort_get_querystring();
  return $ts;
}

function tablesort_pager() {
  $cgi = $_SERVER['REQUEST_METHOD'] == 'GET' ? $_GET : $_POST;
  unset($cgi['q'], $cgi['from']);
  return $cgi;
}

function tablesort_sql($header) {
  $ts = tablesort_init($header);
  if ($ts['field']) {
    $sql = check_query($ts['sql']);
    $sort = strtoupper(check_query($ts['sort']));
    return " ORDER BY $sql $sort";
  }
}

function tablesort_header($cell, $header, $ts) {
  // special formatting for the currently sorted column header
  if (is_array($cell) && $cell['field']) {
    if ($cell['data'] == $ts['name']) {
      $ts['sort'] = (($ts['sort'] == 'asc') ? 'desc' : 'asc');
      $cell['class'] = 'active';
      $image = '&nbsp;<img src="' . theme('image', 'arrow-' . $ts['sort'] . '.gif') . '" alt="'. t('sort icon') .'" />';
      $title = ($ts['sort'] == 'asc' ? t('sort ascending') : t('sort descending'));
    }
    else {
      // If the user clicks a different header, we want to sort ascending initially.
      $ts['sort'] = 'asc';
    }
    $title = t('sort by %s', array('%s' => $cell['data']));
    $cell['data'] = l($cell['data'] . $image, $_GET['q'], array('title' => $title), "sort=". $ts['sort']. "&amp;order=". urlencode($cell['data']). $ts['query_string']);

    unset($cell['field'], $cell['sort']);
  }
  return $cell;
}

function tablesort_cell($cell, $header, $ts, $i) {
  if ($header[$i]['data'] == $ts['name'] && $header[$i]['field']) {
    if (is_array($cell)) {
      $cell['class'] .= ' active';
    }
    else {
      $cell = array('data' => $cell, 'class' => 'active');
    }
  }
  return $cell;
}

function tablesort_get_querystring() {
  $cgi = $_SERVER['REQUEST_METHOD'] == 'GET' ? $_GET : $_POST;
  foreach ($cgi as $key => $val) {
    if ($key != 'order' && $key != 'sort' && $key != 'q') {
      $query_string .= '&amp;'. $key .'='. $val;
    }
  }
  return $query_string;
}

function tablesort_get_order($headers) {
  $order = $_GET['order'];
  foreach ($headers as $header) {
    if ($order == $header['data']) {
      return array('name' => $header['data'], 'sql' => $header['field']);
    }

    if ($header['sort'] == 'asc' || $header['sort'] == 'desc') {
      $default = array('name' => $header['data'], 'sql' => $header['field']);
    }
    $i++;
  }

  if ($default) {
    return $default;
  }
  else {
    // The first column specified is initial 'order by' field unless otherwise specified
    if (is_array($headers[0])) {
      return array('name' => $headers[0]['data'], 'sql' => $headers[0]['field']);
    }
    else {
      return array('name' => $headers[0]);
    }
  }
}

function tablesort_get_sort($headers) {
  if ($_GET['sort']) {
    return ($_GET['sort'] == 'desc') ? 'desc' : 'asc';
  }
  // user has not specified a sort.  check module for default and if none, use 'asc'
  else {
    foreach ($headers as $header) {
      if (is_array($header) && isset($header['sort'])) {
        return $header['sort'];
      }
    }
  }
  return 'asc';
}

?>