michael@0: /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
michael@0: /* vim: set ts=2 sw=2 et tw=80: */
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: 
michael@0: #include "nsIAtom.h"
michael@0: #include "CParserContext.h"
michael@0: #include "nsToken.h"
michael@0: #include "prenv.h"  
michael@0: #include "nsIHTMLContentSink.h"
michael@0: #include "nsHTMLTokenizer.h"
michael@0: #include "nsMimeTypes.h"
michael@0: #include "nsHTMLTokenizer.h"
michael@0: 
michael@0: CParserContext::CParserContext(CParserContext* aPrevContext,
michael@0:                                nsScanner* aScanner, 
michael@0:                                void *aKey, 
michael@0:                                eParserCommands aCommand,
michael@0:                                nsIRequestObserver* aListener, 
michael@0:                                eAutoDetectResult aStatus, 
michael@0:                                bool aCopyUnused)
michael@0:   : mListener(aListener),
michael@0:     mKey(aKey),
michael@0:     mPrevContext(aPrevContext),
michael@0:     mScanner(aScanner),
michael@0:     mDTDMode(eDTDMode_unknown),
michael@0:     mStreamListenerState(eNone),
michael@0:     mContextType(eCTNone),
michael@0:     mAutoDetectStatus(aStatus),
michael@0:     mParserCommand(aCommand),
michael@0:     mMultipart(true),
michael@0:     mCopyUnused(aCopyUnused)
michael@0: { 
michael@0:   MOZ_COUNT_CTOR(CParserContext); 
michael@0: } 
michael@0: 
michael@0: CParserContext::~CParserContext()
michael@0: {
michael@0:   // It's ok to simply ingore the PrevContext.
michael@0:   MOZ_COUNT_DTOR(CParserContext);
michael@0: }
michael@0: 
michael@0: void
michael@0: CParserContext::SetMimeType(const nsACString& aMimeType)
michael@0: {
michael@0:   mMimeType.Assign(aMimeType);
michael@0: 
michael@0:   mDocType = ePlainText;
michael@0: 
michael@0:   if (mMimeType.EqualsLiteral(TEXT_HTML))
michael@0:     mDocType = eHTML_Strict;
michael@0:   else if (mMimeType.EqualsLiteral(TEXT_XML)              ||
michael@0:            mMimeType.EqualsLiteral(APPLICATION_XML)       ||
michael@0:            mMimeType.EqualsLiteral(APPLICATION_XHTML_XML) ||
michael@0:            mMimeType.EqualsLiteral(TEXT_XUL)              ||
michael@0:            mMimeType.EqualsLiteral(IMAGE_SVG_XML)         ||
michael@0:            mMimeType.EqualsLiteral(APPLICATION_MATHML_XML) ||
michael@0:            mMimeType.EqualsLiteral(APPLICATION_RDF_XML)   ||
michael@0:            mMimeType.EqualsLiteral(TEXT_RDF))
michael@0:     mDocType = eXML;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: CParserContext::GetTokenizer(nsIDTD* aDTD,
michael@0:                              nsIContentSink* aSink,
michael@0:                              nsITokenizer*& aTokenizer)
michael@0: {
michael@0:   nsresult result = NS_OK;
michael@0:   int32_t type = aDTD ? aDTD->GetType() : NS_IPARSER_FLAG_HTML;
michael@0: 
michael@0:   if (!mTokenizer) {
michael@0:     if (type == NS_IPARSER_FLAG_HTML || mParserCommand == eViewSource) {
michael@0:       mTokenizer = new nsHTMLTokenizer;
michael@0:     }
michael@0:     else if (type == NS_IPARSER_FLAG_XML) {
michael@0:       mTokenizer = do_QueryInterface(aDTD, &result);
michael@0:     }
michael@0:   }
michael@0: 
michael@0:   aTokenizer = mTokenizer;
michael@0: 
michael@0:   return result;
michael@0: }