AutoCompletionManager.java

/**   
 * License Agreement for OpenSearchServer
 *
 * Copyright (C) 2012-2013 Emmanuel Keller / Jaeksoft
 * 
 * http://www.open-search-server.com
 * 
 * This file is part of OpenSearchServer.
 *
 * OpenSearchServer is free software: you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 * OpenSearchServer is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with OpenSearchServer. 
 *  If not, see <http://www.gnu.org/licenses/>.
 **/

package com.jaeksoft.searchlib.autocompletion;

import java.io.Closeable;
import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.util.Collection;
import java.util.InvalidPropertiesFormatException;
import java.util.List;
import java.util.TreeSet;

import org.apache.commons.io.filefilter.SuffixFileFilter;

import com.jaeksoft.searchlib.Logging;
import com.jaeksoft.searchlib.SearchLibException;
import com.jaeksoft.searchlib.config.Config;
import com.jaeksoft.searchlib.util.IOUtils;
import com.jaeksoft.searchlib.util.ReadWriteLock;

public class AutoCompletionManager implements Closeable {

	private final ReadWriteLock rwl = new ReadWriteLock();

	private final Config config;

	private TreeSet<AutoCompletionItem> autoCompItems;

	private final static String autoCompletionSubDirectory = "autocompletion";

	private final File autoCompletionDirectory;

	public AutoCompletionManager(Config config) throws SearchLibException,
			IOException {
		this.config = config;
		autoCompletionDirectory = new File(config.getDirectory(),
				autoCompletionSubDirectory);
		if (!autoCompletionDirectory.exists())
			autoCompletionDirectory.mkdir();
		autoCompItems = new TreeSet<AutoCompletionItem>();
		File[] autoCompFiles = autoCompletionDirectory
				.listFiles((FileFilter) new SuffixFileFilter(".xml"));
		if (autoCompFiles == null)
			return;
		for (File autoCompFile : autoCompFiles) {
			try {
				add(new AutoCompletionItem(config, autoCompFile));
			} catch (InvalidPropertiesFormatException e) {
				Logging.warn("Invalid autocompletion file: "
						+ autoCompFile.getAbsolutePath());
			}
		}
	}

	@Override
	public void close() {
		rwl.w.lock();
		try {
			IOUtils.close(autoCompItems);
		} finally {
			rwl.w.unlock();
		}
	}

	public Collection<AutoCompletionItem> getItems() {
		rwl.r.lock();
		try {
			return autoCompItems;
		} finally {
			rwl.r.unlock();
		}
	}

	private AutoCompletionItem find(AutoCompletionItem searchItem) {
		AutoCompletionItem foundItem = autoCompItems.ceiling(searchItem);
		if (foundItem == null)
			return null;
		return searchItem.compareTo(foundItem) == 0 ? foundItem : null;
	}

	public AutoCompletionItem getItem(String name) throws SearchLibException {
		rwl.r.lock();
		try {
			return find(new AutoCompletionItem(config, name));
		} finally {
			rwl.r.unlock();
		}
	}

	public void add(AutoCompletionItem currentItem) throws SearchLibException {
		rwl.w.lock();
		try {
			if (find(currentItem) != null)
				throw new SearchLibException(
						"This name is already taken by another item");
			autoCompItems.add(currentItem);
			currentItem.save();
		} finally {
			rwl.w.unlock();
		}
	}

	public File getDirectory() {
		return autoCompletionDirectory;
	}

	public void delete(AutoCompletionItem selectedItem) throws IOException {
		rwl.w.lock();
		try {
			selectedItem.delete();
			autoCompItems.remove(selectedItem);
		} finally {
			rwl.w.unlock();
		}
	}

	public void toNameList(List<String> nameList) {
		rwl.r.lock();
		try {
			for (AutoCompletionItem item : autoCompItems)
				nameList.add(item.getName());
		} finally {
			rwl.r.unlock();
		}
	}
}