From f53ce5c8c68838d336922a0d48246fe792f772d6 Mon Sep 17 00:00:00 2001
From: Fawad Ali <fawad@puffer.fi>
Date: Tue, 18 Feb 2025 11:57:17 +0100
Subject: [PATCH] Fix error with docs build and add CarrotStaker to main client

---
 example/index.html                                        | 1 -
 lib/api/puffer-client.ts                                  | 8 ++++++++
 ...ker-handler.test.ts => carrot-staking-handler.test.ts} | 8 ++++----
 ...carrot-staker-handler.ts => carrot-staking-handler.ts} | 2 +-
 4 files changed, 13 insertions(+), 6 deletions(-)
 rename lib/contracts/handlers/{carrot-staker-handler.test.ts => carrot-staking-handler.test.ts} (93%)
 rename lib/contracts/handlers/{carrot-staker-handler.ts => carrot-staking-handler.ts} (98%)

diff --git a/example/index.html b/example/index.html
index 4ffc659..e91283b 100644
--- a/example/index.html
+++ b/example/index.html
@@ -47,7 +47,6 @@ <h2>Example Script</h2>
         PufferClientHelpers,
         Chain,
         Token,
-        PufToken,
       } from '../dist/main.js';
 
       const depositInput = document.querySelector('#depositInput');
diff --git a/lib/api/puffer-client.ts b/lib/api/puffer-client.ts
index 9cae39f..8613283 100644
--- a/lib/api/puffer-client.ts
+++ b/lib/api/puffer-client.ts
@@ -19,6 +19,7 @@ import { NucleusTellerHandler } from '../contracts/handlers/nucleus-teller-handl
 import { NucleusAccountantHandler } from '../contracts/handlers/nucleus-accountant-handler';
 import { NucleusAtomicQueueHandler } from '../contracts/handlers/nucleus-atomic-queue-handler';
 import { MtwCarrotHandler } from '../contracts/handlers/mtw-carrot-handler';
+import { CarrotStakingHandler } from '../contracts/handlers/carrot-staking-handler';
 
 /**
  * The core class and the main entry point of the Puffer SDK.
@@ -54,6 +55,8 @@ export class PufferClient {
   public nucleusAtomicQueue: NucleusAtomicQueueHandler;
   /** Handler for the `mtwCARROT` (Merkl Token Wrapper) contract. */
   public mtwCarrot: MtwCarrotHandler;
+  /** Handler for the `CarrotStaker` contract. */
+  public carrotStaker: CarrotStakingHandler;
 
   /**
    * Create the Puffer Client.
@@ -150,6 +153,11 @@ export class PufferClient {
       this.walletClient,
       this.publicClient,
     );
+    this.carrotStaker = new CarrotStakingHandler(
+      chain,
+      this.walletClient,
+      this.publicClient,
+    );
   }
 
   /**
diff --git a/lib/contracts/handlers/carrot-staker-handler.test.ts b/lib/contracts/handlers/carrot-staking-handler.test.ts
similarity index 93%
rename from lib/contracts/handlers/carrot-staker-handler.test.ts
rename to lib/contracts/handlers/carrot-staking-handler.test.ts
index eb90a37..6351b6c 100644
--- a/lib/contracts/handlers/carrot-staker-handler.test.ts
+++ b/lib/contracts/handlers/carrot-staking-handler.test.ts
@@ -5,12 +5,12 @@ import {
   setupTestPublicClient,
 } from '../../../test/setup-test-clients';
 import { mockAccount, testingUtils } from '../../../test/setup-tests';
-import { CarrotStakerHandler } from './carrot-staker-handler';
+import { CarrotStakingHandler } from './carrot-staking-handler';
 import { CarrotStaker } from '../abis/mainnet/CarrotStaker';
 
-describe('CarrotStakerHandler', () => {
+describe('CarrotStakingHandler', () => {
   const contractTestingUtils = testingUtils.generateContractUtils(CarrotStaker);
-  let handler: CarrotStakerHandler;
+  let handler: CarrotStakingHandler;
   let walletClient: WalletClient;
   let publicClient: PublicClient;
 
@@ -18,7 +18,7 @@ describe('CarrotStakerHandler', () => {
     walletClient = setupTestWalletClient();
     publicClient = setupTestPublicClient();
 
-    handler = new CarrotStakerHandler(
+    handler = new CarrotStakingHandler(
       Chain.Holesky,
       walletClient,
       publicClient,
diff --git a/lib/contracts/handlers/carrot-staker-handler.ts b/lib/contracts/handlers/carrot-staking-handler.ts
similarity index 98%
rename from lib/contracts/handlers/carrot-staker-handler.ts
rename to lib/contracts/handlers/carrot-staking-handler.ts
index 6291e2c..00d7495 100644
--- a/lib/contracts/handlers/carrot-staker-handler.ts
+++ b/lib/contracts/handlers/carrot-staking-handler.ts
@@ -12,7 +12,7 @@ import { CarrotStaker } from '../abis/mainnet/CarrotStaker';
 /**
  * Handler for the `CarrotStaker` contract.
  */
-export class CarrotStakerHandler {
+export class CarrotStakingHandler {
   private viemChain: ViemChain;
 
   /**