/*
 * Copyright 2002-2013 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT license.
 *
 * Authors:
 *		Ithamar R. Adema <ithamar@unet.nl>
 *		Stephan Aßmus <superstippi@gmx.de>
 */
#ifndef MAIN_WINDOW_H
#define MAIN_WINDOW_H


#include <DiskDeviceRoster.h>
#include <FilePanel.h>
#include <Notification.h>
#include <PopUpMenu.h>
#include <Window.h>

#include "Support.h"


class BColumnListView;
class BDiskDevice;
class BMenu;
class BMenuBar;
class BMenuItem;
class BPartition;
class BRow;
class DiskView;
class PartitionListView;


enum {
	MSG_SELECTED_PARTITION_ID	= 'spid',
	MSG_UPDATE_ZOOM_LIMITS		= 'uzls'
};


class MainWindow : public BWindow {
public:
								MainWindow();
	virtual						~MainWindow();

	// BWindow interface
	virtual	bool				QuitRequested();
	virtual	void				MessageReceived(BMessage* message);

	// MainWindow
			status_t			StoreSettings(BMessage* archive) const;
			status_t			RestoreSettings(BMessage* archive);
			void				ApplyDefaultSettings();

			status_t			RegisterFileDiskDevice(const char* fileName);
			status_t			UnRegisterFileDiskDevice(const char* fileName);
	static	int32				SaveDiskImage(void* data);
	static	int32				WriteDiskImage(void* data);

private:
	static	void				_WriteDiskImage(BMessenger messenger,
									BFile source, BFile target,
									const char* targetpath,
									BString title, BString status);

			void				_ScanDrives();

			void				_AdaptToSelectedPartition();
			void				_SetToDiskAndPartition(partition_id diskID,
									partition_id partitionID,
									partition_id parentID);
			void				_UpdateMenus(BDiskDevice* disk,
									partition_id selectedPartition,
									partition_id parentID);

			void				_DisplayPartitionError(BString message,
									const BPartition* partition = NULL,
									status_t error = B_OK) const;

			void				_Mount(BDiskDevice* disk,
									partition_id selectedPartition);
			void				_Unmount(BDiskDevice* disk,
									partition_id selectedPartition);
			void				_MountAll();

			void				_Initialize(BDiskDevice* disk,
									partition_id selectedPartition,
									const BString& diskSystemName);
			void				_Create(BDiskDevice* disk,
									partition_id selectedPartition);
			void				_Delete(BDiskDevice* disk,
									partition_id selectedPartition);
			void				_ChangeParameters(BDiskDevice* disk,
									partition_id selectedPartition);
			float				_ColumnListViewHeight(BColumnListView* list,
									BRow* currentRow);
			void				_UpdateWindowZoomLimits();

private:
			BDiskDeviceRoster	fDiskDeviceRoster;
			BDiskDevice*		fCurrentDisk;
			partition_id		fCurrentPartitionID;

			PartitionListView*	fListView;
			DiskView*			fDiskView;

			SpaceIDMap			fSpaceIDMap;

			BMenu*				fDiskMenu;
			BMenu*				fDiskInitMenu;

			BMenu*				fPartitionMenu;
			BMenu*				fFormatMenu;
			BMenu*				fDiskImageMenu;

			BMenuBar* 			fMenuBar;

			BMenuItem*			fWipeMenuItem;
			BMenuItem*			fEjectMenuItem;
			BMenuItem*			fSurfaceTestMenuItem;
			BMenuItem*			fRescanMenuItem;

			BMenuItem*			fCreateMenuItem;
			BMenuItem*			fChangeMenuItem;
			BMenuItem*			fDeleteMenuItem;
			BMenuItem*			fMountMenuItem;
			BMenuItem*			fUnmountMenuItem;
			BMenuItem*			fMountAllMenuItem;
			BMenuItem*			fOpenDiskProbeMenuItem;

			BMenu*				fFormatContextMenuItem;
			BMenuItem*			fCreateContextMenuItem;
			BMenuItem*			fChangeContextMenuItem;
			BMenuItem*			fDeleteContextMenuItem;
			BMenuItem*			fMountContextMenuItem;
			BMenuItem*			fUnmountContextMenuItem;
			BMenuItem*			fOpenDiskProbeContextMenuItem;
			BPopUpMenu*			fContextMenu;

			BMenuItem*			fRegisterMenuItem;
			BMenuItem*			fUnRegisterMenuItem;
			BMenuItem*			fSaveMenuItem;
			BMenuItem*			fWriteMenuItem;

			BFilePanel*			fRegisterFilePanel;
			BFilePanel*			fWriteImageFilePanel;
			BFilePanel*			fReadImageFilePanel;

			BNotification*		fNotification;
};


#endif // MAIN_WINDOW_H