added getAssetList() and getFBX()

interface/src/scripting/GooglePolyScriptingInterface.cpp

@@ -10,6 +10,7 @@
 //
 
 #include <QEventLoop>
+#include <QGlobal.h>
 #include <QJsonArray>
 #include <QJsonDocument>
 #include <QJsonObject>
@@ -18,7 +19,9 @@
 #include <QNetworkRequest>
 //#include <QScopedPointer>
 #include <QString>
+#include <QTime>
 #include <QUrl>
+#include <random>
 
 #include "GooglePolyScriptingInterface.h"
 #include "ScriptEngineLogging.h"
@@ -26,28 +29,81 @@
 QString listPolyUrl = "https://poly.googleapis.com/v1/assets?";
 QString getPolyUrl = "https://poly.googleapis.com/v1/assets/model?";
 
+//authCode = "broke";
+
+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::testPrint() {
+QJsonArray GooglePolyScriptingInterface::getAssetList(QString keyword, QString category, QString format) {
-    qCDebug(scriptengine) << "Google Poly interface exists";
+    QUrl url = formatURLQuery(keyword, category, format);
+    if (!url.isEmpty()) {
+        QByteArray jsonString = getHTTPRequest(url);
+        QJsonArray json = parseJSON(&jsonString, 0).toJsonArray();
+        qCDebug(scriptengine) << "first asset name: " << json.at(0).toObject().value("displayName");
+        return json;
+    } else {
+        qCDebug(scriptengine) << "Invalid filters were specified.";
+        return QJsonArray();
+    }
+}
+
+QString GooglePolyScriptingInterface::getFBX(QString keyword, QString category) {
+    QUrl url = formatURLQuery(keyword, category, "FBX");
+    qCDebug(scriptengine) << "google url: " << url;
+    if (!url.isEmpty()) {
+        QByteArray jsonString = getHTTPRequest(url);
+        qCDebug(scriptengine) << "the list: " << jsonString;
+        QString modelURL = parseJSON(&jsonString, 1).toString();
+        
+        qCDebug(scriptengine) << "the model url: " << modelURL;
+        return modelURL;
+    } else {
+        qCDebug(scriptengine) << "Invalid filters were specified.";
+        return "";
+    }
+
+}
+
+void GooglePolyScriptingInterface::testPrint(QString input) {
+    qCDebug(scriptengine) << "Google message: " << input;
 }
 
 void GooglePolyScriptingInterface::setAPIKey(QString key) {
     authCode = key;
 }
 
-void GooglePolyScriptingInterface::getAssetList() {
+int GooglePolyScriptingInterface::getRandIntInRange(int length) {
-    //authCode = "broke";
+    QTime time = QTime::currentTime();
-    QUrl url(listPolyUrl + "key=" + authCode);
+    qsrand((uint)time.msec());
-    QByteArray jsonString = getHTTPRequest(url);
+    return qrand() % length;
-    qCDebug(scriptengine) << "the list: " << jsonString;
-    QJsonObject json = makeJSON(&jsonString, true).toJsonObject;
-
 }
 
-// FIXME: synchronous = not good code
+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);
+    }
+}
+
+// FIXME: synchronous
 QByteArray GooglePolyScriptingInterface::getHTTPRequest(QUrl url) {
     QNetworkAccessManager manager;
     QNetworkReply *response = manager.get(QNetworkRequest(url));
@@ -56,31 +112,46 @@ QByteArray GooglePolyScriptingInterface::getHTTPRequest(QUrl url) {
     event.exec();
 
     return response->readAll();
-    
+
 }
 
 // since the list is a QJsonArray and a single model is a QJsonObject
-QVariant GooglePolyScriptingInterface::makeJSON(QByteArray* response, bool isList) {
+QVariant GooglePolyScriptingInterface::parseJSON(QByteArray* response, int fileType) {
     //QString firstItem = QString::fromUtf8(response->readAll());
     QJsonDocument doc = QJsonDocument::fromJson(*response);
-    qCDebug(scriptengine) << "json doc is empty: " << doc.isEmpty();
     QJsonObject obj = doc.object();
     qCDebug(scriptengine) << "json obj: " << obj;
     qCDebug(scriptengine) << "json list: " << obj.keys();
     if (obj.keys().first() == "error") {
         qCDebug(scriptengine) << "Invalid API key";
+        return QJsonObject();
+    }
+    qCDebug(scriptengine) << "the assets: " << obj.value("assets");
+    if (fileType == 0 || fileType == 1) {
+        QJsonArray arr = obj.value("assets").toArray();
+        qCDebug(scriptengine) << "in array: " << arr;
+        // return model url
+        if (fileType == 1) {
+            int random = getRandIntInRange(arr.size());
+            QJsonObject json = arr.at(random).toObject();
+            qCDebug(scriptengine) << random << "th object: " << json;
+            qCDebug(scriptengine) << random << "th asset name: " << json.value("displayName");
+            // nested JSONs
+            return json.value("formats").toArray().at(0).toObject().value("root").toObject().value("url");
+        }
+        // return whole asset list
+        return arr;
+    // return specific object
+    } else {
         return obj;
     }
-    qCDebug(scriptengine) << "total size: " << obj.value("totalSize");
-    qCDebug(scriptengine) << "the assets: " << obj.value("assets");
-    QJsonArray arr = obj.value("assets").toArray();
-    qCDebug(scriptengine) << "in array: " << arr;
-    QJsonObject first = arr.takeAt(0).toObject();
-    qCDebug(scriptengine) << "first asset: " << first;
-    qCDebug(scriptengine) << "first asset description: " << first.value("description");
-    return obj;
 }
 
+
+
+
+
+
 /*void GooglePolyScriptingInterface::getAssetList() {
     authCode = "AIzaSyDamk7Vth52j7aU9JVKn3ungFS0kGJYc8A";
     QUrl url(listPolyUrl + "key=" + authCode);

interface/src/scripting/GooglePolyScriptingInterface.h

@@ -21,18 +21,20 @@ public:
     GooglePolyScriptingInterface();
 
 public slots:
-    void testPrint();
+    void testPrint(QString input);
     void setAPIKey(QString key);
 
-    //QJsonArray GooglePolyScriptingInterface::getAssetList();
+    QJsonArray getAssetList(QString keyword, QString category, QString format);
-    void getAssetList();
+    QString getFBX(QString keyword, QString category);
 
 private:
+    QUrl formatURLQuery(QString keyword, QString category, QString format);
     QByteArray getHTTPRequest(QUrl url);
-    QVariant makeJSON(QByteArray* response, bool isList);
+    QVariant parseJSON(QByteArray* response, int fileType);
+    int getRandIntInRange(int length);
     //void onResult(QNetworkReply* reply);
 
-    QString authCode;
+    //QString authCode;
     
 
 };