michael@0: /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
michael@0:  *
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: #include "nsRDFXMLParser.h"
michael@0: 
michael@0: #include "nsIComponentManager.h"
michael@0: #include "nsIParser.h"
michael@0: #include "nsCharsetSource.h"
michael@0: #include "nsIRDFContentSink.h"
michael@0: #include "nsParserCIID.h"
michael@0: #include "nsStringStream.h"
michael@0: #include "nsNetUtil.h"
michael@0: 
michael@0: static NS_DEFINE_CID(kParserCID, NS_PARSER_CID);
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(nsRDFXMLParser, nsIRDFXMLParser)
michael@0: 
michael@0: nsresult
michael@0: nsRDFXMLParser::Create(nsISupports* aOuter, REFNSIID aIID, void** aResult)
michael@0: {
michael@0:     if (aOuter)
michael@0:         return NS_ERROR_NO_AGGREGATION;
michael@0: 
michael@0:     nsRDFXMLParser* result = new nsRDFXMLParser();
michael@0:     if (! result)
michael@0:         return NS_ERROR_OUT_OF_MEMORY;
michael@0: 
michael@0:     nsresult rv;
michael@0:     NS_ADDREF(result);
michael@0:     rv = result->QueryInterface(aIID, aResult);
michael@0:     NS_RELEASE(result);
michael@0:     return rv;
michael@0: }
michael@0: 
michael@0: nsRDFXMLParser::nsRDFXMLParser()
michael@0: {
michael@0:     MOZ_COUNT_CTOR(nsRDFXMLParser);
michael@0: }
michael@0: 
michael@0: nsRDFXMLParser::~nsRDFXMLParser()
michael@0: {
michael@0:     MOZ_COUNT_DTOR(nsRDFXMLParser);
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsRDFXMLParser::ParseAsync(nsIRDFDataSource* aSink, nsIURI* aBaseURI, nsIStreamListener** aResult)
michael@0: {
michael@0:     nsresult rv;
michael@0: 
michael@0:     nsCOMPtr<nsIRDFContentSink> sink =
michael@0:         do_CreateInstance("@mozilla.org/rdf/content-sink;1", &rv);
michael@0: 
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     rv = sink->Init(aBaseURI);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     // We set the content sink's data source directly to our in-memory
michael@0:     // store. This allows the initial content to be generated "directly".
michael@0:     rv = sink->SetDataSource(aSink);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     nsCOMPtr<nsIParser> parser = do_CreateInstance(kParserCID, &rv);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     parser->SetDocumentCharset(NS_LITERAL_CSTRING("UTF-8"),
michael@0:                                kCharsetFromDocTypeDefault);
michael@0:     parser->SetContentSink(sink);
michael@0: 
michael@0:     rv = parser->Parse(aBaseURI);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     return CallQueryInterface(parser, aResult);
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsRDFXMLParser::ParseString(nsIRDFDataSource* aSink, nsIURI* aBaseURI, const nsACString& aString)
michael@0: {
michael@0:     nsresult rv;
michael@0: 
michael@0:     nsCOMPtr<nsIRDFContentSink> sink =
michael@0:         do_CreateInstance("@mozilla.org/rdf/content-sink;1", &rv);
michael@0: 
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     rv = sink->Init(aBaseURI);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     // We set the content sink's data source directly to our in-memory
michael@0:     // store. This allows the initial content to be generated "directly".
michael@0:     rv = sink->SetDataSource(aSink);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     nsCOMPtr<nsIParser> parser = do_CreateInstance(kParserCID, &rv);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     parser->SetDocumentCharset(NS_LITERAL_CSTRING("UTF-8"),
michael@0:                                kCharsetFromOtherComponent);
michael@0:     parser->SetContentSink(sink);
michael@0: 
michael@0:     rv = parser->Parse(aBaseURI);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     nsCOMPtr<nsIStreamListener> listener =
michael@0:         do_QueryInterface(parser);
michael@0: 
michael@0:     if (! listener)
michael@0:         return NS_ERROR_FAILURE;
michael@0: 
michael@0:     nsCOMPtr<nsIInputStream> stream;
michael@0:     rv = NS_NewCStringInputStream(getter_AddRefs(stream), aString);
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     nsCOMPtr<nsIChannel> channel;
michael@0:     rv = NS_NewInputStreamChannel(getter_AddRefs(channel), aBaseURI, stream,
michael@0:                                   NS_LITERAL_CSTRING("text/xml"));
michael@0:     if (NS_FAILED(rv)) return rv;
michael@0: 
michael@0:     listener->OnStartRequest(channel, nullptr);
michael@0:     listener->OnDataAvailable(channel, nullptr, stream, 0, aString.Length());
michael@0:     listener->OnStopRequest(channel, nullptr, NS_OK);
michael@0: 
michael@0:     return NS_OK;
michael@0: }