/*
 * Copyright 2011, Axel Dörfler, axeld@pinc-software.de.
 * Distributed under the terms of the MIT License.
 */


#include "BootDrive.h"

#include <DiskDevice.h>
#include <DiskDeviceRoster.h>
#include <Volume.h>
#include <VolumeRoster.h>


BootDrive::BootDrive(const char* path)
        :
        fPath(path)
{
}


BootDrive::~BootDrive()
{
}


BootMenu*
BootDrive::InstalledMenu(const BootMenuList& menus) const
{
        for (int32 i = 0; i < menus.CountItems(); i++) {
                BootMenu* menu = menus.ItemAt(i);
                if (menu->IsInstalled(*this))
                        return menu;
        }
        return NULL;
}


status_t
BootDrive::CanMenuBeInstalled(const BootMenuList& menus) const
{
        status_t status = B_ERROR;

        for (int32 i = 0; i < menus.CountItems(); i++) {
                status = menus.ItemAt(i)->CanBeInstalled(*this);
                if (status == B_OK)
                        return status;
        }
        return status;
}


/*!     Adds all boot menus from the list \a from that support the drive to \a to.
*/
void
BootDrive::AddSupportedMenus(const BootMenuList& from, BootMenuList& to)
{
        for (int32 i = 0; i < from.CountItems(); i++) {
                BootMenu* menu = from.ItemAt(i);
                if (menu->CanBeInstalled(*this))
                        to.AddItem(menu);
        }
}


const char*
BootDrive::Path() const
{
        return fPath.Path();
}


bool
BootDrive::IsBootDrive() const
{
        BVolumeRoster volumeRoster;
        BVolume volume;
        if (volumeRoster.GetBootVolume(&volume) != B_OK)
                return false;

        BDiskDeviceRoster roster;
        BDiskDevice device;
        if (roster.FindPartitionByVolume(volume, &device, NULL) == B_OK) {
                BPath path;
                if (device.GetPath(&path) == B_OK && path == fPath)
                        return true;
        }

        return false;
}


status_t
BootDrive::GetDiskDevice(BDiskDevice& device) const
{
        BDiskDeviceRoster roster;
        return roster.GetDeviceForPath(fPath.Path(), &device);
}