1 files changed, 82 insertions, 0 deletions
diff --git a/dom/html/CustomStateSet.cpp b/dom/html/CustomStateSet.cpp
new file mode 100644
index 0000000000..4e24551f5e
--- /dev/null
+++ b/dom/html/CustomStateSet.cpp
@@ -0,0 +1,82 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "CustomStateSet.h"
+#include "mozilla/dom/ElementInternalsBinding.h"
+#include "mozilla/dom/HTMLElement.h"
+#include "mozilla/dom/Document.h"
+#include "mozilla/PresShell.h"
+
+namespace mozilla::dom {
+
+NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(CustomStateSet, mTarget);
+
+NS_IMPL_CYCLE_COLLECTING_ADDREF(CustomStateSet)
+NS_IMPL_CYCLE_COLLECTING_RELEASE(CustomStateSet)
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(CustomStateSet)
+  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
+NS_INTERFACE_MAP_END
+
+CustomStateSet::CustomStateSet(HTMLElement* aTarget) : mTarget(aTarget) {}
+
+// WebIDL interface
+nsISupports* CustomStateSet::GetParentObject() const {
+  return ToSupports(mTarget);
+}
+
+JSObject* CustomStateSet::WrapObject(JSContext* aCx,
+                                     JS::Handle<JSObject*> aGivenProto) {
+  return CustomStateSet_Binding::Wrap(aCx, this, aGivenProto);
+}
+
+void CustomStateSet::Clear(ErrorResult& aRv) {
+  CustomStateSet_Binding::SetlikeHelpers::Clear(this, aRv);
+  if (aRv.Failed()) {
+    return;
+  }
+
+  mTarget->EnsureCustomStates().Clear();
+  InvalidateStyleFromCustomStateSetChange();
+}
+
+void CustomStateSet::InvalidateStyleFromCustomStateSetChange() const {
+  Document* doc = mTarget->OwnerDoc();
+
+  PresShell* presShell = doc->GetPresShell();
+  if (!presShell) {
+    return;
+  }
+
+  // TODO: make this more efficient?
+  presShell->DestroyFramesForAndRestyle(mTarget);
+}
+
+bool CustomStateSet::Delete(const nsAString& aState, ErrorResult& aRv) {
+  if (!CustomStateSet_Binding::SetlikeHelpers::Delete(this, aState, aRv) ||
+      aRv.Failed()) {
+    return false;
+  }
+
+  RefPtr<nsAtom> atom = NS_AtomizeMainThread(aState);
+  bool deleted = mTarget->EnsureCustomStates().RemoveElement(atom);
+  if (deleted) {
+    InvalidateStyleFromCustomStateSetChange();
+  }
+  return deleted;
+}
+
+void CustomStateSet::Add(const nsAString& aState, ErrorResult& aRv) {
+  CustomStateSet_Binding::SetlikeHelpers::Add(this, aState, aRv);
+  if (aRv.Failed()) {
+    return;
+  }
+
+  RefPtr<nsAtom> atom = NS_AtomizeMainThread(aState);
+  mTarget->EnsureCustomStates().AppendElement(atom);
+  InvalidateStyleFromCustomStateSetChange();
+}
+
+}  // namespace mozilla::dom