Updated interfaces (getTLBCode, getTLBCodeByAST, add initial expressi…

…on to TLBExprMathType and other utils), release v1.1.0

index.ts

@@ -1,4 +1,6 @@
 import { generateCode } from "./src/main";
-export { generateCode, generateCodeByAST, generateCodeWithGenerator } from "./src/main";
+export * from "./src/ast";
+export { generateCode, getTLBCode, getTLBCodeByAST, generateCodeByAST, generateCodeWithGenerator } from "./src/main";
+export { isBigInt as isBigIntForJson, isBigIntExpr as isBigIntExprForJson } from "./src/generators/typescript/utils";
 export { CodeGenerator } from "./src/generators/generator";
 export { TypescriptGenerator } from "./src/generators/typescript/generator";

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ton-community/tlb-codegen",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "description": "",
   "main": "build/index.js",
   "scripts": {

src/ast.ts

@@ -70,6 +70,7 @@ export type TLBNamedType = {
 export type TLBMathExprType = {
   kind: "TLBExprMathType";
   expr: TLBMathExpr;
+  initialExpr: TLBMathExpr;
 };
 
 export type TLBBoolType = {

src/astbuilder/handle_type.ts

@@ -278,13 +278,14 @@ export function getType(
         return {
           kind: "TLBExprMathType",
           expr: getCalculatedExpression(new TLBVarExpr(expr.name), constructor),
+          initialExpr: new TLBVarExpr(expr.name),
         };
       } else {
         return { kind: "TLBNamedType", name: expr.name, arguments: [] };
       }
     }
   } else if (expr instanceof NumberExpr) {
-    return { kind: "TLBExprMathType", expr: new TLBNumberExpr(expr.num) };
+    return { kind: "TLBExprMathType", expr: new TLBNumberExpr(expr.num), initialExpr: new TLBNumberExpr(expr.num) };
   } else if (expr instanceof NegateExpr && expr.expr instanceof NameExpr) {
     return { kind: "TLBNegatedType", variableName: expr.expr.name };
   } else if (expr instanceof CellRefExpr) {
@@ -309,6 +310,7 @@ export function getType(
       return {
         kind: "TLBExprMathType",
         expr: getCalculatedExpression(convertToMathExpr(expr), constructor),
+        initialExpr: convertToMathExpr(expr),
       };
     }
   } else if (expr instanceof CondExpr) {

src/main.ts

@@ -11,15 +11,29 @@ import { CodeGenerator, CommonGenDeclaration } from "./generators/generator";
 import { TypescriptGenerator } from "./generators/typescript/generator";
 import fs from 'fs'
 
-
-export function generateCodeByAST(tree: Program, input: string, getGenerator: (tlbCode: TLBCode) => CodeGenerator) {
+export function getTLBCodeByAST(tree: Program, input: string) {
   let oldTlbCode: TLBCodeBuild = { types: new Map<string, TLBTypeBuild>() };
-
   let splittedInput = input.split("\n");
-
   fillConstructors(tree.declarations, oldTlbCode, splittedInput);
   let tlbCode: TLBCode = convertCodeToReadonly(oldTlbCode);
 
+  return tlbCode;
+}
+
+export function getTLBCode(inputPath: string) {
+  const input = fs.readFileSync(
+    inputPath,
+    'utf-8',
+  )
+
+  const tree = ast(input)
+
+  return getTLBCodeByAST(tree, input);
+}
+
+
+export function generateCodeByAST(tree: Program, input: string, getGenerator: (tlbCode: TLBCode) => CodeGenerator) {
+  let tlbCode = getTLBCodeByAST(tree, input)
   let codeGenerator: CodeGenerator = getGenerator(tlbCode);
 
   codeGenerator.addTonCoreClassUsage("Builder");