haiku.git

/*
 * Copyright 2009-2010 Haiku, Inc.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Ankur Sethi (get.me.ankur@gmail.com)
 *		Clemens Zeidler <haiku@clemens-zeidler.de>
 */

#include "BeaconSearcher.h"

#include <cstring>

#include <Alert.h>
#include <VolumeRoster.h>

#include "IndexServerPrivate.h"


using namespace lucene::document ;
using namespace lucene::search ;
using namespace lucene::index ;
using namespace lucene::queryParser ;


BeaconSearcher::BeaconSearcher()
{
	BVolumeRoster volumeRoster ;
	BVolume volume ;
	IndexSearcher *indexSearcher ;

	while(volumeRoster.GetNextVolume(&volume) == B_OK) {
		BPath indexPath = GetIndexPath(&volume);
		if(IndexReader::indexExists(indexPath.Path())) {
			indexSearcher = new IndexSearcher(indexPath.Path());
			fSearcherList.AddItem(indexSearcher);
		}
	}
}


BeaconSearcher::~BeaconSearcher()
{
	IndexSearcher *indexSearcher;
	while(fSearcherList.CountItems() > 0) {
		indexSearcher = (IndexSearcher*)fSearcherList.ItemAt(0);
		indexSearcher->close();
		delete indexSearcher;
		fSearcherList.RemoveItem((int32)0);
	}
}


BPath
BeaconSearcher::GetIndexPath(BVolume *volume)
{
	BDirectory dir;
	volume->GetRootDirectory(&dir);
	BPath path(&dir);
	path.Append(kIndexServerDirectory);
	path.Append("FullTextAnalyser");
	
	return path;
}


void
BeaconSearcher::Search(const char* stringQuery)
{
	// CLucene expects wide characters everywhere.
	int size = strlen(stringQuery) * sizeof(wchar_t) ;
	wchar_t *wStringQuery = new wchar_t[size] ;
	if (mbstowcs(wStringQuery, stringQuery, size) == -1)
		return ;

	IndexSearcher *indexSearcher ;
	Hits *hits ;
	Query *luceneQuery ;
	Document doc ;
	Field *field ;
	wchar_t *path ;
	
	/*
	luceneQuery = QueryParser::parse(wStringQuery, _T("contents"),
		&fStandardAnalyzer) ;

	hits = fMultiSearcher->search(luceneQuery) ;
	for(int j = 0 ; j < hits->length() ; j++) {
		doc = hits->doc(j) ;
		field = doc.getField(_T("path")) ;
		path = new wchar_t[B_PATH_NAME_LENGTH * sizeof(wchar_t)] ;
		wcscpy(path, field->stringValue()) ;
		fHits.AddItem(path) ;
	}*/
	
	for(int i = 0 ; (indexSearcher = (IndexSearcher*)fSearcherList.ItemAt(i))
		!= NULL ; i++) {
		luceneQuery = QueryParser::parse(wStringQuery, _T("contents"),
			&fStandardAnalyzer) ;

		hits = indexSearcher->search(luceneQuery) ;

		for(int j = 0 ; j < hits->length() ; j++) {
			doc = hits->doc(j) ;
			field = doc.getField(_T("path")) ;
			path = new wchar_t[B_PATH_NAME_LENGTH * sizeof(wchar_t)] ;
			wcscpy(path, field->stringValue()) ;
			fHits.AddItem(path) ;
		}
	}
}


wchar_t*
BeaconSearcher::GetNextHit()
{
	if(fHits.CountItems() != 0) {
		wchar_t* path = (wchar_t*)fHits.ItemAt(0) ;
		fHits.RemoveItem((int32)0) ;
		return path ;
	}
	
	return NULL ;
}