AutoCompletionComposer.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.web.controller.schema;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.InvalidPropertiesFormatException;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;

import org.zkoss.bind.annotation.AfterCompose;
import org.zkoss.bind.annotation.BindingParam;
import org.zkoss.bind.annotation.Command;
import org.zkoss.bind.annotation.ContextParam;
import org.zkoss.bind.annotation.ContextType;
import org.zkoss.bind.annotation.NotifyChange;
import org.zkoss.zk.ui.event.InputEvent;
import org.zkoss.zul.ListModel;
import org.zkoss.zul.ListModelArray;
import org.zkoss.zul.Messagebox;

import com.jaeksoft.searchlib.Client;
import com.jaeksoft.searchlib.SearchLibException;
import com.jaeksoft.searchlib.autocompletion.AutoCompletionBuildThread;
import com.jaeksoft.searchlib.autocompletion.AutoCompletionItem;
import com.jaeksoft.searchlib.autocompletion.AutoCompletionManager;
import com.jaeksoft.searchlib.request.RequestTypeEnum;
import com.jaeksoft.searchlib.result.AbstractResultSearch;
import com.jaeksoft.searchlib.result.ResultDocument;
import com.jaeksoft.searchlib.web.controller.AlertController;
import com.jaeksoft.searchlib.web.controller.CommonController;

@AfterCompose(superclass = true)
public class AutoCompletionComposer extends CommonController {

	private AutoCompletionItem selectedItem;

	private String name;

	private int rows;

	private String selectedField;

	private final Set<String> fields;

	private String searchRequest;

	private ListModel<String> comboList;

	public AutoCompletionComposer() throws SearchLibException {
		super();
		fields = new TreeSet<String>();
	}

	public List<String> getFieldList() throws SearchLibException {
		synchronized (this) {
			Client client = getClient();
			if (client == null)
				return null;
			List<String> fieldList = new ArrayList<String>();
			client.getSchema().getFieldList().toNameList(fieldList);
			return fieldList;
		}
	}

	public Collection<AutoCompletionItem> getAutoCompletionItems()
			throws SearchLibException {
		Client client = getClient();
		if (client == null)
			return null;
		return client.getAutoCompletionManager().getItems();
	}

	public boolean isRunning() throws SearchLibException {
		Client client = getClient();
		if (client == null)
			return false;
		for (AutoCompletionItem item : client.getAutoCompletionManager()
				.getItems()) {
			AutoCompletionBuildThread thread = item.getBuildThread();
			if (thread == null)
				continue;
			if (thread.isRunning())
				return true;
		}
		return false;
	}

	@Override
	protected void reset() throws SearchLibException {
		comboList = null;
		selectedItem = null;
		name = null;
		rows = 10;
		selectedField = null;
		searchRequest = null;
	}

	@Command
	@NotifyChange("*")
	public void onBuild(@BindingParam("item") AutoCompletionItem item)
			throws SearchLibException, InvalidPropertiesFormatException,
			IOException {
		item.build(null, 1000, null);
	}

	@Command
	@NotifyChange("*")
	public void onSave() throws SearchLibException,
			InvalidPropertiesFormatException, IOException {
		Client client = getClient();
		if (client == null)
			return;
		AutoCompletionManager manager = client.getAutoCompletionManager();
		AutoCompletionItem autoCompItem = selectedItem != null ? selectedItem
				: new AutoCompletionItem(client, name);
		autoCompItem.setFields(fields);
		autoCompItem.setRows(rows);
		autoCompItem.setSearchRequest(searchRequest);
		if (selectedItem == null)
			manager.add(autoCompItem);
		else
			autoCompItem.save();
		onCancel();
	}

	private class DeleteAlert extends AlertController {

		protected DeleteAlert() throws InterruptedException {
			super(
					"Please, confirm that you want to delete the autocompletion: "
							+ selectedItem.getName(), Messagebox.YES
							| Messagebox.NO, Messagebox.QUESTION);
		}

