cc17035a9100a9656c26191a084d76d8cf464ef2
[WebKit-https.git] / Source / WebCore / editing / WebContentReader.h
1 /*
2  * Copyright (C) 2013-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. 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 #pragma once
27
28 #include "Frame.h"
29 #include "Pasteboard.h"
30 #include "Range.h"
31
32 namespace WebCore {
33
34 class ArchiveResource;
35 class Blob;
36
37 class FrameWebContentReader : public PasteboardWebContentReader {
38 public:
39     Frame& frame;
40
41     FrameWebContentReader(Frame& frame)
42         : frame(frame)
43     {
44     }
45
46 protected:
47     bool shouldSanitize() const;
48 };
49
50 class WebContentReader final : public FrameWebContentReader {
51 public:
52     Range& context;
53     const bool allowPlainText;
54
55     RefPtr<DocumentFragment> fragment;
56     bool madeFragmentFromPlainText;
57
58     WebContentReader(Frame& frame, Range& context, bool allowPlainText)
59         : FrameWebContentReader(frame)
60         , context(context)
61         , allowPlainText(allowPlainText)
62         , madeFragmentFromPlainText(false)
63     {
64     }
65
66     void addFragment(Ref<DocumentFragment>&&);
67
68 private:
69 #if PLATFORM(COCOA)
70     bool readWebArchive(SharedBuffer&) override;
71     bool readFilenames(const Vector<String>&) override;
72     bool readHTML(const String&) override;
73     bool readRTFD(SharedBuffer&) override;
74     bool readRTF(SharedBuffer&) override;
75     bool readImage(Ref<SharedBuffer>&&, const String& type) override;
76     bool readURL(const URL&, const String& title) override;
77 #endif
78     bool readPlainText(const String&) override;
79 };
80
81 class WebContentMarkupReader final : public FrameWebContentReader {
82 public:
83     String markup;
84
85     explicit WebContentMarkupReader(Frame& frame)
86         : FrameWebContentReader(frame)
87     {
88     }
89
90 private:
91 #if PLATFORM(COCOA)
92     bool readWebArchive(SharedBuffer&) override;
93     bool readFilenames(const Vector<String>&) override { return false; }
94     bool readHTML(const String&) override;
95     bool readRTFD(SharedBuffer&) override;
96     bool readRTF(SharedBuffer&) override;
97     bool readImage(Ref<SharedBuffer>&&, const String&) override { return false; }
98     bool readURL(const URL&, const String&) override { return false; }
99 #endif
100     bool readPlainText(const String&) override { return false; }
101 };
102
103 #if PLATFORM(COCOA) && defined(__OBJC__)
104 struct FragmentAndResources {
105     RefPtr<DocumentFragment> fragment;
106     Vector<Ref<ArchiveResource>> resources;
107 };
108
109 RefPtr<DocumentFragment> createFragmentAndAddResources(Frame&, NSAttributedString *);
110 #endif
111
112 }