add yaml "requirements" completion on pattern/path keys

Author: Haehnchen

src/fr/adrienbrault/idea/symfony2plugin/config/yaml/YamlCompletionContributor.java

@@ -40,6 +40,9 @@
 import org.jetbrains.yaml.psi.YAMLKeyValue;
 import org.jetbrains.yaml.psi.YAMLSequence;
 
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
 /**
  * @author Daniel Espendiller <daniel@espendiller.net>
  */
@@ -116,6 +119,7 @@ public void addCompletions(@NotNull CompletionParameters parameters,
 
  extend(CompletionType.BASIC, YamlElementPatternHelper.getSuperParentArrayKey("services"), new YamlCompletionProvider(new String[] {"class", "public", "tags", "calls", "arguments", "scope"}));
  extend(CompletionType.BASIC, YamlElementPatternHelper.getWithFirstRootKey(), new RouteKeyNameYamlCompletionProvider(new String[] {"pattern", "defaults", "path", "requirements", "methods", "condition", "resource", "prefix"}));
+ extend(CompletionType.BASIC, YamlElementPatternHelper.getParentKeyName("requirements"), new RouteRequirementsCompletion());
 
  extend(CompletionType.BASIC, StandardPatterns.and(
  YamlElementPatternHelper.getInsideKeyValue("tags"),
@@ -405,5 +409,34 @@ public void addCompletions(@NotNull CompletionParameters parameters, ProcessingC
  super.addCompletions(parameters, context, resultSet);
  }
  }
+
+ /**
+ * "requirements" on "path/pattern: /hello/{name}"
+ */
+ private class RouteRequirementsCompletion extends CompletionProvider<CompletionParameters> {
+ @Override
+ protected void addCompletions(@NotNull CompletionParameters completionParameters, ProcessingContext processingContext, @NotNull CompletionResultSet completionResultSet) {
+ YAMLKeyValue yamlKeyValue = PsiTreeUtil.getParentOfType(completionParameters.getOriginalPosition(), YAMLKeyValue.class);
+ if(yamlKeyValue != null) {
+ PsiElement compoundValue = yamlKeyValue.getParent();
+ if(compoundValue instanceof YAMLCompoundValue) {
+
+ // path and pattern are valid
+ String pattern = YamlHelper.getYamlKeyValueAsString((YAMLCompoundValue) compoundValue, "path", false);
+ if(pattern == null) {
+ pattern = YamlHelper.getYamlKeyValueAsString((YAMLCompoundValue) compoundValue, "pattern", false);
+ }
+
+ if(pattern != null) {
+ Matcher matcher = Pattern.compile("\\{(\\w+)}").matcher(pattern);
+ while(matcher.find()){
+ completionResultSet.addElement(LookupElementBuilder.create(matcher.group(1)));
+ }
+ }
+
+ }
+ }
+ }
+ }
 }
 

src/fr/adrienbrault/idea/symfony2plugin/config/yaml/YamlElementPatternHelper.java

@@ -115,6 +115,66 @@ public static ElementPattern<PsiElement> getOrmParentLookup(String keyName) {
  );
  }
 
+ /**
+ * provides auto complete on
+ *
+ * keyName:
+ * refer|
+ * refer|: xxx
+ * refer|
+ *
+ * @param keyName key name
+ */
+ public static ElementPattern<PsiElement> getParentKeyName(String keyName) {
+ return PlatformPatterns.or(
+ // match
+ //
+ // keyName:
+ // refer|: xxx
+ PlatformPatterns
+ .psiElement(YAMLTokenTypes.SCALAR_KEY)
+ .withParent(PlatformPatterns
+ .psiElement(YAMLKeyValue.class)
+ .withParent(PlatformPatterns
+ .psiElement(YAMLElementTypes.COMPOUND_VALUE)
+ .withParent(PlatformPatterns
+ .psiElement(YAMLKeyValue.class)
+ .withName(
+ PlatformPatterns.string().equalTo(keyName)
+ )
+ )
+ )
+ )
+ .withLanguage(YAMLLanguage.INSTANCE),
+
+ // match
+ //
+ // keyName:
+ // xxx: xxx
+ // refer|
+ PlatformPatterns
+ .psiElement(YAMLTokenTypes.TEXT)
+ .withParent(PlatformPatterns
+ .psiElement(YAMLElementTypes.COMPOUND_VALUE)
+ .withParent(PlatformPatterns
+ .psiElement(YAMLKeyValue.class)
+ .withName(
+ PlatformPatterns.string().equalTo(keyName)
+ )
+ )
+ )
+ .withLanguage(YAMLLanguage.INSTANCE),
+
+ // match
+ //
+ // keyName:
+ // refer|
+ // xxx: xxx
+ getKeyPattern(keyName)
+ .withLanguage(YAMLLanguage.INSTANCE)
+ );
+ }
+
  /**
  * provides auto complete on
  *