MentaContainer

Rev

Rev 4 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | RSS feed

package org.mentacontainer.util;

import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.text.DateFormat;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import java.util.Map;

public class InjectionUtils {

        /**
         * The character used to separate the prefix from the value name when you
         * are using the getObject method with a prefix. You can change the value of
         * this prefix if you want to by changing this static variable.
         *
         * Ex: getObject(User.class, "user") will get all values that begin with
         * "user.".
         */

        public static char PREFIX_SEPARATOR = '.';

        private static Map<Class<? extends Object>, Map<String, Object>> settersMaps = new HashMap<Class<? extends Object>, Map<String, Object>>();

        private static Map<Class<? extends Object>, Map<String, Object>> fieldsMaps = new HashMap<Class<? extends Object>, Map<String, Object>>();

        public static void prepareForInjection(Class<? extends Object> klass,
                        Map<String, Object> setters, Map<String, Object> fields) {

                StringBuffer sb = new StringBuffer(32);

                Method[] methods = klass.getMethods();

                for (int i = 0; i < methods.length; i++) {

                        Method m = methods[i];

                        String name = m.getName();

                        Class[] types = m.getParameterTypes();

                        if (name.startsWith("set") && name.length() > 3
                                        && types.length == 1) {

                                String var = name.substring(3);

                                if (var.length() > 1) {

                                        sb.delete(0, sb.length());

                                        sb.append(var.substring(0, 1).toLowerCase());

                                        sb.append(var.substring(1));

                                        var = sb.toString();

                                } else {

                                        var = var.toLowerCase();
                                }

                                m.setAccessible(true);

                                if (setters.containsKey(var)) {

                                        Object obj = setters.get(var);

                                        if (obj instanceof List) {

                                                List<Method> list = (List<Method>) obj;

                                                list.add(m);

                                        } else if (obj instanceof Method) {

                                                List<Method> list = new ArrayList<Method>();

                                                list.add((Method) obj);

                                                list.add(m);

                                                setters.put(var, list);

                                        }

                                } else {

                                        setters.put(var, m);

                                }
                        }
                }

                if (fields == null)
                        return;

                Field[] f = klass.getDeclaredFields();

                for (int i = 0; i < f.length; i++) {

                        Field field = f[i];

                        field.setAccessible(true);

                        String name = field.getName();

                        if (setters.containsKey(name)) {

                                Object obj = setters.get(name);

                                if (obj instanceof Method) {

                                        Method m = (Method) obj;

                                        Class[] types = m.getParameterTypes();

                                        Class type = field.getType();

                                        if (type.isAssignableFrom(types[0]))
                                                continue; // don't choose a field when we already have
                                                                        // a method...

                                } else if (obj instanceof List) {

                                        List<Method> list = (List<Method>) obj;

                                        Iterator<Method> iter = list.iterator();

                                        boolean found = false;

                                        while (iter.hasNext()) {

                                                Method m = iter.next();

                                                Class[] types = m.getParameterTypes();

                                                Class type = field.getType();

                                                if (type.isAssignableFrom(types[0])) {

                                                        found = true;

                                                        break;
                                                }
                                        }

                                        if (found)
                                                continue; // don't choose a field when we already have
                                                                        // a method...

                                }
                        }

                        fields.put(name, field);

                }
        }

        public static boolean checkPrimitives(Class target,
                        Class<? extends Object> source) {

                if (target.equals(int.class) && source.equals(Integer.class))
                        return true;

                if (target.equals(boolean.class) && source.equals(Boolean.class))
                        return true;

                if (target.equals(byte.class) && source.equals(Byte.class))
                        return true;

                if (target.equals(short.class) && source.equals(Short.class))
                        return true;

                if (target.equals(char.class) && source.equals(Character.class))
                        return true;

                if (target.equals(long.class) && source.equals(Long.class))
                        return true;

                if (target.equals(float.class) && source.equals(Float.class))
                        return true;

                if (target.equals(double.class) && source.equals(Double.class))
                        return true;

                return false;

        }

