listRoots?

[Jan Hanselaer]

"Regan Heath" <regan at netmail.co.nz> schreef in bericht 
news:f27n2v$2oju$1 at digitalmars.com...
> Just realised I could template it and easily support all types.  Attached 
> is a new one.
>


--------------------------------------------------------------------------------


> 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");
> }
>

That does the trick. Thanks a lot!