// ============================================================================
//
// Copyright (C) 2006-2012 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package org.talend.repository.model.migration;

import java.util.Date;
import java.util.GregorianCalendar;

import org.talend.commons.exception.PersistenceException;
import org.talend.commons.ui.runtime.exception.ExceptionHandler;
import org.talend.core.model.migration.AbstractJobMigrationTask;
import org.talend.core.model.properties.Item;
import org.talend.core.repository.model.ProxyRepositoryFactory;
import org.talend.designer.core.model.utils.emf.talendfile.ColumnType;
import org.talend.designer.core.model.utils.emf.talendfile.MetadataType;
import org.talend.designer.core.model.utils.emf.talendfile.NodeType;
import org.talend.designer.core.model.utils.emf.talendfile.ProcessType;

/**
 * <p>
 * This Task check the originalDbColumnName="" in the *.item file, make sure the columnName and the originalDbColumnName
 * are the same.(just remove it, it is OK.)
 * </p>
 * <p>
 * Reason:
 * <li>1. Since TOS 2.2 M1, the DB component schema add the "DbColumn" column, but in the template jet file, it use the
 * columnName to create the SQL statement.</li>
 * <li>2. Since TOS 2.2 M2, the DB component template file use the DbColumn to create the SQL statement.</li>
 * <li>3. If in TOS 2.2 M1, the DbColumn and the column name are not same, the project import to TOS 2.2 M2, it will
 * create the wrong SQL statement.</li>
 */
public class AddDdColumnMigrationTask extends AbstractJobMigrationTask {

    /*
     * (non-Javadoc)
     * 
     * @see org.talend.core.model.migration.IProjectMigrationTask#execute(org.talend.core.model.general.Project)
     */
    @Override
	public ExecutionResult execute(Item item) {
        try {
            removeDbColumn(item);
            return ExecutionResult.SUCCESS_NO_ALERT;
        } catch (Exception e) {
            ExceptionHandler.process(e);
            return ExecutionResult.FAILURE;
        }
    }

    private void removeDbColumn(Item item) throws PersistenceException {
    	ProcessType processType = getProcessType(item);
    	if (processType == null) {
			return;
		}
        ProxyRepositoryFactory factory = ProxyRepositoryFactory.getInstance();

        boolean modified = false;

        for (Object o : processType.getNode()) {

            NodeType node = (NodeType) o;

            for (Object o2 : node.getMetadata()) {

                MetadataType metadata = (MetadataType) o2;

                for (Object o3 : metadata.getColumn()) {
                    ColumnType column = (ColumnType) o3;

                    if (column.getOriginalDbColumnName() != null) {
                        column.setOriginalDbColumnName(null);
                        modified = true;
                    }

                }

            }
        }

        if (modified) {
            factory.save(item,true);
        }
    }
    public Date getOrder() {
        GregorianCalendar gc = new GregorianCalendar(2008, 2, 17, 12, 0, 0);
        return gc.getTime();
    }
}