From a27069f2e5ca512a52d40d9109aa5b5849bd2616 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20K=C4=99dzia?=
 <67273035+KedziaPawel@users.noreply.github.com>
Date: Wed, 13 Mar 2024 08:54:58 +0100
Subject: [PATCH] feat: add comp accrued

---
 .changeset/mean-impalas-compete.md            |  5 ++++
 .../src/Portfolio/Integrations/CompoundV2.ts  | 25 +++++++++++++++++++
 2 files changed, 30 insertions(+)
 create mode 100644 .changeset/mean-impalas-compete.md

diff --git a/.changeset/mean-impalas-compete.md b/.changeset/mean-impalas-compete.md
new file mode 100644
index 00000000..2874fdbd
--- /dev/null
+++ b/.changeset/mean-impalas-compete.md
@@ -0,0 +1,5 @@
+---
+"@enzymefinance/sdk": patch
+---
+
+Add comp accrued
diff --git a/packages/sdk/src/Portfolio/Integrations/CompoundV2.ts b/packages/sdk/src/Portfolio/Integrations/CompoundV2.ts
index af28b7ed..3e7d55f5 100644
--- a/packages/sdk/src/Portfolio/Integrations/CompoundV2.ts
+++ b/packages/sdk/src/Portfolio/Integrations/CompoundV2.ts
@@ -349,6 +349,15 @@ const compoundComptrollerAbi = [
     stateMutability: "view",
     type: "function",
   },
+  {
+    constant: true,
+    inputs: [{ internalType: "address", name: "", type: "address" }],
+    name: "compAccrued",
+    outputs: [{ internalType: "uint256", name: "", type: "uint256" }],
+    payable: false,
+    stateMutability: "view",
+    type: "function",
+  },
 ] as const;
 
 export interface Market {
@@ -507,3 +516,19 @@ export function getMintGuardianPaused(
     address: args.compoundComptroller,
   });
 }
+
+export function getCompAccrued(
+  client: PublicClient,
+  args: Viem.ContractCallParameters<{
+    compoundComptroller: Address;
+    account: Address;
+  }>,
+) {
+  return readContract(client, {
+    ...Viem.extractBlockParameters(args),
+    abi: compoundComptrollerAbi,
+    functionName: "compAccrued",
+    address: args.compoundComptroller,
+    args: [args.account],
+  });
+}