[iOS] Rename WKPDFView to WKLegacyPDFView
[WebKit-https.git] / Source / WebKit / UIProcess / Cocoa / WKWebViewContentProviderRegistry.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 "WKWebViewContentProviderRegistry.h"
28
29 #if WK_API_ENABLED
30
31 #if PLATFORM(IOS)
32
33 #import "WKLegacyPDFView.h"
34 #import "WKSystemPreviewView.h"
35 #import "WKWebViewInternal.h"
36 #import "WebPageProxy.h"
37 #import <WebCore/MIMETypeRegistry.h>
38 #import <wtf/HashCountedSet.h>
39 #import <wtf/HashMap.h>
40 #import <wtf/text/StringHash.h>
41 #import <wtf/text/WTFString.h>
42
43 #if USE(APPLE_INTERNAL_SDK)
44 #import <WebKitAdditions/SystemPreviewTypes.cpp>
45 #endif
46
47 using namespace WebKit;
48
49 @implementation WKWebViewContentProviderRegistry {
50     HashMap<String, Class <WKWebViewContentProvider>, ASCIICaseInsensitiveHash> _contentProviderForMIMEType;
51     HashCountedSet<WebPageProxy*> _pages;
52 }
53
54 - (instancetype)init
55 {
56     if (!(self = [super init]))
57         return nil;
58
59 #if ENABLE(WKPDFVIEW)
60     for (auto& mimeType : WebCore::MIMETypeRegistry::getPDFMIMETypes())
61         [self registerProvider:[WKLegacyPDFView class] forMIMEType:mimeType];
62 #endif
63
64 #if PLATFORM(IOS) && USE(QUICK_LOOK) && USE(APPLE_INTERNAL_SDK)
65     for (auto& mimeType : getSystemPreviewMIMETypes())
66         [self registerProvider:[WKSystemPreviewView class] forMIMEType:mimeType];
67 #endif
68
69     return self;
70 }
71
72 - (void)addPage:(WebPageProxy&)page
73 {
74     ASSERT(!_pages.contains(&page));
75     _pages.add(&page);
76 }
77
78 - (void)removePage:(WebPageProxy&)page
79 {
80     ASSERT(_pages.contains(&page));
81     _pages.remove(&page);
82 }
83
84 - (void)registerProvider:(Class <WKWebViewContentProvider>)contentProvider forMIMEType:(const String&)mimeType
85 {
86     _contentProviderForMIMEType.set(mimeType, contentProvider);
87
88     for (auto& page : _pages)
89         page.key->addMIMETypeWithCustomContentProvider(mimeType);
90 }
91
92 - (Class <WKWebViewContentProvider>)providerForMIMEType:(const String&)mimeType
93 {
94     const auto& representation = _contentProviderForMIMEType.find(mimeType);
95
96     if (representation == _contentProviderForMIMEType.end())
97         return nil;
98
99     return representation->value;
100 }
101
102 - (Vector<String>)_mimeTypesWithCustomContentProviders
103 {
104     return copyToVector(_contentProviderForMIMEType.keys());
105 }
106
107 @end
108
109 #endif // PLATFORM(IOS)
110
111 #endif // WK_API_ENABLED