michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #ifndef nsDOMFileReader_h__
michael@0: #define nsDOMFileReader_h__
michael@0: 
michael@0: #include "mozilla/Attributes.h"
michael@0: #include "nsISupportsUtils.h"      
michael@0: #include "nsString.h"
michael@0: #include "nsWeakReference.h"
michael@0: #include "nsIStreamListener.h"     
michael@0: #include "nsIInterfaceRequestor.h" 
michael@0: #include "nsJSUtils.h"             
michael@0: #include "nsTArray.h"              
michael@0: #include "nsIJSNativeInitializer.h"
michael@0: #include "prtime.h"                
michael@0: #include "nsITimer.h"              
michael@0: 
michael@0: #include "nsIDOMFile.h"
michael@0: #include "nsIDOMFileReader.h"
michael@0: #include "nsIDOMFileList.h"
michael@0: #include "nsIInputStream.h"
michael@0: #include "nsCOMPtr.h"
michael@0: #include "nsIStreamLoader.h"
michael@0: #include "nsIChannel.h"
michael@0: 
michael@0: #include "FileIOObject.h"
michael@0: 
michael@0: class nsDOMFileReader : public mozilla::dom::FileIOObject,
michael@0:                         public nsIDOMFileReader,
michael@0:                         public nsIInterfaceRequestor,
michael@0:                         public nsSupportsWeakReference
michael@0: {
michael@0:   typedef mozilla::ErrorResult ErrorResult;
michael@0:   typedef mozilla::dom::GlobalObject GlobalObject;
michael@0: public:
michael@0:   nsDOMFileReader();
michael@0:   virtual ~nsDOMFileReader();
michael@0: 
michael@0:   NS_DECL_ISUPPORTS_INHERITED
michael@0: 
michael@0:   NS_DECL_NSIDOMFILEREADER
michael@0: 
michael@0:   NS_REALLY_FORWARD_NSIDOMEVENTTARGET(mozilla::DOMEventTargetHelper)
michael@0: 
michael@0:   // nsIInterfaceRequestor 
michael@0:   NS_DECL_NSIINTERFACEREQUESTOR
michael@0: 
michael@0:   // FileIOObject overrides
michael@0:   virtual void DoAbort(nsAString& aEvent) MOZ_OVERRIDE;
michael@0:   NS_IMETHOD DoOnStopRequest(nsIRequest* aRequest, nsISupports* aContext,
michael@0:                              nsresult aStatus, nsAString& aSuccessEvent,
michael@0:                              nsAString& aTerminationEvent) MOZ_OVERRIDE;
michael@0:   NS_IMETHOD DoOnDataAvailable(nsIRequest* aRequest, nsISupports* aContext,
michael@0:                                nsIInputStream* aInputStream, uint64_t aOffset,
michael@0:                                uint32_t aCount) MOZ_OVERRIDE;
michael@0: 
michael@0:   nsPIDOMWindow* GetParentObject() const
michael@0:   {
michael@0:     return GetOwner();
michael@0:   }
michael@0:   virtual JSObject* WrapObject(JSContext* aCx) MOZ_OVERRIDE;
michael@0: 
michael@0:   // WebIDL
michael@0:   static already_AddRefed<nsDOMFileReader>
michael@0:   Constructor(const GlobalObject& aGlobal, ErrorResult& aRv);
michael@0:   void ReadAsArrayBuffer(JSContext* aCx, nsIDOMBlob* aBlob, ErrorResult& aRv)
michael@0:   {
michael@0:     MOZ_ASSERT(aBlob);
michael@0:     ReadFileContent(aCx, aBlob, EmptyString(), FILE_AS_ARRAYBUFFER, aRv);
michael@0:   }
michael@0:   void ReadAsText(nsIDOMBlob* aBlob, const nsAString& aLabel, ErrorResult& aRv)
michael@0:   {
michael@0:     MOZ_ASSERT(aBlob);
michael@0:     ReadFileContent(nullptr, aBlob, aLabel, FILE_AS_TEXT, aRv);
michael@0:   }
michael@0:   void ReadAsDataURL(nsIDOMBlob* aBlob, ErrorResult& aRv)
michael@0:   {
michael@0:     MOZ_ASSERT(aBlob);
michael@0:     ReadFileContent(nullptr, aBlob, EmptyString(), FILE_AS_DATAURL, aRv);
michael@0:   }
michael@0: 
michael@0:   using FileIOObject::Abort;
michael@0: 
michael@0:   // Inherited ReadyState().
michael@0: 
michael@0:   void GetResult(JSContext* aCx, JS::MutableHandle<JS::Value> aResult,
michael@0:                  ErrorResult& aRv);
michael@0: 
michael@0:   using FileIOObject::GetError;
michael@0: 
michael@0:   IMPL_EVENT_HANDLER(loadstart)
michael@0:   using FileIOObject::GetOnprogress;
michael@0:   using FileIOObject::SetOnprogress;
michael@0:   IMPL_EVENT_HANDLER(load)
michael@0:   using FileIOObject::GetOnabort;
michael@0:   using FileIOObject::SetOnabort;
michael@0:   using FileIOObject::GetOnerror;
michael@0:   using FileIOObject::SetOnerror;
michael@0:   IMPL_EVENT_HANDLER(loadend)
michael@0: 
michael@0:   void ReadAsBinaryString(nsIDOMBlob* aBlob, ErrorResult& aRv)
michael@0:   {
michael@0:     MOZ_ASSERT(aBlob);
michael@0:     ReadFileContent(nullptr, aBlob, EmptyString(), FILE_AS_BINARY, aRv);
michael@0:   }
michael@0: 
michael@0: 
michael@0:   nsresult Init();
michael@0: 
michael@0:   NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS_INHERITED(nsDOMFileReader,
michael@0:                                                          FileIOObject)
michael@0:   void RootResultArrayBuffer();
michael@0: 
michael@0: protected:
michael@0:   enum eDataFormat {
michael@0:     FILE_AS_ARRAYBUFFER,
michael@0:     FILE_AS_BINARY,
michael@0:     FILE_AS_TEXT,
michael@0:     FILE_AS_DATAURL
michael@0:   };
michael@0: 
michael@0:   void ReadFileContent(JSContext* aCx, nsIDOMBlob* aBlob,
michael@0:                        const nsAString &aCharset, eDataFormat aDataFormat,
michael@0:                        ErrorResult& aRv);
michael@0:   nsresult GetAsText(nsIDOMBlob *aFile, const nsACString &aCharset,
michael@0:                      const char *aFileData, uint32_t aDataLen, nsAString &aResult);
michael@0:   nsresult GetAsDataURL(nsIDOMBlob *aFile, const char *aFileData, uint32_t aDataLen, nsAString &aResult); 
michael@0: 
michael@0:   void FreeFileData() {
michael@0:     moz_free(mFileData);
michael@0:     mFileData = nullptr;
michael@0:     mDataLen = 0;
michael@0:   }
michael@0: 
michael@0:   char *mFileData;
michael@0:   nsCOMPtr<nsIDOMBlob> mFile;
michael@0:   nsCString mCharset;
michael@0:   uint32_t mDataLen;
michael@0: 
michael@0:   eDataFormat mDataFormat;
michael@0: 
michael@0:   nsString mResult;
michael@0:   nsCOMPtr<nsIPrincipal> mPrincipal;
michael@0:   
michael@0:   JS::Heap<JSObject*> mResultArrayBuffer;
michael@0: };
michael@0: 
michael@0: #endif