/* 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/. */

#include "nsHtml5Module.h"

#include "mozilla/Attributes.h"

#include "mozilla/Preferences.h"

#include "mozilla/Services.h"

#include "mozilla/StaticPrefs.h"

#include "nsHtml5AttributeName.h"

#include "nsHtml5ElementName.h"

#include "nsHtml5HtmlAttributes.h"

#include "nsHtml5NamedCharacters.h"

#include "nsHtml5Portability.h"

#include "nsHtml5StackNode.h"

#include "nsHtml5Tokenizer.h"

#include "nsHtml5TreeBuilder.h"

#include "nsHtml5UTF16Buffer.h"

#include "nsIObserverService.h"

#include "nsIServiceManager.h"

using namespace mozilla;

// static

nsIThread* nsHtml5Module::sStreamParserThread = nullptr;

nsIThread* nsHtml5Module::sMainThread = nullptr;

// static

void

nsHtml5Module::InitializeStatics()

{

  nsHtml5AttributeName::initializeStatics();

  nsHtml5ElementName::initializeStatics();

  nsHtml5HtmlAttributes::initializeStatics();

  nsHtml5NamedCharacters::initializeStatics();

  nsHtml5Portability::initializeStatics();

  nsHtml5StackNode::initializeStatics();

  nsHtml5Tokenizer::initializeStatics();

  nsHtml5TreeBuilder::initializeStatics();

  nsHtml5UTF16Buffer::initializeStatics();

#ifdef DEBUG

  sNsHtml5ModuleInitialized = true;

#endif

}

// static

void

nsHtml5Module::ReleaseStatics()

{

#ifdef DEBUG

  sNsHtml5ModuleInitialized = false;

#endif

  nsHtml5AttributeName::releaseStatics();

  nsHtml5ElementName::releaseStatics();

  nsHtml5HtmlAttributes::releaseStatics();

  nsHtml5NamedCharacters::releaseStatics();

  nsHtml5Portability::releaseStatics();

  nsHtml5StackNode::releaseStatics();

  nsHtml5Tokenizer::releaseStatics();

  nsHtml5TreeBuilder::releaseStatics();

  nsHtml5UTF16Buffer::releaseStatics();

  NS_IF_RELEASE(sStreamParserThread);

  NS_IF_RELEASE(sMainThread);

}

// static

already_AddRefed<nsIParser>

nsHtml5Module::NewHtml5Parser()

{

  MOZ_ASSERT(sNsHtml5ModuleInitialized, "nsHtml5Module not initialized.");

  nsCOMPtr<nsIParser> rv = new nsHtml5Parser();

  return rv.forget();

}

// static

nsresult

nsHtml5Module::Initialize(nsIParser* aParser,

                          nsIDocument* aDoc,

                          nsIURI* aURI,

                          nsISupports* aContainer,

                          nsIChannel* aChannel)

{

  MOZ_ASSERT(sNsHtml5ModuleInitialized, "nsHtml5Module not initialized.");

  nsHtml5Parser* parser = static_cast<nsHtml5Parser*>(aParser);

  return parser->Initialize(aDoc, aURI, aContainer, aChannel);

}

class nsHtml5ParserThreadTerminator final : public nsIObserver

{

public:

  NS_DECL_ISUPPORTS

  explicit nsHtml5ParserThreadTerminator(nsIThread* aThread)

    : mThread(aThread)

  {

  }

  NS_IMETHOD Observe(nsISupports*, const char* topic, const char16_t*) override

  {

    NS_ASSERTION(!strcmp(topic, "xpcom-shutdown-threads"), "Unexpected topic");

    if (mThread) {

      mThread->Shutdown();

      mThread = nullptr;

    }

    return NS_OK;

  }

private:

  ~nsHtml5ParserThreadTerminator() {}

  nsCOMPtr<nsIThread> mThread;

};

NS_IMPL_ISUPPORTS(nsHtml5ParserThreadTerminator, nsIObserver)

// static

nsIThread*

nsHtml5Module::GetStreamParserThread()

{

  if (StaticPrefs::html5_offmainthread()) {

    if (!sStreamParserThread) {

      NS_NewNamedThread("HTML5 Parser", &sStreamParserThread);

      NS_ASSERTION(sStreamParserThread, "Thread creation failed!");

      nsCOMPtr<nsIObserverService> os = mozilla::services::GetObserverService();

      NS_ASSERTION(os, "do_GetService failed");

      os->AddObserver(new nsHtml5ParserThreadTerminator(sStreamParserThread),

                      "xpcom-shutdown-threads",

                      false);

    }

    return sStreamParserThread;

  }

  if (!sMainThread) {

    NS_GetMainThread(&sMainThread);

    NS_ASSERTION(sMainThread, "Main thread getter failed");

  }

  return sMainThread;

}

#ifdef DEBUG

bool nsHtml5Module::sNsHtml5ModuleInitialized = false;

#endif