Home | History | Annotate | Download | only in quota 
      1 /*
      2  * Copyright (C) 2014 Google Inc. All rights reserved.
      3  *
      4  * Redistribution and use in source and binary forms, with or without
      5  * modification, are permitted provided that the following conditions are
      6  * met:
      7  *
      8  *     * Redistributions of source code must retain the above copyright
      9  * notice, this list of conditions and the following disclaimer.
     10  *     * Redistributions in binary form must reproduce the above
     11  * copyright notice, this list of conditions and the following disclaimer
     12  * in the documentation and/or other materials provided with the
     13  * distribution.
     14  *     * Neither the name of Google Inc. nor the names of its
     15  * contributors may be used to endorse or promote products derived from
     16  * this software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
     21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
     22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
     24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     29  */
     30 
     31 #include "config.h"
     32 #include "modules/quota/StorageQuota.h"
     33 
     34 #include "bindings/core/v8/ScriptPromise.h"
     35 #include "bindings/core/v8/ScriptPromiseResolver.h"
     36 #include "core/dom/DOMError.h"
     37 #include "core/dom/ExceptionCode.h"
     38 #include "core/dom/ExecutionContext.h"
     39 #include "modules/quota/StorageQuotaCallbacksImpl.h"
     40 #include "modules/quota/StorageQuotaClient.h"
     41 #include "platform/weborigin/KURL.h"
     42 #include "platform/weborigin/SecurityOrigin.h"
     43 #include "public/platform/Platform.h"
     44 #include "public/platform/WebStorageQuotaCallbacks.h"
     45 #include "public/platform/WebStorageQuotaType.h"
     46 
     47 namespace blink {
     48 
     49 namespace {
     50 
     51 struct StorageTypeMapping {
     52     WebStorageQuotaType type;
     53     const char* const name;
     54 };
     55 
     56 const StorageTypeMapping storageTypeMappings[] = {
     57     { WebStorageQuotaTypeTemporary, "temporary" },
     58     { WebStorageQuotaTypePersistent, "persistent" },
     59 };
     60 
     61 WebStorageQuotaType stringToStorageQuotaType(const String& type)
     62 {
     63     for (size_t i = 0; i < WTF_ARRAY_LENGTH(storageTypeMappings); ++i) {
     64         if (storageTypeMappings[i].name == type)
     65             return storageTypeMappings[i].type;
     66     }
     67     ASSERT_NOT_REACHED();
     68     return WebStorageQuotaTypeTemporary;
     69 }
     70 
     71 } // namespace
     72 
     73 StorageQuota::StorageQuota()
     74 {
     75 }
     76 
     77 Vector<String> StorageQuota::supportedTypes() const
     78 {
     79     Vector<String> types;
     80     for (size_t i = 0; i < WTF_ARRAY_LENGTH(storageTypeMappings); ++i)
     81         types.append(storageTypeMappings[i].name);
     82     return types;
     83 }
     84 
     85 ScriptPromise StorageQuota::queryInfo(ScriptState* scriptState, String type)
     86 {
     87     RefPtr<ScriptPromiseResolver> resolver = ScriptPromiseResolver::create(scriptState);
     88     ScriptPromise promise = resolver->promise();
     89 
     90     SecurityOrigin* securityOrigin = scriptState->executionContext()->securityOrigin();
     91     if (securityOrigin->isUnique()) {
     92         resolver->reject(DOMError::create(NotSupportedError));
     93         return promise;
     94     }
     95 
     96     KURL storagePartition = KURL(KURL(), securityOrigin->toString());
     97     OwnPtrWillBeRawPtr<StorageQuotaCallbacks> callbacks = StorageQuotaCallbacksImpl::create(resolver);
     98     Platform::current()->queryStorageUsageAndQuota(storagePartition, stringToStorageQuotaType(type), callbacks.release());
     99     return promise;
    100 }
    101 
    102 ScriptPromise StorageQuota::requestPersistentQuota(ScriptState* scriptState, unsigned long long newQuota)
    103 {
    104     StorageQuotaClient* client = StorageQuotaClient::from(scriptState->executionContext());
    105     if (!client) {
    106         RefPtr<ScriptPromiseResolver> resolver = ScriptPromiseResolver::create(scriptState);
    107         ScriptPromise promise = resolver->promise();
    108         resolver->reject(DOMError::create(NotSupportedError));
    109         return promise;
    110     }
    111 
    112     return client->requestPersistentQuota(scriptState, newQuota);
    113 }
    114 
    115 } // namespace blink
    116