editor.SymbolCompletionHandler Maven / Gradle / Ivy

Go to download
package editor;

import editor.util.TextComponentUtil;
import gw.lang.parser.IDynamicFunctionSymbol;
import gw.lang.parser.IParseTree;
import gw.lang.parser.ISymbol;
import gw.lang.parser.ISymbolTable;
import gw.lang.parser.Keyword;
import gw.lang.parser.expressions.IMemberAccessExpression;
import gw.lang.reflect.java.GosuTypes;
import gw.util.GosuObjectUtil;

import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

/**
 */
public class SymbolCompletionHandler extends AbstractPathCompletionHandler
{
  @Override
  public boolean handleCompletePath( ISymbolTable transientSymTable )
  {
    final GosuEditor gsEditor = getGosuEditor();
    boolean bDotAtCaret = GosuObjectUtil.equals( TextComponentUtil.getWordAtCaret( gsEditor.getEditor() ), "." ) ||
                          GosuObjectUtil.equals( TextComponentUtil.getWordBeforeCaret( gsEditor.getEditor() ), "." ) ||
                          GosuObjectUtil.equals( TextComponentUtil.getWordAtCaret( gsEditor.getEditor() ), ":" ) ||
                          GosuObjectUtil.equals( TextComponentUtil.getWordBeforeCaret( gsEditor.getEditor() ), ":" );
    if( bDotAtCaret )
    {
      return false;
    }
    IParseTree locationAtCaret = gsEditor.getDeepestLocationAtCaret();
    if( locationAtCaret != null &&
        (locationAtCaret.getParsedElement() instanceof IMemberAccessExpression ||
         InitializerCompletionHandler.isInitializerStart( locationAtCaret.getParsedElement() )) )
    {
      return false;
    }

    String strMemberPath = getSingleNameAtCaret();
    strMemberPath = strMemberPath != null ? strMemberPath.trim() : null;

    String wordBeforeCaret = TextComponentUtil.getWordBeforeCaret( getGosuEditor().getEditor() );
    boolean isAtAsPosition = wordBeforeCaret != null && wordBeforeCaret.trim().equals( Keyword.KW_as.toString() );
    if( gsEditor.isAltDown() || isAtAsPosition )
    {
      return displayTypesPopup( strMemberPath );
    }

    Collection listSymbols = transientSymTable.getSymbols().values();
    filterUnwantedSymbols( listSymbols );

    ISymbol[] symbols = listSymbols.toArray( new ISymbol[listSymbols.size()] );
    SymbolPopup valuePopup = new SymbolPopup( symbols, strMemberPath, gsEditor );
    valuePopup.addNodeChangeListener(
      new ChangeListener()
      {
        @Override
        public void stateChanged( ChangeEvent e )
        {
          String word = (String)e.getSource();
          if( word.endsWith( "()" ) )
          {
            TextComponentUtil.replaceWordAtCaretDynamicAndRemoveEmptyParens( gsEditor.getEditor(),
                                                                             word,
                                                                             gsEditor.getReplaceWordCallback(),
                                                                             true );
          }
          else
          {
            TextComponentUtil.replaceWordAtCaretDynamic( gsEditor.getEditor(),
                                                         word,
                                                         gsEditor.getReplaceWordCallback(),
                                                         true );
          }
          gsEditor.getEditor().requestFocus();
          gsEditor.fixSwingFocusBugWhenPopupCloses();
          gsEditor.getEditor().repaint();
        }
      } );
    gsEditor.setValuePopup( valuePopup );
    gsEditor.displayValuePopup();
    if( !valuePopup.isShowing() && !valuePopup.wasAutoDismissed() )
    {
      return displayTypesPopup( strMemberPath );
    }

    return false;
  }


  private void filterUnwantedSymbols( Collection listSymbols )
  {
    List deleteSyms = new ArrayList();
    for( ISymbol s : listSymbols )
    {
      if( s.getType() == GosuTypes.DEF_CTOR_TYPE() || (s instanceof IDynamicFunctionSymbol && ((IDynamicFunctionSymbol)s).isConstructor()) )
      {
        // Filter constructor symbols (only applicable when editing a gs class)
        deleteSyms.add( s );
      }
      else if( s instanceof IDynamicFunctionSymbol && s.getName().startsWith( "@" ) )
      {
        // This is the getter or setter for a property, and the DPS will be in the list already
        deleteSyms.add( s );
      }
    }
    listSymbols.removeAll( deleteSyms );
  }

  boolean displayTypesPopup( String strPrefix )
  {
    TypePopup popup = new TypePopup( strPrefix, getGosuEditor(), isAnnotationsOnly() );
    getGosuEditor().setValuePopup( popup );
    final boolean addUsesAutomatically = getGosuEditor().acceptsUses();
    popup.addNodeChangeListener(
      new ChangeListener()
      {
        @Override
        public void stateChanged( ChangeEvent e )
        {
          String strQualifedType = (String)e.getSource();
          String strRelativeType = TypePopup.getRelativeTypeName( strQualifedType );
          String strPartialType = strQualifedType;
          if( strPartialType.contains( TextComponentUtil.getWordAtCaret( getGosuEditor().getEditor() ) ) )
          {
            strPartialType = strPartialType.substring( strPartialType.indexOf( TextComponentUtil.getWordAtCaret( getGosuEditor().getEditor() ) ) );
          }

          if( addUsesAutomatically )
          {
            // Need to do this before we replace word because replace word causes
            // reparse, which would leave usesStmts in incomplete state.
            getGosuEditor().addToUses( strQualifedType );
          }

          if( addUsesAutomatically &&
              GosuObjectUtil.equals( TextComponentUtil.getWordBeforeCaret( getGosuEditor().getEditor() ), strRelativeType ) )
          {
            //no need to replace the text, since it is already there
          }
          else
          {
            TextComponentUtil.replaceWordAtCaretDynamic(
              getGosuEditor().getEditor(),
              addUsesAutomatically ? strRelativeType : strPartialType,
              getGosuEditor().getReplaceWordCallback(), false );
          }


          getGosuEditor().getEditor().requestFocus();
          getGosuEditor().fixSwingFocusBugWhenPopupCloses();
          getGosuEditor().getEditor().repaint();
        }
      } );
    getGosuEditor().displayValuePopup();
    return true;
  }

  protected boolean isAnnotationsOnly()
  {
    return false;
  }
}