@irrelon/battlescript 4.0.0 → 4.1.2

package/bin/functionality.ts

@@ -7,18 +7,32 @@ import {
 	fromSourceCodeToBinary,
 	fromSourceCodeToFullAst,
 	fromSourceCodeToMiniAst,
-	miniAstFromBinaryBuffer
+	miniAstFromBinaryBuffer,
+	generateCppFromProtocolAst
 } from "@/services";
 import type {SourceResolverAsync, SourceResolverSync} from "@/types";
+import type {CppGeneratorOptions} from "@/services/cppGenerator";
 import {version} from "../package.json";
 import {readFileSync} from "fs";
 import {writeFileSync} from "node:fs";
 
-const sourceResolverAsync: SourceResolverAsync = (filePath: string) => {
+export {
+	fromFullAstToSourceCode,
+	fromMiniAstToBinary,
+	fromMiniAstToSourceCode,
+	fromSourceCodeToBinary,
+	fromSourceCodeToFullAst,
+	fromSourceCodeToMiniAst,
+	miniAstFromBinaryBuffer
+}
+
+export type {SourceResolverAsync, SourceResolverSync}
+
+export const sourceResolverAsync: SourceResolverAsync = (filePath: string) => {
 	return readFile(filePath, "utf-8")
 };
 
-const sourceResolverSync: SourceResolverSync = (filePath: string): string => {
+export const sourceResolverSync: SourceResolverSync = (filePath: string): string => {
 	return readFileSync(filePath, "utf-8")
 };
 
@@ -169,6 +183,66 @@ export async function fromBinary(inputPath: string, outputPath: string) {
 	}
 }
 
+export async function toCpp(inputPath: string, outputDir?: string, options?: CppGeneratorOptions) {
+	console.log(`[BattleScript ${version}] toCpp: ${inputPath}`);
+	try {
+		const sourceCode = await sourceResolverAsync(inputPath);
+		const ast = fromSourceCodeToFullAst(sourceCode, {
+			filePath: inputPath,
+			includeMetaData: false,
+			sourceResolver: sourceResolverSync
+		});
+
+		const result = generateCppFromProtocolAst(ast, options);
+
+		// Determine output directory
+		const outputDirectory = outputDir ?? dirname(inputPath);
+		const headerPath = join(outputDirectory, result.headerFileName);
+		const sourcePath = join(outputDirectory, result.sourceFileName);
+
+		await writeFile(headerPath, result.headerContent);
+		await writeFile(sourcePath, result.sourceContent);
+
+		console.log(`[BattleScript ${version}] Generated C++ header: ${headerPath}`);
+		console.log(`[BattleScript ${version}] Generated C++ source: ${sourcePath}`);
+
+		return result;
+	} catch (error) {
+		console.error(`[BattleScript ${version}] Error:`, (error as Error).stack);
+		process.exit(1);
+	}
+}
+
+export function toCppSync(inputPath: string, outputDir?: string, options?: CppGeneratorOptions) {
+	console.log(`[BattleScript ${version}] toCppSync: ${inputPath}`);
+	try {
+		const sourceCode = sourceResolverSync(inputPath);
+		const ast = fromSourceCodeToFullAst(sourceCode, {
+			filePath: inputPath,
+			includeMetaData: false,
+			sourceResolver: sourceResolverSync
+		});
+
+		const result = generateCppFromProtocolAst(ast, options);
+
+		// Determine output directory
+		const outputDirectory = outputDir ?? dirname(inputPath);
+		const headerPath = join(outputDirectory, result.headerFileName);
+		const sourcePath = join(outputDirectory, result.sourceFileName);
+
+		writeFileSync(headerPath, result.headerContent);
+		writeFileSync(sourcePath, result.sourceContent);
+
+		console.log(`[BattleScript ${version}] Generated C++ header: ${headerPath}`);
+		console.log(`[BattleScript ${version}] Generated C++ source: ${sourcePath}`);
+
+		return result;
+	} catch (error) {
+		console.error(`[BattleScript ${version}] Error:`, (error as Error).stack);
+		process.exit(1);
+	}
+}
+
 export function changeExtension(filePath: string, newExt: string): string {
 	const dir = dirname(filePath);
 	const base = basename(filePath, extname(filePath));