TadpoleTriggerComposite.java

/*******************************************************************************
 * Copyright (c) 2013 hangum.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser Public License v2.1
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
 * 
 * Contributors:
 *     hangum - initial API and implementation
 ******************************************************************************/
package com.hangum.tadpole.rdb.core.viewers.object.sub.rdb.table.trigger;

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

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.Logger;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.jface.action.MenuManager;
import org.eclipse.jface.action.Separator;
import org.eclipse.jface.viewers.ArrayContentProvider;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.jface.viewers.TableViewer;
import org.eclipse.swt.SWT;
import org.eclipse.swt.custom.CTabFolder;
import org.eclipse.swt.custom.CTabItem;
import org.eclipse.swt.custom.SashForm;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Table;
import org.eclipse.ui.IWorkbenchPartSite;

import com.hangum.tadpole.commons.exception.dialog.ExceptionDetailsErrorDialog;
import com.hangum.tadpole.commons.libs.core.define.PublicTadpoleDefine;
import com.hangum.tadpole.commons.libs.core.define.PublicTadpoleDefine.OBJECT_TYPE;
import com.hangum.tadpole.commons.libs.core.message.CommonMessages;
import com.hangum.tadpole.engine.define.DBGroupDefine;
import com.hangum.tadpole.engine.permission.PermissionChecker;
import com.hangum.tadpole.engine.query.dao.mysql.TableDAO;
import com.hangum.tadpole.engine.query.dao.mysql.TriggerDAO;
import com.hangum.tadpole.engine.query.dao.system.UserDBDAO;
import com.hangum.tadpole.engine.query.sql.DBSystemSchema;
import com.hangum.tadpole.engine.sql.util.tables.TableUtil;
import com.hangum.tadpole.rdb.core.Activator;
import com.hangum.tadpole.rdb.core.Messages;
import com.hangum.tadpole.rdb.core.actions.object.rdb.generate.GenerateViewDDLAction;
import com.hangum.tadpole.rdb.core.actions.object.rdb.object.ObjectCreatAction;
import com.hangum.tadpole.rdb.core.actions.object.rdb.object.ObjectDropAction;
import com.hangum.tadpole.rdb.core.actions.object.rdb.object.ObjectExplorerSelectionToEditorAction;
import com.hangum.tadpole.rdb.core.actions.object.rdb.object.ObjectRefreshAction;
import com.hangum.tadpole.rdb.core.actions.object.rdb.object.OracleObjectCompileAction;
import com.hangum.tadpole.rdb.core.viewers.object.comparator.TriggerComparator;
import com.hangum.tadpole.rdb.core.viewers.object.sub.AbstractObjectComposite;

/**
 * trigger composite
 * 
 * @author hangum
 *
 */
public class TadpoleTriggerComposite extends AbstractObjectComposite {
	/**
	 * Logger for this class
	 */
	private static final Logger logger = Logger.getLogger(TadpoleTriggerComposite.class);
	
	private TableDAO tableDao;
	
	private CTabItem tbtmTriggers;
	private TableViewer triggerTableViewer;
	private TriggerComparator triggerComparator;
	private List<TriggerDAO> showTrigger = new ArrayList<>();
	private TriggerViewFilter triggerFilter;

	private ObjectCreatAction creatAction_Trigger;
	private ObjectDropAction deleteAction_Trigger;
	private ObjectRefreshAction refreshAction_Trigger;
	private GenerateViewDDLAction viewDDLAction;
	private OracleObjectCompileAction objectCompileAction;

	/**
	 * trigger composite
	 * 
	 * @param site
	 * @param tabFolderObject
	 * @param userDB
	 */
	public TadpoleTriggerComposite(IWorkbenchPartSite site, CTabFolder tabFolderObject, UserDBDAO userDB) {
		super(site, tabFolderObject, userDB);
		createWidget(tabFolderObject);
	}
	
