summaryrefslogtreecommitdiff
path: root/common/foundation_request.mm
blob: caf5d13e237eb5f5038ccce0f0da2fb5f58e25b6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#import "foundation_request.h"

#include "TargetConditionals.h"
#if TARGET_OS_IPHONE
#import <UIKit/UIKit.h>
#include <atomic>
#endif

#include <memory>
#include <string>
#include <functional>
#include <llmr/platform/platform.hpp>
#include <uv.h>


uv_once_t request_initialize = UV_ONCE_INIT;
dispatch_queue_t queue = nullptr;
NSURLSession *session = nullptr;


#if TARGET_OS_IPHONE
std::atomic<int> active_tasks;
#endif

void request_initialize_cb() {
    NSString *queueName = [[[NSBundle mainBundle] bundleIdentifier] stringByAppendingString:@".parsing"];

    queue = dispatch_queue_create([queueName UTF8String], DISPATCH_QUEUE_CONCURRENT);

    NSURLSessionConfiguration *sessionConfig = [NSURLSessionConfiguration defaultSessionConfiguration];
    sessionConfig.timeoutIntervalForResource = 6;
    sessionConfig.HTTPMaximumConnectionsPerHost = 8;
    sessionConfig.requestCachePolicy = NSURLRequestUseProtocolCachePolicy;

    session = [NSURLSession sessionWithConfiguration:sessionConfig];

#if TARGET_OS_IPHONE
    active_tasks = 0;
#endif
}

namespace llmr {

class platform::Request {
public:
    Request(NSURLSessionDataTask *task, const std::string &original_url)
        : task(task), original_url(original_url) {
        #if TARGET_OS_IPHONE
            active_tasks++;
            [[UIApplication sharedApplication] setNetworkActivityIndicatorVisible:active_tasks];
        #endif
    }

#if TARGET_OS_IPHONE
    ~Request() {
        active_tasks--;
        [[UIApplication sharedApplication] setNetworkActivityIndicatorVisible:active_tasks];
    }
#endif

    NSURLSessionDataTask *task;
    std::string original_url;
};

std::shared_ptr<platform::Request>
platform::request_http(const std::string &url, std::function<void(Response *)> background_function,
                       std::function<void()> foreground_callback) {

    uv_once(&request_initialize, request_initialize_cb);

    __block std::shared_ptr<Request> req;
    NSURLSessionDataTask *task = [session
                                  dataTaskWithURL:[NSURL URLWithString:@(url.c_str())]
                                  completionHandler:^(NSData *data, NSURLResponse *response, NSError *error) {
        if ([error code] == NSURLErrorCancelled) {
            // We intentionally cancelled this request.
            // Make sure we clear the shared_ptr to resolve the circular reference. We're referencing
            // this shared_ptr by value so that the object stays around until this completion handler is
            // invoked.
            req.reset();

            return;
        }

        dispatch_async(queue, ^(){
            Response res;

            if (!error && [response isKindOfClass:[NSHTTPURLResponse class]]) {
                res.code = [(NSHTTPURLResponse *)response statusCode];
                res.body = {(const char *)[data bytes], [data length]};
            } else {
                res.code = -1;
                res.error_message = [[error localizedDescription] UTF8String];
            }

            background_function(&res);

            dispatch_async(dispatch_get_main_queue(), ^(void) {
                // Make sure we clear the shared_ptr to resolve the circular reference. We're referencing
                // this shared_ptr by value so that the object stays around until this completion handler is
                // invoked.
                req.reset();

                foreground_callback();
            });
        });
    }];

    req = std::make_shared<Request>(task, url);
    [task resume];
    return req;
}

void platform::cancel_request_http(const std::shared_ptr<Request> &req) {
    if (req) {
        [req->task cancel];
    }
}

}