/*
 * Copyright 2010 JBoss Inc
 *
 * 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.
 */

package org.kie.camel.component;

import org.apache.camel.model.dataformat.XStreamDataFormat;
import org.drools.core.command.runtime.BatchExecutionCommandImpl;
import org.drools.core.runtime.help.impl.XStreamXML.AbortWorkItemConverter;
import org.drools.core.runtime.help.impl.XStreamXML.BatchExecutionResultConverter;
import org.drools.core.runtime.help.impl.XStreamXML.CompleteWorkItemConverter;
import org.drools.core.runtime.help.impl.XStreamXML.FactHandleConverter;
import org.drools.core.runtime.help.impl.XStreamXML.FireAllRulesConverter;
import org.drools.core.runtime.help.impl.XStreamXML.GetGlobalConverter;
import org.drools.core.runtime.help.impl.XStreamXML.GetObjectConverter;
import org.drools.core.runtime.help.impl.XStreamXML.GetObjectsConverter;
import org.drools.core.runtime.help.impl.XStreamXML.InsertConverter;
import org.drools.core.runtime.help.impl.XStreamXML.InsertElementsConverter;
import org.drools.core.runtime.help.impl.XStreamXML.ModifyConverter;
import org.drools.core.runtime.help.impl.XStreamXML.QueryConverter;
import org.drools.core.runtime.help.impl.XStreamXML.QueryResultsConverter;
import org.drools.core.runtime.help.impl.XStreamXML.RetractConverter;
import org.drools.core.runtime.help.impl.XStreamXML.SetGlobalConverter;
import org.drools.core.runtime.help.impl.XStreamXML.SignalEventConverter;
import org.drools.core.runtime.help.impl.XStreamXML.StartProcessConvert;

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

public class XStreamXml {

    public static XStreamDataFormat newXStreamMarshaller(XStreamDataFormat xstreamDataFormat) {
        XStreamHelper.setAliases( xstreamDataFormat );

        //xstream.processAnnotations( BatchExecutionCommand.class );

        Map<String, String[]> map = xstreamDataFormat.getImplicitCollections();
        if ( map == null ) {
            map = new HashMap<String, String[]>();
        }
        map.put( BatchExecutionCommandImpl.class.getName(),
                 new String[]{"commands"} );
        xstreamDataFormat.setImplicitCollections( map );

        List<String> converters = xstreamDataFormat.getConverters();
        if ( converters == null ) {
            converters = new ArrayList<String>();
        }

        converters.add( InsertConverter.class.getName() );
        converters.add( RetractConverter.class.getName() );
        converters.add( ModifyConverter.class.getName() );
        converters.add( GetObjectConverter.class.getName() );
        converters.add( InsertElementsConverter.class.getName() );
        converters.add( FireAllRulesConverter.class.getName() );
        converters.add( StartProcessConvert.class.getName() );
        converters.add( SignalEventConverter.class.getName() );
        converters.add( CompleteWorkItemConverter.class.getName() );
        converters.add( AbortWorkItemConverter.class.getName() );
        converters.add( QueryConverter.class.getName() );
        converters.add( SetGlobalConverter.class.getName() );
        converters.add( GetGlobalConverter.class.getName() );
        converters.add( GetObjectsConverter.class.getName() );
        converters.add( BatchExecutionResultConverter.class.getName() );
        converters.add( QueryResultsConverter.class.getName() );
        converters.add( FactHandleConverter.class.getName() );
        xstreamDataFormat.setConverters( converters );

        return xstreamDataFormat;
    }

}
