xml/impl/src/com/intellij/codeInsight/editorActions/DtdSelectioner.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.codeInsight.editorActions;
  17
  18 import com.intellij.psi.PsiElement;
  19 import com.intellij.psi.xml.XmlAttlistDecl;
  20 import com.intellij.psi.xml.XmlElementDecl;
  21 import com.intellij.psi.xml.XmlToken;
  22 import com.intellij.psi.xml.XmlTokenType;
  23 import com.intellij.openapi.util.TextRange;
  24 import com.intellij.openapi.editor.Editor;
  25
  26 import java.util.List;
  27 import java.util.ArrayList;
  28
  29 public class DtdSelectioner implements ExtendWordSelectionHandler {
  30   public boolean canSelect(PsiElement e) {
  31     return e instanceof XmlAttlistDecl || e instanceof XmlElementDecl;
  32   }
  33
  34   public List<TextRange> select(PsiElement e, CharSequence editorText, int cursorOffset, Editor editor) {
  35     PsiElement[] children = e.getChildren();
  36
  37     PsiElement first = null;
  38     PsiElement last = null;
  39     for (PsiElement child : children) {
  40       if (child instanceof XmlToken) {
  41         XmlToken token = (XmlToken)child;
  42         if (token.getTokenType() == XmlTokenType.XML_TAG_END) {
  43           last = token;
  44           break;
  45         }
  46         if (token.getTokenType() == XmlTokenType.XML_ELEMENT_DECL_START ||
  47             token.getTokenType() == XmlTokenType.XML_ATTLIST_DECL_START
  48            ) {
  49           first = token;
  50         }
  51       }
  52     }
  53
  54     List<TextRange> result = new ArrayList<TextRange>(1);
  55     if (first != null && last != null) {
  56       final int offset = last.getTextRange().getEndOffset() + 1;
  57         result.addAll(ExtendWordSelectionHandlerBase.expandToWholeLine(editorText,
  58                                         new TextRange(first.getTextRange().getStartOffset(), offset < editorText.length() ? offset:editorText.length()),
  59                                         false));
  60     }
  61
  62     return result;
  63   }
  64 }