/*******************************************************************************
 * Copyright (c) 2006-2011 Gluster, Inc. <http://www.gluster.com>
 * This file is part of Gluster Management Console.
 *
 * Gluster Management Console 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.
 *
 * Gluster Management Console 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 this program.  If not, see
 * <http://www.gnu.org/licenses/>.
 *******************************************************************************/
package org.gluster.storage.management.console.actions;

import java.net.URI;

import org.eclipse.jface.action.IAction;
import org.eclipse.jface.viewers.ISelection;
import org.gluster.storage.management.client.TasksClient;
import org.gluster.storage.management.client.VolumesClient;
import org.gluster.storage.management.console.GlusterDataModelManager;
import org.gluster.storage.management.console.utils.GUIHelper;
import org.gluster.storage.management.core.model.Status;
import org.gluster.storage.management.core.model.TaskInfo;
import org.gluster.storage.management.core.model.Volume;


public class RebalanceVolumeAction extends AbstractActionDelegate {
	private Volume volume;
	private GUIHelper guiHelper = GUIHelper.getInstance();

	@Override
	protected void performAction(final IAction action) {
		final String actionDesc = action.getDescription();
		try {
			TaskInfo existingTaskInfo = GlusterDataModelManager.getInstance().getTaskByReference(volume.getName());
			if (existingTaskInfo != null && existingTaskInfo.getStatus().getCode() != Status.STATUS_CODE_SUCCESS
					&& existingTaskInfo.getStatus().getCode() != Status.STATUS_CODE_FAILURE) {
				showInfoDialog(actionDesc, "Volume [" + volume.getName()
						+ "] rebalance is already in progress! Try later.");
				return;
			}
			
			URI uri = new VolumesClient().rebalanceStart(volume.getName(), false, false, false);
			// Add the task to model
			TasksClient taskClient = new TasksClient();
			TaskInfo taskInfo = taskClient.getTaskInfo(uri);
			if (taskInfo != null) {
				GlusterDataModelManager.getInstance().addTask(taskInfo);
			}
			if (taskInfo.getStatus().getCode() == Status.STATUS_CODE_ERROR) {
				showWarningDialog(actionDesc, "Volume [" + volume.getName()
						+ "] rebalance started, but following error occured: " + taskInfo.getStatus());
			} else {
				showInfoDialog(actionDesc, "Volume [" + volume.getName() + "] rebalance started successfully!");
			}
			guiHelper.showTaskView();
		} catch (Exception e) {
			showErrorDialog(actionDesc, "Volume rebalance could not be started on [" + volume.getName() + "]! Error: ["
					+ e.getMessage() + "]");
		}
	}

	@Override
	public void selectionChanged(IAction action, ISelection selection) {
		super.selectionChanged(action, selection);

		Volume selectedVolume = guiHelper.getSelectedEntity(getWindow(), Volume.class);
		if (selectedVolume != null) {
			volume = selectedVolume;
			action.setEnabled(true);
		} else {
			action.setEnabled(false);
		}
	}

	@Override
	public void dispose() {
		System.out.println("Disposing [" + this.getClass().getSimpleName() + "]");
	}
}