upei_roblib_search/targets/solr_site/roblib_search_solr_site.module

<?php

/**
 * @file
 * Implementation of Roblib search for searching several targets.
 */


/**
 * Implements hook_menu().
 */
function roblib_search_solr_site_menu() {
  $items = array();

  $items['admin/roblib_search/solr_site_search'] = array(
    'title' => 'Roblib Solr Search Target configuration',
    'description' => 'Configuration for the Roblib Solr site search target',
    'page callback' => 'drupal_get_form',
    'page arguments' => array('roblib_search_solr_site_config_form'),
    'access arguments' => array('access administration pages'),
    'type' => MENU_NORMAL_ITEM,
  );

  $items['roblib_search/solr_site/ajax/%'] = array(
    'title' => 'solr_site ajax',
    'page callback' => 'roblib_search_solr_site_ajax',
    'page arguments' => array(3),
    'type' => MENU_CALLBACK,
    'access arguments' => array('search roblib solr_site'),
  );

  return $items;
}

function roblib_search_solr_site_config_form($form, &$form_state) {
  /*
   * $url = variable_get('roblib_search_solr_site_url', 'http://137.149.200.52');
    $url_suffix = variable_get('roblib_search_solr_site_suffix', '/opac/extras/sru?version=1.1&operation=searchRetrieve&query=');
    $number_of_records = variable_get('roblib_search_solr_site_num_results', '5');
   */
  $form['roblib_search_solr_site_url'] = array(
    '#type' => 'textfield',
    '#title' => t('Solr url'),
    '#default_value' => variable_get('roblib_search_solr_site_url', 'http://localhost:8983/solr'),
    '#description' => t('The base Solr URL, for example http://localhost:8983/solr'),
    '#required' => TRUE,
  );
  $form['roblib_search_solr_site_num_results'] = array(
    '#type' => 'textfield',
    '#title' => t('Number of results to return'),
    '#default_value' => variable_get('roblib_search_solr_site_num_results', '5'),
    '#description' => t('The number of results to display in the Bento box'),
    '#required' => TRUE,
  );

  return system_settings_form($form);
}

/**
 * Implements hook_permission().
 */
function roblib_search_solr_site_permission() {
  return array(
    'search roblib solr_site' => array(
      'title' => t('Search the solr_site target'),
      'description' => t('Search all Roblib solr_site target. This permission exposes the search blocks and allows you to see search results.'),
    ),
    'administer roblib search_solr_site' => array(
      'title' => t('Administer Roblib Search Evergreen'),
      'description' => t('Administer settings for the Roblib solr_site search client.'),
    ),
  );
}

/**
 * Implements hook_theme().
 */
function roblib_search_solr_site_theme() {
  // set path
  $path = drupal_get_path('module', 'roblib_search_solr_site');
  $file = 'theme.inc';

  return array(
    // results page
    'roblib_search_solr_site' => array(
      'path' => $path . '/theme',
      'file' => $file,
      'template' => 'roblib-search-solr-site',
      'variables' => array('results' => NULL),
    )
  );
}

function roblib_search_solr_site_block_info() {

  $blocks['roblib_search_solr_site_results'] = array(
    // info: The name of the block.
    'info' => t('Roblib Solr Search Results block'),
    // Block caching options (per role, per user, etc.)
    'cache' => DRUPAL_CACHE_PER_ROLE, // default
  );

  return $blocks;
}

function roblib_search_solr_site_block_view($delta = '') {
  //The $delta parameter tells us which block is being requested.
  switch ($delta) {
    case 'roblib_search_solr_site_results':
      // The subject is displayed at the top of the block. Note that it
      // should be passed through t() for translation. The title configured
      // for the block using Drupal UI supercedes this one.
      $block['subject'] = t('Library Website');
      // The content of the block is typically generated by calling a custom
      // function.
      $block['content'] = theme('roblib_search_solr_site', array('results' => NULL));
      break;
  }
  return $block;
}

function roblib_search_solr_site_ajax($query) {
  $output = roblib_search_solr_site_get_results($query);
  print $output;
  exit();
}

/**
 *
 * @param string $query
 * @return string
 *   json
 */
function roblib_search_solr_site_get_results($query) {
  $url = roblib_search_solr_site_get_search_url($query);
  $results = drupal_http_request($url);
  if($results->code != '200'){
    return "";
  }
  return $results->data;
}

/**
 * @param $query
 * @return string
 */
function roblib_search_solr_site_get_search_url($query){
  $solr_url = variable_get('roblib_search_solr_site_url', 'http://localhost:8983/solr');
  $num_results = variable_get('roblib_search_solr_site_num_results', '5');
  $data = array(
    'wt' => 'json',
    'q' => $query ,
    'fq' => '-bundle:guide OR -bundle:bestbet OR -bundle:database',
    'rows' => $num_results,
  );
  $url = url($solr_url . '/select', array('query' => $data));
  return $url;
}

/**
 * Implements hook_help().
 */
function roblib_search_solr_site_help($path, $arg) {
  switch ($path) {
    case 'admin/help#roblib_search_solr_site':
      return t(
        '<p>
provides a target for the Roblib search module
</p>'
      );
  }
}