Port checkSourceElement, checkExpression, checkBinaryExpression (#90)

Author: ahejlsberg

internal/ast/ast.go

@@ -170,24 +170,42 @@ func (node *Node) Expression() *Node {
 		return node.AsAsExpression().Expression
 	case KindSatisfiesExpression:
 		return node.AsSatisfiesExpression().Expression
+	case KindTypeOfExpression:
+		return node.AsTypeOfExpression().Expression
 	case KindSpreadAssignment:
 		return node.AsSpreadAssignment().Expression
+	case KindSpreadElement:
+		return node.AsSpreadElement().Expression
 	case KindTemplateSpan:
 		return node.AsTemplateSpan().Expression
+	case KindDeleteExpression:
+		return node.AsDeleteExpression().Expression
+	case KindVoidExpression:
+		return node.AsVoidExpression().Expression
+	case KindAwaitExpression:
+		return node.AsAwaitExpression().Expression
+	case KindYieldExpression:
+		return node.AsYieldExpression().Expression
 	case KindForInStatement, KindForOfStatement:
 		return node.AsForInOrOfStatement().Expression
 	}
 	panic("Unhandled case in Node.Expression")
 }
 
 func (node *Node) Arguments() []*Node {
+	var arguments *NodeList
 	switch node.Kind {
 	case KindCallExpression:
-		return node.AsCallExpression().Arguments.Nodes
+		arguments = node.AsCallExpression().Arguments
 	case KindNewExpression:
-		return node.AsNewExpression().Arguments.Nodes
+		arguments = node.AsNewExpression().Arguments
+	default:
+		panic("Unhandled case in Node.Arguments")
 	}
-	panic("Unhandled case in Node.Arguments")
+	if arguments != nil {
+		return arguments.Nodes
+	}
+	return nil
 }
 
 func (node *Node) ModifierFlags() ModifierFlags {
@@ -677,6 +695,15 @@ func (n *Node) AsJSDocNameReference() *JSDocNameReference {
 func (n *Node) AsTemplateLiteralTypeNode() *TemplateLiteralTypeNode {
 	return n.data.(*TemplateLiteralTypeNode)
 }
+func (n *Node) AsVoidExpression() *VoidExpression {
+	return n.data.(*VoidExpression)
+}
+func (n *Node) AsAwaitExpression() *AwaitExpression {
+	return n.data.(*AwaitExpression)
+}
+func (n *Node) AsYieldExpression() *YieldExpression {
+	return n.data.(*YieldExpression)
+}
 
 // NodeData
 
@@ -1186,8 +1213,15 @@ func (node *ForInOrOfStatement) ForEachChild(v Visitor) bool {
 	return visit(v, node.AwaitModifier) || visit(v, node.Initializer) || visit(v, node.Expression) || visit(v, node.Statement)
 }
 
+func IsForInStatement(node *Node) bool {
+	return node.Kind == KindForInStatement
+}
+func IsForOfStatement(node *Node) bool {
+	return node.Kind == KindForOfStatement
+}
+
 func IsForInOrOfStatement(node *Node) bool {
-	return node.Kind == KindForInStatement || node.Kind == KindForOfStatement
+	return IsForInStatement(node) || IsForOfStatement(node)
 }
 
 // BreakStatement
@@ -1859,6 +1893,10 @@ func (node *EnumMember) Name() *DeclarationName {
 	return node.name
 }
 
+func IsEnumMember(node *Node) bool {
+	return node.Kind == KindEnumMember
+}
+
 // EnumDeclaration
 
 type EnumDeclaration struct {
@@ -2654,6 +2692,10 @@ func (f *NodeFactory) NewOmittedExpression() *Node {
 	return f.newNode(KindOmittedExpression, &OmittedExpression{})
 }
 
+func IsOmittedExpression(node *Node) bool {
+	return node.Kind == KindOmittedExpression
+}
+
 // KeywordExpression
 
 type KeywordExpression struct {
@@ -3223,6 +3265,10 @@ func (node *TaggedTemplateExpression) ForEachChild(v Visitor) bool {
 	return visit(v, node.Tag) || visit(v, node.QuestionDotToken) || visitNodeList(v, node.TypeArguments) || visit(v, node.Template)
 }
 
+func IsTaggedTemplateExpression(node *Node) bool {
+	return node.Kind == KindTaggedTemplateExpression
+}
+
 // ParenthesizedExpression
 
 type ParenthesizedExpression struct {
@@ -3870,6 +3916,10 @@ func (node *TypeLiteralNode) ForEachChild(v Visitor) bool {
 	return visitNodeList(v, node.Members)
 }
 
+func IsTypeLiteralNode(node *Node) bool {
+	return node.Kind == KindTypeLiteral
+}
+
 // TupleTypeNode
 
 type TupleTypeNode struct {