	private void createWidget(final CTabFolder tabFolderObject) {		
		tbtmTriggers = new CTabItem(tabFolderObject, SWT.NONE);
		tbtmTriggers.setText(Messages.get().Triggers);
		tbtmTriggers.setData(TAB_DATA_KEY, PublicTadpoleDefine.OBJECT_TYPE.TRIGGERS.name());

		Composite compositeIndexes = new Composite(tabFolderObject, SWT.NONE);
		tbtmTriggers.setControl(compositeIndexes);
		GridLayout gl_compositeIndexes = new GridLayout(1, false);
		gl_compositeIndexes.verticalSpacing = 2;
		gl_compositeIndexes.horizontalSpacing = 2;
		gl_compositeIndexes.marginHeight = 2;
		gl_compositeIndexes.marginWidth = 2;
		compositeIndexes.setLayout(gl_compositeIndexes);

		SashForm sashForm = new SashForm(compositeIndexes, SWT.NONE);
		sashForm.setOrientation(SWT.VERTICAL);
		sashForm.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1));

		//  SWT.VIRTUAL 일 경우 FILTER를 적용하면 데이터가 보이지 않는 오류수정.
		triggerTableViewer = new TableViewer(sashForm, /* SWT.VIRTUAL | */ SWT.BORDER | SWT.FULL_SELECTION);
		Table tableTableList = triggerTableViewer.getTable();
		tableTableList.setLinesVisible(true);
		tableTableList.setHeaderVisible(true);

		triggerComparator = new TriggerComparator();
		triggerTableViewer.setSorter(triggerComparator);
		triggerComparator.setColumn(0);

		createTriggerColumn(triggerTableViewer, triggerComparator);

		triggerTableViewer.setLabelProvider(new TriggerLabelProvicer());
		triggerTableViewer.setContentProvider(new ArrayContentProvider());

		triggerFilter = new TriggerViewFilter();
		triggerTableViewer.addFilter(triggerFilter);

		sashForm.setWeights(new int[] { 1 });

		// creat action
		createMenu();
	}
	
	private void createMenu() {
		if(getUserDB() == null) return;
		
		creatAction_Trigger = new ObjectCreatAction(getSite().getWorkbenchWindow(), PublicTadpoleDefine.OBJECT_TYPE.TRIGGERS, Messages.get().TadpoleTriggerComposite_1);
		deleteAction_Trigger = new ObjectDropAction(getSite().getWorkbenchWindow(), PublicTadpoleDefine.OBJECT_TYPE.TRIGGERS, Messages.get().TadpoleTriggerComposite_2);
		refreshAction_Trigger = new ObjectRefreshAction(getSite().getWorkbenchWindow(), PublicTadpoleDefine.OBJECT_TYPE.TRIGGERS, CommonMessages.get().Refresh);
		
		viewDDLAction = new GenerateViewDDLAction(getSite().getWorkbenchWindow(), PublicTadpoleDefine.OBJECT_TYPE.TRIGGERS, Messages.get().ViewDDL);
		objectCompileAction = new OracleObjectCompileAction(getSite().getWorkbenchWindow(), PublicTadpoleDefine.OBJECT_TYPE.TRIGGERS, Messages.get().TadpoleTriggerComposite_5);
		
		// object copy to query editor
		objectSelectionToEditorAction = new ObjectExplorerSelectionToEditorAction(getSite().getWorkbenchWindow(), PublicTadpoleDefine.OBJECT_TYPE.TABLES);

		// menu
		final MenuManager menuMgr = new MenuManager("#PopupMenu"); //$NON-NLS-1$
		if(PermissionChecker.isShow(getUserRoleType(), getUserDB())) {
			if(!isDDLLock()) {
				menuMgr.add(creatAction_Trigger);
				menuMgr.add(deleteAction_Trigger);
				menuMgr.add(new Separator());
			}
		}
		menuMgr.add(refreshAction_Trigger);
		menuMgr.add(new Separator());
		menuMgr.add(viewDDLAction);
		if (DBGroupDefine.ORACLE_GROUP == getUserDB().getDBGroup()){
			menuMgr.add(new Separator());
			menuMgr.add(objectCompileAction);
		}
		menuMgr.add(new Separator());
		menuMgr.add(objectSelectionToEditorAction);

		triggerTableViewer.getTable().setMenu(menuMgr.createContextMenu(triggerTableViewer.getTable()));
		getSite().registerContextMenu(menuMgr, triggerTableViewer);
	}

	/**
	 * trigger filter text
	 * @param textSearch
	 */
	public void filter(String textSearch) {
		triggerFilter.setSearchText(textSearch);
		triggerTableViewer.refresh();
	}

	/**
	 * initialize action
	 */
	public void initAction() {
		showTrigger.clear();
		triggerTableViewer.setInput(showTrigger);
		triggerTableViewer.refresh();

		if(getUserDB() == null) return;
		creatAction_Trigger.setUserDB(getUserDB());
		deleteAction_Trigger.setUserDB(getUserDB());
		refreshAction_Trigger.setUserDB(getUserDB());
		
		viewDDLAction.setUserDB(getUserDB());
		objectCompileAction.setUserDB(getUserDB());
		
		objectSelectionToEditorAction.setUserDB(getUserDB());
	}
	
	/**
	 * refresh trigger
	 * 
	 * @param userDB
	 * @param tableDao
	 */
	public void setTable(UserDBDAO userDB, TableDAO tableDao) {
		this.userDB = userDB;
		this.tableDao = tableDao;
		
		refreshTrigger(userDB, true, "");
	}
	
	/**
	 * trigger 정보를 최신으로 갱신 합니다.
	 * @param strObjectName 
	 */
	public void refreshTrigger(final UserDBDAO userDB, boolean boolRefresh, String strObjectName) {
		if(tableDao == null) return;
		if(!boolRefresh) if(!showTrigger.isEmpty()) return;
		this.userDB = userDB;
		
		try {
			showTrigger = DBSystemSchema.getTrigger(userDB, tableDao.getName());

			triggerTableViewer.setInput(showTrigger);
			triggerTableViewer.refresh();
			
			TableUtil.packTable(triggerTableViewer.getTable());

			// select tabitem
			getTabFolderObject().setSelection(tbtmTriggers);
			selectDataOfTable(strObjectName);
		} catch (Exception e) {
			logger.error("showTrigger refresh", e); //$NON-NLS-1$
			Status errStatus = new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); //$NON-NLS-1$
			ExceptionDetailsErrorDialog.openError(getSite().getShell(),CommonMessages.get().Error, Messages.get().ExplorerViewer_76, errStatus); //$NON-NLS-1$
		}
	}

	/**
	 * 지정된 스키마의 모든 trigger 정보를 최신으로 갱신 합니다.
	 * @param strObjectName 
	 */
	public void refreshAllTrigger(final UserDBDAO userDB, boolean boolRefresh, String strObjectName) {
		if(!boolRefresh) if(!showTrigger.isEmpty()) return;
		this.userDB = userDB;
		
		showTrigger = (List<TriggerDAO>)userDB.getDBObject(OBJECT_TYPE.TRIGGERS, userDB.getDefaultSchemanName());
		if(showTrigger == null || showTrigger.isEmpty()) {
			try {
				showTrigger = DBSystemSchema.getTrigger(userDB, "");

				userDB.setDBObject(OBJECT_TYPE.TRIGGERS, userDB.getDefaultSchemanName(), showTrigger);
			} catch (Exception e) {
				logger.error("showTrigger refresh", e); //$NON-NLS-1$
				Status errStatus = new Status(IStatus.ERROR, Activator.PLUGIN_ID, e.getMessage(), e); //$NON-NLS-1$
				ExceptionDetailsErrorDialog.openError(getSite().getShell(),CommonMessages.get().Error, Messages.get().ExplorerViewer_76, errStatus); //$NON-NLS-1$
			}
		}
		triggerTableViewer.setInput(showTrigger);
		triggerTableViewer.refresh();
		
		TableUtil.packTable(triggerTableViewer.getTable());

		// select tabitem
		getTabFolderObject().setSelection(tbtmTriggers);
		selectDataOfTable(strObjectName);
	}

	/**
	 * get tableviewer
	 * @return
	 */
	public TableViewer getTableViewer() {
		return triggerTableViewer;
	}

	@Override
	public void setSearchText(String searchText) {
		triggerFilter.setSearchText(searchText);		
	}
	
	@Override
	public void dispose() {
		super.dispose();
		
		if(creatAction_Trigger != null) creatAction_Trigger.dispose();
		if(deleteAction_Trigger != null) deleteAction_Trigger.dispose();
		if(refreshAction_Trigger != null) refreshAction_Trigger.dispose();
		if(viewDDLAction != null) viewDDLAction.dispose();	
	}

	@Override
	public void selectDataOfTable(String strObjectName) {
		if("".equals(strObjectName) || strObjectName == null) return;
		
		getTableViewer().getTable().setFocus();
		
		// find select object and viewer select
		for(int i=0; i<showTrigger.size(); i++) {
			TriggerDAO tableDao = (TriggerDAO)getTableViewer().getElementAt(i);
			if(StringUtils.equalsIgnoreCase(strObjectName, tableDao.getName())) {
				getTableViewer().setSelection(new StructuredSelection(getTableViewer().getElementAt(i)), true);
				break;
			}
		}
	}

	public void clearList() {
		// TODO Auto-generated method stub
		if(showTrigger != null) this.showTrigger.clear();
	}
	
}