// Copyright (c) Microsoft Corporation. All rights reserved.

// Licensed under the MIT License.

import { EventEmitter } from 'events';

import { Container, decorate, injectable, interfaces } from 'inversify';

import { traceWarning } from '../common/logger';

import { Abstract, IServiceContainer, Newable } from './types';

// This needs to be done once, hence placed in a common location.

// Used by UnitTestSockerServer and also the extension unit tests.

// Place within try..catch, as this can only be done once (it's

// possible another extension would perform this before our extension).

try {

decorate(injectable(), EventEmitter);

} catch (ex) {

traceWarning('Failed to decorate EventEmitter for DI (possibly already decorated by another Extension)', ex);

}

@injectable()

export class ServiceContainer implements IServiceContainer {

constructor(private container: Container) {}

public get<T>(serviceIdentifier: interfaces.ServiceIdentifier<T>, name?: string | number | symbol): T {

return name ? this.container.getNamed<T>(serviceIdentifier, name) : this.container.get<T>(serviceIdentifier);

}

public getAll<T>(

serviceIdentifier: string | symbol | Newable<T> | Abstract<T>,

name?: string | number | symbol | undefined

): T[] {

return name

? this.container.getAllNamed<T>(serviceIdentifier, name)

: this.container.getAll<T>(serviceIdentifier);

}

}