Support SVG in FO
[gpl/argeo-suite.git] / org.argeo.app.servlet.publish / src / org / argeo / app / servlet / publish / FopServlet.java
index bba0d3052238401879ed65df62c61370c8ae3064..ac4a1b78ce68a071570939e0f9fda6e4fa6b509e 100644 (file)
@@ -3,6 +3,8 @@ package org.argeo.app.servlet.publish;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
+import java.io.StringReader;
+import java.io.StringWriter;
 import java.net.URI;
 import java.net.URL;
 import java.net.URLDecoder;
@@ -10,6 +12,7 @@ import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Paths;
 import java.util.Map;
+import java.util.Objects;
 
 import javax.servlet.ServletException;
 import javax.servlet.http.HttpServlet;
@@ -38,9 +41,14 @@ import org.apache.xmlgraphics.io.ResourceResolver;
 import org.argeo.api.acr.Content;
 import org.argeo.api.acr.ContentRepository;
 import org.argeo.api.acr.ContentSession;
+import org.argeo.app.geo.GeoUtils;
+import org.argeo.app.geo.GpxUtils;
 import org.argeo.cms.auth.RemoteAuthUtils;
 import org.argeo.cms.servlet.ServletHttpRequest;
 import org.argeo.util.LangUtils;
+import org.geotools.data.collection.ListFeatureCollection;
+import org.geotools.data.simple.SimpleFeatureCollection;
+import org.opengis.feature.simple.SimpleFeature;
 
 /**
  * A servlet transforming an XML view of the data to either FOP or PDF.
@@ -48,13 +56,15 @@ import org.argeo.util.LangUtils;
 public class FopServlet extends HttpServlet {
        private static final long serialVersionUID = 6906020513498289335L;
 
+       private final static String PROP_ARGEO_FO_XSL = "argeo.fo.xsl";
+
        private ContentRepository contentRepository;
 
        private DocumentBuilderFactory documentBuilderFactory;
        private TransformerFactory transformerFactory;
        private Templates foTemplates;
 
-       private String xslUrl;
+       private URL xslUrl;
 
        @Override
        protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
@@ -68,29 +78,55 @@ public class FopServlet extends HttpServlet {
                ContentSession session = RemoteAuthUtils.doAs(() -> contentRepository.get(), new ServletHttpRequest(req));
                Content content = session.get(path);
                Source xmlInput = content.adapt(Source.class);
-               xmlInput.setSystemId(req.getRequestURI());
+               String systemId = req.getRequestURL().toString();
+               xmlInput.setSystemId(systemId);
 
                // dev only
-               try (InputStream in = new URL(xslUrl).openStream()) {
-                       Source xslSource = new StreamSource(in);
-                       foTemplates = transformerFactory.newTemplates(xslSource);
-                       if (foTemplates == null)
-                               throw new IllegalStateException("Could not instantiate XSL " + xslUrl);
-               } catch (TransformerConfigurationException | IOException e) {
-                       throw new IllegalStateException("Cannot instantiate XSL " + xslUrl, e);
+               final boolean DEV = true;
+               if (DEV) {
+                       try (InputStream in = xslUrl.openStream()) {
+                               Source xslSource = new StreamSource(in);
+                               foTemplates = transformerFactory.newTemplates(xslSource);
+                               if (foTemplates == null)
+                                       throw new IllegalStateException("Could not instantiate XSL " + xslUrl);
+                       } catch (TransformerConfigurationException | IOException e) {
+                               throw new IllegalStateException("Cannot instantiate XSL " + xslUrl, e);
+                       }
                }
 
                URIResolver uriResolver = (href, base) -> {
                        try {
-                               URL url = new URL(href);
-                               if (url.getProtocol().equals("file")) {
-                                       InputStream in = Files.newInputStream(Paths.get(URI.create(url.toString())));
-                                       return new StreamSource(in);
+                               URI url = URI.create(href);
+                               if (url.getScheme() != null) {
+                                       if (url.getScheme().equals("file")) {
+                                               InputStream in = Files.newInputStream(Paths.get(URI.create(url.toString())));
+                                               return new StreamSource(in);
+                                       }
+                                       if (url.getScheme().equals("geo2svg")) {
+                                               String includePath = path + url.getPath();
+                                               String geoExt = includePath.substring(includePath.lastIndexOf('.'));
+                                               Content geoContent = session.get(includePath);
+                                               if (".gpx".equals(geoExt)) {
+                                                       try (InputStream in = geoContent.open(InputStream.class)) {
+                                                               SimpleFeature field = GpxUtils.parseGpxToPolygon(in);
+                                                               SimpleFeatureCollection features = new ListFeatureCollection(field.getType(), field);
+                                                               try (StringWriter writer = new StringWriter()) {
+                                                                       GeoUtils.exportToSvg(features, writer, 100, 100);
+                                                                       StreamSource res = new StreamSource(new StringReader(writer.toString()));
+                                                                       return res;
+                                                               }
+                                                       } 
+                                               } else {
+                                                       throw new UnsupportedOperationException(geoExt + " is not supported");
+                                               }
+                                       }
                                }
                        } catch (IOException e) {
-                               // silent
+                               throw new RuntimeException("Cannot process " + href);
                        }
-                       Content subContent = session.get(href);
+                       
+                       String p = href.startsWith("/")?href:path+'/'+href;
+                       Content subContent = session.get(p);
                        return subContent.adapt(Source.class);
                };
 
@@ -137,6 +173,14 @@ public class FopServlet extends HttpServlet {
 
        }
 
+       @Override
+       public void init() throws ServletException {
+//             for (Enumeration<String> it = getServletConfig().getInitParameterNames(); it.hasMoreElements();)
+//                     System.out.println(it.nextElement());
+//             for (String str : getServletContext().getResourcePaths("/"))
+//                     System.out.println(str);
+       }
+
        public void start(Map<String, Object> properties) {
                documentBuilderFactory = DocumentBuilderFactory.newInstance();
                documentBuilderFactory.setXIncludeAware(true);
@@ -146,13 +190,17 @@ public class FopServlet extends HttpServlet {
                // with DocBook stylesheets
                transformerFactory = new TransformerFactoryImpl();
 //             transformerFactory = TransformerFactory.newDefaultInstance();
+               try {
+                       String xslStr = LangUtils.get(properties, PROP_ARGEO_FO_XSL);
+                       Objects.requireNonNull(xslStr);
+                       xslUrl = new URL(xslStr);
+                       try (InputStream in = xslUrl.openStream()) {
+                               Source xslSource = new StreamSource(in);
+                               foTemplates = transformerFactory.newTemplates(xslSource);
+                               if (foTemplates == null)
+                                       throw new IllegalStateException("Could not instantiate XSL " + xslUrl);
+                       }
 
-               xslUrl = LangUtils.get(properties, "argeo.fo.xsl");
-               try (InputStream in = new URL(xslUrl).openStream()) {
-                       Source xslSource = new StreamSource(in);
-                       foTemplates = transformerFactory.newTemplates(xslSource);
-                       if (foTemplates == null)
-                               throw new IllegalStateException("Could not instantiate XSL " + xslUrl);
                } catch (TransformerConfigurationException | IOException e) {
                        throw new IllegalStateException("Cannot instantiate XSL " + xslUrl, e);
                }