Initialize Disk confirmation dialog added

2 files changed, 169 insertions, 23 deletions

diff --git a/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/InitializeDiskTypeSelection.java b/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/InitializeDiskTypeSelection.java
new file mode 100644
index 00000000..bb700de5
--- /dev/null
+++ b/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/dialogs/InitializeDiskTypeSelection.java
@@ -0,0 +1,148 @@
+/*******************************************************************************
+ * 
+ * InitializeDiskTypeSelection.java
+ * 
+ * Copyright (c) 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 Affero 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 Affero General Public License
+ * for more details.
+ *  
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program. If not, see
+ * <http://www.gnu.org/licenses/>.
+ *******************************************************************************/
+package com.gluster.storage.management.gui.dialogs;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.eclipse.jface.dialogs.Dialog;
+import org.eclipse.swt.SWT;
+import org.eclipse.swt.events.TraverseEvent;
+import org.eclipse.swt.events.TraverseListener;
+import org.eclipse.swt.layout.GridData;
+import org.eclipse.swt.layout.GridLayout;
+import org.eclipse.swt.widgets.Combo;
+import org.eclipse.swt.widgets.Composite;
+import org.eclipse.swt.widgets.Control;
+import org.eclipse.swt.widgets.Label;
+import org.eclipse.swt.widgets.Shell;
+
+import com.gluster.storage.management.core.constants.GlusterConstants;
+import com.gluster.storage.management.gui.utils.GUIHelper;
+
+public class InitializeDiskTypeSelection extends Dialog  {
+	
+	private Combo formatTypeCombo = null;
+	private final GUIHelper guiHelper = GUIHelper.getInstance();
+	private Composite initializeDiskTypeComposite;
+	private Composite composite;
+	private String fsType; 
+
+	public InitializeDiskTypeSelection(Shell parentShell) {
+		super(parentShell);
+		// TODO Auto-generated constructor stub
+	}
+	
+	@Override
+	protected void configureShell(Shell newShell) {
+		super.configureShell(newShell);
+
+		newShell.setText("Gluster Management Console - Select Cluster");
+		addEscapeListener(newShell);
+	}
+	
+	private void addEscapeListener(Shell shell) {
+		shell.addTraverseListener(new TraverseListener() {
+
+			@Override
+			public void keyTraversed(TraverseEvent e) {
+				if (e.keyCode == SWT.ESC) {
+					cancelPressed();
+				}
+			}
+		});
+	}
+	
+	@Override
+	protected Control createDialogArea(Composite parent) {
+		// Makes sure that child composites inherit the same background
+		parent.setBackgroundMode(SWT.INHERIT_FORCE); 
+		
+		composite =  (Composite) super.createDialogArea(parent);
+		configureDialogLayout(composite);
+		createComposite(composite);
+		return composite;
+	}
+	
+	private void configureDialogLayout(Composite composite) {
+		GridLayout layout = (GridLayout) composite.getLayout();
+		layout.numColumns = 3;
+		layout.marginLeft = 20;
+		layout.marginRight = 20;
+		layout.marginTop = 20;
+		layout.horizontalSpacing = 20;
+		layout.verticalSpacing = 20;
+	}
+	
+	private void createComposite(Composite composite) {
+		initializeDiskTypeComposite = new Composite(composite, SWT.NONE);
+		GridLayout layout = new GridLayout(3, false);
+		initializeDiskTypeComposite.setLayout(layout);
+
+		createLabel(initializeDiskTypeComposite, "Format disk using ");
+		createFormatTypeCombo(initializeDiskTypeComposite);
+		createLabel(initializeDiskTypeComposite, " file system");
+	}
+	
+	private void createLabel(Composite composite, String labelText) {
+		Label formatTypeLabel = new Label(composite, SWT.NONE);
+		formatTypeLabel.setText(labelText);
+		formatTypeLabel.setLayoutData(new GridData(GridData.END, GridData.CENTER, false, false));
+	}
+	
+	private void createFormatTypeCombo(Composite composite) {
+		List<String> fsType = new ArrayList<String>();
+		fsType.add(GlusterConstants.FSTYPE_DEFAULT);
+		fsType.add(GlusterConstants.FSTYPE_EXT_3);
+		fsType.add(GlusterConstants.FSTYPE_EXT_4);
+		fsType.add(GlusterConstants.FSTYPE_XFS);
+		formatTypeCombo = new Combo(composite, SWT.READ_ONLY);
+		formatTypeCombo.setItems(fsType.toArray(new String[0]));
+		formatTypeCombo.select(0);
+	}
+	
+	@Override
+	protected void okPressed() {
+		fsType = formatTypeCombo.getText();
+		super.okPressed();
+	}
+	
+	@Override
+	public void cancelPressed() {
+		super.cancelPressed();
+	}
+	
+	/**
+	 * Overriding to make sure that the dialog is centered in screen
+	 */
+	@Override
+	protected void initializeBounds() {
+		super.initializeBounds();
+
+		guiHelper.centerShellInScreen(getShell());
+	}
+	
+	public String getFSType() {
+		return fsType.trim();
+	}
+
+}
diff --git a/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/views/pages/AbstractDisksPage.java b/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/views/pages/AbstractDisksPage.java
index 63cf65ed..cc12415e 100644
--- a/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/views/pages/AbstractDisksPage.java
+++ b/src/com.gluster.storage.management.gui/src/com/gluster/storage/management/gui/views/pages/AbstractDisksPage.java
@@ -18,12 +18,14 @@
  *******************************************************************************/
 package com.gluster.storage.management.gui.views.pages;
 
