Implement quota estimation API from Storage spec

third_party/WebKit/Source/modules/quota/StorageManager.cpp

 // Copyright 2015 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/quota/StorageManager.h"
 
 #include "bindings/core/v8/ScriptPromiseResolver.h"
+#include "bindings/modules/v8/V8StorageEstimate.h"
 #include "core/dom/DOMException.h"
 #include "core/dom/Document.h"
 #include "core/dom/ExceptionCode.h"
 #include "modules/permissions/Permissions.h"
+#include "modules/quota/StorageEstimate.h"
+#include "platform/StorageQuotaCallbacks.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebCallbacks.h"
 #include "public/platform/modules/permissions/WebPermissionClient.h"
 #include "public/platform/modules/permissions/WebPermissionStatus.h"
 
 namespace blink {
 
 namespace {
 
 class DurableStorageQueryCallbacks final : public WebPermissionCallback {
@@ skipping 41 matching lines @@
     }
     void onError() override
     {
         ASSERT_NOT_REACHED();
     }
 
 private:
     Persistent<ScriptPromiseResolver> m_resolver;
 };
 
+class EstimateCallbacks final : public StorageQuotaCallbacks {
+    WTF_MAKE_NONCOPYABLE(EstimateCallbacks);
+public:
+    explicit EstimateCallbacks(ScriptPromiseResolver* resolver)
+        : m_resolver(resolver) {}
+
+    ~EstimateCallbacks() override {}
+
+    void didQueryStorageUsageAndQuota(unsigned long long usageInBytes, unsigned long long quotaInBytes) override
+    {
+        StorageEstimate estimate;
+        estimate.setUsage(usageInBytes);
+        estimate.setQuota(quotaInBytes);
+        m_resolver->resolve(estimate);
+    }
+
+    void didFail(WebStorageQuotaError error) override
+    {
+        m_resolver->reject(DOMException::create(static_cast<ExceptionCode>(error)));
+    }
+
+    DEFINE_INLINE_VIRTUAL_TRACE()
+    {
+        visitor->trace(m_resolver);
+        StorageQuotaCallbacks::trace(visitor);
+    }
+
+private:
+    Member<ScriptPromiseResolver> m_resolver;
+};
+
 } // namespace
 
 ScriptPromise StorageManager::requestPersistent(ScriptState* scriptState)
 {
     ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState);
     ScriptPromise promise = resolver->promise();
     ExecutionContext* executionContext = scriptState->getExecutionContext();
     SecurityOrigin* securityOrigin = executionContext->getSecurityOrigin();
     // TODO(dgrogan): Is the isUnique() check covered by the later
     // isSecureContext() check? If so, maybe remove it. Write a test if it
@@ skipping 24 matching lines @@
     ScriptPromise promise = resolver->promise();
     WebPermissionClient* permissionClient = Permissions::getClient(scriptState->getExecutionContext());
     if (!permissionClient) {
         resolver->reject(DOMException::create(InvalidStateError, "In its current state, the global scope can't query permissions."));
         return promise;
     }
     permissionClient->queryPermission(WebPermissionTypeDurableStorage, KURL(KURL(), scriptState->getExecutionContext()->getSecurityOrigin()->toString()), new DurableStorageQueryCallbacks(resolver));
     return promise;
 }
 
+ScriptPromise StorageManager::estimate(ScriptState* scriptState)
+{
+    ScriptPromiseResolver* resolver = ScriptPromiseResolver::create(scriptState);
+    ScriptPromise promise = resolver->promise();
+    ExecutionContext* executionContext = scriptState->getExecutionContext();
+    SecurityOrigin* securityOrigin = executionContext->getSecurityOrigin();
+    if (securityOrigin->isUnique()) {
+        resolver->reject(DOMException::create(NotSupportedError));
+        return promise;
+    }
+    // IDL has: [SecureContext]
+    String errorMessage;
+    if (!executionContext->isSecureContext(errorMessage)) {
+        resolver->reject(DOMException::create(SecurityError, errorMessage));
+        return promise;
+    }
+
+    KURL storagePartition = KURL(KURL(), securityOrigin->toString());
+    Platform::current()->queryStorageUsageAndQuota(storagePartition, WebStorageQuotaTypeTemporary, new EstimateCallbacks(resolver));
+    return promise;
+}
+
 DEFINE_TRACE(StorageManager)
 {
 }
 
 } // namespace blink