finalize binary

generate_tests.ts

@@ -4,7 +4,7 @@ import { generateCode } from './src/main'
 
 function genCodeForTest(name: string) {
   const fixturesDir = path.resolve(__dirname, 'test')
-  generateCode(path.resolve(fixturesDir, 'tlb', name + '.tlb'), 'test/generated_files/generated_' + name + '.ts')
+  generateCode(path.resolve(fixturesDir, 'tlb', name + '.tlb'), 'test/generated_files/generated_' + name + '.ts', 'typescript')
 }
 
 genCodeForTest('block')

main.ts

@@ -4,21 +4,34 @@ import { generateCode } from './src/main';
 
 const cli = meow(`
 	Usage
-	  $ foo <tlbpath>
+	  $ tlbgen <tlbpath>
 
 	Options
-	  --output, -o Output file, defult = tlbpath + ".tst"
+	  --output, -o Output file, defult = tlbpath + ".ts"
+	  --language, -l Programming language result. Supported languages: ["typescript"]. Default: "typescript"
 `, {
 	flags: {
 		output: {
 			type: 'string',
-			shortFlag: 'o'
+			alias: 'o'
+		},
+		language: {
+			type: 'string',
+			alias: 'l'
 		}
 	}
 });
 
 
 let input = cli.input.at(0)
 if (input) {
-	generateCode(input, input + ".ts");
+	let output = input + '.ts'
+	if (cli.flags.output) {
+		output = cli.flags.output;
+	}
+	let language = 'typescript'
+	if (cli.flags.language) {
+		language = cli.flags.language
+	}
+	generateCode(input, output, language);
 }

src/main.ts

@@ -14,15 +14,15 @@ import { ast } from '@igorivaniuk/tlb-parser'
 import fs from 'fs'
 
 
-export function generate(tree: Program, input: string) {
+export function generate(tree: Program, input: string, getGenerator: (tlbCode: TLBCode) => CodeGenerator) {
   let oldTlbCode: TLBCodeBuild = { types: new Map<string, TLBTypeBuild>() };
 
   let splittedInput = input.split("\n");
 
   fillConstructors(tree.declarations, oldTlbCode, splittedInput);
   let tlbCode: TLBCode = convertCodeToReadonly(oldTlbCode);
 
-  let codeGenerator: CodeGenerator = new TypescriptGenerator(tlbCode);
+  let codeGenerator: CodeGenerator = getGenerator(tlbCode);
 
   codeGenerator.addTonCoreClassUsage("Builder");
   codeGenerator.addTonCoreClassUsage("Slice");
@@ -63,13 +63,19 @@ export function generate(tree: Program, input: string) {
   return generatedCode;
 }
 
-export function generateCode(inputPath: string, outputPath: string) {
+export function generateCode(inputPath: string, outputPath: string, resultLanguage: string) {
   const input = fs.readFileSync(
     inputPath,
     'utf-8',
   )
 
   const tree = ast(input)
 
-  fs.writeFile(outputPath, generate(tree, input), () => { });
+  fs.writeFile(outputPath, generate(tree, input, (tlbCode: TLBCode) => {
+    if (resultLanguage == 'typescript') {
+      return new TypescriptGenerator(tlbCode)
+    } else {
+      throw new Error(`Result language ${resultLanguage} is not supported`)
+    }
+  }), () => { });
 }