package org.orienteer.bpm.camunda.handler;

import com.orientechnologies.orient.core.metadata.schema.OType;

import org.camunda.bpm.engine.impl.cmmn.entity.repository.CaseDefinitionEntity;
import org.camunda.bpm.engine.impl.db.ListQueryParameterObject;
import org.camunda.bpm.engine.impl.persistence.entity.IncidentEntity;
import org.camunda.bpm.engine.runtime.IncidentQuery;
import org.orienteer.bpm.camunda.OPersistenceSession;
import org.orienteer.core.util.OSchemaHelper;

import java.util.List;
import java.util.Map;

/**
 * {@link IEntityHandler} for {@link IncidentEntity} 
 */
public class IncidentEntityHandler extends AbstractEntityHandler<IncidentEntity> {

	public static final String OCLASS_NAME = "BPMIncident";
	
    public IncidentEntityHandler() {
        super(OCLASS_NAME);
    }

    @Override
    public void applySchema(OSchemaHelper helper) {
        super.applySchema(helper);

        helper.oProperty("incidentTimestamp", OType.DATETIME, 20)
                .oProperty("incidentMessage", OType.STRING, 30)
                .oProperty("incidentType", OType.STRING, 40)
                .oProperty("execution", OType.LINK, 50).assignVisualization("listbox")
                .oProperty("activityId", OType.STRING, 60)
                .oProperty("processInstanceId", OType.STRING, 70)
                .oProperty("processDefinition", OType.LINK, 80).assignVisualization("listbox")
                .oProperty("causeIncidentId", OType.STRING, 90)
                .oProperty("rootCauseIncidentId", OType.STRING, 100)
                .oProperty("configuration", OType.STRING, 110)
//                .oProperty("tenantId", OType.STRING, 120) // Tenants are not supported
                .oProperty("jobDefinitionId", OType.STRING, 130);
    }
    
    @Override
    public void create(IncidentEntity entity, OPersistenceSession session) {
    	super.create(entity, session);
    	logger.warn("Incident happened: "+entity);
    }

    @Override
    public void applyRelationships(OSchemaHelper helper) {
        super.applyRelationships(helper);
        helper.setupRelationship(IncidentEntityHandler.OCLASS_NAME, "processDefinition", ProcessDefinitionEntityHandler.OCLASS_NAME);
        helper.setupRelationship(IncidentEntityHandler.OCLASS_NAME, "execution", ExecutionEntityHandler.OCLASS_NAME);
    }

    @Statement
    public List<IncidentEntity> selectIncidentsByExecutionId(OPersistenceSession session, final ListQueryParameterObject parameter) {
        return queryList(session, "select from " + getSchemaClass() + " where execution.id=?", parameter.getParameter());
    }

    @Statement
    public List<IncidentEntity> selectIncidentsByProcessInstanceId(OPersistenceSession session, final ListQueryParameterObject parameter) {
        return queryList(session, "select from " + getSchemaClass() + " where processInstanceId=?", parameter.getParameter());
    }

    @Statement
    public List<IncidentEntity> selectIncidentsByConfiguration(OPersistenceSession session, final ListQueryParameterObject parameter) {
        Map<String, String> map = (Map<String, String>) parameter.getParameter();
        String query = "select from " + getSchemaClass() + " where configuration=?";
        if (map.containsKey("incidentType") && map.get("incidentType") != null) {
            query += " and incidentType=?";
            return queryList(session, query, map.get("configuration"), map.get("incidentType"));
        }
        return queryList(session, query, map.get("configuration"));
    }

    @Statement
    public List<IncidentEntity> selectIncidentsByQueryCriteria(OPersistenceSession session, IncidentQuery query) {
        return query(session, query);
    }

}