diff options
Diffstat (limited to 'solr_search.module')
| -rwxr-xr-x | solr_search.module | 280 |
1 files changed, 280 insertions, 0 deletions
diff --git a/solr_search.module b/solr_search.module new file mode 100755 index 0000000..855134a --- /dev/null +++ b/solr_search.module @@ -0,0 +1,280 @@ +<?php + + include "bootstrap.inc"; + + function solr_search_permission() { + +return array( + "administer solr-search" => array( + "title" => t("administer solr-search"), + "description" => t("administer solr-search.") + ) ); + } + + function solr_search_menu() { + $items = array(); + + // solr search admin settings + $items['admin/settings/solr_search'] = array( + 'title' => 'TBC Solr-Search admin settings', + 'page callback' => 'drupal_get_form', + 'page arguments' => array('tbc_solr_search_settings_form'), + 'access arguments' => array('administer solr-search'), + 'type' => MENU_NORMAL_ITEM, + 'file' => 'settings.inc', + ); + + $items['tbc_solr_search'] = array( + 'title' => 'Search within Textbook Companion Examples', + 'page callback' => 'drupal_get_form', + 'page arguments' => array('tbc_solr_search_form'), + 'access callback' => TRUE, + 'type' => MENU_NORMAL_ITEM, + ); + + $items['add_all_books_to_solr'] = array( + 'title' => 'Add books to Solr', + 'page callback' => 'drupal_get_form', + 'page arguments' => array('add_all_books_to_solr_form'), + 'access arguments' => array('administer solr-search'), + 'type' => MENU_NORMAL_ITEM, + ); + + return $items; + } + + function tbc_solr_search_form($form,&$form_state) { + + + $form['tbc_search_key'] = array( + '#type' => 'textfield', + '#title' => t('Enter your keyword (Note: special characters ^ ( ) { } \" \' [ ] are not allowed)'), + '#required' => TRUE, + ); + $form['submit'] = array( + '#type' => 'submit', + '#value' => t('Search'), + '#ajax' => array( + 'callback' => 'ajax_solr_search_result_callback', + ), + ); + $form['tbc_search_result'] = array( + '#type' => 'item', + '#markup' => '<div id="ajax-search-result-replace"></div>', + + ); + + return $form; + } + + function tbc_solr_search_form_validate($form, &$form_state) { + + if(preg_match('/[^a-zA-Z0-9]/', $form_state['values']['tbc_search_key'])) { + form_set_error('tbc_search_key', t('Special characters not allowed')); + } + return; + } + + function tbc_solr_search_form_submit($form, &$form_state) { + + //drupal_goto('tbc_solr_search', array('key' => $_POST['tbc_search_key'])); + // ajax_solr_search_result_callback($form,$form_state); + } + function add_all_books_to_solr_form($form, &$form_state) { + $form['catid'] = array( + '#type' => 'select', + '#title' => t('Category'), + '#options' => array(0 => 'Please select', + 1 => 'Fluid Mechanics', + 2 => 'Control Theory & Control Systems', + 3 => 'Chemical Engineering', + 4 => 'Thermodynamics', + 5 => 'Mechanical Engineering', + 6 => 'Signal Processing', + 7 => 'Digital Communications', + 8 => 'Electrical Technology', + 9 => 'Mathematics & Pure Science', + 10 => 'Analog Electronics', + 11 => 'Digital Electronics', + 12 => 'Computer Programming', + 13 => 'Others' + ), + ); + + $form['submit'] = array( + '#type' => 'submit', + '#value' => t('Add to Solr'), + // '#suffix' => $output, + ); + + return $form; + } + + function add_all_books_to_solr_form_submit($form, &$form_state) { + + global $user; + ini_set('max_execution_time', 500000); + $root_path = variable_get('tbc_example_file_path', ''); + //var_dump($root_path);die; +$options = array( + 'hostname' => SOLR_SERVER_HOSTNAME, + 'port' => SOLR_SERVER_PORT, + 'login' => SOLR_SERVER_USERNAME, + 'password' => SOLR_SERVER_PASSWORD, + + ); + + $client = new SolrClient($options); + + $book_q = db_query("SELECT * FROM {textbook_companion_preference} WHERE approval_status=1 AND category= :catid AND proposal_id IN (SELECT id from textbook_companion_proposal where proposal_status=3)",array(":catid" => $form_state['values']['catid'])); + + + while($book_data = $book_q->fetchObject()) { + + $chapter_q = db_query("SELECT * FROM {textbook_companion_chapter} WHERE preference_id = :pref_id",array(":pref_id" =>$book_data->id)); + + while($chapter_row = $chapter_q->fetchObject()) { + //$example_q = db_query("SELECT * FROM {textbook_companion_example} WHERE chapter_id = %d AND approval_status = 1", $chapter_row->id); + $query = db_select('textbook_companion_example'); + $query->fields('textbook_companion_example'); + $query->condition('chapter_id', $chapter_row->id); + $query->condition('approval_status', 1); + $example_q = $query->execute(); + while($example_row = $example_q->fetchObject()) { + //$example_files_q = db_query("SELECT * FROM {textbook_companion_example_files} WHERE example_id = %d", $example_row->id); + $query = db_select('textbook_companion_example_files'); + $query->fields('textbook_companion_example_files'); + $query->condition('example_id', $example_row->id); + $example_files_q = $query->execute(); + //$example_dependency_files_q = db_query("SELECT * FROM {textbook_companion_example_dependency} WHERE example_id = %d", $example_row->id); + $query = db_select('textbook_companion_example_dependency'); + $query->fields('textbook_companion_example_dependency'); + $query->condition('example_id', $example_row->id); + $example_dependency_files_q = $query->execute(); + $extensions = array('sce', 'sci'); + $example_data = ''; + while($example_files_row = $example_files_q->fetchObject()) { + if(in_array(end(explode('.', $example_files_row->filepath)), $extensions)) { + // echo $example_files_row->filepath."<br />"; + $my_file = $root_path.$example_files_row->filepath; + $handle = fopen($my_file, 'r'); + $example_data = fread($handle,filesize($my_file)); + fclose($handle); + } + } + /* dependency files */ + $dependency_data = ''; + while($example_dependency_files_row =$example_dependency_files_q->fetchObject()) { + //$dependency_file_data = db_fetch_object(db_query("SELECT * FROM {textbook_companion_dependency_files} WHERE id = %d LIMIT 1", $example_dependency_files_row->dependency_id)); + $query = db_select('textbook_companion_dependency_files'); + $query->fields('textbook_companion_dependency_files'); + $query->condition('id', $example_dependency_files_row->dependency_id); + $query->range(0, 1); + $result = $query->execute(); + $dependency_file_data = $result->fetchObject(); + if($dependency_file_data) { + if(in_array(end(explode('.', $dependency_file_data->filepath)), $extensions)) { + // echo $dependency_file_data->filepath."<br />"; + $my_file = $root_path.$dependency_file_data->filepath; + $handle = fopen($my_file, 'r'); + $dependency_data = fread($handle,filesize($my_file)); + fclose($handle); + } + } + } + $doc = new SolrInputDocument(); + $doc->addField('id', $example_row->id); + $doc->addField('chapter_id', $chapter_row->id); + $doc->addField('book_id', $book_data->id); + $doc->addField('title', mb_convert_encoding($book_data->book, "UTF-8")); + $doc->addField('author', mb_convert_encoding($book_data->author, "UTF-8")); + $doc->addField('chapter', mb_convert_encoding(($chapter_row->number.'. '.$chapter_row->name), "UTF-8")); + $doc->addField('example', mb_convert_encoding(($example_row->number.'. '.$example_row->caption), "UTF-8")); + $doc->addField('content', mb_convert_encoding($dependency_data, "UTF-8").mb_convert_encoding($example_data, "UTF-8")); + $doc->addField('added_by', $user->name); + $doc->addField('timestamp', date('Y-m-d H:i:s')); + $updateResponse = $client->addDocument($doc,false); + $client->commit(); + } + } + } + $cats = array( + 1 => 'Fluid Mechanics', + 2 => 'Control Theory & Control Systems', + 3 => 'Chemical Engineering', + 4 => 'Thermodynamics', + 5 => 'Mechanical Engineering', + 6 => 'Signal Processing', + 7 => 'Digital Communications', + 8 => 'Electrical Technology', + 9 => 'Mathematics & Pure Science', + 10 => 'Analog Electronics', + 11 => 'Digital Electronics', + 12 => 'Computer Programming', + 13 => 'Others' + ); + drupal_set_message('Books under "'.$cats[$_POST['catid']].'" are updated to solr'); + } + + function ajax_solr_search_result_callback($form, $form_state){ +$output = ''; + //$key = ''; + //var_dump($form_state['values']['tbc_search_key']);die; + $path = drupal_get_path('module', 'solr_search'); + drupal_add_css($path.'/solr_search.css', 'module'); + drupal_add_js($path.'/solr_search.js', 'module'); + if(isset($form_state['values']['tbc_search_key'])) { + $key = $form_state['values']['tbc_search_key']; + $output = ''; + $numfound = ''; + if($key != '') { + $options = array( + 'hostname' => SOLR_SERVER_HOSTNAME, + 'login' => SOLR_SERVER_USERNAME, + 'password' => SOLR_SERVER_PASSWORD, + 'port' => SOLR_SERVER_PORT, + ); + $client = new SolrClient($options); + + $query = new SolrQuery(); + + $query->setQuery($key.'*'); + + $query->setStart(0); + + $query->setRows(variable_get('tbc_solr_max_results', '')); + + $query->addField('id')->addField('book_id')->addField('title')->addField('author')->addField('chapter')->addField('example')->addField('content')->addSortField('title', 0); + + $query_response = $client->query($query); + + $response = $query_response->getResponse(); + + $sno = 0; + $output = '<p><b>'.$response['response']->numFound.' result(s) found</b></p>'; + if($response['response']->numFound > 0) { + $output .= '<div class="search-result"><div id="topcontrol" title="Scroll Back to Top"><a href="#top"></a></div><ol>'; + $download_link= ''; + foreach($response['response']->docs as $doc) { + $sno++; + $output .= '<li>'; + $output .= '<p><b>Book: </b>'.$doc->title.' ('.$doc->author.')</p>'; + $output .= '<p><b>Chapter: </b>'.$doc->chapter.'</p>'; + $output .= '<p><b>Example: </b>'.$doc->example.'</p>'; + $output .= '<p><b>Links: </b> <a href="'.base_path().'textbook_companion/generate_book/'.$doc->book_id.'"><span style="font-weight:bold;font-size:16px;">⤵</span> Download code for the entire book</a> <a href="https://cloud.scilab.in/index?eid='.$doc->id.'" target="_blank"><span style="font-weight:bold;font-size:20px;">»</span> View code for this example</a></p>'; + $output .= '</li>'; + } + $output .= '<ol></div>'; + } + }else { + $output .= '<p><b>0 result(s) found</b></p>'; + } + } + +$commands[] = ajax_command_html("#ajax-search-result-replace", $output); + +return array('#type' => 'ajax', '#commands' => $commands); + +} + + |