package org.nwapw.abacus.plugin;

import org.nwapw.abacus.function.*;
import org.nwapw.abacus.function.interfaces.NumberFunction;
import org.nwapw.abacus.function.interfaces.NumberOperator;
import org.nwapw.abacus.function.interfaces.TreeValueFunction;
import org.nwapw.abacus.function.interfaces.TreeValueOperator;
import org.nwapw.abacus.number.NumberInterface;

/**
 * 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 {

    /**
     * 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;
        enabled = false;
    }

    /**
     * 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();
        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, NumberFunction toRegister) {
        manager.registerFunction(name, toRegister);
    }

    /**
     * To be used in load(). Registers a tree value 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 tree value function implementation.
     */
    protected final void registerTreeValueFunction(String name, TreeValueFunction toRegister) {
        manager.registerTreeValueFunction(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, NumberOperator operator) {
        manager.registerOperator(name, operator);
    }

    /**
     * To be used in load(). Registers an operator
     * with the plugin internally, which makes it accessible
     * to the plugin manager.
     *
     * @param name     the name of the tree value operator.
     * @param operator the tree value operator to register.
     */
    protected final void registerTreeValueOperator(String name, TreeValueOperator operator) {
        manager.registerTreeValueOperator(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) {
        manager.registerNumberImplementation(name, implementation);
    }

    /**
     * To be used in load(). Registers a documentation instance
     * used to explain some element of the plugin to the user.
     *
     * @param documentation the documentation instance.
     */
    protected final void registerDocumentation(Documentation documentation) {
        manager.registerDocumentation(documentation);
    }

    /**
     * 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 NumberFunction functionFor(String name) {
        return manager.functionFor(name);
    }

    /**
     * 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 tree value function, or null if none was found for that name.
     */
    protected final TreeValueFunction treeValueFunctionFor(String name) {
        return manager.treeValueFunctionFor(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 NumberOperator operatorFor(String name) {
        return manager.operatorFor(name);
    }

    /**
     * Searches the PluginManager for the given tree value 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 tree value operator, or null if none was found for that name.
     */
    protected final TreeValueOperator treeValueOperatorFor(String name) {
        return manager.treeValueOperatorFor(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 PluginManager for the given documentation name and type.
     *
     * @param name the name for which to search.
     * @param type the type of documentation to search for.
     * @return the found documentation, or null if none was found.
     */
    protected final Documentation documentationFor(String name, DocumentationType type) {
        return manager.documentationFor(name, type);
    }

    /**
     * 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 piFor(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();

}