        public static Object tryToConvert(Object source, Class targetType,
                        Locale loc) {

                return tryToConvert(source, targetType, loc, false);
        }

        public static Object tryToConvert(Object source, Class targetType,
                        Locale loc, boolean tryNumber) {

                String value = null;

                if (source instanceof String) {

                        value = (String) source;

                } else if (tryNumber && source instanceof Number) {

                        value = source.toString();

                } else {

                        return null;
                }

                Object newValue = null;

                String className = targetType.getName();

                if (className.equals("int") || className.equals("java.lang.Integer")) {
                        int x = -1;
                        try {
                                x = Integer.parseInt(value);
                        } catch (Exception e) {
                                return null;
                        }
                        newValue = new Integer(x);
                } else if (className.equals("short")
                                || className.equals("java.lang.Short")) {
                        short x = -1;
                        try {
                                x = Short.parseShort(value);
                        } catch (Exception e) {
                                return null;
                        }
                        newValue = new Short(x);

                } else if (className.equals("char")
                                || className.equals("java.lang.Character")) {

                        if (value.length() != 1) {
                                return null;
                        }

                        newValue = new Character(value.charAt(0));

                } else if (className.equals("long")
                                || className.equals("java.lang.Long")) {
                        long x = -1;
                        try {
                                x = Long.parseLong(value);
                        } catch (Exception e) {
                                return null;
                        }
                        newValue = new Long(x);
                } else if (className.equals("float")
                                || className.equals("java.lang.Float")) {
                        float x = -1;
                        try {
                                x = Float.parseFloat(value);
                        } catch (Exception e) {
                                return null;
                        }
                        newValue = new Float(x);
                } else if (className.equals("double")
                                || className.equals("java.lang.Double")) {
                        double x = -1;
                        try {
                                x = Double.parseDouble(value);
                        } catch (Exception e) {
                                return null;
                        }
                        newValue = new Double(x);
                } else if (className.equals("boolean")
                                || className.equals("java.lang.Boolean")) {
                        try {
                                int x = Integer.parseInt(value);
                                if (x == 1) {
                                        newValue = Boolean.TRUE;
                                } else if (x == 0) {
                                        newValue = Boolean.FALSE;
                                } else {
                                        return null;
                                }
                        } catch (Exception e) {
                                if (value.equalsIgnoreCase("true") || value.equals("on")) {
                                        newValue = Boolean.TRUE;
                                } else if (value.equalsIgnoreCase("false")) {
                                        newValue = Boolean.FALSE;
                                } else {
                                        return null;
                                }
                        }
                } else if (className.equals("java.util.Date") && loc != null) {
                        DateFormat df = DateFormat.getDateInstance(DateFormat.SHORT, loc); // change
                        // this.
                        df.setLenient(false);
                        try {
                                newValue = df.parse(value);
                        } catch (Exception e) {
                                return null;
                        }
                } else if (targetType.isEnum()) {

                        try {

                                newValue = Enum.valueOf(targetType, value);

                        } catch (Exception e) {

                                return null;
                        }

                }

                return newValue;

        }

        public static Object shouldConvertToNull(Object value,
                        Class<? extends Object> targetType) {

                if (targetType.equals(String.class)) {

                        return value;

                } else if (targetType.isPrimitive()) {

                        return value;
                }

                return null;
        }

        public static Class getPrimitiveFrom(Object w) {
                if (w instanceof Boolean) {
                        return Boolean.TYPE;
                } else if (w instanceof Byte) {
                        return Byte.TYPE;
                } else if (w instanceof Short) {
                        return Short.TYPE;
                } else if (w instanceof Character) {
                        return Character.TYPE;
                } else if (w instanceof Integer) {
                        return Integer.TYPE;
                } else if (w instanceof Long) {
                        return Long.TYPE;
                } else if (w instanceof Float) {
                        return Float.TYPE;
                } else if (w instanceof Double) {
                        return Double.TYPE;
                }
                return null;
        }

