package fr.ifremer.adagio.synchro.intercept.internal;
   
   /*
    * #%L
    * SIH-Adagio :: Synchronization
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2012 - 2014 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 java.sql.SQLException;
  
  import com.google.common.base.Preconditions;
  
  import fr.ifremer.adagio.synchro.SynchroTechnicalException;
  import fr.ifremer.adagio.synchro.dao.SynchroTableDao;
  import fr.ifremer.adagio.synchro.intercept.SynchroInterceptorBase;
  import fr.ifremer.adagio.synchro.intercept.SynchroWriteBuffer;
  
  public class RemoteIdWriteInterceptor extends SynchroInterceptorBase {
  
  	private final String tableName;
  
  	private final int columnIndex;
  
  	private final boolean isNullable;
  
  	public RemoteIdWriteInterceptor(fr.ifremer.adagio.synchro.intercept.SynchroInterceptor next, String tableName, int columnIndex, boolean isNullable) {
  		super(next);
  		Preconditions.checkArgument(columnIndex >= 0);
  		this.tableName = tableName;
  		this.columnIndex = columnIndex;
  		this.isNullable = isNullable;
  	}
  
  	public RemoteIdWriteInterceptor(String tableName, int columnIndex, boolean isNullable) {
  		super();
  		Preconditions.checkArgument(columnIndex >= 0);
  		this.tableName = tableName;
  		this.columnIndex = columnIndex;
  		this.isNullable = isNullable;
  	}
  
  	protected void doOnWrite(Object[] data, SynchroTableDao dao, SynchroWriteBuffer buffer) throws SQLException {
  		Integer remoteId = (Integer) data[columnIndex];
  		if (remoteId == null) {
  			return;
  		}
  
  		Integer localId = dao.getIdFromRemoteId(tableName, remoteId);
  		if (localId == null) {
  			// Mandatory column, could not continu
  			if (!isNullable) {
  				throw new SynchroTechnicalException(String.format("Could not retrieve ID corresponding to REMOTE_ID=%s, for table %s",
  						remoteId, tableName));
  			}
  
  			// Optional column : add to pending changes (will be processed later)
  			buffer.addMissingRemoteId(tableName, columnIndex, null, remoteId);
  			data[columnIndex] = null;
  		}
  		else {
  			data[columnIndex] = localId;
  		}
  	}
  
  	@Override
  	public boolean enableOnWrite() {
  		return true;
  	}
  }