FastibatisCodeActionDelegate.java

package net.sf.fast.ibatis.popup.actions;

import java.lang.reflect.InvocationTargetException;
import java.text.MessageFormat;
import java.util.Locale;
import java.util.Map;

import net.sf.fast.ibatis.build.AbstractCodeBuilder;
import net.sf.fast.ibatis.build.DialogType;
import net.sf.fast.ibatis.build.HandleType;
import net.sf.fast.ibatis.dialog.Ibatis2Dialog;
import net.sf.fast.ibatis.dialog.IbatisDialog;
import net.sf.fast.ibatis.dialog.MybatisDialog;
import net.sf.fast.ibatis.i18n.Fasti18n;
import net.sf.fast.ibatis.model.FastIbatisConfig;
import net.sf.fast.ibatis.util.Utils;
import net.sf.fast.ibatis.util.XmlUtil;
import net.sf.fast.ibatis2.build.dao.DAOCustomBuilder;
import net.sf.fast.ibatis2.build.dao.DAODeleteBuilder;
import net.sf.fast.ibatis2.build.dao.DAOInsertBuilder;
import net.sf.fast.ibatis2.build.dao.DAOIntegerBuilder;
import net.sf.fast.ibatis2.build.dao.DAOListBuilder;
import net.sf.fast.ibatis2.build.dao.DAOUpdateBuilder;
import net.sf.fast.ibatis2.build.dao.DAOVoidBuilder;
import net.sf.fast.ibatis2.build.service.ServiceCustomBuilder;
import net.sf.fast.ibatis2.build.service.ServiceDeleteBuilder;
import net.sf.fast.ibatis2.build.service.ServiceInsertBuilder;
import net.sf.fast.ibatis2.build.service.ServiceIntegerBuilder;
import net.sf.fast.ibatis2.build.service.ServiceListBuilder;
import net.sf.fast.ibatis2.build.service.ServiceUpdateBuilder;
import net.sf.fast.ibatis2.build.service.ServiceVoidBuilder;
import net.sf.fast.ibatis3.build.MybatisCodeBuilder;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jface.action.IAction;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.jface.text.TextSelection;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IActionDelegate;
import org.eclipse.ui.IEditorActionDelegate;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.IViewActionDelegate;
import org.eclipse.ui.IViewPart;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.part.FileEditorInput;
/**
 * <p>
 * generate the code action.
 * </p>
 * @author dan.zheng
 * @version 1.0
 */
