Handle numbers

Author: JohnnyMorganz

examples/codemods/identity.luau

@@ -11,3 +11,6 @@ end
 codemod.apply("examples/a.luau", identityTransform)
 codemod.apply("examples/b.luau", identityTransform)
 codemod.apply("examples/json.luau", identityTransform)
+codemod.apply("examples/main.luau", identityTransform)
+codemod.apply("examples/parsing.luau", identityTransform)
+codemod.apply("examples/time_example.luau", identityTransform)

luau/src/luau.cpp

@@ -489,9 +489,9 @@ struct AstSerialize : public Luau::AstVisitor
 
     void serialize(Luau::AstExprConstantNumber* node)
     {
-        lua_rawcheckstack(L, 2);
-        lua_createtable(L, 0, preambleSize + 1);
+        const auto cstNode = lookupCstNode<Luau::CstExprConstantNumber>(node);
 
+        serializeToken(node->location.begin, cstNode ? cstNode->value.data : std::to_string(node->value).data(), preambleSize + 1);
         serializeNodePreamble(node, "number");
 
         lua_pushnumber(L, node->value);

std/codemod/ast_types.luau

@@ -45,6 +45,7 @@ export type AstLocal = {
 
 export type AstExprConstantBool = Token<"true" | "false"> & { tag: "boolean", value: boolean }
 
+export type AstExprConstantNumber = Token<string> & { tag: "number", value: number }
 
 export type AstExprConstantString = Token<string> & {
 	tag: "string",
@@ -118,6 +119,7 @@ export type AstExprBinary = {
 
 export type AstExpr =
 	| AstExprConstantBool
+	| AstExprConstantNumber
 	| AstExprConstantString
 	| AstExprLocal
 	| AstExprGlobal

std/codemod/visitor.luau

@@ -18,6 +18,7 @@ type Visitor = {
 	visitToken: (T.Token) -> boolean,
 	visitString: (T.AstExprConstantString) -> boolean,
 	visitBoolean: (T.AstExprConstantBool) -> boolean,
+	visitNumber: (T.AstExprConstantNumber) -> boolean,
 	visitLocal: (T.AstLocal) -> boolean,
 }
 
@@ -41,6 +42,7 @@ local defaultVisitor: Visitor = {
 	visitToken = alwaysVisit :: any,
 	visitString = alwaysVisit :: any,
 	visitBoolean = alwaysVisit :: any,
+	visitNumber = alwaysVisit :: any,
 	visitLocal = alwaysVisit :: any,
 }
 
@@ -105,6 +107,12 @@ local function visitBoolean(node: T.AstExprConstantBool, visitor: Visitor)
 	end
 end
 
+local function visitNumber(node: T.AstExprConstantNumber, visitor: Visitor)
+	if visitor.visitNumber(node) then
+		visitToken(node, visitor)
+	end
+end
+
 local function visitLocalReference(node: T.AstExprLocal, visitor: Visitor)
 	if visitor.visitLocalReference(node) then
 		visitor.visitToken(node.token)
@@ -183,6 +191,8 @@ end
 function visitExpression(expression: T.AstExpr, visitor: Visitor)
 	if expression.tag == "boolean" then
 		visitBoolean(expression, visitor)
+	elseif expression.tag == "number" then
+		visitNumber(expression, visitor)
 	elseif expression.tag == "string" then
 		visitString(expression, visitor)
 	elseif expression.tag == "local" then