use buffer when it's divisible by 8

ton-community · Nov 7, 2024 · dacd907 · dacd907
1 parent dfea500
commit dacd907
Show file tree

Hide file tree

Showing 3 changed files with 208 additions and 197 deletions.
diff --git a/src/generators/typescript/generator.ts b/src/generators/typescript/generator.ts
@@ -98,6 +98,7 @@ import {
   getParamVarExpr,
   getTypeParametersExpression,
   isBigInt,
+  useBuffer,
 } from "./utils";
 
 /*
@@ -731,16 +732,18 @@ export function storeBool(bool: Bool): (builder: Builder) => void {
           fieldStoreSuffix: "Bit"
         }
       } else {
+        let isBuffer = useBuffer(fieldType);
+        let suffix = isBuffer ? "Buffer" : "Bits";
         exprForParam = {
           argLoadExpr: convertToAST(fieldType.bits, ctx.constructor),
           argStoreExpr: convertToAST(
             fieldType.bits,
             ctx.constructor,
             id(ctx.name)
           ),
-          paramType: "BitString",
-          fieldLoadSuffix: "Bits",
-          fieldStoreSuffix: "Bits",
+          paramType: isBuffer ? "Buffer" : "BitString",
+          fieldLoadSuffix: suffix,
+          fieldStoreSuffix: suffix,
         };
       }
     } else if (fieldType.kind == "TLBCellType") {

diff --git a/src/generators/typescript/utils.ts b/src/generators/typescript/utils.ts
@@ -1,5 +1,6 @@
 import {
   TLBBinaryOp,
+  TLBBitsType,
   TLBCode,
   TLBConstructor,
   TLBMathExpr,
@@ -234,3 +235,10 @@ export function isBigInt(fieldType: TLBNumberType) {
   }
   return true;
 }
+
+export function useBuffer(bitsType: TLBBitsType) {
+  if (bitsType.bits instanceof TLBNumberExpr && (bitsType.bits.n % 8) == 0) {
+    return true;
+  }
+  return false;
+}