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
16using namespace Response;
17
18bool DownloadHandler::canHandle(const char* method, const char* url) {
19    return 0 == strcmp(method, MHD_HTTP_METHOD_GET) &&
20           0 == strcmp(url, "/download");
21}
22
23int 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