Index: tools/skiaserve/urlhandlers/DownloadHandler.cpp |
diff --git a/tools/skiaserve/urlhandlers/DownloadHandler.cpp b/tools/skiaserve/urlhandlers/DownloadHandler.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..719a38249472223df7a531f6b8f558680ecbf77b |
--- /dev/null |
+++ b/tools/skiaserve/urlhandlers/DownloadHandler.cpp |
@@ -0,0 +1,51 @@ |
+/* |
+ * Copyright 2016 Google Inc. |
+ * |
+ * Use of this source code is governed by a BSD-style license that can be |
+ * found in the LICENSE file. |
+ */ |
+ |
+#include "UrlHandler.h" |
+ |
+#include "microhttpd.h" |
+#include "SkPictureRecorder.h" |
+#include "SkPixelSerializer.h" |
+#include "../Request.h" |
+#include "../Response.h" |
+ |
+using namespace Response; |
+ |
+bool DownloadHandler::canHandle(const char* method, const char* url) { |
+ return 0 == strcmp(method, MHD_HTTP_METHOD_GET) && |
+ 0 == strcmp(url, "/download"); |
+} |
+ |
+int DownloadHandler::handle(Request* request, MHD_Connection* connection, |
+ const char* url, const char* method, |
+ const char* upload_data, size_t* upload_data_size) { |
+ if (!request->fPicture.get()) { |
+ return MHD_NO; |
+ } |
+ |
+ // TODO move to a function |
+ // Playback into picture recorder |
+ SkPictureRecorder recorder; |
+ SkCanvas* canvas = recorder.beginRecording(Request::kImageWidth, |
+ Request::kImageHeight); |
+ |
+ request->fDebugCanvas->draw(canvas); |
+ |
+ SkAutoTUnref<SkPicture> picture(recorder.endRecording()); |
+ |
+ SkDynamicMemoryWStream outStream; |
+ |
+ SkAutoTUnref<SkPixelSerializer> serializer(SkImageEncoder::CreatePixelSerializer()); |
+ picture->serialize(&outStream, serializer); |
+ |
+ SkAutoTUnref<SkData> data(outStream.copyToData()); |
+ |
+ // TODO fancier name handling |
+ return SendData(connection, data, "application/octet-stream", true, |
+ "attachment; filename=something.skp;"); |
+} |
+ |