1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34 package info.magnolia.module.googlesitemap.service;
35
36 import info.magnolia.jcr.util.PropertyUtil;
37 import info.magnolia.module.googlesitemap.GoogleSiteMapConfiguration;
38 import info.magnolia.module.googlesitemap.bean.SiteMapEntry;
39 import info.magnolia.module.googlesitemap.bean.SiteMapEntryList;
40 import info.magnolia.module.googlesitemap.config.SiteMapType;
41
42 import java.io.StringWriter;
43 import java.util.ArrayList;
44 import java.util.Iterator;
45 import java.util.List;
46
47 import javax.inject.Provider;
48 import javax.inject.Singleton;
49 import javax.jcr.Node;
50 import javax.jcr.RepositoryException;
51 import javax.xml.bind.JAXBContext;
52 import javax.xml.bind.JAXBException;
53 import javax.xml.bind.Marshaller;
54 import javax.xml.bind.PropertyException;
55 import javax.xml.stream.XMLOutputFactory;
56 import javax.xml.stream.XMLStreamException;
57 import javax.xml.stream.XMLStreamWriter;
58
59 import org.slf4j.Logger;
60 import org.slf4j.LoggerFactory;
61
62 import com.google.inject.Inject;
63
64
65
66
67 @Singleton
68 public class SiteMapXMLUtilImpl implements SiteMapXMLUtil {
69
70 public static final String MOBILE_SCHEMA_URL = "http://www.google.com/schemas/sitemap-mobile/1.0";
71
72 public static final String MOBILE_ELEMENT_NAME = "mobile";
73
74 public static final String MOBILE_PREFIX = "mobile";
75
76 public static final String URL_ELEMENT_NAME = "url";
77
78 private Logger log = LoggerFactory.getLogger(getClass());
79
80 private Provider<SiteMapService> service;
81
82 @Inject
83 public SiteMapXMLUtilImpl(Provider<SiteMapService> service) {
84 this.service = service;
85 }
86
87 @Override
88 public String generateSiteMapXML(Node siteMapNode) throws RepositoryException, JAXBException {
89 List<SiteMapEntry> entries = new ArrayList<SiteMapEntry>();
90 Iterator<SiteMapEntry> entryIt = service.get().getSiteMapBeans(siteMapNode);
91 while (entryIt.hasNext()) {
92 entries.add(entryIt.next());
93 }
94 SiteMapType type = SiteMapType.valueOf(PropertyUtil.getString(siteMapNode, GoogleSiteMapConfiguration.TYPE_PROPERTY, SiteMapType.Standard.name()));
95 return marshalSiteMapEntries(entries, type);
96 }
97
98 private String marshalSiteMapEntries(List<SiteMapEntry> entries, final SiteMapType type) throws JAXBException {
99 JAXBContext jaxbContext = JAXBContext.newInstance(SiteMapEntry.class, SiteMapEntryList.class);
100 StringWriter sw = new StringWriter();
101 Marshaller marshaller = jaxbContext.createMarshaller();
102 marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
103 marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
104
105 try {
106 SiteMapEntryList list = new SiteMapEntryList(entries);
107 final XMLStreamWriter writer = XMLOutputFactory.newInstance().createXMLStreamWriter(sw);
108 final XMLStreamWriter delWriter = new DelegatingXMLStreamWriter(writer) {
109 @Override
110 public void writeStartElement(String prefix, String namespace, String localName) throws XMLStreamException {
111 super.writeStartElement(prefix, namespace, localName);
112 if (URL_ELEMENT_NAME.equals(namespace) && SiteMapType.Mobile == type) {
113 writeEmptyElement(MOBILE_PREFIX, MOBILE_ELEMENT_NAME, "");
114 }
115 }
116
117 @Override
118 public void writeNamespace(String prefix, String localName) throws XMLStreamException {
119 super.writeNamespace(prefix, localName);
120 if (type == SiteMapType.Mobile) {
121 super.writeNamespace(MOBILE_PREFIX, MOBILE_SCHEMA_URL);
122 }
123 }
124 };
125 marshaller.marshal(list, delWriter);
126 } catch (PropertyException e) {
127 log.error("Failed to rebind marshaller's namespace prefix mapper, did the implementation of JAXB change?", e);
128 } catch (XMLStreamException e) {
129 log.error("Failed to marshal sitemap", e);
130 }
131 return sw.toString();
132 }
133
134 }