org.exoplatform.commons.notification.template.TemplateResourceBundle.java Source code

Introduction

Here is the source code for org.exoplatform.commons.notification.template.TemplateResourceBundle.java

Source

/*
 * Copyright (C) 2003-2013 eXo Platform SAS.
 *
 * This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Affero General Public License
* as published by the Free Software Foundation; either version 3
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, see<http://www.gnu.org/licenses/>.
 */
package org.exoplatform.commons.notification.template;

import java.io.IOException;
import java.io.InputStream;
import java.util.Locale;
import java.util.ResourceBundle;
import java.util.Scanner;

import org.apache.commons.lang.StringUtils;
import org.exoplatform.commons.notification.NotificationUtils;
import org.exoplatform.commons.utils.CommonsUtils;
import org.exoplatform.container.configuration.ConfigurationManager;
import org.exoplatform.services.log.ExoLogger;
import org.exoplatform.services.log.Log;
import org.exoplatform.services.resources.ResourceBundleData;
import org.exoplatform.services.resources.ResourceBundleService;

public class TemplateResourceBundle {
    private static final Log LOG = ExoLogger.getLogger(TemplateResourceBundle.class);

    private static final String CONF_LOCATION = "war:/classes/";

    private String language;

    private String bundlePath;

    public TemplateResourceBundle(String language, String bundlePath) {
        this.language = language;
        this.bundlePath = bundlePath;
    }

    /**
     * @return the language
     */
    public String getLanguage() {
        return language;
    }

    /**
     * @param language the language to set
     */
    public void setLanguage(String language) {
        this.language = language;
    }

    /**
     * @return the bundlePath
     */
    public String getBundlePath() {
        return bundlePath;
    }

    /**
     * @param bundlePath the bundlePath to set
     */
    public void setBundlePath(String bundlePath) {
        this.bundlePath = bundlePath;
    }

    public String appRes(String key) {
        return getResourceBundle(key, NotificationUtils.getLocale(language), bundlePath);
    }

    public String appRes(String key, String... strs) {
        String value = appRes(key);
        if (strs != null && strs.length > 0) {
            for (int i = 0; i < strs.length; ++i) {
                value = StringUtils.replace(value, "{" + i + "}", strs[i]);
            }
        }
        return value;
    }

    private static ResourceBundle addResourceBundle(ResourceBundleService bundleService, String resourceLocale,
            Locale locale) {
        String id = new StringBuffer(CONF_LOCATION).append(resourceLocale.replace(".", "/")).append("_")
                .append(locale.getLanguage()).append(".properties").toString();
        try {
            ConfigurationManager configurationManager = CommonsUtils.getService(ConfigurationManager.class);
            InputStream inputStream = configurationManager.getInputStream(id);
            if (inputStream != null) {
                String data = getContent(inputStream);
                ResourceBundleData bundleData = new ResourceBundleData(data);
                bundleData.setLanguage(locale.getLanguage());
                bundleData.setName(resourceLocale);
                bundleData.setCountry("");
                bundleData.setVariant("");
                //
                bundleService.saveResourceBundle(bundleData);
                return bundleService.getResourceBundle(resourceLocale, locale);
            }
        } catch (Exception e) {
            LOG.warn("Can not add resource bundle of locale " + resourceLocale + "\n" + e.getMessage());
        }
        return null;
    }

    public static String getResourceBundle(String key, Locale locale, String resourceLocale) {
        if (key == null || key.trim().length() == 0) {
            return "";
        }

        if (locale == null || locale.getLanguage().isEmpty()) {
            locale = Locale.ENGLISH;
        }

        ResourceBundle res = null;
        ResourceBundleService bundleService = CommonsUtils.getService(ResourceBundleService.class);
        if (bundleService != null) {
            res = bundleService.getResourceBundle(resourceLocale, locale);
            // if null, try another way
            if (res == null) {
                //
                res = addResourceBundle(bundleService, resourceLocale, locale);
            }
        }
        // still null
        if (res == null || res.containsKey(key) == false) {
            if (key.indexOf(".digest.") < 0) {
                LOG.warn("Can not get resource bundle by key: " + key);
            }
            return key;
        }

        return res.getString(key);
    }

    static private String getContent(InputStream input) throws IOException {
        StringBuilder content = new StringBuilder();
        Scanner scanner = new Scanner(input, "UTF-8");
        try {
            while (scanner.hasNextLine()) {
                if (content.length() > 0) {
                    content.append("\n");
                }
                String s = scanner.nextLine();
                content.append(s);
            }
        } finally {
            scanner.close();
            input.close();
        }
        return content.toString();
    }
}