Add exports for NPM package

index.ts

@@ -0,0 +1,4 @@
+import { generateCode } from "./src/main";
+export { generateCode, generateCodeByAST, generateCodeWithGenerator } from "./src/main";
+export { CodeGenerator } from "./src/generators/generator";
+export { TypescriptGenerator } from "./src/generators/typescript/generator";

package.json

@@ -1,8 +1,8 @@
 {
   "name": "@polyprogrammist_test/tlbgen",
-  "version": "1.0.2",
+  "version": "1.0.13",
   "description": "",
-  "main": "index.js",
+  "main": "build/index.js",
   "scripts": {
     "test": "jest",
     "build": "npm run compile",

src/main.ts

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