com.asual.summer.core.resource.MessageResource.java Source code

Introduction

Here is the source code for com.asual.summer.core.resource.MessageResource.java

Source

/*
 * Licensed 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
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.asual.summer.core.resource;

import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Locale;

import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.context.support.ReloadableResourceBundleMessageSource;
import org.springframework.core.Ordered;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.util.ResourceUtils;

/**
 * 
 * @author Rostislav Hristov
 * @author Rostislav Georgiev
 *
 */
public class MessageResource extends LocationResource {

    private ReloadableResourceBundleMessageSource rbms;
    private final Log logger = LogFactory.getLog(getClass());

    public MessageResource() {
        setOrder(Ordered.HIGHEST_PRECEDENCE);
        rbms = new ReloadableResourceBundleMessageSource();
    }

    public String getMessage(String code, Object args[], Locale locale) {
        return rbms.getMessage(code, args, locale);
    }

    public void setLocations(String[] locations) {
        super.setLocations(locations);
        rbms.clearCache();
        rbms.setBasenames(locations);
    }

    public void setWildcardLocations(String[] locations) {

        super.setLocations(locations);

        PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
        List<Resource[]> resourceLocations = new ArrayList<Resource[]>();

        List<String> fileBasenames = new ArrayList<String>();
        List<String> jarBasenames = new ArrayList<String>();

        for (String location : locations) {
            try {
                Resource[] wildcard = (Resource[]) ArrayUtils.addAll(
                        resolver.getResources(location + "*.properties"),
                        resolver.getResources(location + "*.xml"));
                if (wildcard != null && wildcard.length > 0) {
                    resourceLocations.add(wildcard);
                }
            } catch (IOException e) {
                logger.error(e.getMessage(), e);
            }
        }

        int i = 0;
        boolean entries = true;

        while (entries) {
            entries = false;
            for (Resource[] location : resourceLocations) {
                if (location.length > i) {
                    try {
                        URL url = location[i].getURL();
                        boolean isJar = ResourceUtils.isJarURL(url);
                        String basename = "classpath:" + url.getFile()
                                .replaceFirst(isJar ? "^.*" + ResourceUtils.JAR_URL_SEPARATOR
                                        : "^(.*/test-classes|.*/classes|.*/resources)/", "")
                                .replaceAll("(_\\w+){0,3}\\.(properties|xml)", "");
                        if (isJar) {
                            if (!jarBasenames.contains(basename)) {
                                jarBasenames.add(basename);
                            }
                        } else {
                            if (!fileBasenames.contains(basename)) {
                                fileBasenames.add(basename);
                            }
                        }
                    } catch (IOException e) {
                        logger.error(e.getMessage(), e);
                    }
                    entries = true;
                }
            }
            i++;
        }

        fileBasenames.addAll(jarBasenames);

        rbms.clearCache();
        rbms.setBasenames(fileBasenames.toArray(new String[fileBasenames.size()]));
    }

}