feat(property): now can match more new properties and add to results …

…if it is possible
nanoexpress · Oct 18, 2021 · d9a9bb6 · d9a9bb6
1 parent f9487ab
commit d9a9bb6
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 2 deletions.
diff --git a/src/finders/request/index.ts b/src/finders/request/index.ts
@@ -71,6 +71,16 @@ export default function caseRequestModeFinder(
     if (res !== undefined) {
       return res;
     }
+
+    if (_key && res !== null) {
+      return {
+        link: _key,
+        linked: false,
+        line_index: index,
+        key: _key,
+        mode: 'property'
+      };
+    }
   } else {
     // Method matching
     const reqIndex = line.indexOf('req');

diff --git a/src/finders/request/property.ts b/src/finders/request/property.ts
@@ -6,7 +6,7 @@ export default function requestMethodFinder(
   _key: string,
   line: string,
   index: number
-): IBlock | void {
+): IBlock | null | void {
   let key;
   let link;
   const extracted = variableKeyExtractor(line);
@@ -29,7 +29,7 @@ export default function requestMethodFinder(
   }
   if (_key.includes('getMethod(') || _key.includes('getUrl(')) {
     // uWebSockets.js has native `getMethod` and `getUrl` support
-    return undefined;
+    return null;
   }
   if (
     (_key === 'req;' || _key === 'request;') &&

diff --git a/tests/analyze.spec.js b/tests/analyze.spec.js
@@ -115,6 +115,21 @@ describe('property parse', () => {
       }
     ]);
   });
+  it('req.ip', () => {
+    expect(
+      analyze((request, response) => {
+        response.end(request.ip);
+      })
+    ).toEqual([
+      {
+        key: 'ip',
+        line_index: 1,
+        link: 'ip',
+        linked: false,
+        mode: 'property'
+      }
+    ]);
+  });
 });
 
 describe('request property parse', () => {