com.wedian.site.common.template.method.AbbreviateMethod.java Source code

Introduction

Here is the source code for com.wedian.site.common.template.method.AbbreviateMethod.java

Source

/*
 * Copyright 2005-2013 shopxx.net. All rights reserved.
 * Support: http://www.shopxx.net
 * License: http://www.shopxx.net/license
 */
package com.wedian.site.common.template.method;

import java.util.List;
import java.util.regex.Pattern;

import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Component;

import freemarker.template.SimpleScalar;
import freemarker.template.TemplateMethodModel;
import freemarker.template.TemplateModelException;

/**
 * ? - 
 * 
 * @author SHOP++ Team
 * @version 3.0
 */
@Component("abbreviateMethod")
public class AbbreviateMethod implements TemplateMethodModel {

    /** ? */
    private static final Pattern PATTERN = Pattern.compile("[\\u4e00-\\u9fa5\\ufe30-\\uffa0]+$");

    @SuppressWarnings("rawtypes")
    public Object exec(List arguments) throws TemplateModelException {
        if (arguments != null && !arguments.isEmpty() && arguments.get(0) != null
                && StringUtils.isNotEmpty(arguments.get(0).toString())) {
            Integer width = null;
            String ellipsis = null;
            if (arguments.size() == 2) {
                if (arguments.get(1) != null) {
                    width = Integer.valueOf(arguments.get(1).toString());
                }
            } else if (arguments.size() > 2) {
                if (arguments.get(1) != null) {
                    width = Integer.valueOf(arguments.get(1).toString());
                }
                if (arguments.get(2) != null) {
                    ellipsis = arguments.get(2).toString();
                }
            }
            return new SimpleScalar(abbreviate(arguments.get(0).toString(), width, ellipsis));
        }
        return null;
    }

    /**
     * 
     * 
     * 
     * @param str
     *            
     * @param width
     *            
     * @param ellipsis
     *            ?
     * @return 
     */
    private String abbreviate(String str, Integer width, String ellipsis) {
        if (width != null) {
            int strLength = 0;
            for (int strWidth = 0; strLength < str.length(); strLength++) {
                strWidth = PATTERN.matcher(String.valueOf(str.charAt(strLength))).find() ? strWidth + 2
                        : strWidth + 1;
                if (strWidth >= width) {
                    break;
                }
            }
            if (strLength < str.length()) {
                if (ellipsis != null) {
                    return str.substring(0, strLength + 1) + ellipsis;
                } else {
                    return str.substring(0, strLength + 1);
                }
            } else {
                return str;
            }
        } else {
            if (ellipsis != null) {
                return str + ellipsis;
            } else {
                return str;
            }
        }
    }

}