l10ntools/source/help/HelpSearch.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * Version: MPL 1.1 / GPLv3+ / LGPLv3+
   4  *
   5  * The contents of this file are subject to the Mozilla Public License Version
   6  * 1.1 (the "License"); you may not use this file except in compliance with
   7  * the License or as specified alternatively below. You may obtain a copy of
   8  * the License at http://www.mozilla.org/MPL/
   9  *
  10  * Software distributed under the License is distributed on an "AS IS" basis,
  11  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  12  * for the specific language governing rights and limitations under the
  13  * License.
  14  *
  15  * Major Contributor(s):
  16  * Copyright (C) 2012 Gert van Valkenhoef <g.h.m.van.valkenhoef@rug.nl>
  17  *  (initial developer)
  18  *
  19  * All Rights Reserved.
  20  *
  21  * For minor contributions see the git repository.
  22  *
  23  * Alternatively, the contents of this file may be used under the terms of
  24  * either the GNU General Public License Version 3 or later (the "GPLv3+"), or
  25  * the GNU Lesser General Public License Version 3 or later (the "LGPLv3+"),
  26  * in which case the provisions of the GPLv3+ or the LGPLv3+ are applicable
  27  * instead of those above.
  28  */
  29
  30 #include <l10ntools/HelpSearch.hxx>
  31 #include <osl/file.hxx>
  32 #include <osl/thread.hxx>
  33
  34 #include "LuceneHelper.hxx"
  35
  36 HelpSearch::HelpSearch(rtl::OUString const &lang, rtl::OUString const &indexDir)
  37     : d_lang(lang)
  38 {
  39     rtl::OUString ustrSystemPath;
  40     osl::File::getSystemPathFromFileURL(indexDir, ustrSystemPath);
  41     d_indexDir = rtl::OUStringToOString(ustrSystemPath, osl_getThreadTextEncoding());
  42 }
  43
  44 bool HelpSearch::query(rtl::OUString const &queryStr, bool captionOnly,
  45         std::vector<rtl::OUString> &rDocuments, std::vector<float> &rScores) {
  46
  47     lucene::index::IndexReader *reader = lucene::index::IndexReader::open(d_indexDir.getStr());
  48     lucene::search::IndexSearcher searcher(reader);
  49
  50     TCHAR captionField[] = L"caption";
  51     TCHAR contentField[] = L"content";
  52     TCHAR *field = captionOnly ? captionField : contentField;
  53
  54     bool isWildcard = queryStr[queryStr.getLength() - 1] == L'*';
  55     std::vector<TCHAR> aQueryStr(OUStringToTCHARVec(queryStr));
  56     lucene::search::Query *pQuery;
  57     if (isWildcard)
  58         pQuery = _CLNEW lucene::search::WildcardQuery(_CLNEW lucene::index::Term(field, &aQueryStr[0]));
  59     else
  60         pQuery = _CLNEW lucene::search::TermQuery(_CLNEW lucene::index::Term(field, &aQueryStr[0]));
  61
  62     lucene::search::Hits *hits = searcher.search(pQuery);
  63     for (unsigned i = 0; i < hits->length(); ++i) {
  64         lucene::document::Document &doc = hits->doc(i); // Document* belongs to Hits.
  65         wchar_t const *path = doc.get(L"path");
  66         rDocuments.push_back(TCHARArrayToOUString(path != 0 ? path : L""));
  67         rScores.push_back(hits->score(i));
  68     }
  69
  70     _CLDELETE(hits);
  71     _CLDELETE(pQuery);
  72
  73     reader->close();
  74     _CLDELETE(reader);
  75
  76     return true;
  77 }
  78
  79 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */