feat: add simple query support

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stanimirovv/tsoogle",
-  "version": "1.1.0",
+  "version": "1.3.0",
   "description": "Find functions or methods by approximate signature - return type, argument types or both. Supports optional arguments, rest arguments and partial type checking.",
   "main": "dist/index.js",
   "bin": "dist/index.js",

src/lexer.ts

@@ -3,6 +3,8 @@ import type { SearchQuery } from './searchQuery.type'
 export function parse (input: string): SearchQuery {
   input = input.replace(/\s+/g, '')
 
+  input = simpleQueryToFullQuery(input)
+
   const fnType = input.split(':')[0]
   const kind = calculateKind(fnType)
 
@@ -30,6 +32,22 @@ export function parseTypeQuery (input: string): string[] {
   return input.replace('{', '').replace('}', '').split('&')
 }
 
+function simpleQueryToFullQuery (input: string): string {
+  if (!input.includes(':') && !input.includes(',')) {
+    input = `:${input}`
+  }
+
+  if (!input.includes('?') && !input.includes(',')) {
+    input = `${input}?`
+  }
+
+  if (input.includes(',') && !input.includes(':') && !input.includes('?')) {
+    return `:*?${input}`
+  }
+
+  return input
+}
+
 function calculateKind (match: string): 'function' | 'method' | 'both' {
   if (match === '') {
     return 'both'

tests/lexer.spec.ts

@@ -36,4 +36,49 @@ describe('parseTypeString', () => {
       parameterTypes: [['Product'], ['Service', 'number']]
     })
   })
+
+  it('should correctly parse "number"', () => {
+    const result = parse('number')
+    expect(result).toEqual({
+      kind: 'both',
+      returnTypes: ['number'],
+      parameterTypes: []
+    })
+  })
+
+  it('should correctly parse ":number"', () => {
+    const result = parse(':number')
+    expect(result).toEqual({
+      kind: 'both',
+      returnTypes: ['number'],
+      parameterTypes: []
+    })
+  })
+
+  it('should correctly parse "number?"', () => {
+    const result = parse(':number')
+    expect(result).toEqual({
+      kind: 'both',
+      returnTypes: ['number'],
+      parameterTypes: []
+    })
+  })
+
+  it('should correctly parse "number?number"', () => {
+    const result = parse('number?number')
+    expect(result).toEqual({
+      kind: 'both',
+      returnTypes: ['number'],
+      parameterTypes: [['number']]
+    })
+  })
+
+  it('should correctly parse "number,number"', () => {
+    const result = parse('number,number')
+    expect(result).toEqual({
+      kind: 'both',
+      returnTypes: ['*'],
+      parameterTypes: [['number'], ['number']]
+    })
+  })
 })