storage/mozStorageStatementRow.cpp

   1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  * vim: sw=2 ts=2 et lcs=trail\:.,tab\:>~ :
   3  * This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "nsMemory.h"
   8 #include "nsString.h"
   9
  10 #include "mozStorageStatementRow.h"
  11 #include "mozStorageStatement.h"
  12
  13 #include "jsapi.h"
  14
  15 namespace mozilla {
  16 namespace storage {
  17
  18 ////////////////////////////////////////////////////////////////////////////////
  19 //// StatementRow
  20
  21 StatementRow::StatementRow(Statement *aStatement)
  22 : mStatement(aStatement)
  23 {
  24 }
  25
  26 NS_IMPL_ISUPPORTS(
  27   StatementRow,
  28   mozIStorageStatementRow,
  29   nsIXPCScriptable
  30 )
  31
  32 ////////////////////////////////////////////////////////////////////////////////
  33 //// nsIXPCScriptable
  34
  35 #define XPC_MAP_CLASSNAME StatementRow
  36 #define XPC_MAP_QUOTED_CLASSNAME "StatementRow"
  37 #define XPC_MAP_WANT_GETPROPERTY
  38 #define XPC_MAP_WANT_RESOLVE
  39 #define XPC_MAP_FLAGS nsIXPCScriptable::ALLOW_PROP_MODS_DURING_RESOLVE
  40 #include "xpc_map_end.h"
  41
  42 NS_IMETHODIMP
  43 StatementRow::GetProperty(nsIXPConnectWrappedNative *aWrapper,
  44                           JSContext *aCtx,
  45                           JSObject *aScopeObj,
  46                           jsid aId,
  47                           JS::Value *_vp,
  48                           bool *_retval)
  49 {
  50   NS_ENSURE_TRUE(mStatement, NS_ERROR_NOT_INITIALIZED);
  51
  52   JS::RootedObject scope(aCtx, aScopeObj);
  53   if (JSID_IS_STRING(aId)) {
  54     ::JSAutoByteString idBytes(aCtx, JSID_TO_STRING(aId));
  55     NS_ENSURE_TRUE(!!idBytes, NS_ERROR_OUT_OF_MEMORY);
  56     nsDependentCString jsid(idBytes.ptr());
  57
  58     uint32_t idx;
  59     nsresult rv = mStatement->GetColumnIndex(jsid, &idx);
  60     NS_ENSURE_SUCCESS(rv, rv);
  61     int32_t type;
  62     rv = mStatement->GetTypeOfIndex(idx, &type);
  63     NS_ENSURE_SUCCESS(rv, rv);
  64
  65     if (type == mozIStorageValueArray::VALUE_TYPE_INTEGER ||
  66         type == mozIStorageValueArray::VALUE_TYPE_FLOAT) {
  67       double dval;
  68       rv = mStatement->GetDouble(idx, &dval);
  69       NS_ENSURE_SUCCESS(rv, rv);
  70       *_vp = ::JS_NumberValue(dval);
  71     }
  72     else if (type == mozIStorageValueArray::VALUE_TYPE_TEXT) {
  73       uint32_t bytes;
  74       const char16_t *sval = reinterpret_cast<const char16_t *>(
  75         static_cast<mozIStorageStatement *>(mStatement)->
  76           AsSharedWString(idx, &bytes)
  77       );
  78       JSString *str = ::JS_NewUCStringCopyN(aCtx, sval, bytes / 2);
  79       if (!str) {
  80         *_retval = false;
  81         return NS_OK;
  82       }
  83       _vp->setString(str);
  84     }
  85     else if (type == mozIStorageValueArray::VALUE_TYPE_BLOB) {
  86       uint32_t length;
  87       const uint8_t *blob = static_cast<mozIStorageStatement *>(mStatement)->
  88         AsSharedBlob(idx, &length);
  89       JSObject *obj = ::JS_NewArrayObject(aCtx, length);
  90       if (!obj) {
  91         *_retval = false;
  92         return NS_OK;
  93       }
  94       _vp->setObject(*obj);
  95
  96       // Copy the blob over to the JS array.
  97       for (uint32_t i = 0; i < length; i++) {
  98         if (!::JS_DefineElement(aCtx, scope, i, blob[i], JSPROP_ENUMERATE)) {
  99           *_retval = false;
 100           return NS_OK;
 101         }
 102       }
 103     }
 104     else if (type == mozIStorageValueArray::VALUE_TYPE_NULL) {
 105       _vp->setNull();
 106     }
 107     else {
 108       NS_ERROR("unknown column type returned, what's going on?");
 109     }
 110   }
 111
 112   return NS_OK;
 113 }
 114
 115 NS_IMETHODIMP
 116 StatementRow::Resolve(nsIXPConnectWrappedNative *aWrapper,
 117                       JSContext *aCtx,
 118                       JSObject *aScopeObj,
 119                       jsid aId,
 120                       bool *aResolvedp,
 121                       bool *_retval)
 122 {
 123   JS::Rooted<JSObject*> scopeObj(aCtx, aScopeObj);
 124
 125   NS_ENSURE_TRUE(mStatement, NS_ERROR_NOT_INITIALIZED);
 126   // We do not throw at any point after this because we want to allow the
 127   // prototype chain to be checked for the property.
 128
 129   if (JSID_IS_STRING(aId)) {
 130     ::JSAutoByteString idBytes(aCtx, JSID_TO_STRING(aId));
 131     NS_ENSURE_TRUE(!!idBytes, NS_ERROR_OUT_OF_MEMORY);
 132     nsDependentCString name(idBytes.ptr());
 133
 134     uint32_t idx;
 135     nsresult rv = mStatement->GetColumnIndex(name, &idx);
 136     if (NS_FAILED(rv)) {
 137       // It's highly likely that the name doesn't exist, so let the JS engine
 138       // check the prototype chain and throw if that doesn't have the property
 139       // either.
 140       *aResolvedp = false;
 141       return NS_OK;
 142     }
 143
 144     JS::Rooted<jsid> id(aCtx, aId);
 145     *_retval = ::JS_DefinePropertyById(aCtx, scopeObj, id, JS::UndefinedHandleValue,
 146                                        JSPROP_RESOLVING);
 147     *aResolvedp = true;
 148     return NS_OK;
 149   }
 150
 151   return NS_OK;
 152 }
 153
 154 } // namespace storage
 155 } // namespace mozilla