Abacus/src/main/java/org/nwapw/abacus/plugin/Plugin.java

package org.nwapw.abacus.plugin;

import org.nwapw.abacus.function.Function;
import org.nwapw.abacus.function.Operator;
import org.nwapw.abacus.number.NumberInterface;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/**
 * A plugin class that can be externally implemented and loaded via the
 * plugin manager. Plugins provide functionality to the calculator
 * with the "hasFunction" and "getFunction" functions,
 * and can use "registerFunction" and "functionFor" for
 * loading internally.
 */
public abstract class Plugin {

    /**
     * A hash map of functions mapped to their string names.
     */
    private Map<String, Function> functions;
    /**
     * A hash map of operators mapped to their string names.
     */
    private Map<String, Operator> operators;
    /**
     * The map of the number implementations this plugin provides.
     */
    private Map<String, NumberImplementation> numberImplementations;
    /**
     * The plugin manager in which to search for functions
     * not inside this package,
     */
    private PluginManager manager;
    /**
     * Whether this plugin has been loaded.
     */
    private boolean enabled;

    private Plugin() {
    }

    /**
     * Creates a new plugin with the given PluginManager.
     *
     * @param manager the manager controlling this plugin.
     */
    public Plugin(PluginManager manager) {
        this.manager = manager;
        functions = new HashMap<>();
        operators = new HashMap<>();
        numberImplementations = new HashMap<>();
        enabled = false;
    }

    /**
     * Gets the list of functions provided by this plugin.
     *
     * @return the list of registered functions.
     */
    public final Set<String> providedFunctions() {
        return functions.keySet();
    }

    /**
     * Gets the list of functions provided by this plugin.
     *
     * @return the list of registered functions.
     */
    public final Set<String> providedOperators() {
        return operators.keySet();
    }

    /**
     * Gets the list of number implementations provided by this plugin.
     *
     * @return the list of registered number implementations.
     */
    public final Set<String> providedNumberImplementations() {
        return numberImplementations.keySet();
    }

    /**
     * Gets a function under the given function name.
     *
     * @param functionName the name of the function to get
     * @return the function, or null if this plugin doesn't provide it.
     */
    public final Function getFunction(String functionName) {
        return functions.get(functionName);
    }

    /**
     * Gets an operator under the given operator name.
     *
     * @param operatorName the name of the operator to get.
     * @return the operator, or null if this plugin doesn't provide it.
     */
    public final Operator getOperator(String operatorName) {
        return operators.get(operatorName);
    }

    /**
     * Gets the number implementation under the given name.
     *
     * @param name the name of the number implementation to look up.
     * @return the number implementation associated with that name, or null if the plugin doesn't provide it.
     */
    public final NumberImplementation getNumberImplementation(String name) {
        return numberImplementations.get(name);
    }

    /**
     * Enables the function, loading the necessary instances
     * of functions.
     */
    public final void enable() {
        if (enabled) return;
        onEnable();
        enabled = true;
    }

    /**
     * Disables the plugin, clearing loaded data store by default
     * and calling its disable() method.
     */
    public final void disable() {
        if (!enabled) return;
        onDisable();
        functions.clear();
        operators.clear();
        enabled = false;
    }

    /**
     * To be used in load(). Registers a function abstract class with the
     * plugin internally, which makes it accessible to the plugin manager.
     *
     * @param name       the name to register by.
     * @param toRegister the function implementation.
     */
    protected final void registerFunction(String name, Function toRegister) {
        functions.put(name, toRegister);
    }

    /**
     * To be used in load(). Registers an operator abstract class
     * with the plugin internally, which makes it accessible to
     * the plugin manager.
     *
     * @param name     the name of the operator.
     * @param operator the operator to register.
     */
    protected final void registerOperator(String name, Operator operator) {
        operators.put(name, operator);
    }

    /**
     * To be used in load(). Registers a new number implementation with the plugin.
     * This makes it accessible to the plugin manager.
     *
     * @param name           the name of the implementation.
     * @param implementation the actual implementation class to register.
     */
    protected final void registerNumberImplementation(String name, NumberImplementation implementation) {
        numberImplementations.put(name, implementation);
    }

    /**
     * Searches the PluginManager for the given function name.
     * This can be used by the plugins internally in order to call functions
     * they do not provide.
     *
     * @param name the name for which to search
     * @return the resulting function, or null if none was found for that name.
     */
    protected final Function functionFor(String name) {
        return manager.functionFor(name);
    }

    /**
     * Searches the PluginManager for the given operator name.
     * This can be used by the plugins internally in order to call
     * operations they do not provide.
     *
     * @param name the name for which to search
     * @return the resulting operator, or null if none was found for that name.
     */
    protected final Operator operatorFor(String name) {
        return manager.operatorFor(name);
    }

    /**
     * Searches the PluginManager for the given number implementation
     * name. This can be used by the plugins internally in order to find
     * implementations that they do not provide.
     *
     * @param name the name for which to search.
     * @return the resulting number implementation, or null if none was found.
     */
    protected final NumberImplementation numberImplementationFor(String name) {
        return manager.numberImplementationFor(name);
    }

    /**
     * Searches the plugin manager for a Pi value for the given number implementation.
     * This is done so that number implementations with various degrees of precision
     * can provide their own pi values, without losing said precision by
     * promoting NaiveNumbers.
     *
     * @param forClass the class to which to find the pi instance.
     * @return the pi value for the given class.
     */
    protected final NumberInterface getPi(Class<? extends NumberInterface> forClass) {
        return manager.piFor(forClass);
    }

    /**
     * Abstract method to be overridden by plugin implementation, in which the plugins
     * are supposed to register the functions they provide and do any other
     * necessary setup.
     */
    public abstract void onEnable();

    /**
     * Abstract method overridden by the plugin implementation, in which the plugins
     * are supposed to dispose of loaded functions, operators, and macros.
     */
    public abstract void onDisable();

}