gecko-dev/dom/webbrowserpersist/nsWebBrowserPersist.h

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef nsWebBrowserPersist_h__
#define nsWebBrowserPersist_h__

#include "nsCOMPtr.h"
#include "nsWeakReference.h"

#include "nsIInterfaceRequestor.h"
#include "nsIMIMEService.h"
#include "nsIStreamListener.h"
#include "nsIOutputStream.h"
#include "nsIInputStream.h"
#include "nsIChannel.h"
#include "nsIDocumentEncoder.h"
#include "nsITransport.h"
#include "nsIProgressEventSink.h"
#include "nsIFile.h"
#include "nsIContentPolicy.h"
#include "nsIWebProgressListener2.h"
#include "nsIWebBrowserPersist.h"
#include "nsIWebBrowserPersistDocument.h"

#include "mozilla/UniquePtr.h"
#include "nsClassHashtable.h"
#include "nsHashKeys.h"
#include "nsTArray.h"

class nsIStorageStream;
class nsIWebBrowserPersistDocument;

class nsWebBrowserPersist final : public nsIInterfaceRequestor,
                                  public nsIWebBrowserPersist,
                                  public nsIStreamListener,
                                  public nsIProgressEventSink,
                                  public nsSupportsWeakReference {
  friend class nsEncoderNodeFixup;

  // Public members
 public:
  nsWebBrowserPersist();

  NS_DECL_ISUPPORTS
  NS_DECL_NSIINTERFACEREQUESTOR
  NS_DECL_NSICANCELABLE
  NS_DECL_NSIWEBBROWSERPERSIST
  NS_DECL_NSIREQUESTOBSERVER
  NS_DECL_NSISTREAMLISTENER
  NS_DECL_NSIPROGRESSEVENTSINK

  // Private members
 private:
  virtual ~nsWebBrowserPersist();
  nsresult SaveURIInternal(nsIURI *aURI, nsIPrincipal *aTriggeringPrincipal,
                           nsContentPolicyType aContentPolicyType,
                           uint32_t aCacheKey, nsIURI *aReferrer,
                           uint32_t aReferrerPolicy, nsIInputStream *aPostData,
                           const char *aExtraHeaders, nsIURI *aFile,
                           bool aCalcFileExt, bool aIsPrivate);
  nsresult SaveChannelInternal(nsIChannel *aChannel, nsIURI *aFile,
                               bool aCalcFileExt);
  nsresult SaveDocumentInternal(nsIWebBrowserPersistDocument *aDocument,
                                nsIURI *aFile, nsIURI *aDataPath);
  nsresult SaveDocuments();
  void FinishSaveDocumentInternal(nsIURI *aFile, nsIFile *aDataPath);
  nsresult GetExtensionForContentType(const char16_t *aContentType,
                                      char16_t **aExt);

  struct CleanupData;
  struct DocData;
  struct OutputData;
  struct UploadData;
  struct URIData;
  struct WalkData;
  struct URIFixupData;

  class OnWalk;
  class OnWrite;
  class FlatURIMap;
  friend class OnWalk;
  friend class OnWrite;

  nsresult SaveDocumentDeferred(mozilla::UniquePtr<WalkData> &&aData);
  void Cleanup();
  void CleanupLocalFiles();
  nsresult GetValidURIFromObject(nsISupports *aObject, nsIURI **aURI) const;
  static nsresult GetLocalFileFromURI(nsIURI *aURI, nsIFile **aLocalFile);
  static nsresult AppendPathToURI(nsIURI *aURI, const nsAString &aPath,
                                  nsCOMPtr<nsIURI> &aOutURI);
  nsresult MakeAndStoreLocalFilenameInURIMap(
      nsIURI *aURI, nsIWebBrowserPersistDocument *aDoc,
      nsContentPolicyType aContentPolicyType, bool aNeedsPersisting,
      URIData **aData);
  nsresult MakeOutputStream(nsIURI *aFile, nsIOutputStream **aOutputStream);
  nsresult MakeOutputStreamFromFile(nsIFile *aFile,
                                    nsIOutputStream **aOutputStream);
  nsresult MakeOutputStreamFromURI(nsIURI *aURI, nsIOutputStream **aOutStream);
  nsresult CreateChannelFromURI(nsIURI *aURI, nsIChannel **aChannel);
  nsresult StartUpload(nsIStorageStream *aOutStream, nsIURI *aDestinationURI,
                       const nsACString &aContentType);
  nsresult StartUpload(nsIInputStream *aInputStream, nsIURI *aDestinationURI,
                       const nsACString &aContentType);
  nsresult CalculateAndAppendFileExt(nsIURI *aURI, nsIChannel *aChannel,
                                     nsIURI *aOriginalURIWithExtension,
                                     nsCOMPtr<nsIURI> &aOutURI);
  nsresult CalculateUniqueFilename(nsIURI *aURI, nsCOMPtr<nsIURI> &aOutURI);
  nsresult MakeFilenameFromURI(nsIURI *aURI, nsString &aFilename);
  nsresult StoreURI(const char *aURI, nsIWebBrowserPersistDocument *aDoc,
                    nsContentPolicyType aContentPolicyType,
                    bool aNeedsPersisting = true, URIData **aData = nullptr);
  nsresult StoreURI(nsIURI *aURI, nsIWebBrowserPersistDocument *aDoc,
                    nsContentPolicyType aContentPolicyType,
                    bool aNeedsPersisting = true, URIData **aData = nullptr);
  bool DocumentEncoderExists(const char *aContentType);

  nsresult SaveSubframeContent(nsIWebBrowserPersistDocument *aFrameContent,
                               nsIWebBrowserPersistDocument *aParentDocument,
                               const nsCString &aURISpec, URIData *aData);
  nsresult SendErrorStatusChange(bool aIsReadError, nsresult aResult,
                                 nsIRequest *aRequest, nsIURI *aURI);

  nsresult FixRedirectedChannelEntry(nsIChannel *aNewChannel);

  void EndDownload(nsresult aResult);
  void FinishDownload();
  void SerializeNextFile();
  void CalcTotalProgress();

  void SetApplyConversionIfNeeded(nsIChannel *aChannel);

  nsCOMPtr<nsIURI> mCurrentDataPath;
  bool mCurrentDataPathIsRelative;
  nsCString mCurrentRelativePathToData;
  nsCOMPtr<nsIURI> mCurrentBaseURI;
  nsCString mCurrentCharset;
  nsCOMPtr<nsIURI> mTargetBaseURI;
  uint32_t mCurrentThingsToPersist;

  nsCOMPtr<nsIMIMEService> mMIMEService;
  nsCOMPtr<nsIURI> mURI;
  nsCOMPtr<nsIWebProgressListener> mProgressListener;
  /**
   * Progress listener for 64-bit values; this is the same object as
   * mProgressListener, but is a member to avoid having to qi it for each
   * progress notification.
   */
  nsCOMPtr<nsIWebProgressListener2> mProgressListener2;
  nsCOMPtr<nsIProgressEventSink> mEventSink;
  nsClassHashtable<nsISupportsHashKey, OutputData> mOutputMap;
  nsClassHashtable<nsISupportsHashKey, UploadData> mUploadList;
  nsClassHashtable<nsCStringHashKey, URIData> mURIMap;
  nsCOMPtr<nsIWebBrowserPersistURIMap> mFlatURIMap;
  nsTArray<mozilla::UniquePtr<WalkData>> mWalkStack;
  nsTArray<DocData *> mDocList;
  nsTArray<CleanupData *> mCleanupList;
  nsTArray<nsCString> mFilenameList;
  bool mFirstAndOnlyUse;
  bool mSavingDocument;
  bool mCancel;
  bool mCompleted;
  bool mStartSaving;
  bool mReplaceExisting;
  bool mSerializingOutput;
  bool mIsPrivate;
  uint32_t mPersistFlags;
  nsresult mPersistResult;
  int64_t mTotalCurrentProgress;
  int64_t mTotalMaxProgress;
  int16_t mWrapColumn;
  uint32_t mEncodingFlags;
  nsString mContentType;
};

#endif