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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
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 var pathMappings: MutableMap<String, MutableMap<String, String>> = mutableMapOf()
private const val TRIDENT_LIST_KEY = "TridentList"
private const val PATH_MAPPINGS_LIST_KEY = "PathMappings"
private var quickMenu: QuickMenu? = null
var quickMenuSelectedIndex: Int? = null
var quickMenuContent: String = ""
private var pathMappingsMenu: PathMappingsMenu? = null
var pathMappingsContent: 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 pathMappingsList = properties.getList(PATH_MAPPINGS_LIST_KEY)
var map = mutableMapOf<String, String>()
if (pathMappingsList == null) {
map["..."] = project.basePath!!
} else {
for (entry in pathMappingsList) {
val split = entry.split("=")
if (split.size != 2) continue
map[split[0]] = split[1]
}
}
this.pathMappings[project.name] = map
val list = properties.getList(TRIDENT_LIST_KEY)
if (list == null) {
projectFiles[project.name] = mutableListOf()
} else {
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)
val pathMappingsString = pathMappings[project.name]!!.entries
.joinToString("\n") { "${it.key}=${it.value}" }
.split("\n")
properties.setList(PATH_MAPPINGS_LIST_KEY, pathMappingsString)
}
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 = QuickMenu(content)
val result = quickMenu!!.showAndGet()
if (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 {
loadProject(project)
val map = pathMappings[project.name] ?: throw IllegalStateException("Path mappings not found")
println("map: $map")
var content = projectFiles[project.name]!!
.joinToString("\n") { it.path }
.trim()
println("content: $content")
for ((key, value) in map) {
println("key: $key | value: $value")
content = content.replace(value, key)
}
println("content: $content")
return content
}
private fun retrieveProjectFiles(project: Project, content: String) {
loadProject(project)
val files = content
.split("\n")
.asSequence()
.map { it.trim() }
.map { line ->
pathMappings[project.name]!!.entries.fold(line) { acc, (key, value) ->
acc.replace(key, value)
}
}
// .map {
// pathMappings[project.name]!!.entries.forEach { (key, value) -> it.replace(key, value) }
// it
// }
.filter { it.isNotBlank() }
.map { LocalFileSystem.getInstance().findFileByPath(it) }
.filterNotNull()
.filter { it.isValid }
.toMutableList()
projectFiles[project.name] = files
}
fun togglePathMappingsMenu(project: Project) {
if (pathMappingsMenu != null && pathMappingsMenu!!.isShowing) {
pathMappingsMenu!!.doCancelAction()
return
}
loadProject(project)
val content = pathMappings[project.name]!!.entries
.joinToString("\n") { "${it.key}=${it.value}" }
pathMappingsMenu = PathMappingsMenu(content)
pathMappingsMenu!!.showAndGet()
if (content != pathMappingsContent) {
pathMappingsContent.split("\n").forEach {
val split = it.split("=")
if (split.size != 2) return@forEach
this.pathMappings[project.name]!![split[0]] = split[1]
}
saveProject(project)
}
}
}
|
