Refactor WebContentReader out of EditorMac and EditorIOS
[WebKit-https.git] / Source / WebCore / editing / ios / WebContentReaderIOS.mm
1 /*
2  * Copyright (C) 2006-2017 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. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27 #import "WebContentReader.h"
28
29 #import "ArchiveResource.h"
30 #import "DOMURL.h"
31 #import "Document.h"
32 #import "DocumentFragment.h"
33 #import "DocumentLoader.h"
34 #import "Editor.h"
35 #import "EditorClient.h"
36 #import "FragmentScriptingPermission.h"
37 #import "FrameLoader.h"
38 #import "FrameLoaderClient.h"
39 #import "HTMLAnchorElement.h"
40 #import "HTMLNames.h"
41 #import "LegacyWebArchive.h"
42 #import "MIMETypeRegistry.h"
43 #import "Text.h"
44 #import "UTIUtilities.h"
45 #import "WebNSAttributedStringExtras.h"
46 #import "markup.h"
47 #import <MobileCoreServices/MobileCoreServices.h>
48 #import <pal/spi/cocoa/NSAttributedStringSPI.h>
49 #import <wtf/unicode/CharacterNames.h>
50
51 namespace WebCore {
52
53 void WebContentReader::addFragment(RefPtr<DocumentFragment>&& newFragment)
54 {
55     if (!newFragment)
56         return;
57
58     if (!fragment) {
59         fragment = WTFMove(newFragment);
60         return;
61     }
62
63     while (auto* firstChild = newFragment->firstChild()) {
64         if (fragment->appendChild(*firstChild).hasException())
65             break;
66     }
67 }
68
69 bool WebContentReader::readWebArchive(SharedBuffer* buffer)
70 {
71     if (!frame.document())
72         return false;
73
74     if (!buffer)
75         return false;
76
77     auto archive = LegacyWebArchive::create(URL(), *buffer);
78     if (!archive)
79         return false;
80
81     auto* mainResource = archive->mainResource();
82     if (!mainResource)
83         return false;
84
85     auto& type = mainResource->mimeType();
86     if (!frame.loader().client().canShowMIMETypeAsHTML(type))
87         return false;
88
89     // FIXME: The code in createFragmentAndAddResources calls setDefersLoading(true). Don't we need that here?
90     if (auto* loader = frame.loader().documentLoader())
91         loader->addAllArchiveResources(*archive);
92
93     auto markupString = String::fromUTF8(mainResource->data().data(), mainResource->data().size());
94     addFragment(createFragmentFromMarkup(*frame.document(), markupString, mainResource->url(), DisallowScriptingAndPluginContent));
95     return true;
96 }
97
98 bool WebContentReader::readFilenames(const Vector<String>&)
99 {
100     return false;
101 }
102
103 bool WebContentReader::readHTML(const String& string)
104 {
105     if (!frame.document())
106         return false;
107
108     addFragment(createFragmentFromMarkup(*frame.document(), string, emptyString(), DisallowScriptingAndPluginContent));
109     return true;
110 }
111
112 bool WebContentReader::readRTFD(SharedBuffer& buffer)
113 {
114     addFragment(createFragmentAndAddResources(frame, adoptNS([[NSAttributedString alloc] initWithRTFD:buffer.createNSData().get() documentAttributes:nullptr]).get()));
115     return fragment;
116 }
117
118 bool WebContentReader::readRTF(SharedBuffer& buffer)
119 {
120     addFragment(createFragmentAndAddResources(frame, adoptNS([[NSAttributedString alloc] initWithRTF:buffer.createNSData().get() documentAttributes:nullptr]).get()));
121     return fragment;
122 }
123
124 bool WebContentReader::readImage(Ref<SharedBuffer>&& buffer, const String& type)
125 {
126     RetainPtr<CFStringRef> stringType = type.createCFString();
127     RetainPtr<NSString> filenameExtension = adoptNS((NSString *)UTTypeCopyPreferredTagWithClass(stringType.get(), kUTTagClassFilenameExtension));
128     NSString *relativeURLPart = [@"image" stringByAppendingString:filenameExtension.get()];
129     String mimeType = MIMETypeFromUTI(type);
130
131     // FIXME: Use a blob URL instead.
132     auto archive = ArchiveResource::create(WTFMove(buffer), URL::fakeURLWithRelativePart(relativeURLPart), mimeType, emptyString(), emptyString());
133     ASSERT(archive);
134     addFragment(createFragmentForImageResourceAndAddResource(frame, *archive));
135     return fragment;
136 }
137
138 bool WebContentReader::readURL(const URL& url, const String& title)
139 {
140     if (url.isEmpty())
141         return false;
142
143     // FIXME: This code shoudln't be accessing selection and changing the behavior.
144     if (!frame.editor().client()->hasRichlyEditableSelection()) {
145         if (readPlainText([(NSURL *)url absoluteString]))
146             return true;
147     }
148
149     if ([(NSURL *)url isFileURL])
150         return false;
151
152     auto anchor = HTMLAnchorElement::create(*frame.document());
153     anchor->setAttributeWithoutSynchronization(HTMLNames::hrefAttr, url.string());
154
155     String linkText = title.length() ? title : String([[(NSURL *)url absoluteString] precomposedStringWithCanonicalMapping]);
156     anchor->appendChild(frame.document()->createTextNode(linkText));
157
158     auto newFragment = frame.document()->createDocumentFragment();
159     if (fragment)
160         newFragment->appendChild(Text::create(*frame.document(), { &space, 1 }));
161     newFragment->appendChild(anchor);
162     addFragment(WTFMove(newFragment));
163     return true;
164 }
165
166 bool WebContentReader::readPlainText(const String& text)
167 {
168     if (!allowPlainText)
169         return false;
170
171     addFragment(createFragmentFromText(context, [text precomposedStringWithCanonicalMapping]));
172     if (!fragment)
173         return false;
174
175     madeFragmentFromPlainText = true;
176     return true;
177 }
178
179 }