/**
 * Contains the implementation of the dependency container.
 *
 * Part of the Poodinis Dependency Injection framework.
 *
 * Authors:
 *  Mike Bierlee, m.bierlee@lostmoment.com
 * Copyright: 2014-2022 Mike Bierlee
 * License:
 *  This software is licensed under the terms of the MIT license.
 *  The full terms of the license can be found in the LICENSE file.
 */

module poodinis.container;

import poodinis.registration : Registration, singleInstance,
    toConcreteTypeListString, initializeFactoryType;
import poodinis.autowire : AutowiredRegistration, AutowireInstantiationContext;
import poodinis.factory : ConstructorInjectingInstanceFactory;
import poodinis.valueinjection : ValueInjectionException;
import poodinis.altphobos : isFunction;
import poodinis.imports : createImportsString;

import std.string : format;
import std.algorithm : canFind;
import std.traits : fullyQualifiedName, hasUDA, BaseTypeTuple;
import std.meta : AliasSeq;

debug
{
    import std.stdio;
}

/**
 * Exception thrown when errors occur while resolving a type in a dependency container.
 */
class ResolveException : Exception
{
    this(string message, TypeInfo resolveType)
    {
        super(format("Exception while resolving type %s: %s", resolveType.toString(), message));
    }

    this(Throwable cause, TypeInfo resolveType)
    {
        super(format("Exception while resolving type %s", resolveType.toString()), cause);
    }
}

/**
 * Exception thrown when errors occur while registering a type in a dependency container.
 */
class RegistrationException : Exception
{
    this(string message, TypeInfo registrationType)
    {
        super(format("Exception while registering type %s: %s",
                registrationType.toString(), message));
    }
}

/**
 * Options which influence the process of registering dependencies
 */
public enum RegistrationOption
{
    none = 0,
    /**
     * Prevent a concrete type being registered on itself. With this option you will always need
     * to use the supertype as the type of the dependency.
     */
    doNotAddConcreteTypeRegistration = 1 << 0,
}

/**
 * Options which influence the process of resolving dependencies
 */
public enum ResolveOption
{
    none = 0,
    /**
     * Registers the type you're trying to resolve before returning it.
     * This essentially makes registration optional for resolving by concerete types.
     * Resolinvg will still fail when trying to resolve a dependency by supertype.
     */
    registerBeforeResolving = 1 << 0,

    /**
     * Does not throw a resolve exception when a type is not registered but will
     * return null instead. If the type is an array, an empty array is returned instead.
     */
    noResolveException = 1 << 1
}

/**
 * Methods marked with this UDA within dependencies are called after that dependency
 * is constructed by the dependency container.
 *
 * Multiple methods can be marked and will all be called after construction. The order in which
 * methods are called is undetermined. Methods should have the signature void(void).
 */
struct PostConstruct
{
}

/**
 * Methods marked with this UDA within dependencies are called before the container
 * loses the dependency's registration.
 *
 * This method is called when removeRegistration or clearAllRegistrations is called.
 * It will also be called when the container's destructor is called.
 */
struct PreDestroy
{
}

/**
 * The dependency container maintains all dependencies registered with it.
 *
 * Dependencies registered by a container can be resolved as long as they are still registered with the container.
 * Upon resolving a dependency, an instance is fetched according to a specific scope which dictates how instances of
 * dependencies are created. Resolved dependencies will be autowired before being returned.
 *
 * In most cases you want to use a global singleton dependency container provided by getInstance() to manage all dependencies.
 * You can still create new instances of this class for exceptional situations.
 */
