001//
002// MIT License
003//
004// Copyright (c) 2021 Alexander Söderberg & Contributors
005//
006// Permission is hereby granted, free of charge, to any person obtaining a copy
007// of this software and associated documentation files (the "Software"), to deal
008// in the Software without restriction, including without limitation the rights
009// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
010// copies of the Software, and to permit persons to whom the Software is
011// furnished to do so, subject to the following conditions:
012//
013// The above copyright notice and this permission notice shall be included in all
014// copies or substantial portions of the Software.
015//
016// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
017// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
018// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
019// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
020// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
021// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
022// SOFTWARE.
023//
024package cloud.commandframework.annotations.suggestions;
025
026import cloud.commandframework.context.CommandContext;
027import org.checkerframework.checker.nullness.qual.NonNull;
028
029import java.lang.invoke.MethodHandle;
030import java.lang.invoke.MethodHandles;
031import java.lang.reflect.Method;
032import java.util.List;
033import java.util.function.BiFunction;
034
035/**
036 * Represents a method annotated with {@link Suggestions}
037 *
038 * @param <C> Command sender type
039 * @since 1.3.0
040 */
041public final class MethodSuggestionsProvider<C> implements BiFunction<CommandContext<C>, String, List<String>> {
042
043    private final MethodHandle methodHandle;
044
045    /**
046     * Create a new provider
047     *
048     * @param instance Instance that owns the method
049     * @param method   The annotated method
050     * @throws Exception If the method lookup fails
051     */
052    public MethodSuggestionsProvider(
053            final @NonNull Object instance,
054            final @NonNull Method method
055    ) throws Exception {
056        this.methodHandle = MethodHandles.lookup().unreflect(method).bindTo(instance);
057    }
058
059    @Override
060    @SuppressWarnings("unchecked")
061    public List<String> apply(final CommandContext<C> context, final String s) {
062        try {
063            return (List<String>) this.methodHandle.invokeWithArguments(context, s);
064        } catch (final Throwable t) {
065            throw new RuntimeException(t);
066        }
067    }
068
069}