Add missing visitor of template literal type (#40738)

* Add missing visitor of template literal type * make linter happy
microsoft · Sep 24, 2020 · fc03982 · fc03982
1 parent 3b9eb1e
commit fc03982
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/src/compiler/factory/nodeTests.ts b/src/compiler/factory/nodeTests.ts
@@ -233,6 +233,10 @@ namespace ts {
         return node.kind === SyntaxKind.ImportType;
     }
 
+    export function isTemplateLiteralTypeSpan(node: Node): node is TemplateLiteralTypeSpan {
+        return node.kind === SyntaxKind.TemplateLiteralTypeSpan;
+    }
+
     // Binding patterns
 
     export function isObjectBindingPattern(node: Node): node is ObjectBindingPattern {

diff --git a/src/compiler/visitorPublic.ts b/src/compiler/visitorPublic.ts
@@ -573,6 +573,16 @@ namespace ts {
                 return factory.updateLiteralTypeNode(<LiteralTypeNode>node,
                     nodeVisitor((<LiteralTypeNode>node).literal, visitor, isExpression));
 
+            case SyntaxKind.TemplateLiteralType:
+                return factory.updateTemplateLiteralType(<TemplateLiteralTypeNode>node,
+                    nodeVisitor((<TemplateLiteralTypeNode>node).head, visitor, isTemplateHead),
+                    nodesVisitor((<TemplateLiteralTypeNode>node).templateSpans, visitor, isTemplateLiteralTypeSpan));
+
+            case SyntaxKind.TemplateLiteralTypeSpan:
+                return factory.updateTemplateLiteralTypeSpan(<TemplateLiteralTypeSpan>node,
+                    nodeVisitor((<TemplateLiteralTypeSpan>node).type, visitor, isTypeNode),
+                    nodeVisitor((<TemplateLiteralTypeSpan>node).literal, visitor, isTemplateMiddleOrTemplateTail));
+
             // Binding patterns
             case SyntaxKind.ObjectBindingPattern:
                 return factory.updateObjectBindingPattern(<ObjectBindingPattern>node,

diff --git a/tests/baselines/reference/api/tsserverlibrary.d.ts b/tests/baselines/reference/api/tsserverlibrary.d.ts
@@ -4368,6 +4368,7 @@ declare namespace ts {
     function isMappedTypeNode(node: Node): node is MappedTypeNode;
     function isLiteralTypeNode(node: Node): node is LiteralTypeNode;
     function isImportTypeNode(node: Node): node is ImportTypeNode;
+    function isTemplateLiteralTypeSpan(node: Node): node is TemplateLiteralTypeSpan;
     function isObjectBindingPattern(node: Node): node is ObjectBindingPattern;
     function isArrayBindingPattern(node: Node): node is ArrayBindingPattern;
     function isBindingElement(node: Node): node is BindingElement;

diff --git a/tests/baselines/reference/api/typescript.d.ts b/tests/baselines/reference/api/typescript.d.ts
@@ -4368,6 +4368,7 @@ declare namespace ts {
     function isMappedTypeNode(node: Node): node is MappedTypeNode;
     function isLiteralTypeNode(node: Node): node is LiteralTypeNode;
     function isImportTypeNode(node: Node): node is ImportTypeNode;
+    function isTemplateLiteralTypeSpan(node: Node): node is TemplateLiteralTypeSpan;
     function isObjectBindingPattern(node: Node): node is ObjectBindingPattern;
     function isArrayBindingPattern(node: Node): node is ArrayBindingPattern;
     function isBindingElement(node: Node): node is BindingElement;