Handle AstExprVarargs

Author: JohnnyMorganz

batteries/codemod/ast_types.luau

@@ -74,6 +74,8 @@ export type AstExprGlobal = {
 	name: Token,
 }
 
+export type AstExprVarargs = Token<"..."> & { tag: "vararg" }
+
 export type AstExprCall = {
 	tag: "call",
 	func: AstExpr, -- TODO: stricter?
@@ -158,6 +160,7 @@ export type AstExpr =
 	| AstExprConstantString
 	| AstExprLocal
 	| AstExprGlobal
+	| AstExprVarargs
 	| AstExprCall
 	| AstExprIndexName
 	| AstExprIndexExpr

batteries/codemod/visitor.luau

@@ -26,6 +26,7 @@ type Visitor = {
 	visitBoolean: (T.AstExprConstantBool) -> boolean,
 	visitNumber: (T.AstExprConstantNumber) -> boolean,
 	visitLocal: (T.AstLocal) -> boolean,
+	visitVarargs: (T.AstExprVarargs) -> boolean,
 }
 
 local function alwaysVisit(...: any)
@@ -56,6 +57,7 @@ local defaultVisitor: Visitor = {
 	visitBoolean = alwaysVisit :: any,
 	visitNumber = alwaysVisit :: any,
 	visitLocal = alwaysVisit :: any,
+	visitVarargs = alwaysVisit :: any,
 }
 
 local function exhaustiveMatch(value: never): never
@@ -160,6 +162,12 @@ local function visitGlobal(node: T.AstExprGlobal, visitor: Visitor)
 	end
 end
 
+local function visitVarargs(node: T.AstExprVarargs, visitor: Visitor)
+	if visitor.visitVarargs(node) then
+		visitToken(node)
+	end
+end
+
 local function visitCall(node: T.AstExprCall, visitor: Visitor)
 	if visitor.visitCall(node) then
 		visitExpression(node.func, visitor)
@@ -278,6 +286,8 @@ function visitExpression(expression: T.AstExpr, visitor: Visitor)
 		visitLocalReference(expression, visitor)
 	elseif expression.tag == "global" then
 		visitGlobal(expression, visitor)
+	elseif expression.tag == "vararg" then
+		visitVarargs(expression, visitor)
 	elseif expression.tag == "call" then
 		visitCall(expression, visitor)
 	elseif expression.tag == "unary" then

luau/src/luau.cpp

@@ -571,9 +571,7 @@ struct AstSerialize : public Luau::AstVisitor
 
     void serialize(Luau::AstExprVarargs* node)
     {
-        lua_rawcheckstack(L, 2);
-        lua_createtable(L, 0, preambleSize);
-
+        serializeToken(node->location.begin, "...", preambleSize);
         serializeNodePreamble(node, "vararg");
     }