From 36da94a623924f425d77f669624dbcdb9641b9cb Mon Sep 17 00:00:00 2001
From: Shinwoo PARK <shinwoo.park@psw.kr>
Date: Fri, 15 Dec 2023 01:37:35 +0900
Subject: [PATCH] feat: add Returns decorator

---
 src/reflected.ts | 45 +++++++++++++++++++++++++++++++++++++--------
 1 file changed, 37 insertions(+), 8 deletions(-)

diff --git a/src/reflected.ts b/src/reflected.ts
index 62327d2..ce08c66 100644
--- a/src/reflected.ts
+++ b/src/reflected.ts
@@ -1,3 +1,11 @@
+export type Path = string | string[];
+export type Paths = Path[];
+
+export interface IncludeExcludePath {
+  includePath?: Paths;
+  excludePath?: Paths;
+}
+
 export interface LoggedParamReflectData {
   name: string;
   index: number;
@@ -12,10 +20,16 @@ export interface ScopeKeyReflectData {
   priority?: number;
 }
 
+export interface ReturnsReflectData {
+  name: string;
+  path: string;
+}
+
 export const scopedLogger = Symbol("nlogdec-scopedLogger");
 export const loggedParam = Symbol("nlogdec-loggedParam");
 export const scopeKey = Symbol("nlogdec-scopeKey");
 export const forceScopeKey = Symbol("nlogdec-forceScopeKey");
+export const returns = Symbol("nlogdec-returns");
 
 export function InjectLogger(
   target: any,
@@ -25,13 +39,7 @@ export function InjectLogger(
   Reflect.defineMetadata(scopedLogger, parameterIndex, target, propertyKey);
 }
 
-export function LoggedParam(
-  name: string,
-  options?: {
-    includePath?: (string | string[])[];
-    excludePath?: (string | string[])[];
-  }
-) {
+export function LoggedParam(name: string, options?: IncludeExcludePath) {
   return (
     target: any,
     propertyKey: string | symbol,
@@ -65,7 +73,7 @@ export function LoggedParam(
 
 export function ScopeKey(
   name: string,
-  options?: { path?: string | string[]; priority?: number }
+  options?: { path?: Path; priority?: number }
 ) {
   return (
     target: any,
@@ -90,6 +98,27 @@ export function ScopeKey(
   };
 }
 
+export function Returns<F extends Array<any>, R>(namePaths?: {
+  [name: string]: string;
+}) {
+  return (
+    _target: any,
+    _key: string | symbol,
+    descriptor: TypedPropertyDescriptor<(...args: F) => Promise<R>>
+  ) => {
+    Reflect.defineMetadata(
+      returns,
+      namePaths
+        ? Object.entries(namePaths).reduce<ReturnsReflectData[]>(
+            (prev, curr) => [...prev, { name: curr[0], path: curr[1] }],
+            []
+          )
+        : true,
+      descriptor.value
+    );
+  };
+}
+
 export function ShouldScoped(
   _target: any,
   _key: string,