JamInterceptorRef.java example

Explorer
jstestdriver-idea-plugin-master
/*
 * Copyright 2009 The authors
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.intellij.struts2.model.jam.convention;

import com.intellij.codeInsight.lookup.LookupElement;
import com.intellij.codeInsight.lookup.LookupElementBuilder;
import com.intellij.jam.*;
import com.intellij.jam.annotations.JamPsiConnector;
import com.intellij.jam.annotations.JamPsiValidity;
import com.intellij.jam.reflect.*;
import com.intellij.javaee.model.common.CommonModelElement;
import com.intellij.openapi.util.Comparing;
import com.intellij.openapi.util.Condition;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.pom.PomTarget;
import com.intellij.psi.PsiAnnotation;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiMember;
import com.intellij.struts2.dom.struts.model.StrutsModel;
import com.intellij.struts2.dom.struts.strutspackage.InterceptorOrStackBase;
import com.intellij.struts2.model.jam.StrutsJamUtils;
import com.intellij.util.Consumer;
import com.intellij.util.PairConsumer;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.xml.DomUtil;
import com.intellij.util.xml.ElementPresentationManager;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.Collection;
import java.util.Collections;

/**
 * {@code org.apache.struts2.convention.annotation.InterceptorRef(s)}.
 *
 * @author Yann Cébron
 */
@SuppressWarnings({"AbstractClassNeverImplemented"})
public abstract class JamInterceptorRef extends CommonModelElement.PsiBase implements JamElement {

  public static final String ANNOTATION_NAME = "org.apache.struts2.convention.annotation.InterceptorRef";
  public static final String ANNOTATION_NAME_LIST = "org.apache.struts2.convention.annotation.InterceptorRefs";

  private static final JamConverter<InterceptorOrStackBase> VALUE_CONVERTER =
    new JamSimpleReferenceConverter<InterceptorOrStackBase>() {

      @Override
      public InterceptorOrStackBase fromString(@Nullable final String s, final JamStringAttributeElement<InterceptorOrStackBase> context) {
        if (s == null) {
          return null;
        }

        final StrutsModel strutsModel = StrutsJamUtils.getStrutsModel(context);
        if (strutsModel == null) {
          return null;
        }

        return ContainerUtil.find(strutsModel.getAllInterceptorsAndStacks(), new Condition<InterceptorOrStackBase>() {
          public boolean value(final InterceptorOrStackBase interceptorOrStackBase) {
            return Comparing.strEqual(s, interceptorOrStackBase.getName().getStringValue());
          }
        });
      }

      @Override
      public Collection<InterceptorOrStackBase> getVariants(final JamStringAttributeElement<InterceptorOrStackBase> context) {
        final StrutsModel strutsModel = StrutsJamUtils.getStrutsModel(context);
        if (strutsModel == null) {
          return Collections.emptyList();
        }

        return strutsModel.getAllInterceptorsAndStacks();
      }

      @NotNull
      @Override
      protected LookupElement createLookupElementFor(@NotNull final InterceptorOrStackBase target) {
        return LookupElementBuilder.create(StringUtil.notNullize(target.getName().getStringValue()))
          .setIcon(ElementPresentationManager.getIcon(target))
          .setTypeText(DomUtil.getFile(target).getName());
      }
    };

  
  private static final JamStringAttributeMeta.Single<InterceptorOrStackBase> VALUE_ATTRIBUTE =
    JamAttributeMeta.singleString(PsiAnnotation.DEFAULT_REFERENCED_METHOD_NAME, VALUE_CONVERTER);

  private static final JamAnnotationMeta INTERCEPTOR_REF_META =
    new JamAnnotationMeta(ANNOTATION_NAME)
      .addAttribute(VALUE_ATTRIBUTE);

  public static final JamClassMeta<JamInterceptorRef> META_CLASS =
    new JamClassMeta<JamInterceptorRef>(JamInterceptorRef.class).addAnnotation(INTERCEPTOR_REF_META);


  private static final JamAnnotationAttributeMeta.Collection<JamInterceptorRef> INTERCEPTOR_REFS_VALUE_ATTRIBUTE =
    JamAttributeMeta.annoCollection(PsiAnnotation.DEFAULT_REFERENCED_METHOD_NAME, INTERCEPTOR_REF_META, JamInterceptorRef.class)
      .addPomTargetProducer(new PairConsumer<JamInterceptorRef, Consumer<PomTarget>>() {
        public void consume(final JamInterceptorRef named, final Consumer<PomTarget> pomTargetConsumer) {
          pomTargetConsumer.consume(named.getPomTarget());
        }
      });

  private static final JamAnnotationMeta INTERCEPTOR_REFS_META =
    new JamAnnotationMeta(ANNOTATION_NAME_LIST).addAttribute(INTERCEPTOR_REFS_VALUE_ATTRIBUTE);

  public static final JamClassMeta<JamInterceptorRef> META_CLASS_LIST =
    new JamClassMeta<JamInterceptorRef>(JamInterceptorRef.class).addAnnotation(INTERCEPTOR_REFS_META);

  @JamPsiConnector
  public abstract PsiMember getOwner();

  @JamPsiValidity
  public abstract boolean isPsiValid();

  @NotNull
  @Override
  public PsiElement getPsiElement() {
    return getOwner();
  }

  public PomTarget getPomTarget() {
    return new JamPomTarget(this, getValue());
  }

  /**
   * Returns "value" attribute.
   *
   * @return JAM-Attribute.
   */
  public JamStringAttributeElement<InterceptorOrStackBase> getValue() {
    return INTERCEPTOR_REF_META.getAttribute(getOwner(), VALUE_ATTRIBUTE);
  }

}