synchronized class DependencyContainer
{
    private Registration[][TypeInfo] registrations;

    private Registration[] autowireStack;

    private RegistrationOption persistentRegistrationOptions;
    private ResolveOption persistentResolveOptions;

    ~this()
    {
        clearAllRegistrations();
    }

    /**
     * Register a dependency by concrete class type.
     *
     * A dependency registered by concrete class type can only be resolved by concrete class type.
     * No qualifiers can be used when resolving dependencies which are registered by concrete type.
     *
     * The default registration scope is "single instance" scope.
     *
     * Returns:
     * A registration is returned which can be used to change the registration scope.
     *
     * Examples:
     * Register and resolve a class by concrete type:
     * ---
     * class Cat : Animal { ... }
     * container.register!Cat;
     * ---
     *
     * See_Also: singleInstance, newInstance, existingInstance
     */
    public Registration register(ConcreteType)(RegistrationOption options = RegistrationOption.none)
    {
        return register!(ConcreteType, ConcreteType)(options);
    }

    /**
     * Register a dependency by super type.
     *
     * A dependency registered by super type can only be resolved by super type. A qualifier is typically
     * used to resolve dependencies registered by super type.
     *
     * The default registration scope is "single instance" scope.
     *
     * Examples:
     * Register and resolve by super type
     * ---
     * class Cat : Animal { ... }
     * container.register!(Animal, Cat);
     * ---
     *
     * See_Also: singleInstance, newInstance, existingInstance, RegistrationOption
     */
    public Registration register(SuperType, ConcreteType:
        SuperType)(RegistrationOption options = RegistrationOption.none)
            if (!is(ConcreteType == struct))
    {
        TypeInfo registeredType = typeid(SuperType);
        TypeInfo_Class concreteType = typeid(ConcreteType);

        debug (poodinisVerbose)
        {
            writeln(format("DEBUG: Register type %s (as %s)",
                    concreteType.toString(), registeredType.toString()));
        }

        auto existingRegistration = getExistingRegistration(registeredType, concreteType);
        if (existingRegistration)
        {
            return existingRegistration;
        }

        auto instanceFactory = new ConstructorInjectingInstanceFactory!ConcreteType(this);
        auto newRegistration = new AutowiredRegistration!ConcreteType(registeredType,
            instanceFactory, this);
        newRegistration.initializeFactoryType().singleInstance();

        static if (!is(SuperType == ConcreteType))
        {
            if (!hasOption(options, persistentRegistrationOptions,
                    RegistrationOption.doNotAddConcreteTypeRegistration))
            {
                auto concreteTypeRegistration = register!ConcreteType;
                concreteTypeRegistration.linkTo(newRegistration);
            }
        }

        registrations[registeredType] ~= cast(shared(Registration)) newRegistration;
        return newRegistration;
    }

    private bool hasOption(OptionType)(OptionType options,
        OptionType persistentOptions, OptionType option)
    {
        return ((options | persistentOptions) & option) != 0;
    }

    private OptionType buildFlags(OptionType)(OptionType[] options)
    {
        OptionType flags;
        foreach (option; options)
        {
            flags |= option;
        }
        return flags;
    }

    private Registration getExistingRegistration(TypeInfo registrationType, TypeInfo qualifierType)
    {
        auto existingCandidates = registrationType in registrations;
        if (existingCandidates)
        {
            return getRegistration(cast(Registration[])*existingCandidates, qualifierType);
        }

        return null;
    }

    private Registration getRegistration(Registration[] candidates, TypeInfo concreteType)
    {
        foreach (existingRegistration; candidates)
        {
            if (existingRegistration.instanceType == concreteType)
            {
                return existingRegistration;
            }
        }

        return null;
    }

    /**
     * Resolve dependencies.
     *
     * Dependencies can only resolved using this method if they are registered by concrete type or the only
     * concrete type registered by super type.
     *
     * Resolved dependencies are automatically autowired before being returned.
     *
     * Returns:
     * An instance is returned which is created according to the registration scope with which they are registered.
     *
     * Throws:
     * ResolveException when type is not registered.
     *
     * Examples:
     * Resolve dependencies registered by super type and concrete type:
     * ---
     * class Cat : Animal { ... }
     * class Dog : Animal { ... }
     *
     * container.register!(Animal, Cat);
     * container.register!Dog;
     *
     * container.resolve!Animal;
     * container.resolve!Dog;
     * ---
     * You cannot resolve a dependency when it is registered by multiple super types:
     * ---
     * class Cat : Animal { ... }
     * class Dog : Animal { ... }
     *
     * container.register!(Animal, Cat);
     * container.register!(Animal, Dog);
     *
     * container.resolve!Animal; // Error: multiple candidates for type "Animal"
     * container.resolve!Dog; // Error: No type is registered by concrete type "Dog", only by super type "Animal"
     * ---
     * You need to use the resolve method which allows you to specify a qualifier.
     */
    public RegistrationType resolve(RegistrationType)(
        ResolveOption resolveOptions = ResolveOption.none)
            if (!is(RegistrationType == struct))
    {
        return resolve!(RegistrationType, RegistrationType)(resolveOptions);
    }

    /**
     * Resolve dependencies using a qualifier.
     *
     * Dependencies can only resolved using this method if they are registered by super type.
     *
     * Resolved dependencies are automatically autowired before being returned.
     *
     * Returns:
     * An instance is returned which is created according to the registration scope with which they are registered.
     *
     * Throws:
     * ResolveException when type is not registered or there are multiple candidates available for type.
     *
     * Examples:
     * Resolve dependencies registered by super type:
     * ---
     * class Cat : Animal { ... }
     * class Dog : Animal { ... }
     *
     * container.register!(Animal, Cat);
     * container.register!(Animal, Dog);
     *
     * container.resolve!(Animal, Cat);
     * container.resolve!(Animal, Dog);
     * ---
     */
    public QualifierType resolve(RegistrationType, QualifierType:
        RegistrationType)(ResolveOption resolveOptions = ResolveOption.none)
            if (!is(QualifierType == struct))
    {
        TypeInfo resolveType = typeid(RegistrationType);
        TypeInfo qualifierType = typeid(QualifierType);

        debug (poodinisVerbose)
        {
            writeln("DEBUG: Resolving type " ~ resolveType.toString() ~ " with qualifier " ~ qualifierType.toString());
        }

        auto candidates = resolveType in registrations;
        if (!candidates)
        {
            static if (is(typeof(typeid(QualifierType)) == TypeInfo_Class) && !__traits(isAbstractClass, QualifierType))
            {
                if (hasOption(resolveOptions, persistentResolveOptions, ResolveOption
                        .registerBeforeResolving))
                {
                    register!(RegistrationType, QualifierType)();
                    return resolve!(RegistrationType, QualifierType)(resolveOptions);
                }
            }

            if (hasOption(resolveOptions, persistentResolveOptions,
                    ResolveOption.noResolveException))
            {
                return null;
            }

            throw new ResolveException("Type not registered.", resolveType);
        }

        Registration registration = getQualifiedRegistration(resolveType,
            qualifierType, cast(Registration[])*candidates);

        try
        {
            QualifierType newInstance = resolveAutowiredInstance!QualifierType(registration);
            callPostConstructors(newInstance);
            return newInstance;
        }
        catch (ValueInjectionException e)
        {
            throw new ResolveException(e, resolveType);
        }
    }

    bool isRegistered(RegistrationType)()
    {
        TypeInfo typeInfo = typeid(RegistrationType);
        auto candidates = typeInfo in registrations;
        return candidates !is null;
    }

    private QualifierType resolveAutowiredInstance(QualifierType)(Registration registration)
    {
        QualifierType instance;
        if (!(cast(Registration[]) autowireStack).canFind(registration))
        {
            autowireStack ~= cast(shared(Registration)) registration;
            instance = cast(QualifierType) registration.getInstance(
                new AutowireInstantiationContext());
            autowireStack = autowireStack[0 .. $ - 1];
        }
        else
        {
            auto autowireContext = new AutowireInstantiationContext();
            autowireContext.autowireInstance = false;
            instance = cast(QualifierType) registration.getInstance(autowireContext);
        }
        return instance;
    }

    /**
     * Resolve all dependencies registered to a super type.
     *
     * Returns:
     * An array of autowired instances is returned. The order is undetermined.
     *
     * Examples:
     * ---
     * class Cat : Animal { ... }
     * class Dog : Animal { ... }
     *
     * container.register!(Animal, Cat);
     * container.register!(Animal, Dog);
     *
     * Animal[] animals = container.resolveAll!Animal;
     * ---
     */
    public RegistrationType[] resolveAll(RegistrationType)(
        ResolveOption resolveOptions = ResolveOption.none)
    {
        RegistrationType[] instances;
        TypeInfo resolveType = typeid(RegistrationType);

        auto qualifiedRegistrations = resolveType in registrations;
        if (!qualifiedRegistrations)
        {
            if (hasOption(resolveOptions, persistentResolveOptions,
                    ResolveOption.noResolveException))
            {
                return [];
            }

            throw new ResolveException("Type not registered.", resolveType);
        }

        foreach (registration; cast(Registration[])*qualifiedRegistrations)
        {
            instances ~= resolveAutowiredInstance!RegistrationType(registration);
        }

        return instances;
    }

    private Registration getQualifiedRegistration(TypeInfo resolveType,
        TypeInfo qualifierType, Registration[] candidates)
    {
        if (resolveType == qualifierType)
        {
            if (candidates.length > 1)
            {
                string candidateList = candidates.toConcreteTypeListString();
                throw new ResolveException(
                    "Multiple qualified candidates available: " ~ candidateList ~ ". Please use a qualifier.",
                    resolveType);
            }

            return candidates[0];
        }

        return getRegistration(candidates, qualifierType);
    }

    private void callPostConstructors(Type)(Type instance)
    {
        foreach (memberName; __traits(allMembers, Type))
        {
            mixin(createImportsString!Type);
            enum QualifiedName = fullyQualifiedName!Type ~ `.` ~ memberName;
            static if (__traits(compiles, __traits(getProtection, __traits(getMember, instance, memberName)))
                && __traits(getProtection, __traits(getMember, instance, memberName)) == "public"
                && isFunction!(mixin(QualifiedName))
                && hasUDA!(__traits(getMember, instance, memberName), PostConstruct))
            {
                __traits(getMember, instance, memberName)();
            }
        }
    }

    /**
     * Clears all dependency registrations managed by this container.
     */
    public void clearAllRegistrations()
    {
        foreach (registrationsOfType; registrations)
        {
            callPreDestructorsOfRegistrations(registrationsOfType);
        }
        registrations.destroy();
    }

    /**
     * Removes a registered dependency by type.
     *
     * A dependency can be removed either by super type or concrete type, depending on how they are registered.
     *
     * Examples:
     * ---
     * container.removeRegistration!Animal;
     * ---
     */
    public void removeRegistration(RegistrationType)()
    {
        auto registrationsOfType = *(typeid(RegistrationType) in registrations);
        callPreDestructorsOfRegistrations(registrationsOfType);
        registrations.remove(typeid(RegistrationType));
    }

    private void callPreDestructorsOfRegistrations(shared(Registration[]) registrations)
    {
        foreach (registration; registrations)
        {
            Registration unsharedRegistration = cast(Registration) registration;
            if (unsharedRegistration.preDestructor !is null)
            {
                unsharedRegistration.preDestructor()();
            }
        }
    }

    /**
     * Apply persistent registration options which will be used everytime register() is called.
     */
    public void setPersistentRegistrationOptions(RegistrationOption options)
    {
        persistentRegistrationOptions = options;
    }

    /**
     * Unsets all applied persistent registration options
     */
    public void unsetPersistentRegistrationOptions()
    {
        persistentRegistrationOptions = RegistrationOption.none;
    }

    /**
     * Apply persistent resolve options which will be used everytime resolve() is called.
     */
    public void setPersistentResolveOptions(ResolveOption options)
    {
        persistentResolveOptions = options;
    }

    /**
     * Unsets all applied persistent resolve options
     */
    public void unsetPersistentResolveOptions()
    {
        persistentResolveOptions = ResolveOption.none;
    }

}