feat: add FeatureSet activate/deactivate to Node bindings (LiteSVM#289)

* add additional methods to FeatureSet class

* chore: remove unintended yarn.lock changes

* added full definitions for FeatureSet for Node.js

Author: naturespowerr

crates/node-litesvm/litesvm/internal.d.ts

@@ -251,6 +251,12 @@ export declare class FeatureSet {
   static allEnabled(): FeatureSet
   isActive(featureId: Uint8Array): boolean
   activatedSlot(featureId: Uint8Array): bigint | null
+  activate(featureId: Uint8Array, slot: bigint): void
+  deactivate(featureId: Uint8Array): void
+  getActiveFeatures(): Array<Buffer>
+  getInactiveFeatures(): Array<Buffer>
+  getActiveFeaturesCount(): number
+  getInactiveFeaturesCount(): number
   toString(): string
 }
 

crates/node-litesvm/src/feature_set.rs

@@ -1,6 +1,10 @@
 use {
-    crate::{to_string_js, util::convert_pubkey},
+    crate::{
+        to_string_js,
+        util::{bigint_to_u64, convert_pubkey},
+    },
     agave_feature_set::FeatureSet as FeatureSetOriginal,
+    napi::bindgen_prelude::*,
 };
 
 #[derive(Debug, Clone)]
@@ -28,6 +32,45 @@ impl FeatureSet {
     pub fn activated_slot(&self, feature_id: &[u8]) -> Option<u64> {
         self.0.activated_slot(&convert_pubkey(feature_id))
     }
+
+    #[napi]
+    pub fn activate(&mut self, feature_id: &[u8], slot: BigInt) {
+        let slot_u64 = bigint_to_u64(&slot).unwrap_or(0);
+        self.0.activate(&convert_pubkey(feature_id), slot_u64);
+    }
+
+    #[napi]
+    pub fn deactivate(&mut self, feature_id: &[u8]) {
+        self.0.deactivate(&convert_pubkey(feature_id));
+    }
+
+    #[napi]
+    pub fn get_active_features(&self) -> Vec<Buffer> {
+        self.0
+            .active()
+            .keys()
+            .map(|pubkey| Buffer::from(pubkey.to_bytes().to_vec()))
+            .collect()
+    }
+
+    #[napi]
+    pub fn get_inactive_features(&self) -> Vec<Buffer> {
+        self.0
+            .inactive()
+            .iter()
+            .map(|pubkey| Buffer::from(pubkey.to_bytes().to_vec()))
+            .collect()
+    }
+
+    #[napi]
+    pub fn get_active_features_count(&self) -> u32 {
+        self.0.active().len() as u32
+    }
+
+    #[napi]
+    pub fn get_inactive_features_count(&self) -> u32 {
+        self.0.inactive().len() as u32
+    }
 }
 
 to_string_js!(FeatureSet);

crates/node-litesvm/tests/featureSet.test.ts

@@ -0,0 +1,46 @@
+import { describe, it } from "node:test"
+import assert from "node:assert"
+import { FeatureSet } from "../litesvm"
+import { PublicKey } from "@solana/web3.js"
+
+describe("FeatureSet", () => {
+    it("should create default feature set", () => {
+        const fs = new FeatureSet()
+        assert.ok(fs)
+    })
+
+    it("should create all_enabled feature set", () => {
+        const fs = FeatureSet.allEnabled()
+        assert.ok(fs)
+    })
+
+    it("should activate and check feature", () => {
+        const fs = new FeatureSet()
+        const featureId = PublicKey.unique()
+        
+        const isActiveBeforeActivation = fs.isActive(featureId.toBuffer())
+        assert.strictEqual(isActiveBeforeActivation, false)
+        
+        fs.activate(featureId.toBuffer(), 100n)
+        
+        const isActiveAfterActivation = fs.isActive(featureId.toBuffer())
+        assert.strictEqual(isActiveAfterActivation, true)
+        
+        const slot = fs.activatedSlot(featureId.toBuffer())
+        assert.strictEqual(slot, 100n)
+    })
+
+    it("should deactivate feature", () => {
+        const fs = FeatureSet.allEnabled()
+        const featureId = PublicKey.unique()
+        
+        fs.activate(featureId.toBuffer(), 50n)
+        assert.strictEqual(fs.isActive(featureId.toBuffer()), true)
+        
+        fs.deactivate(featureId.toBuffer())
+        assert.strictEqual(fs.isActive(featureId.toBuffer()), false)
+        
+        const slot = fs.activatedSlot(featureId.toBuffer())
+        assert.strictEqual(slot, null)
+    })
+})