deviceenumerator.h

////////////////////////////////////////////////////////////////////////////////
//      This file is part of Unraid USB Creator - https://github.com/limetech/usb-creator
//      Copyright (C) 2013-2015 RasPlex project
//      Copyright (C) 2016 Team LibreELEC
//      Copyright (C) 2018-2020 Lime Technology, Inc
//
//  Unraid USB Creator is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 2 of the License, or
//  (at your option) any later version.
//
//  Unraid USB Creator is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
//
//  You should have received a copy of the GNU General Public License
//  along with Unraid USB Creator.  If not, see <http://www.gnu.org/licenses/>.
////////////////////////////////////////////////////////////////////////////////

#ifndef DEVICEENUMERATOR_H
#define DEVICEENUMERATOR_H

#include <QStringList>
#include <QVariantMap>

// interface class
class DeviceEnumerator
{
public:
    enum loadEject {
      LOADEJECT_LOAD = 0,
      LOADEJECT_EJECT
    };

    virtual ~DeviceEnumerator() {}
    virtual QStringList getRemovableDeviceNames() const = 0;
    virtual QStringList getUserFriendlyNames(const QStringList& devices) const = 0;
    virtual bool unmountDevicePartitions(const QString &device) const = 0;
    virtual qint64 getSizeOfDevice(const QString &device) const = 0;
    virtual int loadEjectDrive(const QString &device, const loadEject action) const = 0;
    virtual int removeDrive(const QString &device) const = 0;
    virtual QList<QVariantMap> listBlockDevices() const = 0;
    virtual bool supportsGuid() const = 0; 

    static QString sizeToHuman(const qint64 size)
    {
        const QStringList suffixes = QStringList() << " B" << " KB" << " MB" << " GB" << " TB";

        if (size <= 0)
            return "0 MB";

        int idx = 0;
        qreal fSize = size;
        while ((fSize > 1024.0) && ((idx + 1) < suffixes.size())) {
            ++idx;
            fSize /= 1024.0;
        }

        return QString::number(fSize, 'f', 1) + suffixes[idx];
    }
};

#endif // DEVICEENUMERATOR_H