Added budget_service.mojom

third_party/WebKit/Source/modules/budget/BudgetService.cpp

 // Copyright 2016 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.
 
 #include "modules/budget/BudgetService.h"
 
 #include "bindings/core/v8/ScriptPromise.h"
 #include "bindings/core/v8/ScriptPromiseResolver.h"
 #include "bindings/core/v8/ScriptState.h"
 #include "core/dom/DOMException.h"
 #include "core/dom/ExceptionCode.h"
+#include "modules/budget/BudgetState.h"
+#include "public/platform/InterfaceProvider.h"
+#include "public/platform/Platform.h"
+#include "public/platform/modules/budget_service/budget_service.mojom-blink.h"
 
 namespace blink {
 
-BudgetService::BudgetService() {}
+BudgetService::BudgetService()
+{
+    Platform::current()->interfaceProvider()->getInterface(mojo::GetProxy(&m_service));

[haraken, 2016/08/23 15:02:59]

Per discussion on this thread
(https://groups.google.com/a/chromium.org/d/topic/chromium-mojo/QpKZPmdWpYY/di...),
it seems better to pass in LocalFrame& to the constructor and use a per-frame
InterfaceProvider instead of the platform-wide InterfaceProvider.

[harkness, 2016/08/23 15:10:46]

This is going to be used in both worker (next CL) and document, so there may not
be a frame.

[haraken, 2016/08/23 15:21:34]

Then the platform-wide InterfaceProvider makes sense.

+}
 
-ScriptPromise BudgetService::getCost(ScriptState* scriptState, const AtomicString& actionType)
+BudgetService::~BudgetService()
 {
-    // TODO(harkness): Trigger the cost calculation.
-    return ScriptPromise::rejectWithDOMException(scriptState, DOMException::create(NotSupportedError, "Not yet implemented"));
+}
+
+ScriptPromise BudgetService::getCost(ScriptState* scriptState, const AtomicString& /* actionType */)
+{
+    DCHECK(m_service);
+
+    ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState);
+    ScriptPromise promise = resolver->promise();
+
+    // TODO(harkness): Map the actionType to BudgetOperationType.
+    // Get the cost for the action from the browser BudgetService.
+    mojom::blink::BudgetOperationType type = mojom::blink::BudgetOperationType::SILENT_PUSH;
+    m_service->GetCost(type, convertToBaseCallback(WTF::bind(&BudgetService::gotCost, wrapPersistent(this), wrapPersistent(resolver))));
+    return promise;
+}
+
+void BudgetService::gotCost(ScriptPromiseResolver* resolver, double cost) const
+{
+    resolver->resolve(cost);
 }
 
 ScriptPromise BudgetService::getBudget(ScriptState* scriptState)
 {
-    // TODO(harkness): Trigger the budget calculation.
-    return ScriptPromise::rejectWithDOMException(scriptState, DOMException::create(NotSupportedError, "Not yet implemented"));
+    DCHECK(m_service);
+
+    ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState);
+    ScriptPromise promise = resolver->promise();
+
+    // Get the budget from the browser BudgetService.
+    RefPtr<SecurityOrigin> origin(scriptState->getExecutionContext()->getSecurityOrigin());

[haraken, 2016/08/23 11:24:12]

scriptState->getExecutionContext() may return false if the script is being
executed on a detached frame. You need a nullptr check.

[harkness, 2016/08/23 15:10:46]

Done.

+    m_service->GetBudget(origin, convertToBaseCallback(WTF::bind(&BudgetService::gotBudget, wrapPersistent(this), wrapPersistent(resolver))));
+    return promise;
+}
+
+void BudgetService::gotBudget(ScriptPromiseResolver* resolver, const mojo::WTFArray<mojom::blink::BudgetStatePtr> expectations) const
+{
+    // Copy the chunks into the budget array.
+    HeapVector<Member<BudgetState>> budget(expectations.size());
+    for (size_t i = 0; i < expectations.size(); i++)
+        budget[i] = new BudgetState(expectations[i]->budget_at, expectations[i]->time);
+
+    resolver->resolve(budget);
 }
 
 } // namespace blink