website.automate.waml.io.deserializer.ActionDeserializer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of waml-io Show documentation
(De)Serializer of the web automation markup language (WAML) for Java
There is a newer version: 2.1.3
package website.automate.waml.io.deserializer;

import static java.text.MessageFormat.format;
import static java.util.Collections.singletonMap;

import java.io.IOException;
import java.util.Iterator;
import java.util.Map.Entry;

import website.automate.waml.io.model.ActionType;
import website.automate.waml.io.model.CriterionType;
import website.automate.waml.io.model.action.Action;
import website.automate.waml.io.model.action.StoreAction;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.deser.std.StdDeserializer;
import com.fasterxml.jackson.databind.exc.UnrecognizedPropertyException;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;

public class ActionDeserializer extends StdDeserializer {

    private static final long serialVersionUID = 8037140456765531389L;

    public ActionDeserializer() {
        super(Action.class);
    }

    @Override
    public Action deserialize(JsonParser jsonParser,
            DeserializationContext deserializationContext) throws IOException,
            JsonProcessingException {
        ObjectMapper mapper = (ObjectMapper) jsonParser.getCodec();
        ObjectNode root = (ObjectNode) mapper.readTree(jsonParser);

        Class actionClass = null;
        Iterator> elementsIterator = root.fields();
        String key = null;
        while (elementsIterator.hasNext()) {
            Entry element = elementsIterator.next();
            String name = element.getKey();
            if(elementsIterator.hasNext()){
                String anotherName = elementsIterator.next().getKey();
                throw new TooManyActionsException(format("Single step name expected, but found at least: {0}, {1}", name, anotherName));
            }
            ActionType actionType = ActionType.findByName(name);
            if (actionType == null){
                throw new UnknownActionException(format("Unknown step {0}. Correct it to match one of {1}.", name, ActionType.ACTION_TYPES));
            }
            
            actionClass = actionType.getClazz();
            key = name;
        }
        
        JsonNode object = root.get(key);
        if (isShortNotated(actionClass, object)) {
            ActionType actionType = ActionType.findByName(key);
            JsonNode wrapper = new ObjectNode(JsonNodeFactory.instance,
                    singletonMap(actionType.getDefaultCriteriaType().getName(),
                            object));
            object = wrapper;
        }
        
        try {
        	return mapper.convertValue(object, actionClass);
        } catch (IllegalArgumentException e){
        	Throwable cause = e.getCause();
        	if(cause instanceof UnrecognizedPropertyException){
        		UnrecognizedPropertyException unknownCriterionException = (UnrecognizedPropertyException)cause; 
        		throw new UnknownCriterionException(format("Unknown criterion {0} for step {1}. Valid criteria are {2}.", 
        				unknownCriterionException.getPropertyName(),
        				key,
        				unknownCriterionException.getKnownPropertyIds()),
        				e);
        	}
        	throw e;
        }
    }
    
    private boolean isShortNotated(Class actionClass, JsonNode object){
        if(object.isTextual()){
            return true;
        }
        if(actionClass == StoreAction.class){
            return !object.has(CriterionType.IF.getName())
                    && !object.has(CriterionType.UNLESS.getName())
                    && !object.has(CriterionType.VALUE.getName());
        }
        return !object.isObject();
    }
}