/* -*- Mode: C++; tab-width: 2; 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 nsSAXXMLReader_h__

#define nsSAXXMLReader_h__

#include "nsCOMPtr.h"

#include "nsIContentSink.h"

#include "nsIExpatSink.h"

#include "nsIParser.h"

#include "nsIURI.h"

#include "nsISAXXMLReader.h"

#include "nsISAXContentHandler.h"

#include "nsISAXErrorHandler.h"

#include "nsCycleCollectionParticipant.h"

#include "mozilla/Attributes.h"

#include "mozilla/NotNull.h"

#define NS_SAXXMLREADER_CONTRACTID "@mozilla.org/saxparser/xmlreader;1"

#define NS_SAXXMLREADER_CID  \

{ 0xab1da296, 0x6125, 0x40ba, \

{ 0x96, 0xd0, 0x47, 0xa8, 0x28, 0x2a, 0xe3, 0xdb} }

class nsSAXXMLReader final : public nsISAXXMLReader,

                             public nsIExpatSink,

                             public nsIContentSink

{

public:

  NS_DECL_CYCLE_COLLECTING_ISUPPORTS

  NS_DECL_CYCLE_COLLECTION_CLASS_AMBIGUOUS(nsSAXXMLReader, nsISAXXMLReader)

  NS_DECL_NSIEXPATSINK

  NS_DECL_NSISAXXMLREADER

  NS_DECL_NSIREQUESTOBSERVER

  NS_DECL_NSISTREAMLISTENER

  nsSAXXMLReader();

  //nsIContentSink

  NS_IMETHOD WillParse() override

  {

    return NS_OK;

  }

  NS_IMETHOD WillBuildModel(nsDTDMode aDTDMode) override;

  NS_IMETHOD DidBuildModel(bool aTerminated) override;

  NS_IMETHOD SetParser(nsParserBase* aParser) override;

  NS_IMETHOD WillInterrupt() override

  {

    return NS_OK;

  }

  NS_IMETHOD WillResume() override

  {

    return NS_OK;

  }

  virtual void FlushPendingNotifications(mozilla::FlushType aType) override

  {

  }

  virtual void SetDocumentCharset(NotNull<const Encoding*> aEncoding) override

  {

  }

  virtual nsISupports *GetTarget() override

  {

    return nullptr;

  }

private:

  ~nsSAXXMLReader() {}

  nsCOMPtr<nsISAXContentHandler> mContentHandler;

  nsCOMPtr<nsISAXErrorHandler> mErrorHandler;

  nsCOMPtr<nsIURI> mBaseURI;

  nsCOMPtr<nsIStreamListener> mListener;

  nsCOMPtr<nsIRequestObserver> mParserObserver;

  bool mIsAsyncParse;

  static bool TryChannelCharset(nsIChannel *aChannel,

                                int32_t& aCharsetSource,

                                NotNull<const Encoding*>& aEncoding);

  nsresult EnsureBaseURI();

  nsresult InitParser(nsIRequestObserver *aListener, nsIChannel *aChannel);

  nsresult SplitExpatName(const char16_t *aExpatName,

                          nsString &aURI,

                          nsString &aLocalName,

                          nsString &aQName);

};

#endif // nsSAXXMLReader_h__