Hey, I'm trying to create a /contract command that allows players to create a contract and send it to a player, the typical steps to make one are

/contract create myContract mymod:forage_contract
/contract edit myContract [some key] [some value] OR /contract edit myContract [add | remove] [some value]

The mymod:forage_contract is a resource location that points to a registered Criteria object, and I want the arguments past edit myContract to be determined by the criteria the player selected

I've figured out that I can use ArgumentBuilder#fork to redirect the command, and have created this abstract function in the Criteria class

public abstract CommandNode<CommandSourceStack> getCommand();

Issue is, I don't know how to obtain the selected Criteria that the player has written in the chat so far, in order to redirect/fork it in the first place. This is my current code

public static void register(CommandDispatcher<CommandSourceStack> dispatcher) {
    RequiredArgumentBuilder<CommandSourceStack, ?> criteriaNode = Commands.argument("criteria", ResourceLocationArgument.id())
        .suggests((s, v) -> {
            CriteriaType.TYPE_REGISTRY.get().getKeys().forEach(r -> v.suggest(r.toString()));
            return v.buildFuture();
        }
    );
        
    dispatcher.register(
                Commands.literal("contract").requires(CommandSourceStack::isPlayer).then(
                        Commands.literal("edit").then(
                                Commands.argument("name", StringArgumentType.string()).then(criteriaNode)
            )
        )
    );
}

 

The same way you would get any other argument. Since you want it for specifically criteria triggers, you'll most likely need to create your own static wrapper around ResourceLocationArgument#getId. See the static methods in the argument class.