Index: ios/chrome/browser/web/dom_altering_lock.mm |
diff --git a/ios/chrome/browser/web/dom_altering_lock.mm b/ios/chrome/browser/web/dom_altering_lock.mm |
new file mode 100644 |
index 0000000000000000000000000000000000000000..2d5aa49698664ec69a1b8ed741b182d070ed443b |
--- /dev/null |
+++ b/ios/chrome/browser/web/dom_altering_lock.mm |
@@ -0,0 +1,48 @@ |
+// Copyright 2014 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#import "ios/chrome/browser/web/dom_altering_lock.h" |
+ |
+#include "base/logging.h" |
+#include "ios/web/public/web_thread.h" |
+ |
+DEFINE_WEB_STATE_USER_DATA_KEY(DOMAlteringLock); |
+ |
+DOMAlteringLock::DOMAlteringLock(web::WebState* web_state) { |
+} |
+ |
+DOMAlteringLock::~DOMAlteringLock() { |
+} |
+ |
+void DOMAlteringLock::Acquire(id<DOMAltering> feature, |
+ ProceduralBlockWithBool lockAction) { |
+ DCHECK_CURRENTLY_ON_WEB_THREAD(web::WebThread::UI); |
+ if (current_dom_altering_feature_.get() == feature) { |
+ lockAction(YES); |
+ return; |
+ } |
+ if (current_dom_altering_feature_) { |
+ if (![current_dom_altering_feature_ canReleaseDOMLock]) { |
+ lockAction(NO); |
+ return; |
+ } |
+ [current_dom_altering_feature_ releaseDOMLockWithCompletionHandler:^() { |
+ DCHECK_CURRENTLY_ON_WEB_THREAD(web::WebThread::UI); |
+ DCHECK(current_dom_altering_feature_.get() == nil) |
+ << "The lock must be released before calling the completion handler."; |
+ current_dom_altering_feature_.reset(feature); |
+ lockAction(YES); |
+ }]; |
+ return; |
+ } |
+ current_dom_altering_feature_.reset(feature); |
+ lockAction(YES); |
+} |
+ |
+// Release the lock on the DOM tree. |
+void DOMAlteringLock::Release(id<DOMAltering> feature) { |
+ DCHECK_CURRENTLY_ON_WEB_THREAD(web::WebThread::UI); |
+ if (current_dom_altering_feature_.get() == feature) |
+ current_dom_altering_feature_.reset(); |
+} |