add filtering based on account type to getProgramAccounts

Author: jhunsaker

pc/net_socket.cpp

@@ -1471,6 +1471,12 @@ void json_wtr::add_key_verbatim( str key, str val )
   add( val );
 }
 
+void json_wtr::add_key_enc_base58( str key, str val )
+{
+  add_key_only( key );
+  add_enc_base58( val );
+}
+
 void json_wtr::add_key( str key, type_t t )
 {
   add_key_only( key );

pc/net_socket.hpp

@@ -446,6 +446,7 @@ namespace pc
     void add_key( str key, jtrue );
     void add_key( str key, jfalse );
     void add_key_verbatim( str key, str );
+    void add_key_enc_base58( str key, str val );
 
     // add array value
     void add_val( str val );

pc/rpc_client.cpp

@@ -630,6 +630,7 @@ bool rpc::account_subscribe::notify( const jtree& jt )
 
 rpc::program_subscribe::program_subscribe()
 : account_update{}
+, pgm_{ nullptr }
 {
 }
 
@@ -686,7 +687,9 @@ bool rpc::program_subscribe::notify( const jtree& jt )
 // get_program_accounts
 
 rpc::get_program_accounts::get_program_accounts()
-: pgm_{ nullptr }
+: account_update{}
+, pgm_{ nullptr }
+, acct_type_{ 0 }
 {
 }
 
@@ -695,6 +698,11 @@ void rpc::get_program_accounts::set_program( pub_key *pkey )
   pgm_ = pkey;
 }
 
+void rpc::get_program_accounts::set_account_type( uint32_t const acct_type )
+{
+  acct_type_ = acct_type;
+}
+
 void rpc::get_program_accounts::request( json_wtr& msg )
 {
   msg.add_key( "method", "getProgramAccounts" );
@@ -703,6 +711,17 @@ void rpc::get_program_accounts::request( json_wtr& msg )
   msg.add_val( json_wtr::e_obj );
   msg.add_key( "encoding", "base64+zstd" );
   msg.add_key( "commitment", commitment_to_str( cmt_ ) );
+  if ( acct_type_ ) {
+    msg.add_key( "filters", json_wtr::e_arr );
+    msg.add_val( json_wtr::e_obj );
+    msg.add_key( "memcmp", json_wtr::e_obj );
+    msg.add_key( "offset", offsetof( pc_acc_t, type_ ) );
+    msg.add_key_enc_base58(
+      "bytes", str( ( char* )&acct_type_, sizeof( acct_type_ ) ) );
+    msg.pop();
+    msg.pop();
+    msg.pop();
+  }
   msg.add_key( "withContext", json_wtr::jtrue{} );
   msg.pop();
   msg.pop();

pc/rpc_client.hpp

@@ -378,6 +378,7 @@ namespace pc
     public:
       // parameters
       void set_program( pub_key * );
+      void set_account_type( uint32_t );
 
       get_program_accounts();
       void request( json_wtr& ) override;
@@ -387,6 +388,7 @@ namespace pc
 
     private:
       pub_key    *pgm_;
+      uint32_t    acct_type_;
     };
 
     // get all (pyth) transactions in a slot