feat: add .storeBit and .loadBit functions to stdlib

tact-lang · Aug 22, 2024 · 656651d · 656651d
1 parent 8112eeb
commit 656651d
Show file tree

Hide file tree

Showing 5 changed files with 58 additions and 26 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - The `exists` method for the `Map` type: PR [#581](https://github.com/tact-lang/tact/pull/581)
+- The `storeBit` method for `Builder` type and the `loadBit` method for `Slice` type: PR [#699](https://github.com/tact-lang/tact/pull/699)
 
 ### Changed
 

diff --git a/src/imports/stdlib.ts b/src/imports/stdlib.ts
@@ -126,32 +126,34 @@ files['std/cells.tact'] =
     'Ly8KLy8gQnVpbGRlcgovLwoKQG5hbWUoYmVnaW5fY2VsbCkKbmF0aXZlIGJlZ2luQ2VsbCgpOiBCdWlsZGVyOwoKQG5hbWUoc3RvcmVfaW50KQpleHRlbmRzIG5hdGl2' +
     'ZSBzdG9yZUludChzZWxmOiBCdWlsZGVyLCB2YWx1ZTogSW50LCBiaXRzOiBJbnQpOiBCdWlsZGVyOwoKQG5hbWUoc3RvcmVfdWludCkKZXh0ZW5kcyBuYXRpdmUgc3Rv' +
     'cmVVaW50KHNlbGY6IEJ1aWxkZXIsIHZhbHVlOiBJbnQsIGJpdHM6IEludCk6IEJ1aWxkZXI7CgpAbmFtZShfX3RhY3Rfc3RvcmVfYm9vbCkKZXh0ZW5kcyBuYXRpdmUg' +
-    'c3RvcmVCb29sKHNlbGY6IEJ1aWxkZXIsIHZhbHVlOiBCb29sKTogQnVpbGRlcjsKCkBuYW1lKHN0b3JlX2NvaW5zKQpleHRlbmRzIG5hdGl2ZSBzdG9yZUNvaW5zKHNl' +
-    'bGY6IEJ1aWxkZXIsIHZhbHVlOiBJbnQpOiBCdWlsZGVyOwoKQG5hbWUoc3RvcmVfcmVmKQpleHRlbmRzIG5hdGl2ZSBzdG9yZVJlZihzZWxmOiBCdWlsZGVyLCBjZWxs' +
-    'OiBDZWxsKTogQnVpbGRlcjsKCkBuYW1lKHN0b3JlX3NsaWNlKQpleHRlbmRzIG5hdGl2ZSBzdG9yZVNsaWNlKHNlbGY6IEJ1aWxkZXIsIGNlbGw6IFNsaWNlKTogQnVp' +
-    'bGRlcjsKCkBuYW1lKF9fdGFjdF9zdG9yZV9hZGRyZXNzKQpleHRlbmRzIG5hdGl2ZSBzdG9yZUFkZHJlc3Moc2VsZjogQnVpbGRlciwgYWRkcmVzczogQWRkcmVzcyk6' +
-    'IEJ1aWxkZXI7CgpAbmFtZShlbmRfY2VsbCkKZXh0ZW5kcyBuYXRpdmUgZW5kQ2VsbChzZWxmOiBCdWlsZGVyKTogQ2VsbDsKCkBuYW1lKGJ1aWxkZXJfcmVmcykKZXh0' +
-    'ZW5kcyBuYXRpdmUgcmVmcyhzZWxmOiBCdWlsZGVyKTogSW50OwoKQG5hbWUoYnVpbGRlcl9iaXRzKQpleHRlbmRzIG5hdGl2ZSBiaXRzKHNlbGY6IEJ1aWxkZXIpOiBJ' +
-    'bnQ7CgovLwovLyBTbGljZQovLwoKQG5hbWUoYmVnaW5fcGFyc2UpCmV4dGVuZHMgbmF0aXZlIGJlZ2luUGFyc2Uoc2VsZjogQ2VsbCk6IFNsaWNlOwoKQG5hbWUobG9h' +
-    'ZF9yZWYpCmV4dGVuZHMgbXV0YXRlcyBuYXRpdmUgbG9hZFJlZihzZWxmOiBTbGljZSk6IENlbGw7CgpAbmFtZShwcmVsb2FkX3JlZikKZXh0ZW5kcyBuYXRpdmUgcHJl' +
-    'bG9hZFJlZihzZWxmOiBTbGljZSk6IENlbGw7CgpAbmFtZShsb2FkX2JpdHMpCmV4dGVuZHMgbXV0YXRlcyBuYXRpdmUgbG9hZEJpdHMoc2VsZjogU2xpY2UsIGw6IElu' +
-    'dCk6IFNsaWNlOwoKQG5hbWUocHJlbG9hZF9iaXRzKQpleHRlbmRzIG5hdGl2ZSBwcmVsb2FkQml0cyhzZWxmOiBTbGljZSwgbDogSW50KTogU2xpY2U7CgpAbmFtZShs' +
-    'b2FkX2ludCkKZXh0ZW5kcyBtdXRhdGVzIG5hdGl2ZSBsb2FkSW50KHNlbGY6IFNsaWNlLCBsOiBJbnQpOiBJbnQ7CgpAbmFtZShwcmVsb2FkX2ludCkKZXh0ZW5kcyBu' +
-    'YXRpdmUgcHJlbG9hZEludChzZWxmOiBTbGljZSwgbDogSW50KTogSW50OwoKQG5hbWUobG9hZF91aW50KQpleHRlbmRzIG11dGF0ZXMgbmF0aXZlIGxvYWRVaW50KHNl' +
-    'bGY6IFNsaWNlLCBsOiBJbnQpOiBJbnQ7CgpAbmFtZShwcmVsb2FkX3VpbnQpCmV4dGVuZHMgbmF0aXZlIHByZWxvYWRVaW50KHNlbGY6IFNsaWNlLCBsOiBJbnQpOiBJ' +
-    'bnQ7CgpAbmFtZShfX3RhY3RfbG9hZF9ib29sKQpleHRlbmRzIG11dGF0ZXMgbmF0aXZlIGxvYWRCb29sKHNlbGY6IFNsaWNlKTogQm9vbDsKCkBuYW1lKGxvYWRfY29p' +
-    'bnMpCmV4dGVuZHMgbXV0YXRlcyBuYXRpdmUgbG9hZENvaW5zKHNlbGY6IFNsaWNlKTogSW50OwoKQG5hbWUoX190YWN0X2xvYWRfYWRkcmVzcykKZXh0ZW5kcyBtdXRh' +
-    'dGVzIG5hdGl2ZSBsb2FkQWRkcmVzcyhzZWxmOiBTbGljZSk6IEFkZHJlc3M7CgpAbmFtZShza2lwX2JpdHMpCmV4dGVuZHMgbXV0YXRlcyBuYXRpdmUgc2tpcEJpdHMo' +
-    'c2VsZjogU2xpY2UsIGw6IEludCk7CgpAbmFtZShlbmRfcGFyc2UpCmV4dGVuZHMgbmF0aXZlIGVuZFBhcnNlKHNlbGY6IFNsaWNlKTsKCi8vCi8vIFNsaWNlIHNpemUK' +
-    'Ly8KCkBuYW1lKHNsaWNlX3JlZnMpCmV4dGVuZHMgbmF0aXZlIHJlZnMoc2VsZjogU2xpY2UpOiBJbnQ7CgpAbmFtZShzbGljZV9iaXRzKQpleHRlbmRzIG5hdGl2ZSBi' +
-    'aXRzKHNlbGY6IFNsaWNlKTogSW50OwoKQG5hbWUoc2xpY2VfZW1wdHk/KQpleHRlbmRzIG5hdGl2ZSBlbXB0eShzZWxmOiBTbGljZSk6IEJvb2w7CgpAbmFtZShzbGlj' +
-    'ZV9kYXRhX2VtcHR5PykKZXh0ZW5kcyBuYXRpdmUgZGF0YUVtcHR5KHNlbGY6IFNsaWNlKTogQm9vbDsKCkBuYW1lKHNsaWNlX3JlZnNfZW1wdHk/KQpleHRlbmRzIG5h' +
-    'dGl2ZSByZWZzRW1wdHkoc2VsZjogU2xpY2UpOiBCb29sOwoKLy8KLy8gQ29udmVyc2lvbnMKLy8KCmlubGluZSBleHRlbmRzIGZ1biBhc1NsaWNlKHNlbGY6IEJ1aWxk' +
-    'ZXIpOiBTbGljZSB7CiAgICByZXR1cm4gc2VsZi5lbmRDZWxsKCkuYmVnaW5QYXJzZSgpOwp9CgppbmxpbmUgZXh0ZW5kcyBmdW4gYXNTbGljZShzZWxmOiBDZWxsKTog' +
-    'U2xpY2UgewogICAgcmV0dXJuIHNlbGYuYmVnaW5QYXJzZSgpOwp9CgppbmxpbmUgZXh0ZW5kcyBmdW4gYXNDZWxsKHNlbGY6IFNsaWNlKTogQ2VsbCB7CiAgICByZXR1' +
-    'cm4gYmVnaW5DZWxsKCkKICAgICAgICAuc3RvcmVTbGljZShzZWxmKQogICAgICAgIC5lbmRDZWxsKCk7Cn0KCmlubGluZSBleHRlbmRzIGZ1biBhc0NlbGwoc2VsZjog' +
-    'QnVpbGRlcik6IENlbGwgewogICAgcmV0dXJuIHNlbGYuZW5kQ2VsbCgpOwp9CgppbmxpbmUgZnVuIGVtcHR5Q2VsbCgpOiBDZWxsIHsKICAgIHJldHVybiBiZWdpbkNl' +
-    'bGwoKS5lbmRDZWxsKCk7Cn0KCmlubGluZSBmdW4gZW1wdHlTbGljZSgpOiBTbGljZSB7CiAgICByZXR1cm4gZW1wdHlDZWxsKCkuYXNTbGljZSgpOwp9';
+    'c3RvcmVCb29sKHNlbGY6IEJ1aWxkZXIsIHZhbHVlOiBCb29sKTogQnVpbGRlcjsKCkBuYW1lKF9fdGFjdF9zdG9yZV9ib29sKQpleHRlbmRzIG5hdGl2ZSBzdG9yZUJp' +
+    'dChzZWxmOiBCdWlsZGVyLCB2YWx1ZTogQm9vbCk6IEJ1aWxkZXI7CgpAbmFtZShzdG9yZV9jb2lucykKZXh0ZW5kcyBuYXRpdmUgc3RvcmVDb2lucyhzZWxmOiBCdWls' +
+    'ZGVyLCB2YWx1ZTogSW50KTogQnVpbGRlcjsKCkBuYW1lKHN0b3JlX3JlZikKZXh0ZW5kcyBuYXRpdmUgc3RvcmVSZWYoc2VsZjogQnVpbGRlciwgY2VsbDogQ2VsbCk6' +
+    'IEJ1aWxkZXI7CgpAbmFtZShzdG9yZV9zbGljZSkKZXh0ZW5kcyBuYXRpdmUgc3RvcmVTbGljZShzZWxmOiBCdWlsZGVyLCBjZWxsOiBTbGljZSk6IEJ1aWxkZXI7CgpA' +
+    'bmFtZShfX3RhY3Rfc3RvcmVfYWRkcmVzcykKZXh0ZW5kcyBuYXRpdmUgc3RvcmVBZGRyZXNzKHNlbGY6IEJ1aWxkZXIsIGFkZHJlc3M6IEFkZHJlc3MpOiBCdWlsZGVy' +
+    'OwoKQG5hbWUoZW5kX2NlbGwpCmV4dGVuZHMgbmF0aXZlIGVuZENlbGwoc2VsZjogQnVpbGRlcik6IENlbGw7CgpAbmFtZShidWlsZGVyX3JlZnMpCmV4dGVuZHMgbmF0' +
+    'aXZlIHJlZnMoc2VsZjogQnVpbGRlcik6IEludDsKCkBuYW1lKGJ1aWxkZXJfYml0cykKZXh0ZW5kcyBuYXRpdmUgYml0cyhzZWxmOiBCdWlsZGVyKTogSW50OwoKLy8K' +
+    'Ly8gU2xpY2UKLy8KCkBuYW1lKGJlZ2luX3BhcnNlKQpleHRlbmRzIG5hdGl2ZSBiZWdpblBhcnNlKHNlbGY6IENlbGwpOiBTbGljZTsKCkBuYW1lKGxvYWRfcmVmKQpl' +
+    'eHRlbmRzIG11dGF0ZXMgbmF0aXZlIGxvYWRSZWYoc2VsZjogU2xpY2UpOiBDZWxsOwoKQG5hbWUocHJlbG9hZF9yZWYpCmV4dGVuZHMgbmF0aXZlIHByZWxvYWRSZWYo' +
+    'c2VsZjogU2xpY2UpOiBDZWxsOwoKQG5hbWUobG9hZF9iaXRzKQpleHRlbmRzIG11dGF0ZXMgbmF0aXZlIGxvYWRCaXRzKHNlbGY6IFNsaWNlLCBsOiBJbnQpOiBTbGlj' +
+    'ZTsKCkBuYW1lKHByZWxvYWRfYml0cykKZXh0ZW5kcyBuYXRpdmUgcHJlbG9hZEJpdHMoc2VsZjogU2xpY2UsIGw6IEludCk6IFNsaWNlOwoKQG5hbWUobG9hZF9pbnQp' +
+    'CmV4dGVuZHMgbXV0YXRlcyBuYXRpdmUgbG9hZEludChzZWxmOiBTbGljZSwgbDogSW50KTogSW50OwoKQG5hbWUocHJlbG9hZF9pbnQpCmV4dGVuZHMgbmF0aXZlIHBy' +
+    'ZWxvYWRJbnQoc2VsZjogU2xpY2UsIGw6IEludCk6IEludDsKCkBuYW1lKGxvYWRfdWludCkKZXh0ZW5kcyBtdXRhdGVzIG5hdGl2ZSBsb2FkVWludChzZWxmOiBTbGlj' +
+    'ZSwgbDogSW50KTogSW50OwoKQG5hbWUocHJlbG9hZF91aW50KQpleHRlbmRzIG5hdGl2ZSBwcmVsb2FkVWludChzZWxmOiBTbGljZSwgbDogSW50KTogSW50OwoKQG5h' +
+    'bWUoX190YWN0X2xvYWRfYm9vbCkKZXh0ZW5kcyBtdXRhdGVzIG5hdGl2ZSBsb2FkQm9vbChzZWxmOiBTbGljZSk6IEJvb2w7CgpAbmFtZShfX3RhY3RfbG9hZF9ib29s' +
+    'KQpleHRlbmRzIG11dGF0ZXMgbmF0aXZlIGxvYWRCaXQoc2VsZjogU2xpY2UpOiBCb29sOwoKQG5hbWUobG9hZF9jb2lucykKZXh0ZW5kcyBtdXRhdGVzIG5hdGl2ZSBs' +
+    'b2FkQ29pbnMoc2VsZjogU2xpY2UpOiBJbnQ7CgpAbmFtZShfX3RhY3RfbG9hZF9hZGRyZXNzKQpleHRlbmRzIG11dGF0ZXMgbmF0aXZlIGxvYWRBZGRyZXNzKHNlbGY6' +
+    'IFNsaWNlKTogQWRkcmVzczsKCkBuYW1lKHNraXBfYml0cykKZXh0ZW5kcyBtdXRhdGVzIG5hdGl2ZSBza2lwQml0cyhzZWxmOiBTbGljZSwgbDogSW50KTsKCkBuYW1l' +
+    'KGVuZF9wYXJzZSkKZXh0ZW5kcyBuYXRpdmUgZW5kUGFyc2Uoc2VsZjogU2xpY2UpOwoKLy8KLy8gU2xpY2Ugc2l6ZQovLwoKQG5hbWUoc2xpY2VfcmVmcykKZXh0ZW5k' +
+    'cyBuYXRpdmUgcmVmcyhzZWxmOiBTbGljZSk6IEludDsKCkBuYW1lKHNsaWNlX2JpdHMpCmV4dGVuZHMgbmF0aXZlIGJpdHMoc2VsZjogU2xpY2UpOiBJbnQ7CgpAbmFt' +
+    'ZShzbGljZV9lbXB0eT8pCmV4dGVuZHMgbmF0aXZlIGVtcHR5KHNlbGY6IFNsaWNlKTogQm9vbDsKCkBuYW1lKHNsaWNlX2RhdGFfZW1wdHk/KQpleHRlbmRzIG5hdGl2' +
+    'ZSBkYXRhRW1wdHkoc2VsZjogU2xpY2UpOiBCb29sOwoKQG5hbWUoc2xpY2VfcmVmc19lbXB0eT8pCmV4dGVuZHMgbmF0aXZlIHJlZnNFbXB0eShzZWxmOiBTbGljZSk6' +
+    'IEJvb2w7CgovLwovLyBDb252ZXJzaW9ucwovLwoKaW5saW5lIGV4dGVuZHMgZnVuIGFzU2xpY2Uoc2VsZjogQnVpbGRlcik6IFNsaWNlIHsKICAgIHJldHVybiBzZWxm' +
+    'LmVuZENlbGwoKS5iZWdpblBhcnNlKCk7Cn0KCmlubGluZSBleHRlbmRzIGZ1biBhc1NsaWNlKHNlbGY6IENlbGwpOiBTbGljZSB7CiAgICByZXR1cm4gc2VsZi5iZWdp' +
+    'blBhcnNlKCk7Cn0KCmlubGluZSBleHRlbmRzIGZ1biBhc0NlbGwoc2VsZjogU2xpY2UpOiBDZWxsIHsKICAgIHJldHVybiBiZWdpbkNlbGwoKQogICAgICAgIC5zdG9y' +
+    'ZVNsaWNlKHNlbGYpCiAgICAgICAgLmVuZENlbGwoKTsKfQoKaW5saW5lIGV4dGVuZHMgZnVuIGFzQ2VsbChzZWxmOiBCdWlsZGVyKTogQ2VsbCB7CiAgICByZXR1cm4g' +
+    'c2VsZi5lbmRDZWxsKCk7Cn0KCmlubGluZSBmdW4gZW1wdHlDZWxsKCk6IENlbGwgewogICAgcmV0dXJuIGJlZ2luQ2VsbCgpLmVuZENlbGwoKTsKfQoKaW5saW5lIGZ1' +
+    'biBlbXB0eVNsaWNlKCk6IFNsaWNlIHsKICAgIHJldHVybiBlbXB0eUNlbGwoKS5hc1NsaWNlKCk7Cn0=';
 files['std/config.tact'] =
     'QG5hbWUoY29uZmlnX3BhcmFtKQpuYXRpdmUgZ2V0Q29uZmlnUGFyYW0oaWQ6IEludCk6IENlbGw/Ow==';
 files['std/context.tact'] =

diff --git a/src/test/e2e-emulated/contracts/stdlib.tact b/src/test/e2e-emulated/contracts/stdlib.tact
@@ -21,8 +21,20 @@ contract StdlibTest {
     get fun sliceRefs(sc: Slice): Int {
         return sc.refs();
     }
+
+    get fun storeBool(bl: Builder, b: Bool): Builder {
+        return bl.storeBool(b);
+    }
 
     get fun loadBool(sc: Slice): Bool {
         return sc.loadBool();
     }
+
+    get fun storeBit(bl: Builder, b: Bool): Builder {
+        return bl.storeBit(b);
+    }
+
+    get fun loadBit(sc: Slice): Bool {
+        return sc.loadBit();
+    }
 }
diff --git a/src/test/e2e-emulated/stdlib.spec.ts b/src/test/e2e-emulated/stdlib.spec.ts
@@ -22,5 +22,16 @@ describe("stdlib", () => {
         expect(await contract.getSliceRefs(slice)).toBe(1n);
         expect(await contract.getSliceEmpty(slice)).toBe(false);
         expect(await contract.getLoadBool(slice)).toBe(true);
+        expect(await contract.getLoadBit(slice)).toBe(true);
+        expect(
+            (await contract.getStoreBool(beginCell(), true))
+                .endCell()
+                .toString(),
+        ).toBe(beginCell().storeBit(true).endCell().toString());
+        expect(
+            (await contract.getStoreBit(beginCell(), true))
+                .endCell()
+                .toString(),
+        ).toBe(beginCell().storeBit(true).endCell().toString());
     });
 });
diff --git a/stdlib/std/cells.tact b/stdlib/std/cells.tact
@@ -14,6 +14,9 @@ extends native storeUint(self: Builder, value: Int, bits: Int): Builder;
 @name(__tact_store_bool)
 extends native storeBool(self: Builder, value: Bool): Builder;
 
+@name(__tact_store_bool)
+extends native storeBit(self: Builder, value: Bool): Builder;
+
 @name(store_coins)
 extends native storeCoins(self: Builder, value: Int): Builder;
 
@@ -69,6 +72,9 @@ extends native preloadUint(self: Slice, l: Int): Int;
 @name(__tact_load_bool)
 extends mutates native loadBool(self: Slice): Bool;
 
+@name(__tact_load_bool)
+extends mutates native loadBit(self: Slice): Bool;
+
 @name(load_coins)
 extends mutates native loadCoins(self: Slice): Int;