Add support for .luau (roblox-ts#2802)

Depends on:
- roblox-ts/path-translator#2
- roblox-ts/rojo-resolver#496

Allows for both emitting .luau files to `outDir` and importing packages
contain `.luau` files (or `.lua` for backwards compatibility).

Unit tests pass with the above two PR branches installed locally.

Author: osyrisrblx

package-lock.json

@@ -49,8 +49,8 @@
 	"license": "MIT",
 	"dependencies": {
 		"@roblox-ts/luau-ast": "=2.0.0",
-		"@roblox-ts/path-translator": "=1.0.0",
-		"@roblox-ts/rojo-resolver": "=1.0.7",
+		"@roblox-ts/path-translator": "^1.1.0",
+		"@roblox-ts/rojo-resolver": "^1.1.0",
 		"chokidar": "^3.6.0",
 		"fs-extra": "^11.2.0",
 		"kleur": "^4.1.5",

package.json

@@ -111,6 +111,10 @@ export = ts.identity<yargs.CommandModule<object, BuildFlags & Partial<ProjectOpt
 			.option("allowCommentDirectives", {
 				boolean: true,
 				hidden: true,
+			})
+			.option("luau", {
+				boolean: true,
+				describe: "emit files with .luau extension",
 			}),
 
 	handler: async argv => {
@@ -134,7 +138,7 @@ export = ts.identity<yargs.CommandModule<object, BuildFlags & Partial<ProjectOpt
 				setupProjectWatchProgram(data, projectOptions.usePolling);
 			} else {
 				const program = createProjectProgram(data);
-				const pathTranslator = createPathTranslator(program);
+				const pathTranslator = createPathTranslator(program, data);
 				cleanup(pathTranslator);
 				copyInclude(data);
 				copyFiles(data, pathTranslator, new Set(getRootDirs(program.getCompilerOptions())));

src/CLI/commands/build.ts

@@ -27,7 +27,7 @@ describe("should compile tests project", () => {
 		}),
 	);
 	const program = createProjectProgram(data);
-	const pathTranslator = createPathTranslator(program);
+	const pathTranslator = createPathTranslator(program, data);
 
 	// clean outDir between test runs
 	fs.removeSync(program.getCompilerOptions().outDir!);

src/CLI/test.ts

@@ -110,7 +110,7 @@ export class VirtualProject {
 		this.typeChecker = this.program.getTypeChecker();
 
 		const services = createTransformServices(this.typeChecker);
-		const pathTranslator = new PathTranslator(ROOT_DIR, OUT_DIR, undefined, false);
+		const pathTranslator = new PathTranslator(ROOT_DIR, OUT_DIR, undefined, false, this.data.projectOptions.luau);
 
 		const sourceFile = this.program.getSourceFile(PLAYGROUND_PATH);
 		assert(sourceFile);

src/Project/classes/VirtualProject.ts

@@ -1,10 +1,11 @@
 import { PathTranslator } from "@roblox-ts/path-translator";
 import path from "path";
+import { ProjectData } from "Shared/types";
 import { findAncestorDir } from "Shared/util/findAncestorDir";
 import { getRootDirs } from "Shared/util/getRootDirs";
 import ts from "typescript";
 
-export function createPathTranslator(program: ts.BuilderProgram) {
+export function createPathTranslator(program: ts.BuilderProgram, data: ProjectData) {
 	const compilerOptions = program.getCompilerOptions();
 	const rootDir = findAncestorDir([program.getProgram().getCommonSourceDirectory(), ...getRootDirs(compilerOptions)]);
 	const outDir = compilerOptions.outDir!;
@@ -13,5 +14,5 @@ export function createPathTranslator(program: ts.BuilderProgram) {
 		buildInfoPath = path.normalize(buildInfoPath);
 	}
 	const declaration = compilerOptions.declaration === true;
-	return new PathTranslator(rootDir, outDir, buildInfoPath, declaration);
+	return new PathTranslator(rootDir, outDir, buildInfoPath, declaration, data.projectOptions.luau);
 }

src/Project/functions/createPathTranslator.ts

@@ -75,7 +75,7 @@ export function setupProjectWatchProgram(data: ProjectData, usePolling: boolean)
 	const createProgram = createProgramFactory(data, options);
 	function refreshProgram() {
 		program = createProgram([...fileNamesSet], options);
-		pathTranslator = createPathTranslator(program);
+		pathTranslator = createPathTranslator(program, data);
 	}
 
 	function runInitialCompile() {

src/Project/functions/setupProjectWatchProgram.ts

@@ -51,4 +51,5 @@ export const DEFAULT_PROJECT_OPTIONS: ProjectOptions = {
 	writeTransformedFiles: false,
 	optimizedLoops: true,
 	allowCommentDirectives: false,
+	luau: true,
 };

src/Shared/constants.ts

@@ -14,6 +14,7 @@ export interface ProjectOptions {
 	writeTransformedFiles: boolean;
 	optimizedLoops: boolean;
 	allowCommentDirectives: boolean;
+	luau: boolean;
 }
 
 export interface ProjectData {

src/Shared/types.ts

@@ -18,7 +18,7 @@
 		"ServerScriptService": {
 			"$className": "ServerScriptService",
 			"main": {
-				"$path": "out/main.server.lua"
+				"$path": "out/main.server.luau"
 			},
 			"tests": {
 				"$path": "out/tests"
@@ -36,7 +36,7 @@
 		"StarterGui": {
 			"$className": "StarterGui",
 			"isolated": {
-				"$path": "out/helpers/rojo/isolated.lua"
+				"$path": "out/helpers/rojo/isolated.luau"
 			}
 		}
 	}