//
//  GooglePolyScriptingInterface.cpp
//  interface/src/scripting
//
//  Created by Elisa Lupin-Jimenez on 12/3/2017.
//  Copyright 2017 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 <QEventLoop>
#include <QGlobal.h>
#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>
#include <QNetworkAccessManager>
#include <QNetworkReply>
#include <QNetworkRequest>
#include <QString>
#include <QTime>
#include <QUrl>
#include <random>

#include "GooglePolyScriptingInterface.h"
#include "ScriptEngineLogging.h"

const QString listPolyUrl = "https://poly.googleapis.com/v1/assets?";
const QString getPolyUrl = "https://poly.googleapis.com/v1/assets/model?";

QStringList validFormats = QStringList() << "BLOCKS" << "FBX" << "GLTF" << "GLTF2" << "OBJ" << "TILT" << "";
QStringList validCategories = QStringList() << "animals" << "architecture" << "art" << "food" << 
    "nature" << "objects" << "people" << "scenes" << "technology" << "transport" << "";

GooglePolyScriptingInterface::GooglePolyScriptingInterface() {
    // nothing to be implemented
}

void GooglePolyScriptingInterface::setAPIKey(QString key) {
    authCode = key;
}

QByteArray GooglePolyScriptingInterface::getAssetList(QString keyword, QString category, QString format) {
    QUrl url = formatURLQuery(keyword, category, format);
    if (!url.isEmpty()) {
        //QJsonArray json = parseJSON(url, 0).toJsonArray();
        //qCDebug(scriptengine) << "first asset name: " << json.at(0).toObject().value("displayName");
        QByteArray json = parseJSON(url, 0).toJsonDocument().toJson();
        qCDebug(scriptengine) << "asset list as string is: " << json;
        return json;
    } else {
        qCDebug(scriptengine) << "Invalid filters were specified.";
        //return QJsonArray();
        return "";
    }
}

/*QJsonArray GooglePolyScriptingInterface::getAssetList(QString keyword, QString category, QString format) {
    QUrl url = formatURLQuery(keyword, category, format);
    if (!url.isEmpty()) {
        //QJsonArray json = parseJSON(url, 0).toJsonArray();
        //qCDebug(scriptengine) << "first asset name: " << json.at(0).toObject().value("displayName");
        QJsonArray json = parseJSON(url, 0).toJsonArray();
        qCDebug(scriptengine) << "asset list as array is: " << json;
        return json;
    } else {
        qCDebug(scriptengine) << "Invalid filters were specified.";
        return QJsonArray();
        //return "";
    }
}*/

QString GooglePolyScriptingInterface::getFBX(QString keyword, QString category) {
    QUrl url = formatURLQuery(keyword, category, "FBX");
    return getModelURL(url);
}

QString GooglePolyScriptingInterface::getOBJ(QString keyword, QString category) {
    QUrl url = formatURLQuery(keyword, category, "OBJ");
    return getModelURL(url);
}

QString GooglePolyScriptingInterface::getBlocks(QString keyword, QString category) {
    QUrl url = formatURLQuery(keyword, category, "BLOCKS");
    return getModelURL(url);
}

QString GooglePolyScriptingInterface::getGLTF(QString keyword, QString category) {
    QUrl url = formatURLQuery(keyword, category, "GLTF");
    return getModelURL(url);
}

QString GooglePolyScriptingInterface::getGLTF2(QString keyword, QString category) {
    QUrl url = formatURLQuery(keyword, category, "GLTF2");
    return getModelURL(url);
}

/* 
// This method will not work until we support Tilt models
QString GooglePolyScriptingInterface::getTilt(QString keyword, QString category) {
    QUrl url = formatURLQuery(keyword, category, "TILT");
    return getModelURL(url);
}
*/

// can provide asset name or full URL to model
QString GooglePolyScriptingInterface::getModelInfo(QString name) {
    if (name.contains("poly.googleapis") || name.contains("poly.google.com")) {
        QStringList list = name.split("/");
        name = list[4];
    }
    QString urlString = QString(getPolyUrl);
    urlString = urlString.replace("model", name) + "key=" + authCode;
    qCDebug(scriptengine) << "Google URL request: " << urlString;
    QUrl url(urlString);
    QString json = parseJSON(url, 2).toString();
    return json;
}

void GooglePolyScriptingInterface::testPrint(QString input) {
    qCDebug(scriptengine) << "Google message: " << input;
}

int GooglePolyScriptingInterface::getRandIntInRange(int length) {
    QTime time = QTime::currentTime();
    qsrand((uint)time.msec());
    return qrand() % length;
}

QUrl GooglePolyScriptingInterface::formatURLQuery(QString keyword, QString category, QString format) {
    QString queries;
    if (!validFormats.contains(format) || !validCategories.contains(category)) {
        return QUrl("");
    } else {
        if (!keyword.isEmpty()) {
            keyword.replace(" ", "+");
            queries.append("&keywords=" + keyword);
        }
        if (!category.isEmpty()) {
            queries.append("&category=" + category);
        }
        if (!format.isEmpty()) {
            queries.append("&format=" + format);
        }
        QString urlString = QString(listPolyUrl + "key=" + authCode + queries);
        return QUrl(urlString);
    }
}

QString GooglePolyScriptingInterface::getModelURL(QUrl url) {
    qCDebug(scriptengine) << "Google URL request: " << url;
    if (!url.isEmpty()) {
        return parseJSON(url, 1).toString();
    } else {
        qCDebug(scriptengine) << "Invalid filters were specified.";
        return "";
    }
}

