overte-HifiExperiments/libraries/script-engine/src/BatchLoader.cpp

//
//  BatchLoader.cpp
//  libraries/script-engine/src
//
//  Created by Ryan Huffman on 01/22/15
//  Copyright 2015 High Fidelity, Inc.
//
//  Distributed under the Apache License, Version 2.0.
//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
//

#include <QNetworkAccessManager>
#include <QNetworkReply>

#include <QFile>
#include "BatchLoader.h"
#include <NetworkAccessManager.h>
#include <SharedUtil.h>

BatchLoader::BatchLoader(const QList<QUrl>& urls) 
    : QObject(),
      _started(false),
      _finished(false),
      _urls(urls.toSet()),
      _data() {
}

void BatchLoader::start() {
    if (_started) {
        return;
    }

    _started = true;
    QNetworkAccessManager& networkAccessManager = NetworkAccessManager::getInstance();
    for (QUrl url : _urls) {
        if (url.scheme() == "http" || url.scheme() == "https" || url.scheme() == "ftp") {
            QNetworkRequest request = QNetworkRequest(url);
            request.setHeader(QNetworkRequest::UserAgentHeader, HIGH_FIDELITY_USER_AGENT);
            QNetworkReply* reply = networkAccessManager.get(request);

            qDebug() << "Downloading file at" << url;

            connect(reply, &QNetworkReply::finished, [=]() {
                if (reply->error()) {
                    _data.insert(url, QString());
                } else {
                    _data.insert(url, reply->readAll());
                }
                reply->deleteLater();
                checkFinished();
            });

            // If we end up being destroyed before the reply finishes, clean it up
            connect(this, &QObject::destroyed, reply, &QObject::deleteLater);

        } else {
#ifdef _WIN32
            QString fileName = url.toString();
#else
            QString fileName = url.toLocalFile();
#endif

            qDebug() << "Reading file at " << fileName;

            QFile scriptFile(fileName);
            if (scriptFile.open(QFile::ReadOnly | QFile::Text)) {
                QTextStream in(&scriptFile);
                _data.insert(url, in.readAll());
            } else {
                _data.insert(url, QString());
            }
        }
    }
    checkFinished();
}

void BatchLoader::checkFinished() {
    if (!_finished && _urls.size() == _data.size()) {
        _finished = true;
        emit finished(_data);
    }
}