michael@0: /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
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 "nsIOutputStream.h"
michael@0: #include "nsReadableUtils.h"
michael@0: #include "nsCRT.h"
michael@0: #include "nsCOMPtr.h"
michael@0: #include "nsString.h"
michael@0: #include "nsPrintfCString.h"
michael@0: #include "nsIBufferedStreams.h"
michael@0: #include "nsNetCID.h"
michael@0: #include "nsComponentManagerUtils.h"
michael@0: 
michael@0: #include "rdfISerializer.h"
michael@0: #include "rdfIDataSource.h"
michael@0: #include "rdfITripleVisitor.h"
michael@0: 
michael@0: #include "nsIRDFResource.h"
michael@0: #include "nsIRDFLiteral.h"
michael@0: #include "mozilla/Attributes.h"
michael@0: 
michael@0: class TriplesVisitor MOZ_FINAL : public rdfITripleVisitor
michael@0: {
michael@0: public:
michael@0:     TriplesVisitor(nsIOutputStream* aOut) : mOut(aOut) {}
michael@0:     NS_DECL_RDFITRIPLEVISITOR
michael@0:     NS_DECL_ISUPPORTS
michael@0: protected:
michael@0:     nsresult writeResource(nsIRDFResource* aResource);
michael@0:     nsIOutputStream* mOut;
michael@0: };
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(TriplesVisitor, rdfITripleVisitor)
michael@0: 
michael@0: nsresult
michael@0: TriplesVisitor::writeResource(nsIRDFResource *aResource)
michael@0: {
michael@0:     nsCString res;
michael@0:     uint32_t writeCount, wroteCount;
michael@0:     mOut->Write("<", 1, &wroteCount);
michael@0:     NS_ENSURE_TRUE(wroteCount == 1, NS_ERROR_FAILURE);
michael@0:     nsresult rv = aResource->GetValueUTF8(res);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     writeCount = res.Length();
michael@0:     mOut->Write(res.get(), writeCount, &wroteCount);
michael@0:     NS_ENSURE_TRUE(writeCount == wroteCount, NS_ERROR_FAILURE);
michael@0:     mOut->Write("> ", 2, &wroteCount);
michael@0:     NS_ENSURE_TRUE(wroteCount == 2, NS_ERROR_FAILURE);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: TriplesVisitor::Visit(nsIRDFNode *aSubject, nsIRDFResource *aPredicate,
michael@0:                       nsIRDFNode *aObject, bool aTruthValue)
michael@0: {
michael@0:     nsCOMPtr<nsIRDFResource> subjectRes = do_QueryInterface(aSubject);
michael@0:     nsresult rv = NS_OK;
michael@0:     if (subjectRes) {
michael@0:         rv = writeResource(subjectRes);
michael@0:     }
michael@0:     if (NS_FAILED(rv)) {
michael@0:         return rv;
michael@0:     }
michael@0:     rv = writeResource(aPredicate);
michael@0:     if (NS_FAILED(rv)) {
michael@0:         return rv;
michael@0:     }
michael@0:     nsCOMPtr<nsIRDFResource> res = do_QueryInterface(aObject);
michael@0:     nsCOMPtr<nsIRDFLiteral> lit;
michael@0:     nsCOMPtr<nsIRDFInt> intLit;
michael@0:     uint32_t wroteCount;
michael@0:     if (res) {
michael@0:         rv = writeResource(res);
michael@0:     } else if ((lit = do_QueryInterface(aObject)) != nullptr) {
michael@0:         const char16_t *value;
michael@0:         lit->GetValueConst(&value);
michael@0:         nsAutoCString object;
michael@0:         object.AppendLiteral("\"");
michael@0:         AppendUTF16toUTF8(value, object);
michael@0:         object.AppendLiteral("\" ");
michael@0:         uint32_t writeCount = object.Length();
michael@0:         rv = mOut->Write(object.get(), writeCount, &wroteCount);
michael@0:         NS_ENSURE_TRUE(writeCount == wroteCount, NS_ERROR_FAILURE);
michael@0:     } else if ((intLit = do_QueryInterface(aObject)) != nullptr) {
michael@0:         int32_t value;
michael@0:         intLit->GetValue(&value);
michael@0:         nsPrintfCString
michael@0:             object("\"%i\"^^<http://www.w3.org/2001/XMLSchema#integer> ",
michael@0:                    value);
michael@0:         uint32_t writeCount = object.Length();
michael@0:         rv = mOut->Write(object.get(), writeCount, &wroteCount);
michael@0:         NS_ENSURE_TRUE(writeCount == wroteCount, NS_ERROR_FAILURE);
michael@0:     }
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     return mOut->Write(".\n", 2, &wroteCount);
michael@0: }
michael@0: 
michael@0: class rdfTriplesSerializer MOZ_FINAL : public rdfISerializer
michael@0: {
michael@0: public:
michael@0:   NS_DECL_ISUPPORTS
michael@0:   NS_DECL_RDFISERIALIZER
michael@0: 
michael@0:   rdfTriplesSerializer();
michael@0: 
michael@0: private:
michael@0:   ~rdfTriplesSerializer();
michael@0: 
michael@0: };
michael@0: 
michael@0: nsresult
michael@0: NS_NewTriplesSerializer(rdfISerializer** aResult)
michael@0: {
michael@0:     NS_PRECONDITION(aResult != nullptr, "null ptr");
michael@0:     if (! aResult)
michael@0:         return NS_ERROR_NULL_POINTER;
michael@0: 
michael@0:     *aResult = new rdfTriplesSerializer();
michael@0:     if (! *aResult)
michael@0:         return NS_ERROR_OUT_OF_MEMORY;
michael@0:     NS_ADDREF(*aResult);
michael@0:     return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMPL_ISUPPORTS(rdfTriplesSerializer, rdfISerializer)
michael@0: 
michael@0: rdfTriplesSerializer::rdfTriplesSerializer()
michael@0: {
michael@0: }
michael@0: 
michael@0: rdfTriplesSerializer::~rdfTriplesSerializer()
michael@0: {
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: rdfTriplesSerializer::Serialize(rdfIDataSource *aDataSource,
michael@0:                                 nsIOutputStream *aOut)
michael@0: {
michael@0:     nsresult rv;
michael@0:     nsCOMPtr<nsIBufferedOutputStream> bufout = 
michael@0:         do_CreateInstance(NS_BUFFEREDOUTPUTSTREAM_CONTRACTID, &rv);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     rv = bufout->Init(aOut, 1024);
michael@0:     NS_ENSURE_SUCCESS(rv, rv);
michael@0:     nsCOMPtr<rdfITripleVisitor> tv = new TriplesVisitor(bufout);
michael@0:     NS_ENSURE_TRUE(tv, NS_ERROR_OUT_OF_MEMORY);
michael@0:     return aDataSource->VisitAllTriples(tv);
michael@0: }