// FIXME: synchronous
QByteArray GooglePolyScriptingInterface::getHTTPRequest(QUrl url) {
    QNetworkAccessManager manager;
    QNetworkReply *response = manager.get(QNetworkRequest(url));
    QEventLoop event;
    connect(response, SIGNAL(finished()), &event, SLOT(quit()));
    event.exec();

    return response->readAll();
}

// since the list is a QJsonArray and a single model is a QJsonObject
QVariant GooglePolyScriptingInterface::parseJSON(QUrl url, int fileType) {
    //QString firstItem = QString::fromUtf8(response->readAll());
    QByteArray jsonString = getHTTPRequest(url);
    QJsonDocument doc = QJsonDocument::fromJson(jsonString);
    QJsonObject obj = doc.object();
    if (obj.keys().first() == "error") {
        QString error = obj.value("error").toObject().value("message").toString();
        qCDebug(scriptengine) << error;
        return "";
    }
    if (fileType == 0 || fileType == 1) {
        QJsonArray arr = obj.value("assets").toArray();
        // return model url
        if (fileType == 1) {
            int random = getRandIntInRange(arr.size());
            QJsonObject json = arr.at(random).toObject();
            // nested JSONs
            return json.value("formats").toArray().at(0).toObject().value("root").toObject().value("url");
        }
        // return whole asset list
        //return arr;
        return QJsonDocument(arr);
    // return specific object
    } else {
        return jsonString;
    }
}


/*

// FIXME: synchronous = not good code
QByteArray GooglePolyScriptingInterface::getHTTPRequest(QUrl url) {
    QNetworkAccessManager manager;
    QNetworkReply *response = manager.get(QNetworkRequest(url));
    QEventLoop event;
    connect(response, SIGNAL(finished()), &event, SLOT(quit()));
    event.exec();
    return response->readAll();
}



/* NONE OF THESE HTTP GET METHODS WORK D:
https://stackoverflow.com/questions/28267477/getting-a-page-content-with-qt kinda used rn

this correctly returns the asset list but is apparently synchronous and not a good way to do it 
https://stackoverflow.com/questions/24965972/qt-getting-source-html-code-of-a-web-page-hosted-on-the-internet
void GooglePolyScriptingInterface::getAssetList() {
    authCode = "AIzaSyDamk7Vth52j7aU9JVKn3ungFS0kGJYc8A";
    QUrl url(listPolyUrl + "key=" + authCode);
    QNetworkAccessManager manager;
    QNetworkReply *response = manager.get(QNetworkRequest(url));
    QEventLoop event;
    connect(response, SIGNAL(finished()), &event, SLOT(quit()));
    event.exec();
    QString firstItem = response->readAll();
    //QJsonArray arr;
    //QJsonObject jsonObject = QJsonDocument::fromJson(response->readAll()).object();
    //QString firstItem = jsonObject["assets"].toString();
    qCDebug(scriptengine) << "first item: " << firstItem;
    //qCDebug(scriptengine) << "api key: " << authCode;
    //return arr;
}

this didn't work either https://stackoverflow.com/a/24966317
    QScopedPointer<QNetworkAccessManager> manager(new QNetworkAccessManager);
    QNetworkReply* response = manager->get(QNetworkRequest(url));
    QObject::connect(manager, &QNetworkAccessManager::finished, [manager, response] {
    response->deleteLater();
    manager->deleteLater();
    if (response->error() != QNetworkReply::NoError) return;
    QString contentType =
    response->header(QNetworkRequest::ContentTypeHeader).toString();
    if (!contentType.contains("charset=utf-8")) {
    qWarning() << "Content charsets other than utf-8 are not implemented yet.";
    return;
    }
    QString html = QString::fromUtf8(response->readAll());
    // do something with the data
    }) && manager.take();


or this :( https://stackoverflow.com/questions/39518749/make-get-request-to-web-service-get-json-response-and-update-gui-in-qt
    qCDebug(scriptengine) << "gonna get assets with " << url;
    QNetworkAccessManager manager;
    QNetworkReply* reply = manager.get(QNetworkRequest(url));
    QObject::connect(&manager, &QNetworkAccessManager::finished,
                    [reply]{
        qCDebug(scriptengine) << "boo radley";
        if (reply->error() != QNetworkReply::NoError) {
            qCDebug(scriptengine) << "Poly API failed to respond";
            //manager.clearAccessCache();
        } else {
            QJsonObject jsonObject = QJsonDocument::fromJson(reply->readAll()).object();
            QString firstItem = jsonObject["assets"].toString();
            qCDebug(scriptengine) << "first item: " << firstItem;
        }
        reply->deleteLater();
    });

or this :(( http://blog.mathieu-leplatre.info/access-a-json-webservice-with-qt-c.html
    //QJsonArray arr;
    //QJsonObject jsonObject = QJsonDocument::fromJson(response->readAll()).object();
    //QString firstItem = jsonObject["assets"].toString();
    qCDebug(scriptengine) << "first item: " << firstItem;
    //qCDebug(scriptengine) << "api key: " << authCode;
    //return arr;
    QNetworkAccessManager manager;
    QNetworkReply* response = manager.get(QNetworkRequest(url));
    QObject::connect(&manager, SIGNAL(finished(QNetworkReply*)), this, SLOT(onResult(QNetworkReply*)));

    return "";
    }

    void GooglePolyScriptingInterface::onResult(QNetworkReply* reply) {
    if (reply->error() != QNetworkReply::NoError) {
    qCDebug(scriptengine) << "Poly API failed to respond";
    return;
    }
    QString firstItem = reply->readAll();

    qCDebug(scriptengine) << "first item: " << firstItem;
    }

*/