org.bigmouth.nvwa.utils.url.URLDecoder.java Source code

Introduction

Here is the source code for org.bigmouth.nvwa.utils.url.URLDecoder.java

Source

/*
 * Copyright 2015 big-mouth.cn
 *
 * The Project licenses this file to you 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 org.bigmouth.nvwa.utils.url;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.reflect.MethodUtils;
import org.bigmouth.nvwa.utils.Argument;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

/**
 * 
 * @since 1.0
 * @author Allen 
 */
public final class URLDecoder {

    private static final Logger LOGGER = LoggerFactory.getLogger(URLDecoder.class);

    /**
     * URL???
     * 
     * @param <T>
     * @param url
     * ?URL?
     * <ul>
     * <li>http://www.big-mouth.cn/nvwa-utils/xml/decoder?id=123&json_data=bbb</li>
     * <li>/nvwa-utils/xml/decoder?id=123&json_data=bbb</li>
     * <li>id=123&json_data=bbb</li>
     * </li>
     * </ul>
     * @param cls
     * @return
     * @throws Exception 
     */
    public static <T> T decode(String url, Class<T> cls) throws Exception {
        if (StringUtils.isBlank(url))
            return null;
        T t = cls.newInstance();

        String string = null;
        if (StringUtils.contains(url, "?")) {
            string = StringUtils.substringAfter(url, "?");
        } else {
            int start = StringUtils.indexOf(url, "?") + 1;
            string = StringUtils.substring(url, start);
        }

        Map<String, Object> attrs = Maps.newHashMap();
        String[] kvs = StringUtils.split(string, "&");
        for (String kv : kvs) {
            String[] entry = StringUtils.split(kv, "=");
            if (entry.length <= 1) {
                continue;
            } else if (entry.length == 2) {
                attrs.put(entry[0], entry[1]);
            } else {
                List<String> s = Lists.newArrayList(entry);
                s.remove(0);
                attrs.put(entry[0], StringUtils.join(s.toArray(new String[0]), "="));
            }
        }

        Field[] fields = cls.getDeclaredFields();
        for (Field field : fields) {
            // if the field name is 'appId'
            String fieldName = field.getName();
            String paramName = fieldName;
            if (field.isAnnotationPresent(Argument.class)) {
                paramName = field.getAnnotation(Argument.class).name();
            }
            // select appId node
            Object current = attrs.get(paramName);
            if (null == current) {
                // select appid node
                current = attrs.get(paramName.toLowerCase());
            }
            if (null == current) {
                // select APPID node
                current = attrs.get(paramName.toUpperCase());
            }
            if (null == current) {
                // select app_id node
                String nodename = StringUtils.join(StringUtils.splitByCharacterTypeCamelCase(paramName), "_")
                        .toLowerCase();
                current = attrs.get(nodename);
            }
            if (null == current) {
                // select APP_ID node
                String nodename = StringUtils.join(StringUtils.splitByCharacterTypeCamelCase(paramName), "_")
                        .toUpperCase();
                current = attrs.get(nodename);
            }
            if (null != current) {
                String invokeName = StringUtils.join(new String[] { "set", StringUtils.capitalize(fieldName) });
                try {
                    MethodUtils.invokeMethod(t, invokeName, current);
                } catch (NoSuchMethodException e) {
                    LOGGER.warn("NoSuchMethod-" + invokeName);
                } catch (IllegalAccessException e) {
                    LOGGER.warn("IllegalAccess-" + invokeName);
                } catch (InvocationTargetException e) {
                    LOGGER.warn("InvocationTarget-" + invokeName);
                }
            }
        }
        return t;
    }
}