        public static Class getPrimitiveFrom(Class klass) {

                String s = klass.getName();

                if (s.equals("java.lang.Boolean")) {
                        return Boolean.TYPE;
                } else if (s.equals("java.lang.Byte")) {
                        return Byte.TYPE;
                } else if (s.equals("java.lang.Short")) {
                        return Short.TYPE;
                } else if (s.equals("java.lang.Character")) {
                        return Character.TYPE;
                } else if (s.equals("java.lang.Integer")) {
                        return Integer.TYPE;
                } else if (s.equals("java.lang.Long")) {
                        return Long.TYPE;
                } else if (s.equals("java.lang.Float")) {
                        return Float.TYPE;
                } else if (s.equals("java.lang.Double")) {
                        return Double.TYPE;
                }
                return null;
        }

        public static Field getField(Object target, String name) {
                return getField(target.getClass(), name);
        }

        public static Field getField(Class<? extends Object> target, String name) {
                Field fields[] = target.getDeclaredFields();
                for (int i = 0; i < fields.length; i++) {
                        if (name.equals(fields[i].getName())) {
                                return fields[i];
                        }
                }
                return null;
        }

        public static Method findMethodToGet(Class<? extends Object> target,
                        String name) {

                StringBuffer sb = new StringBuffer(128);

                sb.append("get").append(name.substring(0, 1).toUpperCase());

                if (name.length() > 1)
                        sb.append(name.substring(1));

                try {

                        return target.getMethod(sb.toString(), (Class[]) null);

                } catch (Exception e) {

                }

                sb.setLength(0);

                sb.append("is").append(name.substring(0, 1).toUpperCase());

                if (name.length() > 1) {

                        sb.append(name.substring(1));
                }

                try {

                        return target.getMethod(sb.toString(), (Class[]) null);

                } catch (Exception e) {

                }

                return null;
        }

        public static Method findMethodToInject(Class<? extends Object> target,
                        String name, Class source) {

                StringBuffer sb = new StringBuffer(128);

                sb.append("set").append(name.substring(0, 1).toUpperCase());

                if (name.length() > 1)
                        sb.append(name.substring(1));

                String methodName = sb.toString();

                Method m = null;

                try {

                        m = FindMethod
                                        .getMethod(target, methodName, new Class[] { source });

                } catch (Exception e) {
                }

                if (m == null) {

                        Class primitive = getPrimitiveFrom(source);

                        if (primitive != null) {

                                try {

                                        m = target.getMethod(methodName, new Class[] { primitive });

                                } catch (Exception e) {
                                }

                        }
                }

                if (m != null) {
                        m.setAccessible(true);
                }

                return m;

        }

        public static Field findFieldToInject(Class<? extends Object> target,
                        String name, Class<? extends Object> source) {

                Field f = getField(target, name);

                if (f != null) {

                        Class<?> type = f.getType();

                        if (type.isAssignableFrom(source) || checkPrimitives(type, source)) {

                                f.setAccessible(true);

                                return f;
                        }

                }

                return null;
        }

        private static final boolean isBlank(Object value) {

                if (value != null && value instanceof String) {

                        String s = ((String) value).trim();

                        if (s.length() == 0)
                                return true;
                }

                return false;
        }

