From 2649b9130b32ccaf3697723a49dde7c3233e86f9 Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Wed, 8 Aug 2012 18:33:45 +0200
Subject: [PATCH] TemplateFileReferenceSet extracted to python-openapi

---
 python/openapi/python-openapi.iml             |  1 +
 .../TemplateFileReferenceSet.java             | 95 +++++++++++++++++++
 2 files changed, 96 insertions(+)
 create mode 100644 python/openapi/src/com/jetbrains/python/templateLanguages/TemplateFileReferenceSet.java

diff --git a/python/openapi/python-openapi.iml b/python/openapi/python-openapi.iml
index 00adf90f6d1a..7c59745f3bfc 100644
--- a/python/openapi/python-openapi.iml
+++ b/python/openapi/python-openapi.iml
@@ -10,6 +10,7 @@
     <orderEntry type="module" module-name="python-psi-api" exported="" />
     <orderEntry type="module" module-name="platform-impl" />
     <orderEntry type="module" module-name="lang-api" />
+    <orderEntry type="module" module-name="lang-impl" />
   </component>
 </module>
 
diff --git a/python/openapi/src/com/jetbrains/python/templateLanguages/TemplateFileReferenceSet.java b/python/openapi/src/com/jetbrains/python/templateLanguages/TemplateFileReferenceSet.java
new file mode 100644
index 000000000000..353e64fcba8f
--- /dev/null
+++ b/python/openapi/src/com/jetbrains/python/templateLanguages/TemplateFileReferenceSet.java
@@ -0,0 +1,95 @@
+package com.jetbrains.python.templateLanguages;
+
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleUtilCore;
+import com.intellij.openapi.util.Pair;
+import com.intellij.openapi.util.SystemInfo;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.*;
+import com.intellij.psi.impl.source.resolve.reference.impl.providers.FileReferenceSet;
+import com.intellij.util.containers.ContainerUtil;
+import com.jetbrains.python.PythonStringUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.Collection;
+import java.util.List;
+
+/**
+ * @author traff
+ */
+public abstract class TemplateFileReferenceSet extends FileReferenceSet {
+  private final String mySeparatorString;
+
+  public TemplateFileReferenceSet(PsiElement element, @Nullable PsiReferenceProvider provider) {
+    this(str(element), element, provider);
+  }
+
+  public TemplateFileReferenceSet(String text, PsiElement element,
+                                  @Nullable PsiReferenceProvider provider) {
+    super(text, element, detectShift(element, text), provider,
+          SystemInfo.isFileSystemCaseSensitive);
+    mySeparatorString = detectSeparator(element);
+    reparse();
+  }
+
+  private static String str(PsiElement element) {
+    return PythonStringUtil.stripQuotesAroundValue(element.getText());
+  }
+
+  private static int detectShift(PsiElement element, String text) {
+    String elementText = element.getText();
+    int from = 0;
+    Pair<String, String> quotes = PythonStringUtil.getQuotes(elementText);
+    if (quotes != null) {
+      from = quotes.first.length();
+    }
+
+    return elementText.indexOf(text, from);
+  }
+
+  private static String detectSeparator(PsiElement element) {
+    String winSeparator;
+    if (PythonStringUtil.isRawString(element.getText())) {
+      winSeparator = "\\";
+    }
+    else {
+      winSeparator = "\\\\";
+    }
+    return str(element).contains(winSeparator) ? winSeparator : "/";
+  }
+
+  @Override
+  public String getSeparatorString() {
+    if (mySeparatorString == null) {
+      return super.getSeparatorString();
+    }
+    else {
+      return mySeparatorString;
+    }
+  }
+
+  @NotNull
+  @Override
+  public Collection<PsiFileSystemItem> computeDefaultContexts() {
+    List<PsiFileSystemItem> contexts = ContainerUtil.newArrayList();
+    if (getPathString().startsWith("/") || getPathString().startsWith("\\")) {
+      return contexts;
+    }
+    Module module = ModuleUtilCore.findModuleForPsiElement(getElement());
+    if (module != null) {
+      List<VirtualFile> templatesFolders = getRoots(module);
+      for (VirtualFile folder : templatesFolders) {
+        final PsiDirectory directory = PsiManager.getInstance(module.getProject()).findDirectory(folder);
+        if (directory != null) {
+          contexts.add(directory);
+        }
+      }
+    }
+    return contexts;
+  }
+
+  protected List<VirtualFile> getRoots(Module module) {
+    return TemplatesService.getInstance(module).getTemplateFolders();
+  }
+}