#include "ioutil.hpp"

bool IoUtil::writeAll(QIODevice &dst, QByteArray const & src)
{
    qint64 offset = 0;

    while(offset < src.size())
    {
        qint64 len = dst.write(src.data() + offset, src.size() - offset);
        if(len == 0)
            return false;
        offset += len;
    }

    return true;
}

QString IoUtil::size_human(qint64 size)
{
    if(size < 1024)
        return QString(QObject::tr("%1 B")).arg(size);

    float num = size;
    QStringList list;
    list << QObject::tr("KB") << QObject::tr("MB") << QObject::tr("GB") << QObject::tr("TB");

    QStringListIterator i(list);
    QString unit(QObject::tr("B"));

    while(num >= 1024.0 && i.hasNext())
    {
        unit = i.next();
        num /= 1024.0;
    }
    return QString().setNum(num,'f',2)+" "+unit;
}

QUrl IoUtil::uniformUrl(QUrl url)
{
    // We have to manually strip the root path slash
    // since StripTrailingSlash doesn't strip it for some reason.
    if (url.path() == "/") url.setPath(QString { });

    // This will strip slashes from non-root paths
    return url.adjusted(QUrl::RemoveFragment | QUrl::StripTrailingSlash);
}

QString IoUtil::uniformUrlString(QUrl url)
{
    return IoUtil::uniformUrl(url).toString(QUrl::FullyEncoded);
}