Detect NBT files automatically based on content rather than extension

Author: DenWav

src/main/kotlin/com/demonwav/mcdev/nbt/filetype/NbtFileType.kt

@@ -17,7 +17,7 @@ import com.intellij.openapi.vfs.VirtualFile
 object NbtFileType : FileType {
     override fun getDefaultExtension() = "nbt"
     override fun getIcon() = PlatformAssets.MINECRAFT_ICON
-    override fun getCharset(file: VirtualFile, content: ByteArray) = null
+    override fun getCharset(file: VirtualFile, content: ByteArray): String? = null
     override fun getName() = "NBT"
     override fun getDescription() = "Named Binary Tag"
     override fun isBinary() = true

src/main/kotlin/com/demonwav/mcdev/nbt/filetype/NbtFileTypeDetector.kt

@@ -0,0 +1,34 @@
+/*
+ * Minecraft Dev for IntelliJ
+ *
+ * https://minecraftdev.org
+ *
+ * Copyright (c) 2017 minecraft-dev
+ *
+ * MIT License
+ */
+
+package com.demonwav.mcdev.nbt.filetype
+
+import com.demonwav.mcdev.nbt.MalformedNbtFileException
+import com.demonwav.mcdev.nbt.Nbt
+import com.intellij.openapi.fileTypes.FileType
+import com.intellij.openapi.fileTypes.FileTypeRegistry
+import com.intellij.openapi.util.io.ByteSequence
+import com.intellij.openapi.vfs.VirtualFile
+import java.io.IOException
+
+class NbtFileTypeDetector : FileTypeRegistry.FileTypeDetector {
+    override fun getVersion() = 1
+
+    override fun detect(file: VirtualFile, firstBytes: ByteSequence, firstCharsIfText: CharSequence?): FileType? {
+        return try {
+            Nbt.buildTagTree(file.inputStream, 100)
+            NbtFileType
+        } catch (e: MalformedNbtFileException) {
+            null
+        } catch (e: IOException) {
+            null
+        }
+    }
+}

src/main/kotlin/com/demonwav/mcdev/nbt/lang/NbttFileType.kt

@@ -17,5 +17,5 @@ object NbttFileType : LanguageFileType(NbttLanguage) {
     override fun getIcon() = PlatformAssets.MINECRAFT_ICON
     override fun getName() = "NBTT"
     override fun getDefaultExtension() = "nbtt"
-    override fun getDescription() = "NBT Text Representation"
+    override fun getDescription() = "NBT Text Representation (don't use this one)"
 }

src/main/kotlin/com/demonwav/mcdev/util/AbstractProjectComponent.kt

@@ -13,14 +13,14 @@ package com.demonwav.mcdev.util
 import com.intellij.openapi.components.ProjectComponent
 import com.intellij.openapi.project.Project
 
-abstract class AbstractProjectComponent(var projectRef: Project?) : ProjectComponent {
+abstract class AbstractProjectComponent(private var projectRef: Project?) : ProjectComponent {
 
     protected val project
         get() = projectRef!!
 
     override fun getComponentName(): String = javaClass.name
 
-    override final fun disposeComponent() {
+    final override fun disposeComponent() {
         projectRef = null
     }
 

src/main/resources/META-INF/plugin.xml

@@ -81,7 +81,9 @@
                                  groupId="language"
                                  instance="com.demonwav.mcdev.MinecraftConfigurable"/>
 
+        <!-- NBT -->
         <fileTypeFactory implementation="com.demonwav.mcdev.nbt.filetype.NbtFileTypeFactory"/>
+        <fileTypeDetector implementation="com.demonwav.mcdev.nbt.filetype.NbtFileTypeDetector"/>
         <fileEditorProvider implementation="com.demonwav.mcdev.nbt.editor.NbtFileEditorProvider" />
 
         <!-- Text representation of NBT -->