package fr.ifremer.quadrige3.synchro.intercept.administration;
   
   /*-
    * #%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.eventbus.Subscribe;
  import fr.ifremer.common.synchro.meta.SynchroDatabaseMetadata;
  import fr.ifremer.common.synchro.meta.SynchroTableMetadata;
  import fr.ifremer.common.synchro.meta.event.CreateQueryEvent;
  import fr.ifremer.common.synchro.query.SynchroQueryBuilder;
  import fr.ifremer.common.synchro.query.SynchroQueryName;
  import fr.ifremer.common.synchro.query.SynchroQueryOperator;
  import fr.ifremer.quadrige3.synchro.meta.DatabaseColumns;
  import fr.ifremer.quadrige3.synchro.meta.administration.MoratoriumSynchroTables;
  import fr.ifremer.quadrige3.synchro.meta.administration.ProgramStrategySynchroTables;
  import org.apache.commons.collections4.SetUtils;
  import org.hibernate.tool.hbm2ddl.TableMetadata;
  
  /**
   * Manage table on Program/Strategy, with a column 'PROG_CD' :
   * <ul>
   * <li>Limit to program list (if specified in configuration)
   * </ul>
   * 
   * @author Benoit Lavenier <benoit.lavenier@e-is.pro>
   * @since 1.0
   */
  public class TableWithProgCdInterceptor extends AbstractProgramStrategyInterceptor {
  
  	/**
  	 * <p>
  	 * Constructor for TableWithProgCdInterceptor.
  	 * </p>
  	 */
  	public TableWithProgCdInterceptor() {
  		super(SetUtils.union(ProgramStrategySynchroTables.tableNames(), MoratoriumSynchroTables.tableNames()));
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	public boolean doApply(SynchroDatabaseMetadata meta, TableMetadata table) {
  		return super.doApply(meta, table) && hasColumns(table, DatabaseColumns.PROG_CD.name());
  	}
  
  	/**
  	 * <p>
  	 * handleQuery.
  	 * </p>
  	 * 
  	 * @param e
  	 *            a {@link fr.ifremer.common.synchro.meta.event.CreateQueryEvent} object.
  	 */
  	@Subscribe
  	public void handleQuery(CreateQueryEvent e) {
  
  		switch (e.queryName) {
  		case count:
  		case countFromUpdateDate:
  		case select:
  		case selectFromUpdateDate:
  		case selectMaxUpdateDate:
  			// Add restriction
  			e.sql = addRestriction(e.source, e.queryName, e.sql);
  
  		default:
  			break;
  		}
  	}
  
  	/* -- Internal methods -- */
  
  	/**
  	 * <p>
  	 * addRestriction.
  	 * </p>
  	 * 
  	 * @param table
  	 *            a {@link fr.ifremer.common.synchro.meta.SynchroTableMetadata} object.
  	 * @param queryName
 	 *            a {@link fr.ifremer.common.synchro.query.SynchroQueryName} object.
 	 * @param sql
 	 *            a {@link java.lang.String} object.
 	 * @return a {@link java.lang.String} object.
 	 */
 	protected String addRestriction(SynchroTableMetadata table, SynchroQueryName queryName, String sql) {
 		SynchroQueryBuilder queryBuilder = SynchroQueryBuilder.newBuilder(sql);
 
 		// where: add program filter
 		String programWhereClauseOrNull = createProgramWhereClauseOrNull(getConfig(), "t." + DatabaseColumns.PROG_CD.name() + " IN (%s)");
 		if (programWhereClauseOrNull != null) {
 			queryBuilder.addWhere(SynchroQueryOperator.AND, programWhereClauseOrNull);
 		}
 
 		return queryBuilder.build();
 	}
 }