support constraints for both store and load

ton-community · Dec 9, 2023 · a0def47 · a0def47
1 parent 2bde57d
commit a0def47
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 2 deletions.
diff --git a/generated_test.ts b/generated_test.ts
@@ -1318,6 +1318,9 @@ export function storeHmLabel(hmLabel: HmLabel): (builder: Builder) => void {
 			hmLabel.s.forEach((arg: BitString) => {
   				builder.storeBits(arg);
   			});
+			if ((!(hmLabel.n <= hmLabel.m))) {
+  				throw new Error('');
+  			};
   		};
   	};
 	if ((hmLabel.kind == 'HmLabel_hml_long')) {
@@ -1467,5 +1470,8 @@ export function loadImplicitCondition(slice: Slice): ImplicitCondition {
 export function storeImplicitCondition(implicitCondition: ImplicitCondition): (builder: Builder) => void {
   	return (builder: Builder) => {
   		builder.storeUint(implicitCondition.flags, 10);
+		if ((!(implicitCondition.flags <= 100))) {
+  			throw new Error('');
+  		};
   	};
   }
diff --git a/src/codegen/main.ts b/src/codegen/main.ts
@@ -70,6 +70,7 @@ export function generate(tree: Program) {
 
       constructor.constraints.forEach(constraint => {
         constructorLoadStatements.push(tIfStatement(tUnaryOpExpression('!', convertToAST(constraint, constructor, true)), [tExpressionStatement(tIdentifier("throw new Error('')"))]));
+        subStructStoreStatements.push(tIfStatement(tUnaryOpExpression('!', convertToAST(constraint, constructor, true, tIdentifier(variableCombinatorName))), [tExpressionStatement(tIdentifier("throw new Error('')"))]))
       });
 
       constructorLoadStatements.push(tReturnStatement(tObjectExpression(subStructLoadProperties)));

diff --git a/tests/tlbgen.spec.ts b/tests/tlbgen.spec.ts
@@ -162,8 +162,19 @@ describe('Generating tlb code', () => {
     })
 
     test('Exceptions', () => {
-        let implicitCondition: ImplicitCondition = {kind: 'ImplicitCondition', flags: 200}
-        checkThrowOnStoreLoad(implicitCondition, loadImplicitCondition, storeImplicitCondition);
+        let implicitConditionIncorrect: ImplicitCondition = {kind: 'ImplicitCondition', flags: 200}
+        checkThrowOnStoreLoad(implicitConditionIncorrect, loadImplicitCondition, storeImplicitCondition);
+
+        let implicitCondition: ImplicitCondition = {kind: 'ImplicitCondition', flags: 100}
+        checkSameOnStoreLoad(implicitCondition, loadImplicitCondition, storeImplicitCondition);
+
+        let implicitConditionIncorrectCell = beginCell().storeUint(200, 10).endCell().beginParse();
+        expect(() => {
+            loadImplicitCondition(implicitConditionIncorrectCell)
+        }).toThrow(Error);
+
+        let implicitConditionCell = beginCell().storeUint(100, 10).endCell().beginParse();
+        expect(loadImplicitCondition(implicitConditionCell).flags == 100).toBeTruthy()
     })
 
     test('Hashmap', () => {