1 files changed, 77 insertions, 0 deletions
diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/typeEnhancers/SignatureHintProcessor.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/typeEnhancers/SignatureHintProcessor.java
new file mode 100644
index 000000000000..44c8f2ac5094
--- /dev/null
+++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/typeEnhancers/SignatureHintProcessor.java
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * 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 org.jetbrains.plugins.groovy.lang.psi.typeEnhancers;
+
+import com.intellij.openapi.extensions.ExtensionPointName;
+import com.intellij.psi.*;
+import com.intellij.util.ArrayUtil;
+import com.intellij.util.containers.ContainerUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Created by Max Medvedev on 28/02/14
+ */
+public abstract class SignatureHintProcessor {
+  private static final ExtensionPointName<SignatureHintProcessor> EP_NAME = ExtensionPointName.create("org.intellij.groovy.signatureHintProcessor");
+
+  static String[] buildOptions(PsiAnnotation anno) {
+    PsiAnnotationMemberValue options = anno.findAttributeValue("options");
+    if (options instanceof PsiLiteral) {
+      Object value = ((PsiLiteral)options).getValue();
+      if (value instanceof String) {
+        return new String[]{(String)value};
+      }
+    }
+    else if (options instanceof PsiArrayInitializerMemberValue) {
+      PsiAnnotationMemberValue[] initializers = ((PsiArrayInitializerMemberValue)options).getInitializers();
+      ArrayList<String> result = ContainerUtil.newArrayList();
+      for (PsiAnnotationMemberValue initializer : initializers) {
+        if (initializer instanceof PsiLiteral) {
+          Object value = ((PsiLiteral)initializer).getValue();
+          if (value instanceof String) {
+            result.add((String)value);
+          }
+        }
+      }
+
+      return ArrayUtil.toStringArray(result);
+    }
+
+    return ArrayUtil.EMPTY_STRING_ARRAY;
+  }
+
+  public abstract String getHintName();
+
+  @NotNull
+  public abstract List<PsiType[]> inferExpectedSignatures(@NotNull PsiMethod method,
+                                                          @NotNull PsiSubstitutor substitutor,
+                                                          @NotNull String[] options);
+
+  @Nullable
+  public static SignatureHintProcessor getHintProcessor(@NotNull String hint) {
+    for (SignatureHintProcessor processor : EP_NAME.getExtensions()) {
+      if (hint.equals(processor.getHintName())) {
+        return processor;
+      }
+    }
+
+    return null;
+  }
+}