feat: Omit<..., ""> suggestions

Author: zardoy

typescript/src/completions/additionalTypesSuggestions.ts

@@ -1,7 +1,33 @@
-export default (entries: ts.CompletionEntry[], program: ts.Program, leftNode: ts.Node) => {
-    if (!ts.isStringLiteral(leftNode) || !ts.isTypeParameterDeclaration(leftNode.parent.parent) || leftNode.parent.parent.default !== leftNode.parent) return
+export default (entries: ts.CompletionEntry[], program: ts.Program, node: ts.Node) => {
+    if (!ts.isStringLiteral(node)) return
     const typeChecker = program.getTypeChecker()
-    const { constraint } = leftNode.parent.parent
+    if (ts.isLiteralTypeNode(node.parent)) {
+        node = node.parent
+        const previousValues: string[] = []
+        if (node.parent.kind === ts.SyntaxKind.UnionType) {
+            ;(node.parent as ts.UnionTypeNode).types.forEach(type => {
+                if (ts.isLiteralTypeNode(type) && ts.isStringLiteral(type.literal)) {
+                    previousValues.push(type.literal.text)
+                }
+            })
+            node = node.parent
+        }
+        if (ts.isTypeReferenceNode(node.parent) && node.parent.typeName.getText() === 'Omit' && node.parent.typeArguments?.[1] === node) {
+            // Omit<..., '|'> suggestions
+            const type = typeChecker.getTypeFromTypeNode(node.parent.typeArguments[0]!)
+            type.getProperties().forEach(({ name }) => {
+                if (previousValues.includes(name)) return
+                entries.push({
+                    name,
+                    kind: ts.ScriptElementKind.string,
+                    sortText: '',
+                })
+            })
+            return
+        }
+    }
+    if (!ts.isTypeParameterDeclaration(node.parent) || node.parent.default !== node) return
+    const { constraint } = node.parent
     if (!constraint) return
     const type = typeChecker.getTypeAtLocation(constraint)
     if (!(type.flags & ts.TypeFlags.Union)) return

typescript/src/completionsAtPosition.ts

@@ -96,8 +96,8 @@ export const getCompletionsAtPosition = (
         }
         // #endregion
     }
-    if (leftNode && !hasSuggestions && ensurePrior() && prior) {
-        prior.entries = additionalTypesSuggestions(prior.entries, program, leftNode) ?? prior.entries
+    if (node && !hasSuggestions && ensurePrior() && prior) {
+        prior.entries = additionalTypesSuggestions(prior.entries, program, node) ?? prior.entries
     }
     const addSignatureAccessCompletions = hasSuggestions ? [] : indexSignatureAccessCompletions(position, node, scriptSnapshot, sourceFile, program)
     if (addSignatureAccessCompletions.length && ensurePrior() && prior) {

typescript/src/globals.d.ts

@@ -3,6 +3,9 @@ import('ts-expose-internals')
 declare let ts: typeof import('typescript')
 declare let tsFull: typeof import('typescript-full')
 
+declare type FullChecker = import('typescript-full').TypeChecker
+declare type FullSourceFile = import('typescript-full').SourceFile
+
 // declare type ts = import('typescript')
 // export {}
 // export * from 'typescript'