package fr.ifremer.quadrige3.synchro.intercept.data;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 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 com.google.common.base.Joiner;
  import com.google.common.eventbus.Subscribe;
  import fr.ifremer.common.synchro.meta.SynchroDatabaseMetadata;
  import fr.ifremer.common.synchro.meta.event.CreateQueryEvent;
  import fr.ifremer.common.synchro.query.SynchroQueryBuilder;
  import fr.ifremer.common.synchro.query.SynchroQueryOperator;
  import fr.ifremer.quadrige3.synchro.meta.DatabaseColumns;
  import fr.ifremer.quadrige3.synchro.service.SynchroDirection;
  import org.apache.commons.collections4.CollectionUtils;
  import org.hibernate.tool.hbm2ddl.TableMetadata;
  
  import java.util.Set;
  
  /**
   * Interceptor and filter tables with PROG_CD column, when importing from Server DB -> Temp DB
   */
  public class DataTableWithProgCdInterceptor extends AbstractDataInterceptor {
  
  	/**
  	 * <p>
  	 * Constructor for DataTableWithProgCdInterceptor.
  	 * </p>
  	 */
  	public DataTableWithProgCdInterceptor() {
  		super(SynchroDirection.IMPORT_SERVER2TEMP);
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	public boolean doApply(SynchroDatabaseMetadata meta, TableMetadata table) {
  		return hasColumns(table, DatabaseColumns.PROG_CD.name());
  	}
  
  	/**
  	 * <p>
  	 * handleCreateQuery.
  	 * </p>
  	 * 
  	 * @param e
  	 *            a {@link fr.ifremer.common.synchro.meta.event.CreateQueryEvent} object.
  	 */
  	@Subscribe
  	public void handleCreateQuery(CreateQueryEvent e) {
  		switch (e.queryName) {
  		case count:
  		case countFromUpdateDate:
  		case select:
  		case selectFromUpdateDate:
  		case selectMaxUpdateDate:
  
  			e.sql = addRestrictionOnImport(e.sql);
  			break;
  
  		default:
  			break;
  		}
  	}
  
  	/* -- internal methods -- */
  
  	private String addRestrictionOnImport(String sql) {
  		Set<String> programCodes = getConfig().getProgramCodes();
  
  		// If no filter programs : nothing to do
  		if (CollectionUtils.isEmpty(programCodes)) {
  			return sql;
  		}
  
  		StringBuilder sb = new StringBuilder();
  		sb.append('\'');
  		Joiner.on("','").appendTo(sb, programCodes);
  		sb.append('\'');
  
  		// Add restriction on PROG_CD
 		return SynchroQueryBuilder.newBuilder(sql)
 				.addWhere(SynchroQueryOperator.AND, String.format("t.%s in (%s)",
 						DatabaseColumns.PROG_CD.name(),
 						sb.toString()))
 				.build();
 	}
 }