		@Override
		protected void onYes() throws SearchLibException {
			Client client = getClient();
			if (client == null)
				return;
			try {
				AutoCompletionManager manager = client
						.getAutoCompletionManager();
				if (selectedItem != null)
					manager.delete(selectedItem);
				reload();
				onCancel();
			} catch (IOException e) {
				throw new SearchLibException(e);
			}
		}
	}

	@Command
	@NotifyChange("*")
	public void onDelete() throws InterruptedException {
		new DeleteAlert();
	}

	@Command
	@NotifyChange("*")
	public void onCancel() {
		selectedItem = null;
		selectedField = null;
		name = null;
		rows = 10;
		fields.clear();
	}

	@Command
	@NotifyChange("*")
	public void onFieldAdd() {
		fields.add(selectedField);
	}

	@Command
	@NotifyChange("*")
	public void onFieldRemove(@BindingParam("field") String field) {
		fields.remove(field);
	}

	@Command
	@NotifyChange("comboList")
	public void onChanging(
			@ContextParam(ContextType.TRIGGER_EVENT) InputEvent event)
			throws SearchLibException {
		if (selectedItem == null)
			return;
		String[] resultArray = new String[0];
		AbstractResultSearch<?> result = selectedItem.search(event.getValue(),
				selectedItem.getRows());
		if (result != null) {
			if (result.getDocumentCount() > 0) {
				resultArray = new String[result.getDocumentCount()];
				int i = 0;
				for (ResultDocument resDoc : result) {
					resultArray[i++] = resDoc
							.getValueContent(
									AutoCompletionItem.autoCompletionSchemaFieldTerm,
									0);
				}
			}
		}
		comboList = new ListModelArray<String>(resultArray);
	}

	public ListModel<String> getComboList() {
		return comboList;
	}

	/**
	 * @return the selectedItem
	 */
	public AutoCompletionItem getSelectedItem() {
		return selectedItem;
	}

	/**
	 * @param selectedItem
	 *            the selectedItem to set
	 */
	@NotifyChange("*")
	public void setSelectedItem(AutoCompletionItem selectedItem) {
		this.selectedItem = selectedItem;
		fields.clear();
		fields.addAll(selectedItem.getFields());
		this.rows = selectedItem.getRows();
		this.name = selectedItem.getName();
		this.searchRequest = selectedItem.getSearchRequest();
	}

	/**
	 * @return the name
	 */
	public String getName() {
		return name;
	}

	/**
	 * @param name
	 *            the name to set
	 * @throws SearchLibException
	 */
	public void setName(String name) throws SearchLibException {
		if (selectedItem != null)
			throw new SearchLibException("Changing name is not allowed");
		else
			this.name = name;
	}

	/**
	 * @return the rows
	 */
	public int getRows() {
		return rows;
	}

	/**
	 * @param rows
	 *            the rows to set
	 */
	public void setRows(int rows) {
		this.rows = rows;
	}

	/**
	 * @return the selectedField
	 */
	public String getSelectedField() {
		return selectedField;
	}

	/**
	 * @param selectedField
	 *            the selectedField to set
	 */
	public void setSelectedField(String selectedField) {
		this.selectedField = selectedField;
	}

	public Collection<String> getFields() {
		return fields;
	}

	/**
	 * @return the searchRequest
	 */
	public String getSearchRequest() {
		return searchRequest;
	}

	/**
	 * @param searchRequest
	 *            the searchRequest to set
	 */
	public void setSearchRequest(String searchRequest) {
		this.searchRequest = searchRequest;
	}

	public List<String> getSearchRequests() throws SearchLibException {
		Client client = getClient();
		if (client == null)
			return null;
		List<String> nameList = new ArrayList<String>(0);
		client.getRequestMap().getNameList(nameList,
				RequestTypeEnum.SearchFieldRequest,
				RequestTypeEnum.SearchRequest);
		return nameList;
	}
}