Modernize UserScript.h
[WebKit-https.git] / Source / WebCore / page / UserScript.h
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 "URL.h"
29 #include "UserContentTypes.h"
30 #include "UserScriptTypes.h"
31 #include <wtf/Vector.h>
32
33 namespace WebCore {
34
35 class UserScript {
36     WTF_MAKE_FAST_ALLOCATED;
37 public:
38     UserScript() { }
39     ~UserScript() { }
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
61 private:
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
70 template<class Encoder>
71 void 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
81 template<class Decoder>
82 bool 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