import std.stdio, std.string;

alias std.string.toStringz CSTR;

import std.c.windows.windows;

extern(Windows)
{
	DWORD GetLogicalDrives();
	UINT GetDriveTypeA(LPCSTR lpRootPathName);
	UINT GetDriveTypeW(LPCWSTR lpRootPathName);
	const UINT DRIVE_UNKNOWN     = 0; //The drive type cannot be determined. 
	const UINT DRIVE_NO_ROOT_DIR = 1; //The root path is invalid; for example, there is no volume is mounted at the path. 
	const UINT DRIVE_REMOVABLE   = 2; //The drive has removable media; for example, a floppy drive, thumb drive, or flash card reader. 
	const UINT DRIVE_FIXED       = 3; //The drive has fixed media; for example, a hard drive or flash drive. 
	const UINT DRIVE_REMOTE      = 4; //The drive is a remote (network) drive. 
	const UINT DRIVE_CDROM       = 5; //The drive is a CD-ROM drive. 
	const UINT DRIVE_RAMDISK     = 6; //The drive is a RAM disk. 
}

char[][] LocalDrives()
{
	DWORD mask = GetLogicalDrives();
	char[] drive = new char[3];
	char[][] list;	
	
	drive[1..3] = ":\\";	
	for(int i = 0; i < 26; i++)
	{
		if (mask & 0x1<<i)
		{
			drive[0] = 'A'+i;
			list ~= drive.dup;
		}		
	}
	
	return list;
}

template SortDrive(UINT TYPE) { char[][] SortDrive() {
	char[][] list;
	
	foreach(drive; LocalDrives())
	{
		switch(GetDriveTypeA(CSTR(drive)))
		{
			case TYPE:
				list ~= drive;
				break;
			default:
				break;
		}
	}
	
	return list;
}}

alias SortDrive!(DRIVE_FIXED) 		FixedDrives;
alias SortDrive!(DRIVE_REMOVABLE) 	RemovableDrives;
alias SortDrive!(DRIVE_REMOTE) 		RemoteDrives;
alias SortDrive!(DRIVE_CDROM) 		CdDrives;
alias SortDrive!(DRIVE_RAMDISK)		RamDisks;

void main()
{
	writef("Fixed    : ",FixedDrives(),"\n");
	writef("Removable: ",RemovableDrives(),"\n");
	writef("Remote   : ",RemoteDrives(),"\n");
	writef("Cd       : ",CdDrives(),"\n");
	writef("Ramdisk  : ",RamDisks(),"\n");
}