public class FastibatisCodeActionDelegate implements IEditorActionDelegate,
		IViewActionDelegate {

	private Shell shell;	
	private String ibatisSqlMapIdName;
	private DialogType type;
	/**
	 * Constructor for Action1.
	 */
	public FastibatisCodeActionDelegate() {
		super();
	}
	private String getHandleType(HandleType type) {
		if(type==HandleType.INSERT)
			return "insert";
		else if(type==HandleType.SELECT)
			return "select";
		else if(type==HandleType.UPDATE)
			return "update";
		else if(type==HandleType.DELETE)
			return "delete";
		else
			return "none";
	}
	private HandleType convertHandleType(String src) {
		if(src.equals("select"))
			return HandleType.SELECT;
		else if(src.equals("update"))
			return HandleType.UPDATE;
		else if(src.equals("delete"))
			return HandleType.DELETE;
		else if(src.equals("insert"))
			return HandleType.INSERT;
		else return HandleType.NONE;
	}
	/**
	 * @see IActionDelegate#run(IAction)
	 */
	public void run(IAction action) {
		try {
			final IFile file = ((FileEditorInput) PlatformUI.getWorkbench()
					.getActiveWorkbenchWindow().getActivePage()
					.getActiveEditor().getEditorInput()).getFile();
			final StringBuffer hdType = new StringBuffer();
			final StringBuffer sb = new StringBuffer();
			final FastIbatisConfig fc = new FastIbatisConfig();
			populateConfig(fc, file);
			
			HandleType handleType = getHandleType(file);
			hdType.append(getHandleType(handleType));
			handleType = convertHandleType(hdType.toString());
			if (handleType == HandleType.NONE) {
				MessageDialog.openWarning(shell, "Warning", ibatisSqlMapIdName + " " + Fasti18n.getString("not_the_effective_node"));
				return;
			}
			
			IbatisDialog  dialog = null;
			if (type == DialogType.MyBatis) {
				dialog = new MybatisDialog(shell);
			} else if (type == DialogType.Ibatis2) {
				dialog = new Ibatis2Dialog(shell);
				if(sb!=null&&sb.length()>0) {
					boolean b = MessageDialog.openConfirm(shell, Fasti18n.getString("Confirm"),MessageFormat.format(Fasti18n.getString("similar_sql"), sb.toString()));
					if(!b)
						return;
				}
			} else {
				return;
			}
			
			dialog.open(ibatisSqlMapIdName,
					handleType, fc);
			if (fc.isNeedGeneration()) {
				ProgressMonitorDialog progressDlg = new ProgressMonitorDialog(shell);
				progressDlg.run(true, true, new IRunnableWithProgress() {
					public void run(IProgressMonitor monitor)
							throws InvocationTargetException, InterruptedException {
						monitor.beginTask(Fasti18n.getString("start_working")+"...", 1);
						try {
							generateFastIbatisCode(fc, file);
							if (file != null)
								file.getProject().refreshLocal(
										IResource.DEPTH_INFINITE, monitor);
							monitor.worked(1);
						} catch (CoreException e) {
							e.printStackTrace();
						} finally {
							monitor.done();
						}

					}
				});
			}
		} catch (Exception e) {
			e.printStackTrace();
		}

	}
	/**
	 * get from tables to avoid writing resemble sql.
	 * @return
	 */
	public String getSimilarSQLId(String id,IFile file) {
		StringBuffer sb = new StringBuffer();
		try {			
			String xmlFilePath = file.getLocation().makeAbsolute().toFile()
					.getAbsolutePath();
			Map<String,String> sqlIdTables = XmlUtil.getSelectFromtableSets(xmlFilePath);
			String strs = sqlIdTables.get(this.ibatisSqlMapIdName);
			if(strs!=null) {
				String[] arr1 = strs.split(",");
				for(String key:sqlIdTables.keySet()) {
					if(!key.equals(this.ibatisSqlMapIdName)) {
						String tbs = sqlIdTables.get(key);
						String[] arr2 = tbs.split(",");
						if(isArraySameContent(arr1,arr2))
							sb.append(key).append(",");
					}
				}
			}
		}catch(Exception e) {
			
		}
		if (sb.length() > 0) {
			sb.deleteCharAt(sb.length()-1);
		}
		
		return sb.toString();
	}
	private boolean isArraySameContent(String[] arr1,String arr2[]) {
		if(arr1==null||arr2==null||arr1.length!=arr2.length)
			return false;
		for(int i=0;i<arr1.length;i++) {
			String str = arr1[i];
			boolean isfind = false;
			for(int j=0;j<arr2.length;j++) {
				if(str.equalsIgnoreCase(arr2[j])) {
					isfind = true;
					break;
				}
			}
			if(!isfind)
				return false;
		}
		return true;
	}
	private void populateConfig(FastIbatisConfig fc, IFile file) {
		String xmlFileName = file.getName();
		String modelName = Utils.getModelName(xmlFileName);
		fc.setXmlFileName(xmlFileName);
		fc.setModelName(modelName);
		String xmlFilePath = file.getLocation().makeAbsolute().toFile()
				.getAbsolutePath();
		fc.setXmlFilePath(xmlFilePath);
		fc.setProjectPath(file.getProject().getLocation().makeAbsolute().toFile().getAbsolutePath());
		Locale ll = Locale.getDefault();
		if (ll != null && ll.getLanguage().startsWith("zh")) {
			fc.setCharsetName("GBK");
		}
	}
	private void generateFastIbatisCode(FastIbatisConfig fc, IFile file) {
		try {
			if (type == DialogType.MyBatis) {
				AbstractCodeBuilder codeBuilder = new MybatisCodeBuilder();
				String str1 = fc.getMapperFilePath();
				ICompilationUnit icu = JavaCore.createCompilationUnitFrom(Utils.getModelFileFromPath(str1));
				String mapperContent = codeBuilder.generateCode(icu, fc);
				//System.out.println(mapperContent);
				//Utils.writeContent(str1, mapperContent, fc);
				Utils.writeContent(str1, mapperContent);
				
				String str2 = fc.getSrvFilePath();
				if (str2 != null && str2.length() > 0) {
					ICompilationUnit icu1 = JavaCore.createCompilationUnitFrom(Utils.getModelFileFromPath(str2));
					String srvContent = codeBuilder.generateCode(icu1, fc);
					//System.out.println(srvContent);
					Utils.writeContent(str2, srvContent);
				}
				String str3 = fc.getSrvImplFilePath();
				if (str3 != null && str3.length() > 0) {
					ICompilationUnit icu2 = JavaCore.createCompilationUnitFrom(Utils.getModelFileFromPath(str3));
					String srvImplContent = codeBuilder.generateCode(icu2, fc);
					//System.out.println(srvImplContent);
					Utils.writeContent(str3, srvImplContent);
				}
			} else if (type == DialogType.Ibatis2) {
				HandleType handleType = getHandleType(file);
				if (handleType == HandleType.SELECT) {
					handleSelectMethod(fc);
				} else if (handleType == HandleType.UPDATE) {
					handleUpdateMethod(fc);
				} else if (handleType == HandleType.DELETE) {
					handleDeleteMethod(fc);
				} else if (handleType == HandleType.INSERT) {
					handleInsertMethod(fc);
				}
			}

		} catch (Exception e) {
			e.printStackTrace();
		}

	}

	private HandleType getHandleType(IFile file) {
		String xmlFilePath = file.getLocation().makeAbsolute().toFile()
				.getAbsolutePath();
		HandleType t = XmlUtil.getIbatis2SelectedIdType(xmlFilePath, ibatisSqlMapIdName);
		if (t == HandleType.NONE) {
			t = XmlUtil.getMybatisSelectedIdType(xmlFilePath, ibatisSqlMapIdName);
			type = DialogType.MyBatis;
		} else {
			type = DialogType.Ibatis2;
		}
		return t;
	}

	private void handleInsertMethod(FastIbatisConfig fc) {
		AbstractCodeBuilder dao_cb = new DAOInsertBuilder();
		generateCode(dao_cb.generateCode(fc));
		if (fc.isServiceGenerate()) {
			AbstractCodeBuilder cb = new ServiceInsertBuilder();
			generateCode(cb.generateCode(fc));
		}
	}

	private void handleDeleteMethod(FastIbatisConfig fc) {
		AbstractCodeBuilder dao_cb = new DAODeleteBuilder();
		generateCode(dao_cb.generateCode(fc));
		if (fc.isServiceGenerate()) {
			AbstractCodeBuilder cb = new ServiceDeleteBuilder();
			generateCode(cb.generateCode(fc));
		}
	}

	private void handleUpdateMethod(FastIbatisConfig fc) {
		AbstractCodeBuilder dao_cb = new DAOUpdateBuilder();
		generateCode(dao_cb.generateCode(fc));
		if (fc.isServiceGenerate()) {
			AbstractCodeBuilder cb = new ServiceUpdateBuilder();
			generateCode(cb.generateCode(fc));
		}
	}

	private void handleSelectMethod(FastIbatisConfig fc) {
		if (fc.getReturnType().equals("Integer")) {
			AbstractCodeBuilder dao_cb = new DAOIntegerBuilder();
			generateCode(dao_cb.generateCode(fc));
			if (fc.isServiceGenerate()) {
				AbstractCodeBuilder cb = new ServiceIntegerBuilder();
				generateCode(cb.generateCode(fc));
			}
		} else if (fc.getReturnType().equals("List")) {
			AbstractCodeBuilder dao_cb = new DAOListBuilder();
			generateCode(dao_cb.generateCode(fc));
			if (fc.isServiceGenerate()) {
				AbstractCodeBuilder cb = new ServiceListBuilder();
				generateCode(cb.generateCode(fc));
			}
		} else if (fc.getReturnType().equals("void")) {
			AbstractCodeBuilder dao_cb = new DAOVoidBuilder();
			generateCode(dao_cb.generateCode(fc));
			if (fc.isServiceGenerate()) {
				AbstractCodeBuilder cb = new ServiceVoidBuilder();
				generateCode(cb.generateCode(fc));
			}
		} else {
			AbstractCodeBuilder dao_cb = new DAOCustomBuilder();
			generateCode(dao_cb.generateCode(fc));
			if (fc.isServiceGenerate()) {
				AbstractCodeBuilder cb = new ServiceCustomBuilder();
				generateCode(cb.generateCode(fc));
			}
		}
	}

	private void generateCode(Map<String, String> result) {
		if (result != null) {
			for (String key : result.keySet())
				if (key != null && result.get(key) != null)
					Utils.writeContent(key, result.get(key));
		}
	}

	/**
	 * @see IActionDelegate#selectionChanged(IAction, ISelection)
	 */
	public void selectionChanged(IAction action, ISelection selection) {
		TextSelection selectdIbatisText = (TextSelection) selection;
		ibatisSqlMapIdName = selectdIbatisText.getText();
	}

	public void init(IViewPart view) {

	}


	public void setActiveEditor(IAction action, IEditorPart targetEditor) {
		shell = targetEditor.getSite().getShell();
	}
}