dom/indexedDB/IDBFileRequest.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=2 et sw=2 tw=80: */
   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 file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "IDBFileRequest.h"
   8
   9 #include "IDBFileHandle.h"
  10 #include "js/RootingAPI.h"
  11 #include "jsapi.h"
  12 #include "MainThreadUtils.h"
  13 #include "mozilla/Assertions.h"
  14 #include "mozilla/dom/FileHelper.h"
  15 #include "mozilla/dom/IDBFileRequestBinding.h"
  16 #include "mozilla/dom/ProgressEvent.h"
  17 #include "mozilla/EventDispatcher.h"
  18 #include "nsCOMPtr.h"
  19 #include "nsDebug.h"
  20 #include "nsError.h"
  21 #include "nsIDOMEvent.h"
  22 #include "nsIScriptContext.h"
  23 #include "nsLiteralString.h"
  24
  25 namespace mozilla {
  26 namespace dom {
  27 namespace indexedDB {
  28
  29 IDBFileRequest::IDBFileRequest(nsPIDOMWindow* aWindow)
  30   : DOMRequest(aWindow), mWrapAsDOMRequest(false)
  31 {
  32 }
  33
  34 IDBFileRequest::~IDBFileRequest()
  35 {
  36 }
  37
  38 // static
  39 already_AddRefed<IDBFileRequest>
  40 IDBFileRequest::Create(nsPIDOMWindow* aOwner, IDBFileHandle* aFileHandle,
  41                        bool aWrapAsDOMRequest)
  42 {
  43   MOZ_ASSERT(NS_IsMainThread(), "Wrong thread!");
  44
  45   nsRefPtr<IDBFileRequest> request = new IDBFileRequest(aOwner);
  46   request->mFileHandle = aFileHandle;
  47   request->mWrapAsDOMRequest = aWrapAsDOMRequest;
  48
  49   return request.forget();
  50 }
  51
  52 nsresult
  53 IDBFileRequest::PreHandleEvent(EventChainPreVisitor& aVisitor)
  54 {
  55   MOZ_ASSERT(NS_IsMainThread(), "Wrong thread!");
  56
  57   aVisitor.mCanHandle = true;
  58   aVisitor.mParentTarget = mFileHandle;
  59   return NS_OK;
  60 }
  61
  62 void
  63 IDBFileRequest::OnProgress(uint64_t aProgress, uint64_t aProgressMax)
  64 {
  65   FireProgressEvent(aProgress, aProgressMax);
  66 }
  67
  68 nsresult
  69 IDBFileRequest::NotifyHelperCompleted(FileHelper* aFileHelper)
  70 {
  71   MOZ_ASSERT(NS_IsMainThread(), "Wrong thread!");
  72
  73   nsresult rv = aFileHelper->ResultCode();
  74
  75   // If the request failed then fire error event and return.
  76   if (NS_FAILED(rv)) {
  77     FireError(rv);
  78     return NS_OK;
  79   }
  80
  81   // Otherwise we need to get the result from the helper.
  82   nsIScriptContext* sc = GetContextForEventHandlers(&rv);
  83   NS_ENSURE_STATE(sc);
  84
  85   AutoJSContext cx;
  86   MOZ_ASSERT(cx, "Failed to get a context!");
  87
  88   JS::Rooted<JS::Value> result(cx);
  89
  90   JS::Rooted<JSObject*> global(cx, sc->GetWindowProxy());
  91   MOZ_ASSERT(global, "Failed to get global object!");
  92
  93   JSAutoCompartment ac(cx, global);
  94
  95   rv = aFileHelper->GetSuccessResult(cx, &result);
  96   if (NS_FAILED(rv)) {
  97     NS_WARNING("GetSuccessResult failed!");
  98   }
  99
 100   if (NS_SUCCEEDED(rv)) {
 101     FireSuccess(result);
 102   }
 103   else {
 104     FireError(rv);
 105   }
 106   return NS_OK;
 107 }
 108
 109 NS_IMPL_CYCLE_COLLECTION_INHERITED(IDBFileRequest, DOMRequest,
 110                                    mFileHandle)
 111
 112 NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION_INHERITED(IDBFileRequest)
 113 NS_INTERFACE_MAP_END_INHERITING(DOMRequest)
 114
 115 NS_IMPL_ADDREF_INHERITED(IDBFileRequest, DOMRequest)
 116 NS_IMPL_RELEASE_INHERITED(IDBFileRequest, DOMRequest)
 117
 118 // virtual
 119 JSObject*
 120 IDBFileRequest::WrapObject(JSContext* aCx)
 121 {
 122   if (mWrapAsDOMRequest) {
 123     return DOMRequest::WrapObject(aCx);
 124   }
 125   return IDBFileRequestBinding::Wrap(aCx, this);
 126 }
 127
 128
 129 IDBFileHandle*
 130 IDBFileRequest::GetFileHandle() const
 131 {
 132   MOZ_ASSERT(NS_IsMainThread(), "Wrong thread!");
 133
 134   return static_cast<IDBFileHandle*>(mFileHandle.get());
 135 }
 136
 137 void
 138 IDBFileRequest::FireProgressEvent(uint64_t aLoaded, uint64_t aTotal)
 139 {
 140   if (NS_FAILED(CheckInnerWindowCorrectness())) {
 141     return;
 142   }
 143
 144   ProgressEventInit init;
 145   init.mBubbles = false;
 146   init.mCancelable = false;
 147   init.mLengthComputable = false;
 148   init.mLoaded = aLoaded;
 149   init.mTotal = aTotal;
 150
 151   nsRefPtr<ProgressEvent> event =
 152     ProgressEvent::Constructor(this, NS_LITERAL_STRING("progress"), init);
 153   DispatchTrustedEvent(event);
 154 }
 155
 156 } // namespace indexedDB
 157 } // namespace dom
 158 } // namespace mozilla