net.contextfw.web.application.component.ObjectBuilder Maven / Gradle / Ivy

Go to download
package net.contextfw.web.application.component;

import java.lang.reflect.Field;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;

public final class ObjectBuilder  {

    private final Set attributeFields = new HashSet();
    private final Set textFields = new HashSet();
    private final String elementName;

    private ObjectBuilder(String elementName) {
        this.elementName = elementName;
    }
    
    public void build(DOMBuilder superb, T t) {
        DOMBuilder b = elementName == null ? superb : superb.descend(elementName);
        try {
            for (Field field : attributeFields) {
                Object obj = field.get(t);
                if (obj != null) {
                    b.attr(field.getName(), obj);
                }
            }
            for (Field field : textFields) {
                Object obj = field.get(t);
                if (obj != null) {
                    b.descend(field.getName()).text(obj);
                }
            }
        }
        catch (IllegalArgumentException e) {
            e.printStackTrace();
        }
        catch (IllegalAccessException e) {
            e.printStackTrace();
        }
    }
    
    public static  ObjectBuilder forClass (Class clazz) {
        return forClass(clazz, clazz.getSimpleName());
    }
    
    public static  ObjectBuilder forClass (Class clazz, String elementName) {
        ObjectBuilder builder = new ObjectBuilder(elementName);
        
        Class currentClass = clazz;

        while (currentClass instanceof Object) {
            for (Field field : currentClass.getDeclaredFields()) {
                if (isUsable(field)) {
                    field.setAccessible(true);
                    builder.attributeFields.add(field);
                }
            }
            currentClass = currentClass.getSuperclass();
        }
        
        return builder;
    }

    private static boolean isUsable(Field field) {
        return !field.getName().equals("serialVersionUID") && !field.getType().isArray() 
                && !Collection.class.isAssignableFrom(field.getType());
    }
    
    public ObjectBuilder asText(String fieldName) {
        
        Field textField = null;
        
        for (Field attr : attributeFields) {
            if (attr.getName().equals(fieldName)) {
                textField = attr;
                break;
            }
        }
        
        if (textField == null) {
            throw new RuntimeException("Could not find field " + fieldName);
        }
        else {
            attributeFields.remove(textField);
            textFields.add(textField);
            return this;
        }
    }
    
    public ObjectBuilder ignore(String fieldName) {
        
        Field textField = null;
        
        for (Field attr : attributeFields) {
            if (attr.getName().equals(fieldName)) {
                textField = attr;
                break;
            }
        }
        
        if (textField == null) {
            throw new RuntimeException("Could not find field " + fieldName);
        }
        else {
            attributeFields.remove(textField);
            textFields.remove(textField);
            return this;
        }
    }
}