+import java.net.URI;
 import java.util.List;
 
 import org.eclipse.jface.dialogs.MessageDialog;
 import org.eclipse.jface.viewers.ArrayContentProvider;
 import org.eclipse.jface.viewers.IContentProvider;
 import org.eclipse.jface.viewers.TableViewer;
+import org.eclipse.jface.window.Window;
 import org.eclipse.swt.SWT;
 import org.eclipse.swt.custom.TableEditor;
 import org.eclipse.swt.events.DisposeEvent;
@@ -39,18 +41,18 @@ import org.eclipse.ui.forms.events.HyperlinkAdapter;
 import org.eclipse.ui.forms.events.HyperlinkEvent;
 import org.eclipse.ui.forms.widgets.ImageHyperlink;
 
+import com.gluster.storage.management.client.GlusterDataModelManager;
 import com.gluster.storage.management.client.GlusterServersClient;
-import com.gluster.storage.management.core.constants.GlusterConstants;
+import com.gluster.storage.management.client.TasksClient;
 import com.gluster.storage.management.core.model.ClusterListener;
 import com.gluster.storage.management.core.model.DefaultClusterListener;
 import com.gluster.storage.management.core.model.Disk;
 import com.gluster.storage.management.core.model.Disk.DISK_STATUS;
 import com.gluster.storage.management.core.model.Entity;
+import com.gluster.storage.management.core.model.TaskInfo;
 import com.gluster.storage.management.gui.Application;
 import com.gluster.storage.management.gui.IEntityListener;
-import com.gluster.storage.management.gui.IImageKeys;
-import com.gluster.storage.management.gui.jobs.InitializeDiskJob;
-import com.gluster.storage.management.gui.utils.GUIHelper;
+import com.gluster.storage.management.gui.dialogs.InitializeDiskTypeSelection;
 
 public abstract class AbstractDisksPage extends AbstractTableViewerPage<Disk> implements IEntityListener {
 	private List<Disk> disks;
@@ -203,30 +205,26 @@ public abstract class AbstractDisksPage extends AbstractTableViewerPage<Disk> im
 
 		@Override
 		public void linkActivated(HyperlinkEvent e) {
-			Integer formatOption = new MessageDialog(getShell(), "Initialize Disk", GUIHelper.getInstance().getImage(
-					IImageKeys.DISK), "Please choose the file system to Initialize the disk?", MessageDialog.QUESTION, new String[] {
-					"Cancel", GlusterConstants.FSTYPE_EXT_3, GlusterConstants.FSTYPE_EXT_4, GlusterConstants.FSTYPE_XFS }, -1).open();
-
-			if (formatOption <= 0) { // By Cancel button(0) or Escape key(-1)
+			InitializeDiskTypeSelection formatDialog = new InitializeDiskTypeSelection(getShell());
+			int userAction = formatDialog.open();
+			if (userAction == Window.CANCEL) {
+				formatDialog.cancelPressed();
 				return;
 			}
 
-			String fsType = null;
-			if (formatOption == 1) {
-				fsType = GlusterConstants.FSTYPE_EXT_3;
-			} else if (formatOption == 2) {
-				fsType = GlusterConstants.FSTYPE_EXT_4;
-			} else if (formatOption == 3) {
-				fsType = GlusterConstants.FSTYPE_XFS;
-			}
-
 			GlusterServersClient serversClient = new GlusterServersClient();
-			serversClient.initializeDisk(disk.getServerName(), disk.getName(), fsType);
+			try {
+				URI uri = serversClient.initializeDisk(disk.getServerName(), disk.getName(), formatDialog.getFSType());
 
-			updateStatus(DISK_STATUS.INITIALIZING, true); 
-			
-			guiHelper.showProgressView();
-			new InitializeDiskJob(disk).schedule();
+				TasksClient taskClient = new TasksClient();
+				TaskInfo taskInfo = taskClient.getTaskInfo(uri);
+				if (taskInfo != null && taskInfo instanceof TaskInfo) {
+					GlusterDataModelManager.getInstance().getModel().getCluster().addTaskInfo(taskInfo);
+				}
+				updateStatus(DISK_STATUS.INITIALIZING, true);
+			} catch (Exception e1) {
+				MessageDialog.openError(getShell(), "Error: Initialize disk", e1.getMessage());
+			}
 		}
 	}