1 files changed, 43 insertions, 0 deletions
diff --git a/tools/skiaserve/urlhandlers/ImgHandler.cpp b/tools/skiaserve/urlhandlers/ImgHandler.cpp
new file mode 100644
index 0000000000..68828267fb
--- /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");
+}
+