#include <Disks/IStoragePolicy.h>
#include <Common/quoteString.h>
#include <Common/Exception.h>

namespace DB
{

namespace ErrorCodes
{
    extern const int UNKNOWN_VOLUME;
    extern const int UNKNOWN_DISK;
}

DiskPtr IStoragePolicy::getDiskByName(const String & disk_name) const
{
    auto disk = tryGetDiskByName(disk_name);
    if (!disk)
        throw Exception(ErrorCodes::UNKNOWN_DISK,
            "No such disk {} in storage policy {}", backQuote(disk_name), backQuote(getName()));

    return disk;
}

VolumePtr IStoragePolicy::getVolumeByName(const String & volume_name) const
{
    auto volume = tryGetVolumeByName(volume_name);
    if (!volume)
        throw Exception(ErrorCodes::UNKNOWN_VOLUME,
            "No such volume {} in storage policy {}", backQuote(volume_name), backQuote(getName()));

    return volume;
}

size_t IStoragePolicy::getVolumeIndexByDiskName(const String & disk_name) const
{
    auto index = tryGetVolumeIndexByDiskName(disk_name);
    if (!index)
        throw Exception(ErrorCodes::UNKNOWN_DISK,
            "No disk {} in policy {}", backQuote(disk_name), backQuote(getName()));

    return *index;
}

VolumePtr IStoragePolicy::tryGetVolumeByDiskName(const String & disk_name) const
{
    auto index = tryGetVolumeIndexByDiskName(disk_name);
    if (!index)
        return nullptr;

    return getVolume(*index);
}

VolumePtr IStoragePolicy::getVolumeByDiskName(const String & disk_name) const
{
    auto volume = tryGetVolumeByDiskName(disk_name);
    if (!volume)
        throw Exception(ErrorCodes::UNKNOWN_DISK,
            "No disk {} in policy {}", backQuote(disk_name), backQuote(getName()));

    return volume;
}

}