/*
 * Copyright (c) 2008-2014 akquinet tech@spree GmbH
 *
 * This file is part of Hibersap.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this software 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.hibersap.execution.jca;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.hibersap.bapi.BapiConstants;
import org.hibersap.execution.UnsafeCastHelper;

import javax.resource.ResourceException;
import javax.resource.cci.IndexedRecord;
import javax.resource.cci.MappedRecord;
import javax.resource.cci.Record;
import javax.resource.cci.RecordFactory;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

/**
 * @author M. Dahm
 */
public class JCAMapper {

    private static final Log LOG = LogFactory.getLog( JCAMapper.class );

    public MappedRecord mapFunctionMapValuesToMappedRecord( final String bapiName, final RecordFactory recordFactory, final Map<String, Object> functionMap )
            throws ResourceException {
        LOG.info( "mapFunctionMapValuesToMappedRecord() functionMap=" + functionMap );

        MappedRecord mappedInputRecord = recordFactory.createMappedRecord( bapiName );

        final Map<String, Object> importMap = UnsafeCastHelper.castToMap( functionMap.get( BapiConstants.IMPORT ) );
        mapToMappedRecord( recordFactory, mappedInputRecord, importMap );
        final Map<String, Object> tableMap = UnsafeCastHelper.castToMap( functionMap.get( BapiConstants.TABLE ) );
        mapToMappedRecord( recordFactory, mappedInputRecord, tableMap );

        LOG.info( "mapFunctionMapValuesToMappedRecord() record=" + mappedInputRecord );

        return mappedInputRecord;
    }

    public void mapRecordToFunctionMap( final Map<String, Object> functionMap, final Map<String, Object> resultRecordMap ) {
        LOG.info( "mapRecordToFunctionMap() recordMap=" + resultRecordMap );

        for ( final Entry<String, Object> entry : resultRecordMap.entrySet() ) {
            final Object recordValue = entry.getValue();
            final String recordKey = entry.getKey();

            LOG.debug( "mapping " + recordValue.getClass().getName() + ": " + recordKey + "=" + recordValue );

            if ( recordValue instanceof MappedRecord ) {
                final MappedRecord mappedResultRecord = (MappedRecord) recordValue;
                final Map<String, Object> resultMap = new HashMap<String, Object>();
                resultMap.put( mappedResultRecord.getRecordName(), mappedResultRecord );
                Map<String, Object> export = UnsafeCastHelper.castToMap( functionMap.get( BapiConstants.EXPORT ) );
                export.put( recordKey, recordValue );
            } else if ( recordValue instanceof IndexedRecord ) {
                final IndexedRecord indexedResultRecord = (IndexedRecord) recordValue;
                List<Map<String, Object>> table = new ArrayList<Map<String, Object>>();

                for ( Object object : indexedResultRecord ) {
                    MappedRecord mr = (MappedRecord) object;
                    final Map<String, Object> line = new HashMap<String, Object>();

                    @SuppressWarnings("unchecked")
                    Set<String> keys = mr.keySet();

                    for ( String key : keys ) {
                        line.put( key, mr.get( key ) );
                    }
                    table.add( line );
                }

                Map<String, Object> tables = UnsafeCastHelper.castToMap( functionMap.get( BapiConstants.TABLE ) );
                tables.put( indexedResultRecord.getRecordName(), table );
            } else {
                Map<String, Object> export = UnsafeCastHelper.castToMap( functionMap.get( BapiConstants.EXPORT ) );
                export.put( recordKey, recordValue );
            }
        }
        LOG.info( "mapRecordToFunctionMap() functionMap=" + functionMap );
    }

    @SuppressWarnings("unchecked")
    private void appendToRecord( final Record record, final String fieldName, final Object value ) {
        if ( record instanceof IndexedRecord ) {
            ( (IndexedRecord) record ).add( value );
        } else {
            ( (MappedRecord) record ).put( fieldName, value );
        }
    }

    @SuppressWarnings("unchecked")
    private void mapToMappedRecord( final RecordFactory recordFactory, final Record record, final Map<String, Object> map ) throws ResourceException {
        for ( final String fieldName : map.keySet() ) {
            final Object value = map.get( fieldName );

            if ( Map.class.isAssignableFrom( value.getClass() ) ) {
                final Map<String, Object> structureMap = UnsafeCastHelper.castToMap( value );
                final Record structure = recordFactory.createMappedRecord( fieldName );

                appendToRecord( record, fieldName, structure );

                mapToMappedRecord( recordFactory, structure, structureMap );
            } else if ( Collection.class.isAssignableFrom( value.getClass() ) ) {
                final Collection<Map<String, Object>> tableMap = UnsafeCastHelper.castToCollectionOfMaps( value );
                final IndexedRecord table = recordFactory.createIndexedRecord( fieldName );

                appendToRecord( record, fieldName, table );

                int i = 0;
                for ( final Map<String, Object> row : tableMap ) {
                    MappedRecord rowRecord = recordFactory.createMappedRecord( fieldName + ":row:" + i );
                    mapToMappedRecord( recordFactory, rowRecord, row );
                    table.add( rowRecord );
                    i++;
                }
            } else {
                appendToRecord( record, fieldName, value );
            }
        }
    }
}