Qv2ray/src/QvUtils.cpp

#include "QvUtils.h"
#include <QTextStream>

namespace Qv2ray
{
    namespace Utils
    {
        static Qv2rayConfig GlobalConfig;
        static QString ConfigDirPath;
        void SetGlobalConfig(Qv2rayConfig conf)
        {
            GlobalConfig = conf;
            QFile config(QV2RAY_CONFIG_FILE_PATH);
            QString str = StructToJsonString(GetGlobalConfig());
            StringToFile(&str, &config);
        }

        Qv2rayConfig GetGlobalConfig()
        {
            return GlobalConfig;
        }

        QString GetConfigDirPath()
        {
            return ConfigDirPath;
        }

        void SetConfigDirPath(const QString *path)
        {
            ConfigDirPath = *path;
        }

        QString Stringify(list<string> list, QString saperator)
        {
            QString out;

            foreach (string item, list) {
                out.append(QSTRING(item));
                out.append(saperator);
            }

            if (out.length() >= 1)
                out = out.remove(out.length() - 1, 1);

            return out;
        }

        QString Stringify(QList<QString> list, QString saperator)
        {
            QString out;

            foreach (QString item, list) {
                out.append(saperator);
                out.append(item);
            }

            if (out.length() >= 1)
                out = out.remove(out.length() - 1, 1);

            return out;
        }

        bool StringToFile(const QString *text, QFile *targetFile)
        {
            bool override = targetFile->exists();
            targetFile->open(QFile::WriteOnly);
            QTextStream stream(targetFile);
            stream << *text << endl;
            stream.flush();
            targetFile->close();
            return override;
        }

        QJsonObject JSONFromFile(QFile *sourceFile)
        {
            QString json = StringFromFile(sourceFile);
            return JsonFromString(json);
        }

        QString JsonToString(QJsonObject json)
        {
            QJsonDocument doc;
            doc.setObject(json);
            return doc.toJson();
        }

        QString VerifyJsonString(const QString *source)
        {
            QJsonParseError error;
            QJsonDocument doc = QJsonDocument::fromJson(source->toUtf8(), &error);

            if (error.error == QJsonParseError::NoError) {
                return "";
            } else {
                LOG(MODULE_CORE, "WARNING: Json parse returns: " + error.errorString().toStdString())
                return error.errorString();
            }
        }

        QJsonObject JsonFromString(QString string)
        {
            QJsonDocument doc = QJsonDocument::fromJson(string.toUtf8());
            return doc.object();
        }

        QString StringFromFile(QFile *sourceFile)
        {
            sourceFile->open(QFile::ReadOnly);
            QTextStream stream(sourceFile);
            QString str = stream.readAll();
            sourceFile->close();
            return str;
        }

        QString Base64Encode(QString string)
        {
            QByteArray ba;
            ba.append(string);
            return ba.toBase64();
        }

        QString Base64Decode(QString string)
        {
            QByteArray ba;
            ba.append(string);
            return QString(QByteArray::fromBase64(ba));
        }

        void LoadGlobalConfig()
        {
            QFile file(QV2RAY_CONFIG_FILE_PATH);
            file.open(QFile::ReadOnly);
            QTextStream stream(&file);
            auto str = stream.readAll();
            auto config  = StructFromJsonString<Qv2rayConfig>(str);
            SetGlobalConfig(config);
            file.close();
        }

        QStringList getFileList(QDir dir)
        {
            return dir.entryList(QStringList() << "*" << "*.*", QDir::Hidden | QDir::Files);
        }

        bool CheckFile(QDir dir, QString fileName)
        {
            return getFileList(dir).indexOf(fileName) >= 0;
        }

        void QvMessageBox(QWidget *parent, QString title, QString text)
        {
            QMessageBox::warning(parent, title, text, QMessageBox::Ok | QMessageBox::Default, 0);
        }

        int QvMessageBoxAsk(QWidget *parent, QString title, QString text, QMessageBox::StandardButton extraButtons)
        {
            return QMessageBox::information(parent, title, text, QMessageBox::Yes | QMessageBox::No | extraButtons);
        }

        QString FormatBytes(long long bytes, char *str)
        {
            const char *sizes[5] = { "B", "KB", "MB", "GB", "TB" };
            int i;
            double dblByte = bytes;

            for (i = 0; i < 5 && bytes >= 1024; i++, bytes /= 1024)
                dblByte = bytes / 1024.0;

            sprintf(str, "%.2f", dblByte);
            return strcat(strcat(str, " "), sizes[i]);
        }


        QTranslator *getTranslator(const QString *lang)
        {
            QTranslator *translator = new QTranslator();
            translator->load(*lang + ".qm", ":/translations/");
            return translator;
        }
    }
}