1 files changed, 81 insertions, 0 deletions
diff --git a/json/src/com/jetbrains/jsonSchema/extension/JsonLikePsiWalker.java b/json/src/com/jetbrains/jsonSchema/extension/JsonLikePsiWalker.java
new file mode 100644
index 00000000..c805a06a
--- /dev/null
+++ b/json/src/com/jetbrains/jsonSchema/extension/JsonLikePsiWalker.java
@@ -0,0 +1,81 @@
+// Copyright 2000-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
+package com.jetbrains.jsonSchema.extension;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.TextRange;
+import com.intellij.psi.PsiElement;
+import com.intellij.util.ThreeState;
+import com.jetbrains.jsonSchema.extension.adapters.JsonPropertyAdapter;
+import com.jetbrains.jsonSchema.extension.adapters.JsonValueAdapter;
+import com.jetbrains.jsonSchema.impl.JsonOriginalPsiWalker;
+import com.jetbrains.jsonSchema.impl.JsonSchemaObject;
+import com.jetbrains.jsonSchema.impl.JsonSchemaVariantsTreeBuilder;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @author Irina.Chernushina on 2/15/2017.
+ */
+public interface JsonLikePsiWalker {
+  JsonOriginalPsiWalker JSON_ORIGINAL_PSI_WALKER = new JsonOriginalPsiWalker();
+
+  /**
+   * Returns YES in place where a property name is expected,
+   *         NO in place where a property value is expected,
+   *         UNSURE where both property name and property value can be present
+   */
+  ThreeState isName(PsiElement element);
+
+  boolean isPropertyWithValue(@NotNull PsiElement element);
+  PsiElement goUpToCheckable(@NotNull final PsiElement element);
+  @Nullable
+  List<JsonSchemaVariantsTreeBuilder.Step> findPosition(@NotNull final PsiElement element, boolean forceLastTransition);
+  boolean isNameQuoted();
+  boolean onlyDoubleQuotesForStringLiterals();
+  default boolean quotesForStringLiterals() { return true; }
+  boolean hasPropertiesBehindAndNoComma(@NotNull PsiElement element);
+  Set<String> getPropertyNamesOfParentObject(@NotNull PsiElement originalPosition, PsiElement computedPosition);
+  @Nullable
+  JsonPropertyAdapter getParentPropertyAdapter(@NotNull PsiElement element);
+  boolean isTopJsonElement(@NotNull PsiElement element);
+  @Nullable
+  JsonValueAdapter createValueAdapter(@NotNull PsiElement element);
+
+  default TextRange adjustErrorHighlightingRange(@NotNull PsiElement element) {
+    return element.getTextRange();
+  }
+
+  @Nullable
+  static JsonLikePsiWalker getWalker(@NotNull final PsiElement element, JsonSchemaObject schemaObject) {
+    if (JSON_ORIGINAL_PSI_WALKER.handles(element)) return JSON_ORIGINAL_PSI_WALKER;
+
+    return JsonLikePsiWalkerFactory.EXTENSION_POINT_NAME.getExtensionList().stream()
+      .filter(extension -> extension.handles(element))
+      .findFirst()
+      .map(extension -> extension.create(schemaObject))
+      .orElse(null);
+  }
+
+  default String getDefaultObjectValue() { return "{}"; }
+  @Nullable default String defaultObjectValueDescription() { return null; }
+  default String getDefaultArrayValue() { return "[]"; }
+  @Nullable default String defaultArrayValueDescription() { return null; }
+
+  default boolean invokeEnterBeforeObjectAndArray() { return false; }
+
+  default String getNodeTextForValidation(PsiElement element) { return element.getText(); }
+
+  default QuickFixAdapter getQuickFixAdapter(Project project) { return null; }
+  interface QuickFixAdapter {
+    @Nullable PsiElement getPropertyValue(PsiElement property);
+    default @NotNull PsiElement adjustValue(@NotNull PsiElement value) { return value; }
+    @Nullable String getPropertyName(PsiElement property);
+    @NotNull PsiElement createProperty(@NotNull final String name, @NotNull final String value);
+    boolean ensureComma(PsiElement backward, PsiElement self, PsiElement newElement);
+    void removeIfComma(PsiElement forward);
+    boolean fixWhitespaceBefore(PsiElement initialElement, PsiElement element);
+  }
+}