com.cnd.greencube.web.base.template.method.AbbreviateMethod.java Source code

Introduction

Here is the source code for com.cnd.greencube.web.base.template.method.AbbreviateMethod.java

Source

/*
 * Copyright 2005-2020 Top Team All rights reserved.
 * Support: 
 * License: top team license
 */
package com.cnd.greencube.web.base.template.method;

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

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

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

/**
 * ? - 
 * 
 * @author Top Team 
 * @version 1.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;
            }
        }
    }

}