package fr.ifremer.quadrige2.synchro.intercept.data.internal;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 Synchro Core
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   * 
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  import fr.ifremer.common.synchro.dao.SynchroTableDao;
  import fr.ifremer.common.synchro.intercept.SynchroInterceptorBase;
  import fr.ifremer.common.synchro.intercept.SynchroOperationRepository;
  
  import java.sql.SQLException;
  import java.util.List;
  import java.util.Map;
  
  /**
   * Interceptor on data table (for import from file only) that remap source FK to local FK.
   * Used with ImportFromFileNumericalPkInterceptor to collect local PK
   * <p/>
   * Created by Ludovic on 30/11/2015.
   */
  public class ReplaceColumnValuesInterceptor extends SynchroInterceptorBase {
  
  	private final int columnIndex;
  
  	private final Map<String, Object> valuesMap;
  
  	/**
  	 * <p>
  	 * Constructor for ReplaceColumnValuesInterceptor.
  	 * </p>
  	 * 
  	 * @param columnIndex
  	 *            a int.
  	 * @param columnValuesMap
  	 *            a {@link java.util.Map} object.
  	 */
  	public ReplaceColumnValuesInterceptor(int columnIndex,
  			Map<String, Object> columnValuesMap) {
  		this.columnIndex = columnIndex;
  		this.valuesMap = columnValuesMap;
  		setEnableOnRead(true);
  		setEnableOnWrite(true);
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	public SynchroInterceptorBase clone() {
  		return new ReplaceColumnValuesInterceptor(columnIndex, valuesMap);
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	protected void doOnRead(Object[] data,
  			SynchroTableDao sourceDao,
  			SynchroTableDao targetDao) throws SQLException {
  
  		Object sourceValue = data[columnIndex];
  		if (sourceValue != null) {
  			Object targetValue = valuesMap.get(sourceValue.toString());
  			if (targetValue != null) {
  				data[columnIndex] = targetValue;
  			}
  		}
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	protected void doOnWrite(Object[] data, List<Object> pk, SynchroTableDao sourceDao, SynchroTableDao targetDao, SynchroOperationRepository buffer,
  			boolean insert) throws SQLException {
  
  		Object sourceValue = data[columnIndex];
  		if (sourceValue != null) {
  			Object targetValue = valuesMap.get(sourceValue.toString());
  			if (targetValue != null) {
  				data[columnIndex] = targetValue;
  			}
  		}
  	}
  
  }