Improve use of NeverDestroyed
[WebKit-https.git] / Source / WebKit / WebProcess / WebCoreSupport / WebPasteboardOverrides.cpp
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 #include "config.h"
27 #include "WebPasteboardOverrides.h"
28
29 #include <wtf/NeverDestroyed.h>
30
31 namespace WebKit {
32
33 WebPasteboardOverrides& WebPasteboardOverrides::sharedPasteboardOverrides()
34 {
35     static NeverDestroyed<WebPasteboardOverrides> sharedOverrides;
36     return sharedOverrides;
37 }
38
39 WebPasteboardOverrides::WebPasteboardOverrides()
40 {
41 }
42
43 void WebPasteboardOverrides::addOverride(const String& pasteboardName, const String& type, const Vector<uint8_t>& data)
44 {
45     auto addResult = m_overridesMap.add(pasteboardName, nullptr);
46
47     if (addResult.isNewEntry) {
48         std::unique_ptr<HashMap<String, Vector<uint8_t>>> typeMap = std::make_unique<HashMap<String, Vector<uint8_t>>>();
49         addResult.iterator->value = WTFMove(typeMap);
50     }
51
52     addResult.iterator->value->set(type, data);
53 }
54
55 void WebPasteboardOverrides::removeOverride(const String& pasteboardName, const String& type)
56 {
57     auto it = m_overridesMap.find(pasteboardName);
58     if (it == m_overridesMap.end())
59         return;
60
61     ASSERT(it->value);
62
63     it->value->remove(type);
64
65     // If this was the last override for this pasteboard, remove its record completely.
66     if (it->value->isEmpty())
67         m_overridesMap.remove(it);
68 }
69
70 Vector<String> WebPasteboardOverrides::overriddenTypes(const String& pasteboardName)
71 {
72     Vector<String> result;
73
74     auto it = m_overridesMap.find(pasteboardName);
75     if (it == m_overridesMap.end())
76         return result;
77
78     ASSERT(it->value);
79
80     for (String& type : it->value->keys())
81         result.append(type);
82
83     return result;
84 }
85
86 bool WebPasteboardOverrides::getDataForOverride(const String& pasteboardName, const String& type, Vector<uint8_t>& data) const
87 {
88     auto pasteboardIterator = m_overridesMap.find(pasteboardName);
89     if (pasteboardIterator == m_overridesMap.end())
90         return false;
91
92     auto typeIterator = pasteboardIterator->value->find(type);
93     if (typeIterator == pasteboardIterator->value->end())
94         return false;
95
96     data = typeIterator->value;
97     return true;
98 }
99
100 bool WebPasteboardOverrides::getDataForOverride(const String& pasteboardName, const String& type, Vector<char>& data) const
101 {
102     Vector<uint8_t> foundBuffer;
103     if (!getDataForOverride(pasteboardName, type, foundBuffer))
104         return false;
105
106     data.clear();
107     data.resize(foundBuffer.size());
108     memcpy(data.data(), foundBuffer.data(), foundBuffer.size());
109
110     return true;
111 }
112
113 } // namespace WebKit