blob: 3b476c2abde4800929f776e756b6d30c61c2ca6a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
package me.kyren223.harpoonforjb
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 HarpoonState {
private var projectFiles: MutableMap<String, MutableList<VirtualFile>> = mutableMapOf()
private const val HARPOON_LIST_KEY = "HarpoonList"
private var quickMenu: HarpoonQuickMenu? = 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(HARPOON_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(HARPOON_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) {
println("Canceling quick menu")
quickMenu!!.doCancelAction()
return
}
loadProject(project)
val content = getProjectContent(project)
quickMenu = HarpoonQuickMenu(content)
val result = quickMenu!!.showAndGet()
if (quickMenuContent != null && content != quickMenuContent) {
// Update the content
// println("Content Before:\n${quickMenuContent}")
retrieveProjectFiles(project, quickMenuContent!!)
// val contentAfter = projectFiles[project.name]!!
// .joinToString("\n") { it.path }
// .trim()
// println("Content After:\n${contentAfter}")
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
}
}
|
