feat(container): add decorate to container

src/Container.spec.ts

@@ -22,6 +22,21 @@ interface Dep3 {
 const Dep3 = Binding<Dep3>('Dep3');
 
 describe('Container', () => {
+  describe('#decorate', () => {
+    it('adds dependencies in decorator', async () => {
+      const container = Container.create()
+        .bindFactory(Dep1, async c => ({
+          a: (await c.resolve(Dep2)).b.toString(),
+        }))
+        .decorate(c =>
+          c.bindFactory(Dep2, () => ({b: 2})).bindConstant(Dep3, {c: true}),
+        );
+
+      expect(await container.resolve(Dep1)).toHaveProperty('a', '2');
+      expect(await container.resolve(Dep3)).toHaveProperty('c', true);
+    });
+  });
+
   describe('#bindFactory', () => {
     it('calls the factory with the latest container container', async () => {
       const container = Container.create()

src/Container.ts

@@ -17,6 +17,12 @@ export class Container<Service> {
     private readonly chain: Array<Resolution<Service>>,
   ) {}
 
+  decorate<S>(
+    decorator: Container.Decorator<S, Service>,
+  ): Container<Service | S> {
+    return decorator(this);
+  }
+
   bindFactory<S>(
     {Tag: tag}: Binding<S>,
     factory: Resolution<S>['factory'],
@@ -119,3 +125,9 @@ export interface Resolution<Service> {
   tag: Binding.Tag<Service>;
   factory: (container: Container<any>) => Service | Promise<Service>;
 }
+
+export namespace Container {
+  export type Decorator<Out, In> = (
+    container: Container<In>,
+  ) => Container<In | Out>;
+}

src/Module.ts

@@ -1,6 +1,9 @@
 import {Binding} from './Binding';
 import {Container, Resolution} from './Container';
 
+/**
+ * @deprecated Use container Decorators instead.
+ */
 export class Module<Export, Service = any> {
   static create<Service>(
     container: Container<Service>,