storage/src/mozStorageStatementParams.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
   2  * vim: sw=4 ts=4 sts=4
   3  * ***** BEGIN LICENSE BLOCK *****
   4  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   5  *
   6  * The contents of this file are subject to the Mozilla Public License Version
   7  * 1.1 (the "License"); you may not use this file except in compliance with
   8  * the License. You may obtain a copy of the License at
   9  * http://www.mozilla.org/MPL/
  10  *
  11  * Software distributed under the License is distributed on an "AS IS" basis,
  12  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  13  * for the specific language governing rights and limitations under the
  14  * License.
  15  *
  16  * The Original Code is Oracle Corporation code.
  17  *
  18  * The Initial Developer of the Original Code is
  19  *  Oracle Corporation
  20  * Portions created by the Initial Developer are Copyright (C) 2004
  21  * the Initial Developer. All Rights Reserved.
  22  *
  23  * Contributor(s):
  24  *   Vladimir Vukicevic <vladimir.vukicevic@oracle.com>
  25  *   Shawn Wilsher <me@shawnwilsher.com>
  26  *
  27  * Alternatively, the contents of this file may be used under the terms of
  28  * either the GNU General Public License Version 2 or later (the "GPL"), or
  29  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  30  * in which case the provisions of the GPL or the LGPL are applicable instead
  31  * of those above. If you wish to allow use of your version of this file only
  32  * under the terms of either the GPL or the LGPL, and not to allow others to
  33  * use your version of this file under the terms of the MPL, indicate your
  34  * decision by deleting the provisions above and replace them with the notice
  35  * and other provisions required by the GPL or the LGPL. If you do not delete
  36  * the provisions above, a recipient may use your version of this file under
  37  * the terms of any one of the MPL, the GPL or the LGPL.
  38  *
  39  * ***** END LICENSE BLOCK ***** */
  40
  41 #include "nsMemory.h"
  42 #include "nsString.h"
  43
  44 #include "mozStorageStatementParams.h"
  45
  46 #include "sqlite3.h"
  47
  48 /*************************************************************************
  49  ****
  50  **** mozStorageStatementParams
  51  ****
  52  *************************************************************************/
  53
  54 NS_IMPL_ISUPPORTS2(mozStorageStatementParams, mozIStorageStatementParams, nsIXPCScriptable)
  55
  56 mozStorageStatementParams::mozStorageStatementParams(mozIStorageStatement *aStatement)
  57     : mStatement(aStatement)
  58 {
  59     NS_ASSERTION(mStatement != nsnull, "mStatement is null");
  60     mStatement->GetParameterCount(&mParamCount);
  61 }
  62
  63 /*
  64  * nsIXPCScriptable impl
  65  */
  66
  67 /* readonly attribute string className; */
  68 NS_IMETHODIMP
  69 mozStorageStatementParams::GetClassName(char * *aClassName)
  70 {
  71     NS_ENSURE_ARG_POINTER(aClassName);
  72     *aClassName = (char *) nsMemory::Clone("mozStorageStatementParams", 26);
  73     if (!*aClassName)
  74         return NS_ERROR_OUT_OF_MEMORY;
  75     return NS_OK;
  76 }
  77
  78 /* readonly attribute PRUint32 scriptableFlags; */
  79 NS_IMETHODIMP
  80 mozStorageStatementParams::GetScriptableFlags(PRUint32 *aScriptableFlags)
  81 {
  82     *aScriptableFlags =
  83         nsIXPCScriptable::WANT_SETPROPERTY |
  84         nsIXPCScriptable::WANT_NEWRESOLVE |
  85         nsIXPCScriptable::ALLOW_PROP_MODS_DURING_RESOLVE;
  86     return NS_OK;
  87 }
  88
  89 /* PRBool getProperty (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal id, in JSValPtr vp); */
  90 NS_IMETHODIMP
  91 mozStorageStatementParams::GetProperty(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
  92                          JSObject * obj, jsval id, jsval * vp, PRBool *_retval)
  93 {
  94     return NS_ERROR_NOT_IMPLEMENTED;
  95 }
  96
  97
  98 /* PRBool setProperty (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal id, in JSValPtr vp); */
  99 NS_IMETHODIMP
 100 mozStorageStatementParams::SetProperty(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 101                          JSObject * obj, jsval id, jsval * vp, PRBool *_retval)
 102 {
 103     NS_ENSURE_TRUE(mStatement, NS_ERROR_NOT_INITIALIZED);
 104
 105     if (JSVAL_IS_INT(id)) {
 106         int idx = JSVAL_TO_INT(id);
 107
 108         *_retval = JSValStorageStatementBinder (cx, mStatement, idx, *vp);
 109     } else if (JSVAL_IS_STRING(id)) {
 110         sqlite3_stmt *stmt = mStatement->GetNativeStatementPointer();
 111
 112         JSString *str = JSVAL_TO_STRING(id);
 113         nsCAutoString name(":");
 114         name.Append(NS_ConvertUTF16toUTF8(nsDependentString((PRUnichar *)::JS_GetStringChars(str),
 115                                                             ::JS_GetStringLength(str))));
 116
 117         // check to see if there's a parameter with this name
 118         if (sqlite3_bind_parameter_index(stmt, name.get()) == 0) {
 119             *_retval = PR_FALSE;
 120             return NS_ERROR_INVALID_ARG;
 121         }
 122
 123         *_retval = PR_TRUE;
 124         // You can use a named parameter more than once in a statement...
 125         int count = sqlite3_bind_parameter_count(stmt);
 126         for (int i = 0; (i < count) && (*_retval); i++) {
 127             // sqlite indices start at 1
 128             const char *pName = sqlite3_bind_parameter_name(stmt, i + 1);
 129             if (name.Equals(pName))
 130                 *_retval = JSValStorageStatementBinder(cx, mStatement, i, *vp);
 131         }
 132     } else {
 133         *_retval = PR_FALSE;
 134     }
 135
 136     return (*_retval) ? NS_OK : NS_ERROR_INVALID_ARG;
 137 }
 138
 139 /* void preCreate (in nsISupports nativeObj, in JSContextPtr cx, in JSObjectPtr globalObj, out JSObjectPtr parentObj); */
 140 NS_IMETHODIMP
 141 mozStorageStatementParams::PreCreate(nsISupports *nativeObj, JSContext * cx,
 142                        JSObject * globalObj, JSObject * *parentObj)
 143 {
 144     return NS_ERROR_NOT_IMPLEMENTED;
 145 }
 146
 147 /* void create (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj); */
 148 NS_IMETHODIMP
 149 mozStorageStatementParams::Create(nsIXPConnectWrappedNative *wrapper, JSContext * cx, JSObject * obj)
 150 {
 151     return NS_ERROR_NOT_IMPLEMENTED;
 152 }
 153
 154 /* void postCreate (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj); */
 155 NS_IMETHODIMP
 156 mozStorageStatementParams::PostCreate(nsIXPConnectWrappedNative *wrapper, JSContext * cx, JSObject * obj)
 157 {
 158     return NS_ERROR_NOT_IMPLEMENTED;
 159 }
 160
 161 /* PRBool addProperty (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal id, in JSValPtr vp); */
 162 NS_IMETHODIMP
 163 mozStorageStatementParams::AddProperty(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 164                                     JSObject * obj, jsval id, jsval * vp, PRBool *_retval)
 165 {
 166     return NS_ERROR_NOT_IMPLEMENTED;
 167 }
 168
 169 /* PRBool delProperty (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal id, in JSValPtr vp); */
 170 NS_IMETHODIMP
 171 mozStorageStatementParams::DelProperty(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 172                                     JSObject * obj, jsval id, jsval * vp, PRBool *_retval)
 173 {
 174     return NS_ERROR_NOT_IMPLEMENTED;
 175 }
 176
 177 /* PRBool enumerate (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj); */
 178 NS_IMETHODIMP
 179 mozStorageStatementParams::Enumerate(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 180                                   JSObject * obj, PRBool *_retval)
 181 {
 182     return NS_ERROR_NOT_IMPLEMENTED;
 183 }
 184
 185 /* PRBool newEnumerate (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in PRUint32 enum_op, in JSValPtr statep, out JSID idp); */
 186 NS_IMETHODIMP
 187 mozStorageStatementParams::NewEnumerate(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 188                                      JSObject * obj, PRUint32 enum_op, jsval * statep, jsid *idp, PRBool *_retval)
 189 {
 190     return NS_ERROR_NOT_IMPLEMENTED;
 191 }
 192
 193 /* PRBool newResolve (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal id, in PRUint32 flags, out JSObjectPtr objp); */
 194 NS_IMETHODIMP
 195 mozStorageStatementParams::NewResolve(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 196                                    JSObject * obj, jsval id, PRUint32 flags, JSObject * *objp, PRBool *_retval)
 197 {
 198     NS_ENSURE_TRUE(mStatement, NS_ERROR_NOT_INITIALIZED);
 199
 200     int idx = -1;
 201
 202     if (JSVAL_IS_INT(id)) {
 203         idx = JSVAL_TO_INT(id);
 204     } else if (JSVAL_IS_STRING(id)) {
 205         JSString *str = JSVAL_TO_STRING(id);
 206         nsCAutoString name(":");
 207         name.Append(NS_ConvertUTF16toUTF8(nsDependentString((PRUnichar *)::JS_GetStringChars(str),
 208                                                             ::JS_GetStringLength(str))));
 209
 210         // check to see if there's a parameter with this name
 211         idx = sqlite3_bind_parameter_index(mStatement->GetNativeStatementPointer(), name.get());
 212         if (idx == 0) {
 213             // nope.
 214             fprintf (stderr, "********** mozStorageStatementWrapper: Couldn't find parameter %s\n", name.get());
 215             *_retval = PR_FALSE;
 216             return NS_OK;
 217         } else {
 218             // set idx, so that the numbered property also gets defined
 219             idx = idx - 1;
 220         }
 221
 222         PRBool success = ::JS_DefineUCProperty(cx, obj, ::JS_GetStringChars(str),
 223                                                ::JS_GetStringLength(str),
 224                                                JSVAL_VOID,
 225                                                nsnull, nsnull, 0);
 226         if (!success) {
 227             *_retval = PR_FALSE;
 228             return NS_ERROR_FAILURE;
 229         }
 230     }
 231
 232     if (idx == -1) {
 233         *_retval = PR_FALSE;
 234         return NS_ERROR_FAILURE;
 235     }
 236
 237     // is it out of range?
 238     if (idx < 0 || idx >= (int)mParamCount) {
 239         *_retval = PR_FALSE;
 240         return NS_OK;
 241     }
 242
 243     *_retval = ::JS_DefineElement(cx, obj, idx, JSVAL_VOID, nsnull, nsnull, 0);
 244     if (*_retval)
 245         *objp = obj;
 246     return NS_OK;
 247 }
 248
 249 /* PRBool convert (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in PRUint32 type, in JSValPtr vp); */
 250 NS_IMETHODIMP
 251 mozStorageStatementParams::Convert(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 252                                 JSObject * obj, PRUint32 type, jsval * vp, PRBool *_retval)
 253 {
 254     return NS_ERROR_NOT_IMPLEMENTED;
 255 }
 256
 257 /* void finalize (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj); */
 258 NS_IMETHODIMP
 259 mozStorageStatementParams::Finalize(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 260                                  JSObject * obj)
 261 {
 262     return NS_ERROR_NOT_IMPLEMENTED;
 263 }
 264
 265 /* PRBool checkAccess (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal id, in PRUint32 mode, in JSValPtr vp); */
 266 NS_IMETHODIMP
 267 mozStorageStatementParams::CheckAccess(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 268                                     JSObject * obj, jsval id, PRUint32 mode, jsval * vp, PRBool *_retval)
 269 {
 270     return NS_ERROR_NOT_IMPLEMENTED;
 271 }
 272
 273 /* PRBool call (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in PRUint32 argc, in JSValPtr argv, in JSValPtr vp); */
 274 NS_IMETHODIMP
 275 mozStorageStatementParams::Call(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 276                              JSObject * obj, PRUint32 argc, jsval * argv, jsval * vp, PRBool *_retval)
 277 {
 278     return NS_ERROR_NOT_IMPLEMENTED;
 279 }
 280
 281 /* PRBool construct (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in PRUint32 argc, in JSValPtr argv, in JSValPtr vp); */
 282 NS_IMETHODIMP
 283 mozStorageStatementParams::Construct(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 284                                   JSObject * obj, PRUint32 argc, jsval * argv, jsval * vp, PRBool *_retval)
 285 {
 286     return NS_ERROR_NOT_IMPLEMENTED;
 287 }
 288
 289 /* PRBool hasInstance (in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal val, out PRBool bp); */
 290 NS_IMETHODIMP
 291 mozStorageStatementParams::HasInstance(nsIXPConnectWrappedNative *wrapper, JSContext * cx,
 292                                     JSObject * obj, jsval val, PRBool *bp, PRBool *_retval)
 293 {
 294     return NS_ERROR_NOT_IMPLEMENTED;
 295 }
 296
 297 /* void trace (in nsIXPConnectWrappedNative wrapper, in JSTracerPtr trc, in JSObjectPtr obj); */
 298 NS_IMETHODIMP
 299 mozStorageStatementParams::Trace(nsIXPConnectWrappedNative *wrapper,
 300                                 JSTracer *trc, JSObject * obj)
 301 {
 302     return NS_ERROR_NOT_IMPLEMENTED;
 303 }
 304
 305 /* PRBool equality(in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj, in JSVal val); */
 306 NS_IMETHODIMP
 307 mozStorageStatementParams::Equality(nsIXPConnectWrappedNative *wrapper,
 308                                     JSContext *cx, JSObject *obj, jsval val,
 309                                     PRBool *_retval)
 310 {
 311     return NS_ERROR_NOT_IMPLEMENTED;
 312 }
 313
 314 /* JSObjectPtr outerObject(in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj); */
 315 NS_IMETHODIMP
 316 mozStorageStatementParams::OuterObject(nsIXPConnectWrappedNative *wrapper,
 317                                        JSContext *cx, JSObject *obj,
 318                                        JSObject **_retval)
 319 {
 320     return NS_ERROR_NOT_IMPLEMENTED;
 321 }
 322
 323 /* JSObjectPtr innerObject(in nsIXPConnectWrappedNative wrapper, in JSContextPtr cx, in JSObjectPtr obj); */
 324 NS_IMETHODIMP
 325 mozStorageStatementParams::InnerObject(nsIXPConnectWrappedNative *wrapper,
 326                                        JSContext *cx, JSObject *obj,
 327                                        JSObject **_retval)
 328 {
 329     return NS_ERROR_NOT_IMPLEMENTED;
 330 }
 331
 332 /* void postCreatePrototype (in JSContextPtr cx, in JSObjectPtr proto); */
 333 NS_IMETHODIMP
 334 mozStorageStatementParams::PostCreatePrototype(JSContext * cx, JSObject * proto)
 335 {
 336     return NS_OK;
 337 }