LEFT | RIGHT |
1 /* | 1 /* |
2 * This file is part of Adblock Plus <https://adblockplus.org/>, | 2 * This file is part of Adblock Plus <https://adblockplus.org/>, |
3 * Copyright (C) 2006-present eyeo GmbH | 3 * Copyright (C) 2006-present eyeo GmbH |
4 * | 4 * |
5 * Adblock Plus is free software: you can redistribute it and/or modify | 5 * Adblock Plus is free software: you can redistribute it and/or modify |
6 * it under the terms of the GNU General Public License version 3 as | 6 * it under the terms of the GNU General Public License version 3 as |
7 * published by the Free Software Foundation. | 7 * published by the Free Software Foundation. |
8 * | 8 * |
9 * Adblock Plus is distributed in the hope that it will be useful, | 9 * Adblock Plus is distributed in the hope that it will be useful, |
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of | 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
12 * GNU General Public License for more details. | 12 * GNU General Public License for more details. |
13 * | 13 * |
14 * You should have received a copy of the GNU General Public License | 14 * You should have received a copy of the GNU General Public License |
15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
16 */ | 16 */ |
17 | 17 |
18 /** | 18 /** |
19 * @fileOverview Element hiding implementation. | 19 * @fileOverview Element hiding implementation. |
20 */ | 20 */ |
21 | 21 |
22 #pragma once | 22 #pragma once |
23 | 23 |
24 #include <unordered_map> | |
25 #include <vector> | 24 #include <vector> |
26 | 25 |
27 #include "bindings/runtime.h" | 26 #include "bindings/runtime.h" |
28 #include "intrusive_ptr.h" | 27 #include "intrusive_ptr.h" |
29 #include "StringMap.h" | 28 #include "StringMap.h" |
30 #include "filter/Filter.h" | 29 #include "filter/Filter.h" |
31 #include "filter/ElemHideBase.h" | 30 #include "filter/ElemHideBase.h" |
32 #include "filter/ElemHideException.h" | 31 #include "filter/ElemHideException.h" |
33 | 32 |
34 class _ElemHide_SelectorList : public ref_counted | 33 class ElemHide_SelectorList : public ref_counted |
35 { | 34 { |
36 std::vector<ElemHideBasePtr> mSelectors; | 35 std::vector<ElemHideBasePtr> mSelectors; |
37 public: | 36 public: |
38 size_t BINDINGS_EXPORTED GetSelectorCount() const | 37 size_t BINDINGS_EXPORTED GetSelectorCount() const |
39 { | 38 { |
40 return mSelectors.size(); | 39 return mSelectors.size(); |
41 } | 40 } |
42 DependentString BINDINGS_EXPORTED SelectorAt(size_t idx) const; | 41 OwnedString BINDINGS_EXPORTED SelectorAt(size_t idx) const; |
43 const String& BINDINGS_EXPORTED FilterKeyAt(size_t idx) const; | 42 const String& BINDINGS_EXPORTED FilterKeyAt(size_t idx) const; |
44 | 43 |
45 void push_back(ElemHideBasePtr filter) | 44 void push_back(const ElemHideBasePtr& filter) |
46 { | 45 { |
47 mSelectors.push_back(filter); | 46 mSelectors.push_back(filter); |
48 } | 47 } |
49 | 48 |
50 void append(const _ElemHide_SelectorList* list) | 49 void append(const ElemHide_SelectorList& list) |
51 { | 50 { |
52 mSelectors.insert(mSelectors.end(), | 51 mSelectors.insert(mSelectors.end(), |
53 list->mSelectors.cbegin(), list->mSelectors.cend()); | 52 list.mSelectors.cbegin(), list.mSelectors.cend()); |
54 } | 53 } |
55 }; | 54 }; |
56 | 55 |
57 class ElemHide : public ref_counted | 56 class ElemHide : public ref_counted |
58 { | 57 { |
59 // All filters. Key is filter text. Exception filters excluded. | 58 // All filters. Key is filter text. Exception filters excluded. |
60 StringMap<ElemHideBasePtr> mFilters; | 59 StringMap<ElemHideBasePtr> mFilters; |
61 // StringMap is non copyable. std::unordered_map<> it is | 60 // Filters by domain. Key is domain. |
62 // Filters by domain. Key is domain. Subkey is filter text. | 61 // In value key is filter text, value is filter or NULL |
63 StringMap<std::unordered_map<DependentString,ElemHideBasePtr,StringHash>> mFil
tersByDomain; | 62 OwnedStringMap<OwnedStringMap<ElemHideBasePtr>> mFiltersByDomain; |
64 | 63 |
65 // Exceptions. The key is the selector. | 64 // Exceptions. The key is the selector. |
66 StringMap<std::vector<ElemHideExceptionPtr>> mExceptions; | 65 OwnedStringMap<std::vector<ElemHideExceptionPtr>> mExceptions; |
67 // Known exceptions. Filter text as keys. | 66 // Known exceptions. Filter text as keys. |
68 StringSet mKnownExceptions; | 67 StringSet mKnownExceptions; |
69 | 68 |
70 // Unconditional selectors. Filter text as key | 69 // Unconditional selectors. Filter selector as key. Filter as value. |
71 StringSet mUnconditionalSelectors; | 70 OwnedStringMap<ElemHideBasePtr> mUnconditionalSelectors; |
72 | 71 |
73 mutable intrusive_ptr<_ElemHide_SelectorList> mUnconditionalSelectorsCache; | 72 mutable intrusive_ptr<ElemHide_SelectorList> mUnconditionalSelectorsCache; |
74 | 73 |
75 static ElemHide* mInstance; | |
76 public: | 74 public: |
| 75 static ElemHide* BINDINGS_EXPORTED Create() |
| 76 { |
| 77 return new ElemHide(); |
| 78 } |
| 79 // Used by GetSelectorsForDomain to narrow down selectors to return. |
| 80 // Keep these value up to date in the ElemHide.js import script. |
77 enum Criteria | 81 enum Criteria |
78 { | 82 { |
| 83 // Return all selectors applying to a particular hostname. |
79 ALL_MATCHING = 0, | 84 ALL_MATCHING = 0, |
| 85 // Exclude selectors which apply to all websites without exception. |
80 NO_UNCONDITIONAL = 1, | 86 NO_UNCONDITIONAL = 1, |
| 87 // Return only selectors for filters which specifically match the |
| 88 // given host name. |
81 SPECIFIC_ONLY = 2, | 89 SPECIFIC_ONLY = 2, |
82 }; | 90 }; |
83 | |
84 static ElemHide* BINDINGS_EXPORTED GetInstance() | |
85 { | |
86 return mInstance; | |
87 } | |
88 | 91 |
89 void BINDINGS_EXPORTED Clear(); | 92 void BINDINGS_EXPORTED Clear(); |
90 void BINDINGS_EXPORTED Add(ElemHideBase& filter); | 93 void BINDINGS_EXPORTED Add(ElemHideBase& filter); |
91 void BINDINGS_EXPORTED Remove(ElemHideBase& filter); | 94 void BINDINGS_EXPORTED Remove(ElemHideBase& filter); |
92 | 95 |
93 _ElemHide_SelectorList* BINDINGS_EXPORTED GetSelectorsForDomain(const String&
domain, Criteria criteria) const; | 96 ElemHide_SelectorList* BINDINGS_EXPORTED GetSelectorsForDomain(const String& d
omain, |
94 _ElemHide_SelectorList* BINDINGS_EXPORTED GetUnconditionalSelectors() const; | 97 Criteria criter
ia) const; |
| 98 ElemHide_SelectorList* BINDINGS_EXPORTED GetUnconditionalSelectors() const; |
95 | 99 |
96 ElemHideException* GetException(const ElemHideBase& filter, | 100 ElemHideException* GetException(const ElemHideBase& filter, |
97 DependentString& docDomain) const; | 101 DependentString& docDomain) const; |
98 | 102 |
99 private: | 103 private: |
100 void AddToFiltersByDomain(ElemHideBase& filter); | 104 void AddToFiltersByDomain(const ElemHideBasePtr & filter); |
101 }; | 105 }; |
102 | 106 |
LEFT | RIGHT |