1/*
2 * Copyright (C) 2009 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#pragma once
27
28#include <wtf/URL.h>
29#include "UserContentTypes.h"
30#include "UserScriptTypes.h"
31#include <wtf/Vector.h>
32
33namespace WebCore {
34
35class UserScript {
36 WTF_MAKE_FAST_ALLOCATED;
37public:
38 UserScript() = default;
39 ~UserScript() = default;
40
41 UserScript(String&& source, URL&& url, Vector<String>&& whitelist, Vector<String>&& blacklist, UserScriptInjectionTime injectionTime, UserContentInjectedFrames injectedFrames)
42 : m_source(WTFMove(source))
43 , m_url(WTFMove(url))
44 , m_whitelist(WTFMove(whitelist))
45 , m_blacklist(WTFMove(blacklist))
46 , m_injectionTime(injectionTime)
47 , m_injectedFrames(injectedFrames)
48 {
49 }
50
51 const String& source() const { return m_source; }
52 const URL& url() const { return m_url; }
53 const Vector<String>& whitelist() const { return m_whitelist; }
54 const Vector<String>& blacklist() const { return m_blacklist; }
55 UserScriptInjectionTime injectionTime() const { return m_injectionTime; }
56 UserContentInjectedFrames injectedFrames() const { return m_injectedFrames; }
57
58 template<class Encoder> void encode(Encoder&) const;
59 template<class Decoder> static bool decode(Decoder&, UserScript&);
60
61private:
62 String m_source;
63 URL m_url;
64 Vector<String> m_whitelist;
65 Vector<String> m_blacklist;
66 UserScriptInjectionTime m_injectionTime { InjectAtDocumentStart };
67 UserContentInjectedFrames m_injectedFrames { InjectInAllFrames };
68};
69
70template<class Encoder>
71void UserScript::encode(Encoder& encoder) const
72{
73 encoder << m_source;
74 encoder << m_url;
75 encoder << m_whitelist;
76 encoder << m_blacklist;
77 encoder.encodeEnum(m_injectionTime);
78 encoder.encodeEnum(m_injectedFrames);
79}
80
81template<class Decoder>
82bool UserScript::decode(Decoder& decoder, UserScript& userScript)
83{
84 String source;
85 if (!decoder.decode(source))
86 return false;
87
88 URL url;
89 if (!decoder.decode(url))
90 return false;
91
92 Vector<String> whitelist;
93 if (!decoder.decode(whitelist))
94 return false;
95
96 Vector<String> blacklist;
97 if (!decoder.decode(blacklist))
98 return false;
99
100 UserScriptInjectionTime injectionTime;
101 if (!decoder.decodeEnum(injectionTime))
102 return false;
103
104 UserContentInjectedFrames injectedFrames;
105 if (!decoder.decodeEnum(injectedFrames))
106 return false;
107
108 userScript = { WTFMove(source), WTFMove(url), WTFMove(whitelist), WTFMove(blacklist), injectionTime, injectedFrames };
109 return true;
110}
111
112} // namespace WebCore
113