aboutsummaryrefslogtreecommitdiff
path: root/src/main/kotlin/moe/nea/firmament/commands/CaseInsensitiveLiteralCommandNode.kt
blob: b819a5a328bdcb7763932fef48ebf3155a78d21c (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
/*
 * SPDX-FileCopyrightText: 2023 Linnea Gräf <nea@nea.moe>
 *
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

package moe.nea.firmament.commands

import com.mojang.brigadier.Command
import com.mojang.brigadier.RedirectModifier
import com.mojang.brigadier.StringReader
import com.mojang.brigadier.builder.LiteralArgumentBuilder
import com.mojang.brigadier.context.CommandContextBuilder
import com.mojang.brigadier.context.StringRange
import com.mojang.brigadier.exceptions.CommandSyntaxException
import com.mojang.brigadier.tree.CommandNode
import com.mojang.brigadier.tree.LiteralCommandNode
import java.util.function.Predicate

class CaseInsensitiveLiteralCommandNode<S>(
    literal: String, command: Command<S>?, requirement: Predicate<S>?,
    redirect: CommandNode<S>?, modifier: RedirectModifier<S>?, forks: Boolean
) : LiteralCommandNode<S>(
    literal.lowercase(), command, requirement, redirect, modifier, forks
) {
    class Builder<S>(literal: String) : LiteralArgumentBuilder<S>(literal) {
        override fun build(): LiteralCommandNode<S> {
            val result = CaseInsensitiveLiteralCommandNode(
                literal,
                command, requirement, redirect, redirectModifier, isFork
            )
            for (argument in arguments) {
                result.addChild(argument)
            }
            return result
        }
    }

    override fun createBuilder(): LiteralArgumentBuilder<S> {
        return Builder<S>(literal).also {
            it.requires(requirement)
            it.forward(redirect, redirectModifier, isFork)
            if (command != null)
                it.executes(command)
        }
    }

    override fun parse(reader: StringReader, contextBuilder: CommandContextBuilder<S>) {
        val start = reader.cursor
        val end = parse0(reader)
        if (end > -1) {
            contextBuilder.withNode(this, StringRange.between(start, end))
            return
        }

        throw CommandSyntaxException.BUILT_IN_EXCEPTIONS.literalIncorrect().createWithContext(reader, literal)
    }

    override fun toString(): String {
        return "<iliteral $literal>"
    }

    private fun parse0(reader: StringReader): Int {
        val start = reader.cursor
        if (reader.canRead(literal.length)) {
            val end = start + literal.length
            if (reader.string.substring(start, end).equals(literal, true)) {
                reader.cursor = end
                if (!reader.canRead() || reader.peek() == ' ') {
                    return end
                } else {
                    reader.cursor = start
                }
            }
        }
        return -1
    }

}