Home | History | Annotate | Download | only in urlhandlers
      1 /*
      2  * Copyright 2016 Google Inc.
      3  *
      4  * Use of this source code is governed by a BSD-style license that can be
      5  * found in the LICENSE file.
      6  */
      7 
      8 #include "UrlHandler.h"
      9 
     10 #include "microhttpd.h"
     11 #include "SkPictureRecorder.h"
     12 #include "SkPixelSerializer.h"
     13 #include "../Request.h"
     14 #include "../Response.h"
     15 
     16 using namespace Response;
     17 
     18 bool DownloadHandler::canHandle(const char* method, const char* url) {
     19     return 0 == strcmp(method, MHD_HTTP_METHOD_GET) &&
     20            0 == strcmp(url, "/download");
     21 }
     22 
     23 int DownloadHandler::handle(Request* request, MHD_Connection* connection,
     24                             const char* url, const char* method,
     25                             const char* upload_data, size_t* upload_data_size) {
     26     if (!request->fPicture.get()) {
     27         return MHD_NO;
     28     }
     29 
     30     // TODO move to a function
     31     // Playback into picture recorder
     32     SkPictureRecorder recorder;
     33     SkCanvas* canvas = recorder.beginRecording(Request::kImageWidth,
     34                                                Request::kImageHeight);
     35 
     36     request->fDebugCanvas->draw(canvas);
     37 
     38     SkAutoTUnref<SkPicture> picture(recorder.endRecording());
     39 
     40     SkDynamicMemoryWStream outStream;
     41 
     42     SkAutoTUnref<SkPixelSerializer> serializer(SkImageEncoder::CreatePixelSerializer());
     43     picture->serialize(&outStream, serializer);
     44 
     45     SkAutoTUnref<SkData> data(outStream.copyToData());
     46 
     47     // TODO fancier name handling
     48     return SendData(connection, data, "application/octet-stream", true,
     49                     "attachment; filename=something.skp;");
     50 }
     51 
     52