import * as DI from './di'
import {repositoryMap} from '../core/exports'
import {ClassType} from '../core/utils/classtype';

let repoSource = function (target: ClassType<any>) {
    let repo;
    if (!target) {
        return null;
    }
    Object.keys(repositoryMap())
        .forEach(x => {
            //// TODO: Decide repository map will hold the prototype or the function
            var path;
            if (target.prototype) {
                path = target.prototype.path;
            }
            else if (target['default']) {
                path = target['default'].prototype.path;
            }
            if (path === x) {
                repo = repositoryMap()[x].repo;
            }
            //if (repositoryMap()[x].fn === target || <Function>repositoryMap()[x].fn === target.prototype|| (target.prototype && x === target.prototype.path)) {
            //    repo = repositoryMap()[x].repo;
            //}
        });
    return repo;
}

DI.Container.addSource(repoSource);

export let Container: DI.IContainer = DI.Container;