feat: Support @field() option 'complexity' through comment line

unlight · Jan 18, 2025 · ef8c327 · ef8c327
1 parent 28eb34c
commit ef8c327
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -372,6 +372,7 @@ For example:
 model Product {
   /// Old description
   /// @deprecated Use new name instead
+  /// @complexity 1
   oldName String
 }
 ```
@@ -388,6 +389,7 @@ export class Product {
   @Field(() => String, {
     description: 'Old description',
     deprecationReason: 'Use new name instead',
+    complexity: 1
   })
   oldName: string;
 }

diff --git a/src/helpers/object-settings.ts b/src/helpers/object-settings.ts
@@ -184,6 +184,16 @@ function createSettingElement({
 
     return result;
   }
+
+  if (line.startsWith('@complexity')) {
+    let n = Number.parseInt(trim(line.slice(11)));
+    if(n !== n || n < 1) n = 1;
+    fieldElement.arguments!['complexity'] = n;
+
+    result.element = fieldElement;
+
+    return result;
+  }
 
   const name = match?.groups?.name;