Add auto-registration to Vault and NCN in Operator Client (#165)

- Fixes #85 

```
   cargo r --bin tip-router-operator-cli  -- \
  --keypair-path ~/.config/solana/id.json \
  --operator-address 2jXCuwLFGZ69FwBRSsrx6AjRhPx1wohB24Lsap7422Zr \
  --ledger-path ./ \
  --backup-snapshots-dir ./backup \
  --snapshot-output-dir ./snapshot \
  --full-snapshots-path ./snapshot \
  --save-path ./ \
  --rpc-url "" \
  run \
  --ncn-address 2jXCuwLFGZ69FwBRSsrx6AjRhPx1wohB24Lsap7422Zr \
  --tip-distribution-program-id 2jXCuwLFGZ69FwBRSsrx6AjRhPx1wohB24Lsap7422Zr  \
  --priority-fee-distribution-program-id 2jXCuwLFGZ69FwBRSsrx6AjRhPx1wohB24Lsap7422Zr  \
  --tip-payment-program-id 2jXCuwLFGZ69FwBRSsrx6AjRhPx1wohB24Lsap7422Zr 
```

Author: aoikurokawa

Cargo.lock

@@ -119,7 +119,7 @@ spl-token-2022 = { version = "=7.0.0", features = ["no-entrypoint"] }
 switchboard-on-demand = "0.3.4"
 syn = "2.0.72"
 thiserror = "1.0.57"
-tokio = { version = "1.36.0", features = ["full"] }
+tokio = { version = "1.47.1", features = ["full"] }
 
 [profile.release]
 overflow-checks = true

Cargo.toml

@@ -7,7 +7,7 @@ description = "CLI for Jito Tip Router"
 [dependencies]
 anchor-lang = { workspace = true }
 anyhow = { workspace = true }
-base64 = "0.13"
+base64 = { workspace = true }
 clap = { workspace = true }
 clap_old = { workspace = true }
 crossbeam-channel = "0.5.15"
@@ -17,6 +17,9 @@ im = "15.1"
 itertools = "0.11"
 jito-bytemuck = { workspace = true }
 jito-priority-fee-distribution-sdk = { workspace = true }
+jito-restaking-client = { workspace = true }
+jito-restaking-core = { workspace = true }
+jito-restaking-program = { workspace = true }
 jito-tip-distribution-sdk = { workspace = true }
 jito-tip-payment-sdk = { workspace = true }
 jito-tip-router-client = { workspace = true }

tip-router-operator-cli/Cargo.toml

@@ -1,6 +1,7 @@
 use std::{fs::File, io::Write, str::FromStr};
 
 use anchor_lang::prelude::*;
+use base64::{engine::general_purpose, Engine};
 use clap::Parser;
 use jito_tip_distribution_sdk::{
     derive_tip_distribution_account_address, TipDistributionAccount, TIP_DISTRIBUTION_SIZE,
@@ -71,7 +72,7 @@ fn main() {
         .expect("Failed to serialize account");
 
     // Encode the binary data as base64
-    let base64_data = base64::encode(binary_data);
+    let base64_data = general_purpose::STANDARD.encode(binary_data);
 
     // Create the JSON structure
     let json_data = json!({

tip-router-operator-cli/src/bin/serialize-accounts.rs

@@ -148,6 +148,9 @@ pub enum Commands {
         #[arg(long, env)]
         tip_router_program_id: Pubkey,
 
+        #[arg(long, env, default_value = "jito_restaking_program::id()")]
+        restaking_program_id: Pubkey,
+
         #[arg(long, env, default_value = "3")]
         num_monitored_epochs: u64,
 

tip-router-operator-cli/src/cli.rs

@@ -12,6 +12,7 @@ pub mod load_and_process_ledger;
 pub mod priority_fees;
 pub mod process_epoch;
 pub mod reclaim;
+pub mod restaking;
 pub mod rpc_utils;
 pub mod solana_cli;
 pub mod submit;

tip-router-operator-cli/src/lib.rs

@@ -2,12 +2,12 @@
 use ::{
     anyhow::Result,
     clap::Parser,
+    jito_restaking_core::config::Config as RestakingConfig,
     log::{error, info},
     solana_metrics::{datapoint_error, datapoint_info, set_host_id},
     solana_rpc_client::nonblocking::rpc_client::RpcClient,
     solana_sdk::{pubkey::Pubkey, signer::keypair::read_keypair_file},
-    std::process::Command,
-    std::{str::FromStr, sync::Arc, time::Duration},
+    std::{process::Command, str::FromStr, sync::Arc, time::Duration},
     tip_router_operator_cli::{
         backup_snapshots::BackupSnapshotMonitor,
         claim::{claim_mev_tips_with_emit, emit_claim_mev_tips_metrics},
@@ -16,6 +16,7 @@ use ::{
         ledger_utils::get_bank_from_snapshot_at_slot,
         load_bank_from_snapshot, meta_merkle_tree_path, process_epoch, read_merkle_tree_collection,
         read_stake_meta_collection, reclaim,
+        restaking::RestakingHandler,
         submit::{submit_recent_epochs_to_ncn, submit_to_ncn},
         tip_distribution_stats::get_tip_distribution_stats,
         tip_router::get_ncn_config,
@@ -96,6 +97,7 @@ async fn main() -> Result<()> {
             priority_fee_distribution_program_id,
             tip_payment_program_id,
             tip_router_program_id,
+            restaking_program_id,
             save_snapshot,
             num_monitored_epochs,
             override_target_slot,
@@ -132,6 +134,21 @@ async fn main() -> Result<()> {
             let rpc_url = cli.rpc_url.clone();
             let claim_tips_epoch_filepath = cli.claim_tips_epoch_filepath.clone();
             let cli_clone: Cli = cli.clone();
+            let operator_address = cli.operator_address.clone();
+            let cluster = cli.cluster.clone();
+            let restaking_config_address =
+                RestakingConfig::find_program_address(&restaking_program_id).0;
+
+            let restaking_handler = RestakingHandler::new(
+                rpc_client.clone(),
+                restaking_program_id,
+                restaking_config_address,
+                ncn_address,
+                Pubkey::from_str(cli.operator_address.as_str()).unwrap(),
+                keypair.clone(),
+            );
+            restaking_handler.warmup_operator().await?;
+            restaking_handler.create_operator_vault_tickets().await?;
 
             if !backup_snapshots_dir.exists() {
                 info!(
@@ -141,9 +158,6 @@ async fn main() -> Result<()> {
                 std::fs::create_dir_all(&backup_snapshots_dir)?;
             }
 
-            let operator_address = cli.operator_address.clone();
-            let cluster = cli.cluster.clone();
-
             let try_catchup = tip_router_operator_cli::solana_cli::catchup(
                 cli.rpc_url.to_owned(),
                 cli.localhost_port,