HQLScratchpadAction.java

/*
 * JBoss, Home of Professional Open Source
 * Copyright 2005, JBoss Inc., and individual contributors as indicated
 * by the @authors tag. See the copyright.txt in the distribution for a
 * full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.hibernate.eclipse.console.actions;

import org.eclipse.jface.viewers.TreePath;
import org.hibernate.console.ConsoleConfiguration;
import org.hibernate.console.ImageConstants;
import org.hibernate.console.node.BaseNode;
import org.hibernate.eclipse.console.HibernateConsoleMessages;
import org.hibernate.eclipse.console.HibernateConsolePlugin;
import org.hibernate.eclipse.console.utils.EclipseImages;
import org.jboss.tools.hibernate.runtime.spi.IPersistentClass;
import org.jboss.tools.hibernate.runtime.spi.IProperty;

public class HQLScratchpadAction extends OpenQueryEditorAction {

	public static final String HQLSCRATCHPAD_ACTIONID = "actionid.hqlscratchpad"; //$NON-NLS-1$
	
	public HQLScratchpadAction() {
		super( HibernateConsoleMessages.HQLScratchpadAction_hql_editor );
		setImageDescriptor(EclipseImages.getImageDescriptor(ImageConstants.HQL_EDITOR));
		setToolTipText(HibernateConsoleMessages.HQLScratchpadAction_open_hql_editor);
		setEnabled( true );
		setId(HQLSCRATCHPAD_ACTIONID);
	}

	protected void openQueryEditor(ConsoleConfiguration config, String query) {
		HibernateConsolePlugin.getDefault().openScratchHQLEditor(config==null?null:config.getName(), query);
	}

	/* (non-Javadoc)
	 * @see org.hibernate.eclipse.console.actions.OpenQueryEditorAction#generateQuery(org.eclipse.jface.viewers.TreePath)
	 */
	protected String generateQuery(TreePath path) {
		Object node = path.getLastSegment();
		if (node instanceof IPersistentClass){
			String name = ((IPersistentClass)node).getEntityName();
			return "from " + name; //$NON-NLS-1$
		} else if (node instanceof IProperty){
			String prName = ((IProperty)node).getName();
			IPersistentClass pClass = ((IProperty)node).getPersistentClass();
			String enName = ""; //$NON-NLS-1$
			if (pClass != null){
				enName = pClass.getEntityName();
				enName = enName.substring(enName.lastIndexOf('.') + 1);
			} else {
				// Generate script for Component property
				for (int i = path.getSegmentCount() - 2; i > 0; i--) {
					if (path.getSegment(i) instanceof IPersistentClass){
						enName = ((IPersistentClass)path.getSegment(i)).getEntityName();
						enName = enName.substring(enName.lastIndexOf('.') + 1);
					} else if (path.getSegment(i) instanceof IProperty){
						prName = ((IProperty)path.getSegment(i)).getName() + "." + prName; //$NON-NLS-1$
					}
				}
			}
			return "select o." + prName + " from " + enName + " o";  //$NON-NLS-1$//$NON-NLS-2$ //$NON-NLS-3$
		}
		else if (node instanceof BaseNode) {
			BaseNode baseNode = (BaseNode)node;
			ConsoleConfiguration consoleConfiguration = baseNode.getConsoleConfiguration();
			if (consoleConfiguration.isSessionFactoryCreated()) {
				return baseNode.getHQL();
			}
		}
		return ""; //$NON-NLS-1$
	}
}