Cranker: Fix the error InvalidAccountData (#29)

* fix: update

* fix: remove base64

* fix: error of Dockerfile

* fix: unhealthy error

Author: aoikurokawa

.github/workflows/build-container-images.yml

@@ -46,7 +46,7 @@ jobs:
       uses: docker/build-push-action@v5
       with:
         context: .
-        file: ./cranker.Dockerfile
+        file: ./cranker/Dockerfile
         platforms: linux/amd64
         push: true
         tags: ${{ steps.meta.outputs.tags }}

cli/Cargo.toml

@@ -5,7 +5,7 @@ version = "0.1.0"
 edition = "2021"
 
 [[bin]]
-name = "spl-stake-pool-interceptor"
+name = "stake-deposit-interceptor-cli"
 path = "src/main.rs"
 
 [dependencies]

cranker.Dockerfile

@@ -0,0 +1,15 @@
+FROM rust:1.81 AS builder
+RUN apt-get update && apt-get install -y libudev-dev libclang-dev clang pkg-config libssl-dev build-essential cmake protobuf-compiler
+RUN update-ca-certificates
+WORKDIR /usr/src/app
+COPY . .
+RUN --mount=type=cache,mode=0777,target=/home/root/app/target \
+    --mount=type=cache,mode=0777,target=/usr/local/cargo/registry \
+    cargo build --release --bin stake-deposit-interceptor-cranker
+
+#########
+FROM ubuntu:22.04 as runtime
+RUN apt-get update && apt-get install -y ca-certificates procps
+ENV APP="stake-deposit-interceptor-cranker"
+COPY --from=builder /usr/src/app/target/release/$APP ./$APP
+ENTRYPOINT ./$APP

cranker/Dockerfile

@@ -115,21 +115,21 @@ impl InterceptorCranker {
 
             info!(
                 "Receipt {} raw bytes:\n\
-                 Interpreted values:\n\
-                 deposit_time: {}\n\
-                 cool_down: {}\n\
-                 current_time: {}",
+                  Interpreted values:\n\
+                  deposit_time: {}\n\
+                  cool_down: {}\n\
+                  current_time: {}",
                 receipt.base, deposit_time, cool_down, now
             );
             emit_deposit_receipt(&receipt, &self.cluster_name);
 
             if deposit_time > now {
                 info!(
-                    "Receipt {} not yet expired (future deposit time). Current time: {}, Deposit time: {}",
-                    receipt.base,
-                    now,
-                    deposit_time
-                );
+                     "Receipt {} not yet expired (future deposit time). Current time: {}, Deposit time: {}",
+                     receipt.base,
+                     now,
+                     deposit_time
+                 );
                 future_deposits += 1;
                 continue;
             }
@@ -171,11 +171,11 @@ impl InterceptorCranker {
                 }
                 None => {
                     emit_error(format!(
-                        "Receipt {} has invalid timing values - would overflow. Deposit time: {}, Cool down: {}",
-                        receipt.base,
-                        deposit_time,
-                        cool_down
-                    ), &self.cluster_name);
+                         "Receipt {} has invalid timing values - would overflow. Deposit time: {}, Cool down: {}",
+                         receipt.base,
+                         deposit_time,
+                         cool_down
+                     ), &self.cluster_name);
                 }
             }
         }
@@ -263,6 +263,35 @@ impl InterceptorCranker {
         let owner_ata =
             get_associated_token_address(&receipt.owner, &stake_pool_deposit_authority.pool_mint);
 
+        // Check if account exists
+        match self.rpc_client.get_account(&owner_ata).await {
+            Ok(_) => {
+                info!("Owner token account exists: {owner_ata}");
+            }
+            Err(_) => {
+                info!("Creating owner token account: {owner_ata}");
+                let create_ata_ix = create_associated_token_account(
+                    &self.payer.pubkey(),
+                    &receipt.owner,
+                    &stake_pool_deposit_authority.pool_mint,
+                    &spl_token::id(),
+                );
+
+                let recent_blockhash = self.rpc_client.get_latest_blockhash().await?;
+                let create_ata_tx = Transaction::new_signed_with_payer(
+                    &[create_ata_ix],
+                    Some(&self.payer.pubkey()),
+                    &[self.payer.as_ref()],
+                    recent_blockhash,
+                );
+
+                self.rpc_client
+                    .send_and_confirm_transaction(&create_ata_tx)
+                    .await?;
+                info!("Created owner ata token account");
+            }
+        }
+
         let fee_wallet_token_account = get_associated_token_address(
             &stake_pool_deposit_authority.fee_wallet,
             &stake_pool_deposit_authority.pool_mint,

cranker/src/lib.rs

@@ -1,10 +1,9 @@
 services:
   interceptor-cranker:
     build:
-      dockerfile: ./cranker.Dockerfile
+      dockerfile: ./cranker/Dockerfile
     environment:
       - RPC_URL=${RPC_URL:-https://api.devnet.solana.com}
-      - WS_URL=${WS_URL:-wss://api.devnet.solana.com}
       - KEYPAIR_PATH=${KEYPAIR_PATH}
       - PROGRAM_ID=${PROGRAM_ID:-5TAiuAh3YGDbwjEruC1ZpXTJWdNDS7Ur7VeqNNiHMmGV}
       - INTERVAL_SECONDS=${INTERVAL_SECONDS:-60}