// ============================================================================
//
// 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.dialog;

import org.eclipse.jface.dialogs.MessageDialogWithToggle;
import org.eclipse.jface.viewers.CheckStateChangedEvent;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.ui.dialogs.ContainerCheckedTreeViewer;
import org.talend.dataprofiler.core.i18n.internal.DefaultMessagesImpl;

/**
 * DOC xqliu class global comment. Detailled comment
 */
public class TableSelectionViewer extends ContainerCheckedTreeViewer {

    private final String tooMuchItemSeleted = DefaultMessagesImpl.getString("TableSelectionViewer.string"); //$NON-NLS-1$

    /**
     * DOC xqliu TableSelectionViewer constructor comment.
     * 
     * @param parent
     * @param style
     */
    public TableSelectionViewer(Composite parent, int style) {
        super(parent, style);
    }

    @Override
    protected void fireCheckStateChanged(CheckStateChangedEvent event) {

        boolean checked = event.getChecked();
        Object element = event.getElement();

        boolean isPackage = element instanceof orgomg.cwm.objectmodel.core.Package;

        if ((checked && isPackage) || !checked) {
            super.fireCheckStateChanged(event);
        }
        if (checked && !isPackage) {
            if (MessageDialogWithToggle.openConfirm(null,
                    DefaultMessagesImpl.getString("TableSelectionViewer.warning"), tooMuchItemSeleted)) { //$NON-NLS-1$
                super.fireCheckStateChanged(event);
            } else {
                event.getCheckable().setChecked(element, false);
            }
        }
    }

}