chore(common): helpers to deal with bigint

Emurgo · Apr 25, 2024 · 7cc0320 · 7cc0320
1 parent d46d28d
commit 7cc0320
Show file tree

Hide file tree

Showing 6 changed files with 138 additions and 20 deletions.
diff --git a/packages/common/src/index.ts b/packages/common/src/index.ts
@@ -25,6 +25,7 @@ export * from './utils/strings'
 
 export * from './numbers/as-atomic-value'
 export * from './numbers/split-bigint'
+export * from './numbers/bigint-formatter'
 
 export * from './observer/observer'
 

diff --git a/packages/common/src/numbers/bigint-formatter.test.ts b/packages/common/src/numbers/bigint-formatter.test.ts
@@ -0,0 +1,21 @@
+import {bigintFormatter} from './bigint-formatter'
+
+describe('bigintFormatter', () => {
+  // NOTE: decimal and thousands separators are locale-dependent
+  it('it works', () => {
+    let value = BigInt(1_234_567_890)
+    let decimalPlaces = 2
+    let formattedValue = bigintFormatter({value, decimalPlaces})
+    expect(formattedValue).toBe('12,345,678.90')
+
+    value = BigInt(189)
+    decimalPlaces = 6
+    formattedValue = bigintFormatter({value, decimalPlaces})
+    expect(formattedValue).toBe('0.000189')
+
+    value = BigInt(1_000_000)
+    decimalPlaces = 6
+    formattedValue = bigintFormatter({value, decimalPlaces})
+    expect(formattedValue).toBe('1.000000')
+  })
+})
diff --git a/packages/common/src/numbers/bigint-formatter.ts b/packages/common/src/numbers/bigint-formatter.ts
@@ -0,0 +1,11 @@
+import {splitBigInt} from './split-bigint'
+
+export function bigintFormatter({
+  value,
+  decimalPlaces,
+}: {
+  value: bigint
+  decimalPlaces: number
+}) {
+  return splitBigInt(value, decimalPlaces).bn.toFormat(decimalPlaces)
+}
diff --git a/packages/common/src/numbers/split-bigint.test.ts b/packages/common/src/numbers/split-bigint.test.ts
@@ -1,32 +1,104 @@
 import BigNumber from 'bignumber.js'
+
 import {splitBigInt} from './split-bigint'
 
 describe('splitBigInt', () => {
   it('should split a bigint into int dec and bn', () => {
     let bigInt = BigInt(123_456_789)
     let decimalPlaces = 3
     let expected = {
-      int: BigInt(123_456),
-      dec: BigInt(789),
+      integer: '123456',
+      fraction: '789',
       bn: new BigNumber('123456.789'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '123456.789',
     }
     expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
 
     bigInt = BigInt(987_654_321)
     decimalPlaces = 2
     expected = {
-      int: BigInt(9_876_543),
-      dec: BigInt(21),
+      integer: '9876543',
+      fraction: '21',
       bn: new BigNumber('9876543.21'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '9876543.21',
     }
     expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
 
     bigInt = BigInt(1_432_116_543)
     decimalPlaces = 5
     expected = {
-      int: BigInt(14_321),
-      dec: BigInt(16_543),
+      integer: '14321',
+      fraction: '16543',
       bn: new BigNumber('14321.16543'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '14321.16543',
+    }
+    expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
+
+    bigInt = BigInt(123)
+    decimalPlaces = 6
+    expected = {
+      integer: '0',
+      fraction: '000123',
+      bn: new BigNumber('0.000123'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '0.000123',
+    }
+    expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
+
+    bigInt = BigInt(123)
+    decimalPlaces = 3
+    expected = {
+      integer: '0',
+      fraction: '123',
+      bn: new BigNumber('0.123'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '0.123',
+    }
+    expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
+  })
+
+  it('should handle negative bigint', () => {
+    let bigInt = BigInt(-123_456_789)
+    let decimalPlaces = 3
+    let expected = {
+      integer: '123456',
+      fraction: '789',
+      bn: new BigNumber('-123456.789'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '-123456.789',
+    }
+    expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
+
+    bigInt = BigInt(-987_654_321)
+    decimalPlaces = 2
+    expected = {
+      integer: '9876543',
+      fraction: '21',
+      bn: new BigNumber('-9876543.21'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '-9876543.21',
+    }
+    expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
+
+    bigInt = BigInt(-1_432_116_543)
+    decimalPlaces = 5
+    expected = {
+      integer: '14321',
+      fraction: '16543',
+      bn: new BigNumber('-14321.16543'),
+      bi: bigInt,
+      decimalPlaces,
+      str: '-14321.16543',
     }
     expect(splitBigInt(bigInt, decimalPlaces)).toEqual(expected)
   })

diff --git a/packages/common/src/numbers/split-bigint.ts b/packages/common/src/numbers/split-bigint.ts
@@ -1,17 +1,31 @@
 import BigNumber from 'bignumber.js'
 import {freeze} from 'immer'
 
-export function splitBigInt(
-  bigInt: bigint,
-  decimalPlaces: number,
-): {int: bigint; dec: bigint} {
+export function splitBigInt(bigInt: bigint, decimalPlaces: number) {
   const scale: bigint = BigInt(10) ** BigInt(decimalPlaces)
-  const integerPart: bigint = bigInt / scale
-  const decimalPart: bigint = bigInt % scale
+
+  const isNegative = bigInt < 0
+  const sign = isNegative ? '-' : ''
+  const signAdjust = isNegative ? -1n : 1n
+
+  const absoluteBigInt = signAdjust * bigInt
+  const integer = (absoluteBigInt / scale).toString()
+  const fraction = (absoluteBigInt % scale)
+    .toString()
+    .padStart(decimalPlaces, '0')
+
+  const str = `${sign}${integer}.${fraction}`
+
+  const bn = new BigNumber(str)
 
   return freeze({
-    int: integerPart,
-    dec: decimalPart,
-    bn: new BigNumber(`${integerPart}.${decimalPart}`),
+    decimalPlaces,
+
+    bi: bigInt,
+    integer,
+    fraction,
+
+    bn,
+    str,
   })
 }
diff --git a/packages/portfolio/package.json b/packages/portfolio/package.json
@@ -133,11 +133,6 @@
       "<rootDir>/jest.setup.js"
     ]
   },
-  "dependencies": {
-    "@yoroi/api": "1.5.1",
-    "@yoroi/common": "1.5.2",
-    "@yoroi/types": "1.5.4"
-  },
   "devDependencies": {
     "@commitlint/config-conventional": "^17.0.2",
     "@react-native-async-storage/async-storage": "^1.19.3",
@@ -148,6 +143,8 @@
     "@types/jest": "^29.5.12",
     "@types/react": "^18.2.55",
     "@types/react-test-renderer": "^18.0.7",
+    "@yoroi/types": "1.5.4",
+    "bignumber.js": "^9.0.1",
     "commitlint": "^17.0.2",
     "del-cli": "^5.0.0",
     "dependency-cruiser": "^13.1.1",
@@ -171,6 +168,8 @@
   },
   "peerDependencies": {
     "@react-native-async-storage/async-storage": "^1.19.3",
+    "@yoroi/api": "1.5.1",
+    "@yoroi/common": "1.5.2",
     "bignumber.js": "^9.0.1",
     "immer": "^10.0.3",
     "react": ">= 16.8.0 <= 19.0.0",