feature(): add scope option to the controller decorator

Author: kamilmysliwiec

packages/common/decorators/core/controller.decorator.ts

@@ -1,13 +1,30 @@
-import { PATH_METADATA } from '../../constants';
-import { isUndefined } from '../../utils/shared.utils';
+import { PATH_METADATA, SCOPE_OPTIONS_METADATA } from '../../constants';
+import { isString, isUndefined } from '../../utils/shared.utils';
+import { ScopeOptions } from './../../interfaces/scope-options.interface';
+
+export interface ControllerOptions extends ScopeOptions {
+  path?: string;
+}
 
 /**
  * Defines the controller. Controller can inject dependencies through constructor.
  * Those dependencies have to belong to the same module.
  */
-export function Controller(prefix?: string): ClassDecorator {
-  const path = isUndefined(prefix) ? '/' : prefix;
+export function Controller();
+export function Controller(prefix: string);
+export function Controller(options: ControllerOptions);
+export function Controller(
+  prefixOrOptions?: string | ControllerOptions,
+): ClassDecorator {
+  const defaultPath = '/';
+  const [path, scopeOptions] = isUndefined(prefixOrOptions)
+    ? [defaultPath, undefined]
+    : isString(prefixOrOptions)
+      ? [prefixOrOptions, undefined]
+      : [prefixOrOptions.path || defaultPath, { scope: prefixOrOptions.scope }];
+
   return (target: object) => {
     Reflect.defineMetadata(PATH_METADATA, path, target);
+    Reflect.defineMetadata(SCOPE_OPTIONS_METADATA, scopeOptions, target);
   };
 }

packages/core/injector/module.ts

@@ -348,6 +348,7 @@ export class Module {
         metatype: controller,
         instance: null,
         isResolved: false,
+        scope: this.getClassScope(controller),
         host: this,
       }),
     );

packages/core/test/injector/module.spec.ts

@@ -1,4 +1,4 @@
-import { Scope } from '@nestjs/common';
+import { Controller, Scope } from '@nestjs/common';
 import { expect } from 'chai';
 import * as sinon from 'sinon';
 import { Injectable } from '../../../common';
@@ -30,14 +30,16 @@ describe('Module', () => {
     const setSpy = sinon.spy(collection, 'set');
     (module as any)._controllers = collection;
 
+    @Controller({ scope: Scope.REQUEST })
     class Test {}
+
     module.addController(Test);
     expect(setSpy.getCall(0).args).to.deep.equal([
       'Test',
       new InstanceWrapper({
         host: module,
         name: 'Test',
-        scope: 0,
+        scope: Scope.REQUEST,
         metatype: Test,
         instance: null,
         isResolved: false,