/*
 *  Copyright 2009 salaboy.
 * 
 *  Licensed 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.
 *  under the License.
 */

package org.drools.grid.remote.internal.commands;

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

import org.drools.command.Context;
import org.drools.command.impl.GenericCommand;
import org.drools.command.impl.KnowledgeCommandContext;
import org.drools.grid.remote.ProcessInstanceRemoteClient;
import org.drools.runtime.ExecutionResults;
import org.drools.runtime.StatefulKnowledgeSession;
import org.drools.runtime.impl.ExecutionResultImpl;
import org.drools.runtime.process.ProcessInstance;

/**
 *
 * @author salaboy
 */
public class StartProcessRemoteCommand
    implements
    GenericCommand<ExecutionResults> {
    private String              processId;
    private Map<String, Object> parameters = new HashMap<String, Object>();
    private List<Object>        data       = null;

    public StartProcessRemoteCommand() {
    }

    public StartProcessRemoteCommand(String processId) {
        this.processId = processId;
    }

    public List<Object> getData() {
        return this.data;
    }

    public void setData(List<Object> data) {
        this.data = data;
    }

    public Map<String, Object> getParameters() {
        return this.parameters;
    }

    public void setParameters(Map<String, Object> parameters) {
        this.parameters = parameters;
    }

    public String getProcessId() {
        return this.processId;
    }

    public void setProcessId(String processId) {
        this.processId = processId;
    }

    public ExecutionResults execute(Context context) {
        StatefulKnowledgeSession ksession = ((KnowledgeCommandContext) context).getStatefulKnowledgesession();

        if ( this.data != null ) {
            for ( Object o : this.data ) {
                ksession.insert( o );
            }
        }
        ProcessInstance processInstance = ksession.startProcess( this.processId,
                                                                 this.parameters );

        ((ExecutionResultImpl) ((KnowledgeCommandContext) context).getExecutionResults()).getResults().put( this.processId,
                                                                                                            getRemoteClient( processInstance ) );

        return (((KnowledgeCommandContext) context).getExecutionResults());
    }

    @Override
    public String toString() {
        String result = "session.startProcess(" + this.processId + ", [";
        if ( this.parameters != null ) {
            int i = 0;
            for ( Map.Entry<String, Object> entry : this.parameters.entrySet() ) {
                if ( i++ > 0 ) {
                    result += ", ";
                }
                result += entry.getKey() + "=" + entry.getValue();
            }
        }
        result += "]);";
        return result;
    }

    private ProcessInstance getRemoteClient(ProcessInstance processInstance) {
        return new ProcessInstanceRemoteClient( processInstance.getId(),
                                                processInstance.getProcessId(),
                                                processInstance.getProcessName(),
                                                processInstance.getState() );
    }
}