CONFIG += mobility
MOBILITY += systeminfo

using namespace QtMobility;

#ifndef SYSTEMSTORAGEINFO_H
#define SYSTEMSTORAGEINFO_H
 
#include <QtGui/QWidget>
#include "ui_SystemStorageInfo.h"
#include <qsysteminfo.h>
 
using namespace QtMobility;
class SystemStorageInfo : public QWidget
{
    Q_OBJECT
 
public:
	SystemStorageInfo(QWidget *parent = 0);
    ~SystemStorageInfo();
 
public slots:
	void listLogicalDrives();
	void listDriveInformation(int aDriveIndex);
	void resetAllFields();
 
private:
    Ui::SystemStorageInfo ui;
    QSystemStorageInfo* storageInfo;
};
 
#endif // SYSTEMSTORAGEINFO_H

#include "SystemStorageInfo.h"
 
SystemStorageInfo::SystemStorageInfo(QWidget *parent)
    : QWidget(parent), storageInfo(NULL)
{
	ui.setupUi(this);
 
	storageInfo = new QSystemStorageInfo();
 
	QObject::connect(ui.driveComboBox, 
					SIGNAL(currentIndexChanged(int)),
					this, 
					SLOT(listDriveInformation(int)));
	QObject::connect(ui.storageInfoButton, 
					SIGNAL(clicked()),
					this, 
					SLOT(listLogicalDrives()));	
	QObject::connect(ui.resetButton, 
					SIGNAL(clicked()),
					this, 
					SLOT(resetAllFields()));
}
 
SystemStorageInfo::~SystemStorageInfo()
{
	delete storageInfo;
}
 
void SystemStorageInfo::listLogicalDrives()
{
	if(ui.driveComboBox->count() > 1)
	{
		//this has been called previously and combo box contains data already
		ui.driveComboBox->showPopup();
		return;
	}
	QStringList list = storageInfo->logicalDrives();
	ui.driveComboBox->addItems(list);
	ui.driveComboBox->showPopup();
}
 
void SystemStorageInfo::listDriveInformation(int aDriveIndex)
{
	if(aDriveIndex == 0)
	{
		resetAllFields();
		return;
	}
	// Set available disc space label
	QString textValue = ui.driveComboBox->itemText(aDriveIndex);
	qlonglong availableSpace = storageInfo->availableDiskSpace(textValue);
	QString str = QString::number(availableSpace);
	ui.availabelDiscSpaceLabel->setText(str);
 
	// Set total disc space label
	qlonglong totalSpace = storageInfo->totalDiskSpace(textValue);
	str = QString::number(totalSpace);
	ui.totalDiscSpaceLabel->setText(str);
 
	// Set drive type label
	QString labelValue;
	QSystemStorageInfo::DriveType drvType = storageInfo->typeForDrive(textValue);
	switch(drvType)
	{
		case QSystemStorageInfo::NoDrive:
		{
			labelValue = ("No drive");
			break;
		}
		case QSystemStorageInfo::InternalDrive:
		{
			labelValue = ("Internal Drive");
			break;
		}
		case QSystemStorageInfo::RemovableDrive:
		{
			labelValue = ("Removable Drive");
			break;
		}
		case QSystemStorageInfo::RemoteDrive:
		{
			labelValue = ("Remote Drive");
			break;
		}
		case QSystemStorageInfo::CdromDrive:
		{
			labelValue = ("CD ROM Drive");
			break;
		}
		default:
		{
			break;
		}
	}
	ui.driveTypeLabel->setText(labelValue);
}
 
void SystemStorageInfo::resetAllFields()
{
	ui.driveComboBox->setCurrentIndex(0);
	ui.availabelDiscSpaceLabel->clear();
	ui.totalDiscSpaceLabel->clear();
	ui.driveTypeLabel->clear();
}