/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */

/*

 * This file is part of the LibreOffice project.

 *

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

 *

 * This file incorporates work covered by the following license notice:

 *

 *   Licensed to the Apache Software Foundation (ASF) under one or more

 *   contributor license agreements. See the NOTICE file distributed

 *   with this work for additional information regarding copyright

 *   ownership. The ASF licenses this file to you under the Apache

 *   License, Version 2.0 (the "License"); you may not use this file

 *   except in compliance with the License. You may obtain a copy of

 *   the License at http://www.apache.org/licenses/LICENSE-2.0 .

 */

#include <sal/config.h>

#include <com/sun/star/graphic/XSvgParser.hpp>

#include <com/sun/star/lang/XServiceInfo.hpp>

#include <com/sun/star/lang/XInitialization.hpp>

#include <comphelper/processfactory.hxx>

#include <cppuhelper/implbase.hxx>

#include <cppuhelper/supportsservice.hxx>

#include <com/sun/star/xml/sax/XParser.hpp>

#include <com/sun/star/xml/sax/Parser.hpp>

#include <com/sun/star/xml/sax/InputSource.hpp>

#include <svgdocumenthandler.hxx>

#include <svgfeimagenode.hxx>

#include <svgimagenode.hxx>

#include <comphelper/sequence.hxx>

#include <comphelper/diagnose_ex.hxx>

#include <rtl/ref.hxx>

#include <tools/stream.hxx>

#include <unotools/streamwrap.hxx>

#include <svgvisitor.hxx>

#include <algorithm>

#include <utility>

using namespace ::com::sun::star;

namespace svgio::svgreader

{

        namespace {

        class XSvgParser : public ::cppu::WeakImplHelper< graphic::XSvgParser, lang::XServiceInfo >

        {

        private:

            std::shared_ptr<SvgDrawVisitor> mpVisitor;

            uno::Reference< uno::XComponentContext > context_;

            bool parseSvgXML(uno::Reference<io::XInputStream> const & xSVGStream,

                             uno::Reference<xml::sax::XDocumentHandler> const & xSvgDocHdl);

        public:

            explicit XSvgParser(

                uno::Reference< uno::XComponentContext > context);

            XSvgParser(const XSvgParser&) = delete;

            XSvgParser& operator=(const XSvgParser&) = delete;

            // XSvgParser

            virtual uno::Sequence< uno::Reference< ::graphic::XPrimitive2D > > SAL_CALL getDecomposition(

                const uno::Reference< ::io::XInputStream >& xSVGStream,

                const OUString& aAbsolutePath) override;

            virtual uno::Any SAL_CALL getDrawCommands(

                uno::Reference<io::XInputStream> const & xSvgStream,

                const OUString& aAbsolutePath) override;

            virtual uno::Sequence<OUString> SAL_CALL getExternalImageUrls(

                uno::Reference<io::XInputStream> const & xSvgStream,

                const OUString& aAbsolutePath) override;

            // XServiceInfo

            virtual OUString SAL_CALL getImplementationName() override;

            virtual sal_Bool SAL_CALL supportsService(const OUString&) override;

            virtual uno::Sequence< OUString > SAL_CALL getSupportedServiceNames() override;

        };

        }

        XSvgParser::XSvgParser(

            uno::Reference< uno::XComponentContext > context):

            context_(std::move(context))

        {

        }

        bool XSvgParser::parseSvgXML(uno::Reference<io::XInputStream> const & xSVGStream, uno::Reference<xml::sax::XDocumentHandler> const & xSvgDocHdl)

        {

            try

            {

                // prepare ParserInputSource

                xml::sax::InputSource myInputSource;

                myInputSource.aInputStream = xSVGStream;

                // get parser

                uno::Reference< xml::sax::XParser > xParser(

                    xml::sax::Parser::create(context_));

                // fdo#60471 need to enable internal entities because

                // certain ... popular proprietary products write SVG files

                // that use entities to define XML namespaces.

                uno::Reference<lang::XInitialization> const xInit(xParser,

                        uno::UNO_QUERY_THROW);

                uno::Sequence<uno::Any> args{ uno::Any(u"DoSmeplease"_ustr) };

                xInit->initialize(args);

                // connect parser and filter

                xParser->setDocumentHandler(xSvgDocHdl);

                // finally, parse the stream to a hierarchy of

                // SVGGraphicPrimitive2D which will be embedded to the

                // primitive sequence. Their decompositions will in the

                // end create local low-level primitives, thus SVG will

                // be processable from all our processors

                xParser->parseStream(myInputSource);

            }

            catch(const uno::Exception&)

            {

                TOOLS_INFO_EXCEPTION( "svg", "Parse error");

                return false;

            }

            return true;

        }

        uno::Sequence< uno::Reference< ::graphic::XPrimitive2D > > XSvgParser::getDecomposition(

            const uno::Reference< ::io::XInputStream >& xSVGStream,

            const OUString& aAbsolutePath )

