From 59312709aeed55e24533dc3c49956fa1f220e316 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stephan=20A=C3=9Fmus?= <superstippi@gmx.de>
Date: Wed, 20 Feb 2008 14:23:13 +0000
Subject: [PATCH] * Maintain the previously selected partition when rescanning
 drives (ie also   when mounting/unmounting a partition). * Set the enabled
 state of the Unmount menu item when a partition is not   mounted (was using
 previous state from last partition).

git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@24034 a95241bf-73f2-0310-859d-f6bbb57e9c96
---
 src/apps/drivesetup/MainWindow.cpp | 27 ++++++++++++++++++++-------
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/src/apps/drivesetup/MainWindow.cpp b/src/apps/drivesetup/MainWindow.cpp
index acd72c6daa..8b4daf58ea 100644
--- a/src/apps/drivesetup/MainWindow.cpp
+++ b/src/apps/drivesetup/MainWindow.cpp
@@ -146,6 +146,7 @@ MainWindow::MainWindow(BRect frame)
 	: BWindow(frame, "DriveSetup", B_DOCUMENT_WINDOW,
 		B_ASYNCHRONOUS_CONTROLS | B_NOT_ZOOMABLE)
 	, fCurrentDisk(NULL)
+	, fCurrentPartitionID(-1)
 	, fSpaceIDMap()
 {
 	BMenuBar* menuBar = new BMenuBar(Bounds(), "root menu");
@@ -179,7 +180,7 @@ fDeleteMI->SetEnabled(false);
 
 	// Parition menu
 	fPartitionMenu = new BMenu("Partition");
-		fCreateMenu = new BMenu("Create");
+		fCreateMenu = new BMenu("Create (not implemented)");
 		fPartitionMenu->AddItem(fCreateMenu);
 
 		fInitMenu = new BMenu("Initialize");
@@ -224,8 +225,6 @@ fDeleteMI->SetEnabled(false);
 
 	// visit all disks in the system and show their contents
 	_ScanDrives();
-
-	_EnabledDisableMenuItems(NULL, -1, -1);
 }
 
 
@@ -368,6 +367,17 @@ MainWindow::_ScanDrives()
 	ListPopulatorVisitor driveVisitor(fListView, diskCount, fSpaceIDMap);
 	fDDRoster.VisitEachPartition(&driveVisitor);
 	fDiskView->SetDiskCount(diskCount);
+
+	// restore selection
+	PartitionListRow* previousSelection
+		= fListView->FindRow(fCurrentPartitionID);
+	if (previousSelection) {
+		fListView->AddToSelection(previousSelection);
+		_EnabledDisableMenuItems(fCurrentDisk, fCurrentPartitionID,
+			previousSelection->ParentID());
+	} else {
+		_EnabledDisableMenuItems(NULL, -1, -1);
+	}
 }
 
 
@@ -483,6 +493,7 @@ fSurfaceTestMI->SetEnabled(false);
 			parentPartition = disk->FindDescendant(parentID);
 
 		if (parentPartition) {
+disk->PrepareModifications();
 			fCreateMenu->SetEnabled(true);
 			BString supportedChildType;
 			int32 cookie = 0;
@@ -500,6 +511,7 @@ fSurfaceTestMI->SetEnabled(false);
 			if (fCreateMenu->CountItems() == 0)
 				fprintf(stderr, "Failed to get supported child types: %s\n",
 					strerror(ret));
+disk->CancelModifications();
 		} else {
 			fCreateMenu->SetEnabled(false);
 		}
@@ -512,17 +524,18 @@ fSurfaceTestMI->SetEnabled(false);
 fDeleteMI->SetEnabled(false);
 			fMountMI->SetEnabled(!partition->IsMounted());
 
+			bool unMountable = false;
 			if (partition->IsMounted()) {
 				// see if this partition is the boot volume
 				BVolume volume; 
 				BVolume bootVolume; 
 				if (BVolumeRoster().GetBootVolume(&bootVolume) == B_OK
 					&& partition->GetVolume(&volume) == B_OK) { 
-					fUnmountMI->SetEnabled(volume != bootVolume);
-				} else {
-					fUnmountMI->SetEnabled(true);
-				}
+					unMountable = volume != bootVolume;
+				} else
+					unMountable = true;
 			}
+			fUnmountMI->SetEnabled(unMountable);
 		} else {
 			fInitMenu->SetEnabled(false);
 			fDeleteMI->SetEnabled(false);