feat: markOrRemoveGlobalCompletions! (disabled by default)

Author: zardoy

src/configurationType.ts

@@ -44,6 +44,10 @@ export type Configuration = {
      * @default true
      *  */
     'removeUselessFunctionProps.enable': boolean
+    /**
+     * @default disable
+     */
+    'removeOrMarkGlobalCompletions.action': 'disable' | 'mark' | 'remove'
     /**
      * Useful for Number types.
      * Patch `toString()`: Removes arg tabstop

typescript/src/completions/markOrRemoveGlobalCompletions.ts

@@ -0,0 +1,30 @@
+import { compact } from '@zardoy/utils'
+import { GetConfig } from '../types'
+
+const isLibCompletion = (symbol: ts.Symbol) => {
+    return symbol.declarations?.[0]?.getSourceFile().fileName.includes('node_modules/typescript/lib/')
+}
+
+export default (entries: ts.CompletionEntry[], position: number, languageService: ts.LanguageService, c: GetConfig) => {
+    const action = c('removeOrMarkGlobalCompletions.action')
+    if (action === 'disable') return
+
+    return compact(
+        entries.map(entry => {
+            if (entry.sourceDisplay) return entry
+            const symbol = entry['symbol'] as ts.Symbol | undefined
+            if (!symbol) return entry
+            if (!isLibCompletion(symbol)) return entry
+            if (action === 'remove') return undefined
+            return {
+                ...entry,
+                sourceDisplay: [
+                    {
+                        kind: 'text',
+                        text: 'global',
+                    },
+                ],
+            }
+        }),
+    )
+}

typescript/src/completionsAtPosition.ts

@@ -19,6 +19,7 @@ import jsdocDefault from './completions/jsdocDefault'
 import defaultHelpers from './completions/defaultHelpers'
 import objectLiteralCompletions from './completions/objectLiteralCompletions'
 import filterJsxElements from './completions/filterJsxComponents'
+import markOrRemoveGlobalCompletions from './completions/markOrRemoveGlobalCompletions'
 
 export type PrevCompletionMap = Record<string, { originalName?: string; documentationOverride?: string | ts.SymbolDisplayPart[] }>
 
@@ -182,6 +183,7 @@ export const getCompletionsAtPosition = (
         })
     }
 
+    prior.entries = markOrRemoveGlobalCompletions(prior.entries, position, languageService, c) ?? prior.entries
     if (exactNode) prior.entries = filterJsxElements(prior.entries, exactNode, position, languageService, c) ?? prior.entries
 
     if (c('correctSorting.enable')) {