// ============================================================================
//
// Copyright (C) 2006-2016 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.dataprofiler.core.ui.action.provider.predefined;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.jface.action.IMenuManager;
import org.eclipse.jface.viewers.TreeSelection;
import org.talend.cwm.relational.TdColumn;
import org.talend.dataprofiler.core.ui.action.actions.predefined.PreviewColumnAction;
import org.talend.dataprofiler.core.ui.action.provider.AbstractCommonActionProvider;
import org.talend.dq.nodes.DBColumnRepNode;

/**
 * DOC Zqin class global comment. Detailled comment
 */
public class PreviewColumnProvider extends AbstractCommonActionProvider {

    @Override
    public void fillContextMenu(IMenuManager menu) {
        // MOD mzhao user readonly role on svn repository mode.
        if (!isShowMenu()) {
            return;
        }

        TreeSelection treeSelection = ((TreeSelection) this.getContext().getSelection());
        List<TdColumn> list = new ArrayList<TdColumn>();
        Object[] selectedObjs = treeSelection.toArray();
        for (Object obj : selectedObjs) {
            // MOD msjian 2011-12-7 TDQ-4091: the tdColumn info is not correct
            if (obj instanceof DBColumnRepNode) {
                list.add(((DBColumnRepNode) obj).getTdColumn());
            }
            // TDQ-4091 ~
        }

        if (!list.isEmpty()) {
            PreviewColumnAction action = new PreviewColumnAction(list.toArray(new TdColumn[list.size()]));
            menu.add(action);
        }
    }
}