feat: enableFileDefinitions (must have to enable, disabled by default) useful for path.join

feat: `removeModuleFileDefinitions` (disabled by default, enable if you need it) removes TS module declaration e.g. *.css

Author: zardoy

src/configurationType.ts

@@ -123,6 +123,19 @@ export type Configuration = {
      * @default true
      *  */
     // 'importUpDefinition.enable': boolean
+    /**
+     * Remove definitions for TS module declarations e.g. *.css
+     * Enable it if your first definition that receives focus is TS module declaration instead of target file itself
+     * Might be really really useful in some cases
+     * @default false
+     */
+    removeModuleFileDefinitions: boolean
+    /**
+     * Enable definitions for strings that appears to be paths (relatively to file)
+     * Also must have and should be enabled if you work with path.join a lot
+     * @default false
+     */
+    enableFileDefinitions: boolean
     /**
      * @default true
      * */

typescript/src/definitions.ts

@@ -1,9 +1,41 @@
+import { join } from 'path'
 import { GetConfig } from './types'
+import { findChildContainingExactPosition } from './utils'
 
 export default (proxy: ts.LanguageService, info: ts.server.PluginCreateInfo, c: GetConfig) => {
     proxy.getDefinitionAndBoundSpan = (fileName, position) => {
         const prior = info.languageService.getDefinitionAndBoundSpan(fileName, position)
-        if (!prior) return
+        if (!prior) {
+            if (c('enableFileDefinitions')) {
+                const sourceFile = info.languageService.getProgram()!.getSourceFile(fileName)!
+                const node = findChildContainingExactPosition(sourceFile, position)
+                if (node && ts.isStringLiteral(node) && ['./', '../'].some(str => node.text.startsWith(str))) {
+                    const file = join(fileName, '..', node.text)
+                    if (info.languageServiceHost.fileExists?.(file)) {
+                        const start = node.pos + node.getLeadingTriviaWidth() + 1 // + 1 for quote
+                        const textSpan = {
+                            start,
+                            length: node.end - start - 1,
+                        }
+                        return {
+                            textSpan,
+                            definitions: [
+                                {
+                                    containerKind: undefined as any,
+                                    containerName: '',
+                                    name: '',
+                                    fileName: file,
+                                    textSpan: { start: 0, length: 0 },
+                                    kind: ts.ScriptElementKind.moduleElement,
+                                    contextSpan: { start: 0, length: 0 },
+                                },
+                            ],
+                        }
+                    }
+                }
+            }
+            return
+        }
         if (__WEB__) {
             // let extension handle it
             // TODO failedAliasResolution
@@ -31,7 +63,18 @@ export default (proxy: ts.LanguageService, info: ts.server.PluginCreateInfo, c:
                 const isFcDef = fileName.endsWith('node_modules/@types/react/index.d.ts') && containerName === 'FunctionComponent'
                 return !isFcDef
             })
+            // 11
         }
+
+        if (
+            c('removeModuleFileDefinitions') &&
+            prior.definitions?.length === 1 &&
+            firstDef.kind === ts.ScriptElementKind.moduleElement &&
+            firstDef.name.slice(1, -1).startsWith('*.')
+        ) {
+            return
+        }
+
         return prior
     }
 }