        {

            drawinglayer::primitive2d::Primitive2DContainer aRetval;

            if(xSVGStream.is())

            {

                // local document handler

                rtl::Reference<SvgDocHdl> pSvgDocHdl = new SvgDocHdl(aAbsolutePath);

                parseSvgXML(xSVGStream, pSvgDocHdl);

                // decompose to primitives

                for(std::unique_ptr<SvgNode> const & pCandidate : pSvgDocHdl->getSvgDocument().getSvgNodeVector())

                {

                    if (Display::None != pCandidate->getDisplay())

                    {

                        pCandidate->decomposeSvgNode(aRetval, false);

                    }

                }

            }

            else

            {

                OSL_ENSURE(false, "Invalid stream (!)");

            }

            return aRetval.toSequence();

        }

        uno::Any SAL_CALL XSvgParser::getDrawCommands(

                uno::Reference<io::XInputStream> const & xSvgStream,

                const OUString& aAbsolutePath)

        {

            uno::Any aAnyResult;

            if (!xSvgStream.is())

                return aAnyResult;

            rtl::Reference<SvgDocHdl> pSvgDocHdl = new SvgDocHdl(aAbsolutePath);

            parseSvgXML(xSvgStream, pSvgDocHdl);

            // decompose to primitives

            for (std::unique_ptr<SvgNode> const & pCandidate : pSvgDocHdl->getSvgDocument().getSvgNodeVector())

            {

                if (Display::None != pCandidate->getDisplay())

                {

                    mpVisitor = std::make_shared<SvgDrawVisitor>();

                    pCandidate->accept(*mpVisitor);

                    std::shared_ptr<gfx::DrawRoot> pDrawRoot(mpVisitor->getDrawRoot());

                    sal_uInt64 nPointer = reinterpret_cast<sal_uInt64>(pDrawRoot.get());

                    aAnyResult <<= sal_uInt64(nPointer);

                }

            }

            return aAnyResult;

        }

        static void collectExternalImageUrls(const SvgNode& rNode, std::vector<OUString>& rUrls)

        {

            OUString aUrl;

            if (auto pImage = dynamic_cast<const SvgImageNode*>(&rNode))

                aUrl = pImage->getExternalImageUrl();

            else if (auto pFeImage = dynamic_cast<const SvgFeImageNode*>(&rNode))

                aUrl = pFeImage->getExternalImageUrl();

            if (!aUrl.isEmpty() && std::find(rUrls.begin(), rUrls.end(), aUrl) == rUrls.end())

                rUrls.push_back(aUrl);

            for (std::unique_ptr<SvgNode> const & pChild : rNode.getChildren())

                collectExternalImageUrls(*pChild, rUrls);

        }

        uno::Sequence<OUString> XSvgParser::getExternalImageUrls(

                uno::Reference<io::XInputStream> const & xSvgStream,

                const OUString& aAbsolutePath)

        {

            std::vector<OUString> aUrls;

            if (!xSvgStream.is())

                return {};

            rtl::Reference<SvgDocHdl> pSvgDocHdl = new SvgDocHdl(aAbsolutePath);

            parseSvgXML(xSvgStream, pSvgDocHdl);

            const SvgDocument& rDocument = pSvgDocHdl->getSvgDocument();

            for (std::unique_ptr<SvgNode> const & pCandidate : rDocument.getSvgNodeVector())

                collectExternalImageUrls(*pCandidate, aUrls);

            return comphelper::containerToSequence(aUrls);

        }

        OUString SAL_CALL XSvgParser::getImplementationName()

        {

            return u"svgio::svgreader::XSvgParser"_ustr;

        }

        sal_Bool SAL_CALL XSvgParser::supportsService(const OUString& rServiceName)

        {

            return cppu::supportsService(this, rServiceName);

        }

        uno::Sequence< OUString > SAL_CALL XSvgParser::getSupportedServiceNames()

        {

            return { u"com.sun.star.graphic.SvgTools"_ustr };

        }

} // end of namespace svgio::svgreader

extern "C" SAL_DLLPUBLIC_EXPORT css::uno::XInterface*

svgio_XSvgParser_get_implementation(

    css::uno::XComponentContext* context, css::uno::Sequence<css::uno::Any> const&)

{

    return cppu::acquire(new svgio::svgreader::XSvgParser(context));

}

extern "C" bool TestImportSVG(SvStream& rStream)

{

    css::uno::Reference<css::io::XInputStream> xStream(new utl::OInputStreamWrapper(rStream));

    rtl::Reference<svgio::svgreader::XSvgParser> xSvgParser(new svgio::svgreader::XSvgParser(comphelper::getProcessComponentContext()));

    return xSvgParser->getDecomposition(xStream, OUString()).getLength() != 0;

}

/* vim:set shiftwidth=4 softtabstop=4 expandtab: */