GabrielTofvesson/SpigotPortals
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implement argparse suggestion priorities

Browse Source
This commit is contained in:
Gabriel Tofvesson 2021-11-02 06:45:19 +01:00
parent 5b129172ae
commit 0ecfcec239
1 changed files with 17 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/main/kotlin/ArgParse.kt
View File

@ -3,6 +3,7 @@ import org.bukkit.OfflinePlayer
import org.bukkit.command.CommandSender import org.bukkit.command.CommandSender
import org.bukkit.permissions.Permission import org.bukkit.permissions.Permission
import java.util.* import java.util.*
import kotlin.collections.LinkedHashMap
const val RESULT_ERROR_NOMATCH = "Unknown command" const val RESULT_ERROR_NOMATCH = "Unknown command"
const val RESULT_ERROR_NOPERMS = "You don't have permission to use this command" const val RESULT_ERROR_NOPERMS = "You don't have permission to use this command"
@ -130,23 +131,31 @@ sealed class NodeParseResult<T> {
data class SuccessResult<T>(val match: T): NodeParseResult<T>() data class SuccessResult<T>(val match: T): NodeParseResult<T>()
} }
class ParseTree { const val PRIORITY_MAX = Int.MIN_VALUE
private val branches = LinkedList<ParseBranch>() const val PRIORITY_MED = 0
const val PRIORITY_MIN = Int.MAX_VALUE
fun branch(vararg nodes: ArgNode<*>) = branch(ParseBranch(*nodes)) class ParseTree {
fun branch(parseBranch: ParseBranch): ParseTree { private val branches = LinkedHashMap<ParseBranch, Int>()
branches += parseBranch
fun branch(vararg nodes: ArgNode<*>) = branch(PRIORITY_MED, *nodes)
fun branch(parseBranch: ParseBranch) = branch(PRIORITY_MED, parseBranch)
fun branch(priority: Int, vararg nodes: ArgNode<*>) = branch(priority, ParseBranch(*nodes))
fun branch(priority: Int, parseBranch: ParseBranch): ParseTree {
branches[parseBranch] = priority
return this return this
} }
fun getSuggestions(args: Array<out String>, sender: CommandSender) = fun getSuggestions(args: Array<out String>, sender: CommandSender) =
branches.filter { it.isEligible(sender) } branches.filter { (k, _) -> k.isEligible(sender) }
.mapNotNull { it.getSuggestions(args, sender) } .mapNotNull { (k, v) -> (k.getSuggestions(args, sender) ?: return@mapNotNull null) to v }
.sortedBy { (_, v) -> v }
.map(Pair<List<String>, *>::first)
.flatten() .flatten()
.toHashSet() .toHashSet()
fun getMatch(args: Array<out String>, sender: CommandSender): ParseResult { fun getMatch(args: Array<out String>, sender: CommandSender): ParseResult {
branches.forEach { branches.keys.forEach {
if (it.isEligible(sender)) { if (it.isEligible(sender)) {
val match = it.match(args, sender) val match = it.match(args, sender)
if (match != null) return ParseResult.SuccessResult(match, it) if (match != null) return ParseResult.SuccessResult(match, it)