        public static boolean inject(Method m, Object target, Object value,
                        Locale loc, boolean tryToConvert, boolean tryingToConvertBoolean)
                        throws Exception {

                Class<?> type = m.getParameterTypes()[0];

                if (tryingToConvertBoolean) {

                        if (value == null && (type.equals(Boolean.class) || type.equals(boolean.class))) {

                                value = Boolean.FALSE;

                        } else {

                                // if trying to convert boolean, convert or don't do anything...

                                return false;

                        }
                }

                if (value == null
                                || (type.isAssignableFrom(value.getClass())
                                                || checkPrimitives(type, value.getClass()) || (tryToConvert && ((isBlank(value) && (value = shouldConvertToNull(
                                                value, type)) == null) || (value = tryToConvert(value,
                                                type, loc)) != null)))) {

                        try {

                                m.invoke(target, new Object[] { value });

                                return true;

                        } catch (Exception e) {

                                System.err.println("Error injecting by method: " + value
                                                + " in " + target + " thru " + m);

                                e.printStackTrace();

                                throw e;

                        }
                }

                return false;

        }

        public static boolean hasDefaultConstructor(Class<? extends Object> klass) {

                try {

                        return klass.getConstructor((Class[]) null) != null;

                } catch (Exception e) {

                        return false;
                }
        }

        /*
         * This method takes setUsername and returns username.
         *
         * If we have a prefix, then it returns prefix.username.
         */

        private static String adjustName(String name, String prefix) {

                StringBuilder sb;

                if (name.length() >= 4
                                && (name.startsWith("get") || name.startsWith("set"))) {

                        sb = new StringBuilder(name.length() - 3);
                        sb.append(name.substring(3, 4).toLowerCase());
                        if (name.length() > 4)
                                sb.append(name.substring(4, name.length()));

                } else if (name.length() >= 3 && name.startsWith("is")) {

                        sb = new StringBuilder(name.length() - 2);
                        sb.append(name.substring(2, 3).toLowerCase());
                        if (name.length() > 3)
                                sb.append(name.substring(3, name.length()));

                } else {

                        throw new IllegalArgumentException("Cannot adjust method: " + name);
                }

                if (prefix != null) {

                        StringBuffer sb2 = new StringBuffer(128);

                        return sb2.append(prefix).append(PREFIX_SEPARATOR).append(
                                        sb.toString()).toString();

                }

                return sb.toString();
        }

        /**
         * Extract the value of a property of a bean!
         *
         * @param bean
         *            the target bean
         * @param nameProperty
         *            the property name
         * @return they value as String. The method toString is always called to
         *         every property!
         * @throws Exception
         */

        public static String getProperty(Object bean, String nameProperty)
                        throws Exception {

                if (nameProperty == null || nameProperty.equals(""))
                        return null;
               
                String methodName = getter(nameProperty);

                Class<? extends Object> clazz = bean.getClass();
                Method[] methods = clazz.getMethods();
                for (Method method : methods) {
                        if (method.getName().equals(methodName)
                                        && method.getParameterTypes().length == 0) {
                                Object value = method.invoke(bean, (Object[]) null);
                                if (value == null)
                                        return null;
                                return value.toString();
                        }
                }

                Field[] fields = clazz.getDeclaredFields();
                for (Field field : fields) {
                        field.setAccessible(true);
                        if (field.getName().equals(nameProperty)) {
                                Object value = field.get(bean);
                                if (value == null)
                                        return null;
                                return value.toString();
                        }
                }

                return null;
        }

        private static String getter(String name) {
                StringBuilder sb = new StringBuilder(name.length() + 3);

                sb.append("get").append(name.substring(0, 1).toUpperCase());

                if (name.length() > 1)
                        sb.append(name.substring(1));
                return sb.toString();
        }

        public static void beanToMap(Object bean, Map<String, String> map)
                        throws IllegalArgumentException, IllegalAccessException,
                        InvocationTargetException {
                if (bean != null) {

                        for (Method method : bean.getClass().getMethods()) {
                                String name = method.getName();

                                if (name.length() > 3 && name.startsWith("get")
                                                && !name.equals("getClass")
                                                && method.getParameterTypes().length == 0) {

                                        method.setAccessible(true);
                                        Object value = method.invoke(bean, new Object[0]);
                                        map.put(name, value.toString());
                                }
                        }
                }
        }

}