source: cpc/trunk/project/apps/frontend/modules/solr/actions/actions.class.php @ 1317

<?php

/**
 * solr actions.
 *
 * @package    cpc
 * @subpackage solr
 * @author     Your name here
 * @version    SVN: $Id: actions.class.php 12479 2008-10-31 10:54:40Z fabien $
 */
class solrActions extends sfActions
{

  private function getLink($obj) {
    sfProjectConfiguration::getActive()->loadHelpers(array('Url'));
    switch(get_class($obj)) {
    case 'Intervention':
      return url_for('@interventions_seance?seance='.$obj->getSeance()->id).'#inter_'.$obj->getMd5();
    case 'QuestionEcrite':
      return url_for('@question_numero?numero='.$obj->numero);
    case 'Amendement':
      return url_for('@amendement?loi='.$obj->loi.'&numero='.$obj->numero);
    case 'Parlementaire':
      return url_for('@parlementaire?slug='.$obj->slug);
    case 'Commentaire':
      return $obj->lien;
    }
  }
  private function getPersonne($obj) {
    sfProjectConfiguration::getActive()->loadHelpers(array('Url'));
    switch(get_class($obj)) {
    case 'Intervention':
      return $obj->getIntervenant()->getNom();
    case 'QuestionEcrite':
      return $obj->getParlementaire()->getNom();
    case 'Amendement':
      return '';
    case 'Parlementaire':
      return '';
    case 'Commentaire':
      return $obj->getCitoyen()->getLogin();
    }
  }
  private function getPhoto($obj) {
    sfProjectConfiguration::getActive()->loadHelpers(array('Url'));
    switch(get_class($obj)) {
    case 'Intervention':
      if ($obj->getParlementaire())
        return url_for('@resized_photo_parlementaire?height=70&slug='.$obj->getIntervenant()->getSlug());
      return '';
    case 'QuestionEcrite':
      return url_for('@resized_photo_parlementaire?height=70&slug='.$obj->getParlementaire()->getSlug());
    case 'Amendement':
      return '';
    case 'Parlementaire':
      return url_for('@resized_photo_parlementaire?height=70&slug='.$obj->getSlug());
    case 'Commentaire':
      return url_for('@photo_citoyen?slug='.$obj->getCitoyen()->getSlug());
    }
  }
 /**
  * Executes index action
  *
  * @param sfRequest $request A request object
  */
  public function executeSearch(sfWebRequest $request)
  {
    if ($search = $request->getParameter('search')) {
      return $this->redirect('solr/search?query='.$search);
    }
    $this->query = $request->getParameter('query');
    
    $query = preg_replace('/\*/', '', $this->query);

    if (!strlen($query)) {
      $query = '*';
    }

    $nb = 20;
    $deb = ($request->getParameter('page', 1) - 1) * $nb ;
    $fq = '';
    $this->facet = array();

    $this->selected = array();
    if ($on = $request->getParameter('object_name')) {
      $this->selected['object_name'][$on] = 1;
      $fq .= " object_name:$on";
    }
    if ($tags = $request->getParameter('tag')) {
      foreach(explode(',', $tags) as $tag) {
        $this->selected['tag'][$tag] = 1;
        $fq .= ' tag:"'.$tag.'"';
      }
    }
    //Récupère les résultats auprès de SolR
    $s = new SolrConnector();
    $params = array('hl'=>'true', 'fl' => 'id,object_id,object_name', 'hl.fragsize'=>500, "facet"=>"true", "facet.field"=>array("object_name","tag"), "facet.date" => "date", "facet.date.start"=>"2007-05-01T00:00:00Z", "facet.date.end"=>"NOW", "facet.date.gap"=>"+1MONTH", 'fq' => $fq);
    $this->sort_type = 'pertinence';
    if ($request->getParameter('sort')) {
      $params['sort'] = "date desc";
      $this->sort_type = 'date';
    }
    if ($date = $request->getParameter('date')) {
      $dates = explode(',', $date);
      $date = array_pop($dates);
      $period = 'MONTH';
      if (count($dates) == 1)
        $period = 'DAY';
      $query .= ' date:['.$date.' TO '.$date.'+1'.$period.']';
      $this->selected['date'][$date] = 1;
      $params['facet.date.start']=$date;
      $params['facet.date.end'] = $date.'+1'.$period;
      $params['facet.date.gap'] = '+1DAY';
    }
    $results = $s->search($query, $params, $deb, $nb);
    //Reconstitut les résultats
    $this->results = $results['response'];
    for($i = 0 ; $i < count($this->results['docs']) ; $i++) {
      $res = $this->results['docs'][$i];
      $obj = Doctrine::getTable($res['object_name'])->find($res['object_id']);
      $this->results['docs'][$i]['link'] = $this->getLink($obj);
      $this->results['docs'][$i]['photo'] = $this->getPhoto($obj);
      $this->results['docs'][$i]['titre'] = $obj->getTitre();
      $this->results['docs'][$i]['personne'] = $this->getPersonne($obj);
      $this->results['docs'][$i]['highlighting'] = preg_replace('/^'.$this->results['docs'][$i]['personne'].'/', '', implode('...', $results['highlighting'][$res['id']]['text']));
      
    }
    $this->results['end'] = $deb + $nb;
    $this->results['page'] = $deb/$nb + 1;
    if ($this->results['end'] > $this->results['numFound']) {
      $this->results['end'] = $this->results['numFound'] + 1;
    }

    //Prépare les facets
    $this->facet['parlementaire']['prefix'] = 'parlementaire=';
    $this->facet['parlementaire']['facet_field'] = 'tag';
    $this->facet['parlementaire']['name'] = 'Parlementaire';

    $this->facet['type']['prefix'] = '';
    $this->facet['type']['facet_field'] = 'object_name';
    $this->facet['type']['name'] = 'Types';
    $this->facet['type']['values'] = $results['facet_counts']['facet_fields']['object_name'];

    $tags = $results['facet_counts']['facet_fields']['tag'];
    $this->facet['tag']['prefix'] = '';
    $this->facet['tag']['facet_field'] = 'tag';
    $this->facet['tag']['name'] = 'Tags';
    foreach($tags as $tag => $nb ) {
      if (!$nb)
        continue;
      if (!preg_match('/=/', $tag))
        $this->facet['tag']['values'][$tag] = $nb;
      if (preg_match('/^parlementaire=(.*)/', $tag, $matches)) {
        $this->facet['parlementaire']['values'][$matches[1]] = $nb;
      }
    }
    if (!$results['response']['numFound']) {
      $this->setTemplate('noresults');
    }else{
      $this->fdates = array();
      $this->fdates['max'] = 1;
      foreach($results['facet_counts']['facet_dates']['date'] as $date => $nb) {
        if (preg_match('/^20/', $date)) {
          $pc = $nb/$results['response']['numFound'];
          $this->fdates['values'][$date] = array('nb' => $nb, 'pc' => $pc);
          if ($this->fdates['max'] < $pc) {
            $this->fdates['max'] = $pc;
          }
        }
      }
    }
  }
}