com.ewcms.publication.uri.UriRule.java Source code

Introduction

Here is the source code for com.ewcms.publication.uri.UriRule.java

Source

/**
 * Copyright (c)2010-2011 Enterprise Website Content Management System(EWCMS), All rights reserved.
 * EWCMS PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.
 * http://www.ewcms.com
 */

package com.ewcms.publication.uri;

import java.text.DateFormat;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.beanutils.PropertyUtils;
import org.apache.commons.lang.StringUtils;
import org.apache.commons.io.FilenameUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.Assert;

import com.ewcms.publication.PublishException;
import com.ewcms.publication.freemarker.GlobalVariable;

/**
 * ??
 * 
 * @author wangwei
 */
public class UriRule implements UriRuleable {

    private static final Logger logger = LoggerFactory.getLogger(UriRule.class);

    private static final DateFormat DEFAULT_DATA_FORMAT = new SimpleDateFormat("yyyy-MM-dd");
    private static final DecimalFormat DEFAULT_NUMBER_FORMAT = new DecimalFormat("#");
    private static final Map<String, String> ALIAS_PARAMETERS = initAliasParameters();

    private final RuleParseable ruleParse;
    private Map<String, Object> parameters = initParameters();

    public UriRule(RuleParseable ruleParse) {
        this.ruleParse = ruleParse;
    }

    @Override
    public void setParameters(Map<String, Object> ps) {
        parameters.putAll(ps);
    }

    @Override
    public void putParameter(String parameter, Object value) {
        parameters.put(parameter, value);
    }

    @Override
    public String getPatter() {
        return ruleParse.getPatter();
    }

    @Override
    public String getUri() throws PublishException {
        Assert.notNull(parameters);

        Map<String, String> variables = ruleParse.getVariables();
        if (variables.isEmpty()) {
            return ruleParse.getPatter();
        }

        String uri = ruleParse.getPatter();
        for (String name : variables.keySet()) {
            String format = variables.get(name);
            String exp = (format == null ? String.format("${%s}", name) : String.format("${%s?%s}", name, format));
            Object value = getVariableValue(name, parameters);
            String formatValue = formatValue(value, format);
            uri = StringUtils.replace(uri, exp, formatValue);
        }
        uri = uri.replace("\\", "/").replace("//", "/");
        return FilenameUtils.normalize(uri);
    }

    /**
     * ??
     */
    private Map<String, Object> initParameters() {
        Map<String, Object> map = new HashMap<String, Object>();
        map.put("now", new Date());

        return map;
    }

    /**
     * ??
     * 
     * @param variable ????
     * @param parameters ??
     * @return ??
     * @throws PublishException
     */
    Object getVariableValue(String variable, Map<String, Object> parameters) throws PublishException {
        logger.debug("Variable is {}", variable);
        String p = StringUtils.splitPreserveAllTokens(variable, ".")[0];
        logger.debug("Parameter name is {}", p);

        String parameter = ALIAS_PARAMETERS.get(p);
        if (parameter == null) {
            logger.warn("\"{}\" parameter is not exist", p);
            parameter = p;
        }
        Object object = parameters.get(parameter);
        if (object == null) {
            logger.error("\"{}\" parameter is not exist", parameter);
            throw new PublishException(variable + " is not exist");
        }
        try {
            if (!p.equals(variable)) {
                String property = StringUtils.removeStart(variable, p + ".");
                logger.debug("Property name is {}", property);
                return PropertyUtils.getProperty(object, property);
            } else {
                return object;
            }
        } catch (Exception e) {
            logger.error("Get variable value is error:{}", e.toString());
            throw new PublishException(e);
        }
    }

    /**
     * ?
     * 
     * @param value 
     * @param patter ??
     * @return ??
     */
    String formatValue(Object value, String patter) {

        logger.debug("Format type is {}", value.getClass().getName());

        if (value instanceof Date) {
            if (patter == null || patter.length() == 0) {
                return DEFAULT_DATA_FORMAT.format(value);
            } else {
                DateFormat dateFormat = new SimpleDateFormat(patter);
                return dateFormat.format(value);
            }
        } else if (value instanceof Number) {
            if (patter == null || patter.length() == 0) {
                return DEFAULT_NUMBER_FORMAT.format(value);
            } else {
                DecimalFormat numberFormat = new DecimalFormat(patter);
                return numberFormat.format(value);
            }
        } else {
            return value.toString();
        }
    }

    private static Map<String, String> initAliasParameters() {
        Map<String, String> map = new HashMap<String, String>();

        map.put("a", GlobalVariable.ARTICLE.toString());
        map.put("c", GlobalVariable.CHANNEL.toString());
        map.put("s", GlobalVariable.SITE.toString());
        map.put("p", GlobalVariable.PAGE_NUMBER.toString());

        map.put("article", GlobalVariable.ARTICLE.toString());
        map.put("channel", GlobalVariable.CHANNEL.toString());
        map.put("site", GlobalVariable.SITE.toString());
        map.put("page", GlobalVariable.PAGE_NUMBER.toString());

        map.put("", GlobalVariable.ARTICLE.toString());
        map.put("?", GlobalVariable.CHANNEL.toString());
        map.put("", GlobalVariable.SITE.toString());
        map.put("", GlobalVariable.PAGE_NUMBER.toString());

        map.put(GlobalVariable.ARTICLE.toString(), GlobalVariable.ARTICLE.toString());
        map.put(GlobalVariable.CHANNEL.toString(), GlobalVariable.CHANNEL.toString());
        map.put(GlobalVariable.SITE.toString(), GlobalVariable.SITE.toString());
        map.put(GlobalVariable.PAGE_NUMBER.toString(), GlobalVariable.PAGE_NUMBER.toString());

        map.put("now", "now");
        map.put("today", "now");

        return map;
    }
}