Provide a way to work with other file name extensions #49

shalupov · Feb 8, 2016 · 95bf005 · 95bf005
1 parent 22d5a99
commit 95bf005
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 4 deletions.
diff --git a/src/main/java/com/intellij/aws/cloudformation/CloudFormationFileType.java b/src/main/java/com/intellij/aws/cloudformation/CloudFormationFileType.java
@@ -2,16 +2,15 @@
 
 import com.google.common.base.Charsets;
 import com.intellij.icons.AllIcons;
+import com.intellij.json.JsonFileType;
 import com.intellij.json.JsonLanguage;
+import com.intellij.openapi.fileTypes.FileTypeManager;
 import com.intellij.openapi.fileTypes.LanguageFileType;
 import com.intellij.openapi.fileTypes.ex.FileTypeIdentifiableByVirtualFile;
-import com.intellij.openapi.util.io.ByteSequence;
-import com.intellij.openapi.util.io.FileUtil;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.openapi.vfs.VirtualFileSystem;
 import com.intellij.openapi.vfs.newvfs.FileSystemInterface;
 import com.intellij.openapi.vfs.newvfs.impl.StubVirtualFile;
-import com.intellij.util.Processor;
 import org.apache.commons.lang.ArrayUtils;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -58,7 +57,12 @@ public Icon getIcon() {
 
   @Override
   public boolean isMyFileType(@NotNull VirtualFile file) {
-    return CloudFormationFileType.EXTENSION.equalsIgnoreCase(file.getExtension()) && detectFromContent(file);
+    String extension = file.getExtension();
+
+    return extension != null &&
+        (FileTypeManager.getInstance().getFileTypeByExtension(extension) == JsonFileType.INSTANCE ||
+            CloudFormationFileType.EXTENSION.equalsIgnoreCase(extension)) &&
+        detectFromContent(file);
   }
 
   private boolean detectFromContent(@NotNull VirtualFile file) {

diff --git a/testData/inspections/expected.xml b/testData/inspections/expected.xml
@@ -10,6 +10,11 @@
     <line>2</line>
     <description>Unknown AWS CloudFormation template version &quot;2010-09-09 + 1&quot;. Supported versions: 2010-09-09</description>
   </problem>
+  <problem>
+    <file>wrongVersion.json</file>
+    <line>2</line>
+    <description>Unknown AWS CloudFormation template version &quot;2010-09-09 + 1&quot;. Supported versions: 2010-09-09</description>
+  </problem>
   <problem>
     <file>noResourceType.template</file>
     <line>6</line>

diff --git a/testData/inspections/src/wrongVersion.json b/testData/inspections/src/wrongVersion.json
@@ -0,0 +1,7 @@
+{
+  "AWSTemplateFormatVersion": "2010-09-09 + 1",
+  "Description": "desc",
+
+  "Resources": {
+  }
+}