fed92095985d1eacf5300a03a4e367303cd8b5a2
[WebKit-https.git] / Source / WebKit2 / UIProcess / API / Cocoa / WKWebsiteDataStore.mm
1 /*
2  * Copyright (C) 2014 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27 #import "WKWebsiteDataStoreInternal.h"
28
29 #if WK_API_ENABLED
30
31 #import "WKNSArray.h"
32 #import "WKWebsiteDataRecordInternal.h"
33
34 @implementation WKWebsiteDataStore
35
36 + (WKWebsiteDataStore *)defaultDataStore
37 {
38     return WebKit::wrapper(*API::WebsiteDataStore::defaultDataStore().get());
39 }
40
41 + (WKWebsiteDataStore *)nonPersistentDataStore
42 {
43     return [WebKit::wrapper(*API::WebsiteDataStore::createNonPersistentDataStore().release().leakRef()) autorelease];
44 }
45
46 - (void)dealloc
47 {
48     _websiteDataStore->API::WebsiteDataStore::~WebsiteDataStore();
49
50     [super dealloc];
51 }
52
53 - (BOOL)isNonPersistent
54 {
55     return _websiteDataStore->isNonPersistent();
56 }
57
58 + (NSSet *)allWebsiteDataTypes
59 {
60     static dispatch_once_t onceToken;
61     static NSSet *allWebsiteDataTypes;
62     dispatch_once(&onceToken, ^{
63         allWebsiteDataTypes = [[NSSet alloc] initWithArray:@[ WKWebsiteDataTypeDiskCache, WKWebsiteDataTypeMemoryCache, WKWebsiteDataTypeOfflineWebApplicationCache, WKWebsiteDataTypeCookies, WKWebsiteDataTypeLocalStorage, WKWebsiteDataTypeWebSQLDatabases ]];
64     });
65
66     return allWebsiteDataTypes;
67 }
68
69 static std::chrono::system_clock::time_point toSystemClockTime(NSDate *date)
70 {
71     ASSERT(date);
72     using namespace std::chrono;
73
74     return system_clock::time_point(duration_cast<system_clock::duration>(duration<double>(date.timeIntervalSince1970)));
75 }
76
77 - (void)fetchDataRecordsOfTypes:(NSSet *)websiteDataTypes completionHandler:(void (^)(NSArray *))completionHandler
78 {
79     auto completionHandlerCopy = Block_copy(completionHandler);
80
81     _websiteDataStore->websiteDataStore().fetchData(WebKit::toWebsiteDataTypes(websiteDataTypes), [completionHandlerCopy](Vector<WebKit::WebsiteDataRecord> websiteDataRecords) {
82         Vector<RefPtr<API::Object>> elements;
83         elements.reserveInitialCapacity(websiteDataRecords.size());
84
85         for (auto& websiteDataRecord : websiteDataRecords)
86             elements.uncheckedAppend(API::WebsiteDataRecord::create(WTF::move(websiteDataRecord)));
87
88         completionHandlerCopy(wrapper(API::Array::create(WTF::move(elements))));
89
90         Block_release(completionHandlerCopy);
91     });
92 }
93
94 - (void)removeDataOfTypes:(NSSet *)websiteDataTypes modifiedSince:(NSDate *)date completionHandler:(void (^)(void))completionHandler
95 {
96     auto completionHandlerCopy = Block_copy(completionHandler);
97     _websiteDataStore->websiteDataStore().removeData(WebKit::toWebsiteDataTypes(websiteDataTypes), toSystemClockTime(date ? date : [NSDate distantPast]), [completionHandlerCopy] {
98         completionHandlerCopy();
99         Block_release(completionHandlerCopy);
100     });
101 }
102
103 static Vector<WebKit::WebsiteDataRecord> toWebsiteDataRecords(NSArray *dataRecords)
104 {
105     Vector<WebKit::WebsiteDataRecord> result;
106
107     for (WKWebsiteDataRecord *dataRecord in dataRecords)
108         result.append(dataRecord->_websiteDataRecord->websiteDataRecord());
109
110     return result;
111 }
112
113 - (void)removeDataOfTypes:(NSSet *)websiteDataTypes forDataRecords:(NSArray *)dataRecords completionHandler:(void (^)(void))completionHandler
114 {
115     auto completionHandlerCopy = Block_copy(completionHandler);
116
117     _websiteDataStore->websiteDataStore().removeData(WebKit::toWebsiteDataTypes(websiteDataTypes), toWebsiteDataRecords(dataRecords), [completionHandlerCopy] {
118         completionHandlerCopy();
119         Block_release(completionHandlerCopy);
120     });
121 }
122
123 #pragma mark WKObject protocol implementation
124
125 - (API::Object&)_apiObject
126 {
127     return *_websiteDataStore;
128 }
129
130 @end
131
132 #endif // WK_API_ENABLED