Rebranded to "Trident"

1 files changed, 123 insertions, 0 deletions

diff --git a/src/main/kotlin/me/kyren223/trident/TridentState.kt b/src/main/kotlin/me/kyren223/trident/TridentState.kt
new file mode 100644
index 0000000..7c60d4b
--- /dev/null
+++ b/src/main/kotlin/me/kyren223/trident/TridentState.kt
@@ -0,0 +1,123 @@
+package me.kyren223.trident
+
+import com.intellij.ide.util.PropertiesComponent
+import com.intellij.openapi.fileEditor.FileEditorManager
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vfs.LocalFileSystem
+import com.intellij.openapi.vfs.VirtualFile
+
+object TridentState {
+    private var projectFiles: MutableMap<String, MutableList<VirtualFile>> = mutableMapOf()
+    private const val TRIDENT_LIST_KEY = "TridentList"
+    private var quickMenu: TridentQuickMenu? = null
+    var quickMenuSelectedIndex: Int? = null
+    var quickMenuContent: String? = ""
+
+    fun appendFile(project: Project, file: VirtualFile) {
+        if (!file.isValid) return
+        loadProject(project)
+        val files = projectFiles[project.name]
+        files!!.add(file)
+        saveProject(project)
+    }
+
+    fun selectFile(project: Project, index: Int): VirtualFile? {
+        if (index < 0) return null
+        loadProject(project)
+        val files = projectFiles[project.name]
+        if (files!!.size <= index) return null
+        return files[index]
+    }
+
+    private fun loadProject(project: Project) {
+        if (project.name in projectFiles) return
+        val properties = PropertiesComponent.getInstance(project)
+        val list = properties.getList(TRIDENT_LIST_KEY)
+        if (list == null) {
+            projectFiles[project.name] = mutableListOf()
+            return
+        }
+        val fs = LocalFileSystem.getInstance()
+        val files = list.stream()
+                .map { fs.findFileByPath(it) }
+                .toList()
+                .filterNotNull()
+                .filter { it.isValid }
+                .toMutableList()
+        projectFiles[project.name] = files
+    }
+
+    private fun saveProject(project: Project) {
+        if (project.name !in projectFiles) return
+        val files = projectFiles[project.name]
+        val properties = PropertiesComponent.getInstance(project)
+        val stringList = files!!.stream()
+                .filter { it.isValid }
+                .map { it.path }
+                .toList()
+        properties.setList(TRIDENT_LIST_KEY, stringList)
+    }
+
+    fun getIndexOfFile(project: Project, data: VirtualFile?): Int {
+        loadProject(project)
+        val files = projectFiles[project.name]
+        if (files == null || data == null) return -1
+        return files.indexOfFirst { it.path == data.path }
+    }
+
+    fun getFileCount(project: Project): Int {
+        loadProject(project)
+        return projectFiles[project.name]!!.size
+    }
+
+    fun toggleQuickMenu(project: Project) {
+        if (quickMenu != null && quickMenu!!.isShowing) {
+            quickMenu!!.doCancelAction()
+            return
+        }
+
+        loadProject(project)
+        val content = getProjectContent(project)
+        quickMenu = TridentQuickMenu(content)
+        val result = quickMenu!!.showAndGet()
+
+        if (quickMenuContent != null && content != quickMenuContent) {
+            retrieveProjectFiles(project, quickMenuContent!!)
+            saveProject(project)
+        }
+
+        if (!result) return
+        val index = quickMenuSelectedIndex ?: return
+        quickMenuSelectedIndex = null
+        val file = selectFile(project, index) ?: return
+        FileEditorManager.getInstance(project).openFile(file, true)
+    }
+
+    fun quickMenuSelect() {
+        if (quickMenu == null || !quickMenu!!.isShowing) return
+        quickMenu!!.select()
+    }
+
+    private fun getProjectContent(project: Project): String {
+        val content = projectFiles[project.name]!!
+                .joinToString("\n") { it.path }
+                .replace(project.basePath!!, "...")
+                .trim()
+        return content
+    }
+
+    private fun retrieveProjectFiles(project: Project, content: String) {
+        loadProject(project)
+        val files = content
+                .split("\n")
+                .asSequence()
+                .map { it.trim() }
+                .map { it.replace("...", project.basePath!!) }
+                .filter { it.isNotBlank() }
+                .map { LocalFileSystem.getInstance().findFileByPath(it) }
+                .filterNotNull()
+                .filter { it.isValid }
+                .toMutableList()
+        projectFiles[project.name] = files
+    }
+}
\ No newline at end of file