001    /**
002     * Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portlet.journal.util;
016    
017    import com.liferay.portal.kernel.log.Log;
018    import com.liferay.portal.kernel.log.LogFactoryUtil;
019    import com.liferay.portal.kernel.templateparser.BaseTransformerListener;
020    import com.liferay.portal.kernel.util.LocaleUtil;
021    import com.liferay.portal.kernel.util.StringUtil;
022    import com.liferay.portal.kernel.xml.Document;
023    import com.liferay.portal.kernel.xml.Element;
024    import com.liferay.portal.kernel.xml.SAXReaderUtil;
025    import com.liferay.portlet.dynamicdatamapping.util.DDMXMLUtil;
026    
027    import java.util.List;
028    import java.util.Map;
029    
030    /**
031     * @author Raymond Aug??
032     */
033    public class LocaleTransformerListener extends BaseTransformerListener {
034    
035            @Override
036            public String onScript(
037                    String script, String xml, String languageId,
038                    Map<String, String> tokens) {
039    
040                    if (_log.isDebugEnabled()) {
041                            _log.debug("onScript");
042                    }
043    
044                    return StringUtil.replace(script, "@language_id@", languageId);
045            }
046    
047            @Override
048            public String onXml(
049                    String xml, String languageId, Map<String, String> tokens) {
050    
051                    if (_log.isDebugEnabled()) {
052                            _log.debug("onXml");
053                    }
054    
055                    return filterByLanguage(xml, languageId);
056            }
057    
058            protected void filterByLanguage(
059                    Element root, String languageId, String defaultLanguageId) {
060    
061                    Element defaultLanguageElement = null;
062    
063                    boolean hasLanguageIdElement = false;
064    
065                    List<Element> elements = root.elements();
066    
067                    int listIndex = elements.size() - 1;
068    
069                    while (listIndex >= 0) {
070                            Element element = elements.get(listIndex);
071    
072                            String tempLanguageId = element.attributeValue(
073                                    "language-id", languageId);
074    
075                            if (StringUtil.equalsIgnoreCase(tempLanguageId, languageId)) {
076                                    hasLanguageIdElement = true;
077    
078                                    filterByLanguage(element, languageId, defaultLanguageId);
079                            }
080                            else {
081                                    if (StringUtil.equalsIgnoreCase(
082                                                    tempLanguageId, defaultLanguageId)) {
083    
084                                            defaultLanguageElement = element;
085                                    }
086    
087                                    root.remove(element);
088                            }
089    
090                            listIndex--;
091                    }
092    
093                    if (!hasLanguageIdElement && (defaultLanguageElement != null)) {
094                            root.add(defaultLanguageElement);
095    
096                            filterByLanguage(
097                                    defaultLanguageElement, languageId, defaultLanguageId);
098                    }
099            }
100    
101            protected String filterByLanguage(String xml, String languageId) {
102                    if (xml == null) {
103                            return xml;
104                    }
105    
106                    try {
107                            Document document = SAXReaderUtil.read(xml);
108    
109                            Element rootElement = document.getRootElement();
110    
111                            String defaultLanguageId = LocaleUtil.toLanguageId(
112                                    LocaleUtil.getSiteDefault());
113    
114                            String[] availableLocales = StringUtil.split(
115                                    rootElement.attributeValue(
116                                            "available-locales", defaultLanguageId));
117    
118                            String defaultLocale = rootElement.attributeValue(
119                                    "default-locale", defaultLanguageId);
120    
121                            boolean supportedLocale = false;
122    
123                            for (String availableLocale : availableLocales) {
124                                    if (StringUtil.equalsIgnoreCase(availableLocale, languageId)) {
125                                            supportedLocale = true;
126    
127                                            break;
128                                    }
129                            }
130    
131                            if (!supportedLocale) {
132                                    filterByLanguage(rootElement, defaultLocale, defaultLanguageId);
133                            }
134                            else {
135                                    filterByLanguage(rootElement, languageId, defaultLanguageId);
136                            }
137    
138                            xml = DDMXMLUtil.formatXML(document);
139                    }
140                    catch (Exception e) {
141                            _log.error(e);
142                    }
143    
144                    return xml;
145            }
146    
147            private static Log _log = LogFactoryUtil.getLog(
148                    LocaleTransformerListener.class);
149    
150    }