create: custom decorator

cacti23 · May 13, 2022 · e6e2f7e · e6e2f7e
1 parent c07c225
commit e6e2f7e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/auth/decorator/get-user.decorator.ts b/src/auth/decorator/get-user.decorator.ts
@@ -0,0 +1,11 @@
+import { createParamDecorator, ExecutionContext } from '@nestjs/common';
+
+export const GetUser = createParamDecorator(
+  (data: string | undefined, ctx: ExecutionContext) => {
+    const request = ctx.switchToHttp().getRequest();
+    if (data) {
+      return request.user[data];
+    }
+    return request.user;
+  },
+);
diff --git a/src/auth/decorator/index.ts b/src/auth/decorator/index.ts
@@ -0,0 +1 @@
+export * from './get-user.decorator';