Index: tools/skiaserve/urlhandlers/ImgHandler.cpp |
diff --git a/tools/skiaserve/urlhandlers/ImgHandler.cpp b/tools/skiaserve/urlhandlers/ImgHandler.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..68828267fbdf6a44384c0f3f4b4765ac60a81e95 |
--- /dev/null |
+++ b/tools/skiaserve/urlhandlers/ImgHandler.cpp |
@@ -0,0 +1,43 @@ |
+/* |
+ * 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 "../Request.h" |
+#include "../Response.h" |
+ |
+using namespace Response; |
+ |
+bool ImgHandler::canHandle(const char* method, const char* url) { |
+ static const char* kBasePath = "/img"; |
+ return 0 == strcmp(method, MHD_HTTP_METHOD_GET) && |
+ 0 == strncmp(url, kBasePath, strlen(kBasePath)); |
+} |
+ |
+int ImgHandler::handle(Request* request, MHD_Connection* connection, |
+ const char* url, const char* method, |
+ const char* upload_data, size_t* upload_data_size) { |
+ SkTArray<SkString> commands; |
+ SkStrSplit(url, "/", &commands); |
+ |
+ if (!request->fPicture.get() || commands.count() > 2) { |
+ return MHD_NO; |
+ } |
+ |
+ int n; |
+ // /img or /img/N |
+ if (commands.count() == 1) { |
+ n = request->fDebugCanvas->getSize() - 1; |
+ } else { |
+ sscanf(commands[1].c_str(), "%d", &n); |
+ } |
+ |
+ SkAutoTUnref<SkData> data(request->drawToPng(n)); |
+ return SendData(connection, data, "image/png"); |
+} |
+ |