/*
    * Copyright 2004-2022 the original author or authors.
    *
    * 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
    *
    *    https://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 com.ibatis.sqlmap.engine.exchange;
  
  import com.ibatis.common.beans.ProbeFactory;
  import com.ibatis.sqlmap.engine.mapping.parameter.ParameterMap;
  import com.ibatis.sqlmap.engine.mapping.parameter.ParameterMapping;
  import com.ibatis.sqlmap.engine.mapping.result.ResultMap;
  import com.ibatis.sqlmap.engine.mapping.result.ResultMapping;
  import com.ibatis.sqlmap.engine.scope.StatementScope;
  
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  /**
   * DataExchange implementation for List objects.
   */
  public class ListDataExchange extends BaseDataExchange implements DataExchange {
  
    /**
     * Instantiates a new list data exchange.
     *
     * @param dataExchangeFactory
     *          the data exchange factory
     */
    protected ListDataExchange(DataExchangeFactory dataExchangeFactory) {
      super(dataExchangeFactory);
    }
  
    public void initialize(Map properties) {
    }
  
    public Object[] getData(StatementScope statementScope, ParameterMap parameterMap, Object parameterObject) {
      ParameterMapping[] mappings = parameterMap.getParameterMappings();
      Object[] data = new Object[mappings.length];
      for (int i = 0; i < mappings.length; i++) {
        String propName = mappings[i].getPropertyName();
  
        // parse on the '.' notation and get nested properties
        String[] propertyArray = propName.split("\\.");
  
        if (propertyArray.length > 0) {
          // iterate list of properties to discover values
  
          Object tempData = parameterObject;
  
          for (int x = 0; x < propertyArray.length; x++) {
  
            // is property an array reference
            int arrayStartIndex = propertyArray[x].indexOf('[');
  
            if (arrayStartIndex == -1) {
  
              // is a normal property
              tempData = ProbeFactory.getProbe().getObject(tempData, propertyArray[x]);
  
            } else {
  
              int index = Integer
                  .parseInt(propertyArray[x].substring(arrayStartIndex + 1, propertyArray[x].length() - 1));
              tempData = ((List) tempData).get(index);
  
            }
  
          }
  
          data[i] = tempData;
  
        } else {
  
          int index = Integer.parseInt((propName.substring(propName.indexOf('[') + 1, propName.length() - 1)));
          data[i] = ((List) parameterObject).get(index);
  
        }
  
      }
      return data;
    }
  
    public Object setData(StatementScope statementScope, ResultMap resultMap, Object resultObject, Object[] values) {
      ResultMapping[] mappings = resultMap.getResultMappings();
      List data = new ArrayList();
      for (int i = 0; i < mappings.length; i++) {
        String propName = mappings[i].getPropertyName();
        int index = Integer.parseInt((propName.substring(1, propName.length() - 1)));
       data.set(index, values[i]);
     }
     return data;
   }
 
   public Object setData(StatementScope statementScope, ParameterMap parameterMap, Object parameterObject,
       Object[] values) {
     ParameterMapping[] mappings = parameterMap.getParameterMappings();
     List data = new ArrayList();
     for (int i = 0; i < mappings.length; i++) {
       if (mappings[i].isOutputAllowed()) {
         String propName = mappings[i].getPropertyName();
         int index = Integer.parseInt((propName.substring(1, propName.length() - 1)));
         data.set(index, values[i]);
       }
     }
 
     return data;
   }
 
 }