From 0999dccfecc8f736c9c628816753b30f5e73e9fc Mon Sep 17 00:00:00 2001
From: Dmitry Leleko <lelekodmitry@gmail.com>
Date: Tue, 17 Jan 2023 17:38:07 +0300
Subject: [PATCH] Add currentPrice to cfdBalanceSchema

---
 package.json                                          |  2 +-
 .../OrionAggregator/ws/schemas/cfdBalancesSchema.ts   | 11 +++++++----
 src/types.ts                                          |  1 +
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/package.json b/package.json
index a83dd8b..d0c5f5c 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@orionprotocol/sdk",
-  "version": "0.16.0-rc.20",
+  "version": "0.16.0-rc.21",
   "description": "Orion Protocol SDK",
   "main": "./lib/esm/index.js",
   "module": "./lib/esm/index.js",
diff --git a/src/services/OrionAggregator/ws/schemas/cfdBalancesSchema.ts b/src/services/OrionAggregator/ws/schemas/cfdBalancesSchema.ts
index 1869c9e..d0d6205 100644
--- a/src/services/OrionAggregator/ws/schemas/cfdBalancesSchema.ts
+++ b/src/services/OrionAggregator/ws/schemas/cfdBalancesSchema.ts
@@ -1,19 +1,21 @@
 import { z } from 'zod';
 
-const cfdBalanceSchema = z.object({
+const cfdBalanceSchema = z
+  .object({
     i: z.string(),
     b: z.string(),
     pnl: z.string(),
     fr: z.string(),
     e: z.string(),
     p: z.string(),
+    cp: z.string(),
     pp: z.string(),
     r: z.string(),
     m: z.string(),
     mu: z.string(),
     fmu: z.string(),
     awb: z.string(),
-})
+  })
   .transform((obj) => ({
     instrument: obj.i,
     balance: obj.b,
@@ -21,14 +23,15 @@ const cfdBalanceSchema = z.object({
     fundingRate: obj.fr,
     equity: obj.e,
     position: obj.p,
+    currentPrice: obj.cp,
     positionPrice: obj.pp,
     reserves: obj.r,
     margin: obj.m,
     marginUSD: obj.mu,
     freeMarginUSD: obj.fmu,
     availableWithdrawBalance: obj.awb,
-}));
+  }));
 
-const cfdBalancesSchema = z.array(cfdBalanceSchema)
+const cfdBalancesSchema = z.array(cfdBalanceSchema);
 
 export default cfdBalancesSchema;
diff --git a/src/types.ts b/src/types.ts
index c64a076..578dc06 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -33,6 +33,7 @@ export type CFDBalance = {
   fundingRate: string,
   equity: string,
   position: string,
+  currentPrice: string,
   positionPrice: string,
   reserves: string,
   margin: string,