From fd2b51cc85bdcbee44124529aa3ca7703f3e24d3 Mon Sep 17 00:00:00 2001
From: DevSrSouza <devsrsouza@gmail.com>
Date: Sat, 29 Aug 2020 17:04:39 -0300
Subject: [PATCH] Showing KotlinBukkitAPI tab in Kotlin Files only if
 KotlinBukkitAPI is available at the Module

---
 .../menu/MenuPreviewFileEditorProvider.kt     | 30 ++++++++++++++-----
 1 file changed, 22 insertions(+), 8 deletions(-)

diff --git a/src/main/kotlin/br/com/devsrsouza/kotlinbukkitapi/tooling/menu/MenuPreviewFileEditorProvider.kt b/src/main/kotlin/br/com/devsrsouza/kotlinbukkitapi/tooling/menu/MenuPreviewFileEditorProvider.kt
index 69fb374..6e433e9 100644
--- a/src/main/kotlin/br/com/devsrsouza/kotlinbukkitapi/tooling/menu/MenuPreviewFileEditorProvider.kt
+++ b/src/main/kotlin/br/com/devsrsouza/kotlinbukkitapi/tooling/menu/MenuPreviewFileEditorProvider.kt
@@ -3,7 +3,9 @@ package br.com.devsrsouza.kotlinbukkitapi.tooling.menu
 import br.com.devsrsouza.kotlinbukkitapi.tooling.bukkript.BukkriptFileType
 import com.intellij.openapi.fileEditor.*
 import com.intellij.openapi.fileEditor.impl.text.TextEditorProvider
+import com.intellij.openapi.module.ModuleUtilCore
 import com.intellij.openapi.project.Project
+import com.intellij.openapi.roots.ModuleRootManager
 import com.intellij.openapi.vfs.VirtualFile
 import org.jetbrains.kotlin.idea.KotlinFileType
 
@@ -13,23 +15,35 @@ class MenuPreviewFileEditorProvider : WeighedFileEditorProvider() {
         const val ID = "kotlinbukkitapi-menu-preview"
     }
 
-    override fun getEditorTypeId(): String =
-            ID
+    override fun getEditorTypeId(): String = ID
 
     override fun accept(project: Project, file: VirtualFile): Boolean {
 
         val type = file.fileType
 
-        if(type != KotlinFileType.INSTANCE && type != BukkriptFileType.INSTANCE) return false
+        if(type == BukkriptFileType.INSTANCE) return true
 
-        return true
-//        return PsiManager.getInstance(project).findFile(file)?.findDescendantOfType<KtCallExpression> {
-//            findMenuDeclaration(it) != null
-//        } != null
+        if(type != KotlinFileType.INSTANCE) return false
+
+        val module = ModuleUtilCore.findModuleForFile(file, project) ?: return false
+
+        val root = ModuleRootManager.getInstance(module)
+        var hasKotlinBukkitAPI = false
+
+        root.orderEntries().withoutSdk().forEachLibrary {
+            if((it.name ?: "").contains("kotlinbukkitapi", true)) {
+                hasKotlinBukkitAPI = true
+
+                false
+            } else {
+                true
+            }
+        }
+
+        return hasKotlinBukkitAPI
     }
 
     override fun createEditor(project: Project, file: VirtualFile): FileEditor {
-        println(file.canonicalPath)
         val editor: TextEditor = TextEditorProvider.getInstance().createEditor(project, file) as TextEditor
         val preview =
                 MenuPreviewFileEditor(project, file, editor)