From db3524a48fcc962ad3ce764d6b1ba2eb20a47a12 Mon Sep 17 00:00:00 2001
From: Stephen Birarda <commit@birarda.com>
Date: Fri, 11 Aug 2017 14:26:27 -0700
Subject: [PATCH] add re-direct of get mapping for baked assets

---
 assignment-client/src/assets/AssetServer.cpp  | 53 +++++++++++++++++--
 .../networking/src/udt/PacketHeaders.cpp      |  2 +
 libraries/networking/src/udt/PacketHeaders.h  |  3 +-
 3 files changed, 54 insertions(+), 4 deletions(-)

diff --git a/assignment-client/src/assets/AssetServer.cpp b/assignment-client/src/assets/AssetServer.cpp
index 911a650cf2..c0729dcf53 100644
--- a/assignment-client/src/assets/AssetServer.cpp
+++ b/assignment-client/src/assets/AssetServer.cpp
@@ -23,6 +23,7 @@
 #include <QtCore/QFileInfo>
 #include <QtCore/QJsonDocument>
 #include <QtCore/QString>
+#include <QtGui/QImageReader>
 
 #include <SharedUtil.h>
 #include <PathUtils.h>
@@ -264,14 +265,61 @@ void AssetServer::handleAssetMappingOperation(QSharedPointer<ReceivedMessage> me
     nodeList->sendPacketList(std::move(replyPacket), *senderNode);
 }
 
+static const QStringList BAKEABLE_MODEL_EXTENSIONS = { ".fbx" };
+static const QString BAKED_MODEL_SIMPLE_NAME = "asset.fbx";
+static const QString BAKED_TEXTURE_SIMPLE_NAME = "texture.ktx";
+
 void AssetServer::handleGetMappingOperation(ReceivedMessage& message, SharedNodePointer senderNode, NLPacketList& replyPacket) {
     QString assetPath = message.readString();
 
     auto it = _fileMappings.find(assetPath);
     if (it != _fileMappings.end()) {
-        auto assetHash = it->toString();
+
+        // check if we should re-direct to a baked asset
+
+        // first, figure out from the mapping extension what type of file this is
+        auto assetPathExtension = assetPath.right(assetPath.lastIndexOf('.')).toLower();
+        QString bakedRootFile;
+
+        if (BAKEABLE_MODEL_EXTENSIONS.contains(assetPathExtension)) {
+            bakedRootFile = BAKED_MODEL_SIMPLE_NAME;
+        } else if (QImageReader::supportedImageFormats().contains(assetPathExtension.toLocal8Bit())) {
+            bakedRootFile = BAKED_TEXTURE_SIMPLE_NAME;
+        }
+
+        auto originalAssetHash = it->toString();
+        QString redirectedAssetHash;
+        QString bakedAssetPath;
+        quint8 wasRedirected = false;
+
+        if (!bakedRootFile.isEmpty()) {
+            // we ran into an asset for which we could have a baked version, let's check if it's ready
+            bakedAssetPath = "/.baked/" + originalAssetHash + "/" + bakedRootFile;
+            auto bakedIt = _fileMappings.find(bakedAssetPath);
+
+            if (bakedIt != _fileMappings.end()) {
+                // we found a baked version of the requested asset to serve, redirect to that
+                redirectedAssetHash = bakedIt->toString();
+                wasRedirected = true;
+            }
+        }
+
         replyPacket.writePrimitive(AssetServerError::NoError);
-        replyPacket.write(QByteArray::fromHex(assetHash.toUtf8()));
+
+        if (wasRedirected) {
+            qDebug() << "Writing re-directed hash for" << originalAssetHash << "to" << redirectedAssetHash;
+            replyPacket.write(QByteArray::fromHex(redirectedAssetHash.toUtf8()));
+
+            // add a flag saying that this mapping request was redirect
+            replyPacket.writePrimitive(wasRedirected);
+
+            // include the re-directed path in case the caller needs to make relative path requests for the baked asset
+            replyPacket.write(bakedAssetPath.toUtf8());
+
+        } else {
+            replyPacket.write(QByteArray::fromHex(originalAssetHash.toUtf8()));
+            replyPacket.writePrimitive(wasRedirected);
+        }
     } else {
         replyPacket.writePrimitive(AssetServerError::AssetNotFound);
     }
@@ -826,7 +874,6 @@ void AssetServer::handleCompletedBake(AssetHash originalAssetHash, QDir temporar
 
             // setup the mapping for this bake file
             auto relativeFilePath = temporaryOutputDir.relativeFilePath(filePath);
-
             static const QString BAKED_ASSET_SIMPLE_NAME = "asset.fbx";
 
             if (relativeFilePath.endsWith(".fbx", Qt::CaseInsensitive)) {
diff --git a/libraries/networking/src/udt/PacketHeaders.cpp b/libraries/networking/src/udt/PacketHeaders.cpp
index 241ccaf5d6..f453d096b6 100644
--- a/libraries/networking/src/udt/PacketHeaders.cpp
+++ b/libraries/networking/src/udt/PacketHeaders.cpp
@@ -42,6 +42,8 @@ PacketVersion versionForPacketType(PacketType packetType) {
             return static_cast<PacketVersion>(MessageDataVersion::TextOrBinaryData);
         case PacketType::ICEServerHeartbeat:
             return 18; // ICE Server Heartbeat signing
+        case PacketType::AssetMappingOperationReply:
+            return static_cast<PacketVersion>(AssetServerPacketVersion::RedirectedMappings);
         case PacketType::AssetGetInfo:
         case PacketType::AssetGet:
         case PacketType::AssetUpload:
diff --git a/libraries/networking/src/udt/PacketHeaders.h b/libraries/networking/src/udt/PacketHeaders.h
index e2304e62f7..ea831bd415 100644
--- a/libraries/networking/src/udt/PacketHeaders.h
+++ b/libraries/networking/src/udt/PacketHeaders.h
@@ -266,7 +266,8 @@ enum class EntityQueryPacketVersion: PacketVersion {
 
 enum class AssetServerPacketVersion: PacketVersion {
     VegasCongestionControl = 19,
-    RangeRequestSupport
+    RangeRequestSupport,
+    RedirectedMappings
 };
 
 enum class AvatarMixerPacketVersion : PacketVersion {