serialization token chainId as Bigint

clbrge · Apr 20, 2023 · 50cdb76 · 50cdb76
1 parent 1dbaaeb
commit 50cdb76
Showing 1 changed file with 29 additions and 11 deletions.
diff --git a/src/index.js b/src/index.js
@@ -67,6 +67,16 @@ export class Token {
     return expandToNDecimals(decimals)(number)
   }
 
+  toJSON() {
+    return {
+      ...this,
+      chainId:
+        typeof this.chainId === 'bigint'
+          ? this.chainId.toString()
+          : this.chainId
+    }
+  }
+
   // if number is decimals
   newAmount(number, decimals) {
     if (number instanceof TokenAmount) {
@@ -178,33 +188,41 @@ export class TokenAmount {
       }
     }
   }
+
   add(any) {
     return toTokenAmount(
       this.number + toTokenAmount(any, this.token).number,
       this.token
     )
   }
 
-  xadd(any) {
-    const other = toTokenAmount(any, this.token)
-    return this.token.newAmount(this.number + other.number)
-  }
-
   sub(any) {
-    const other = toTokenAmount(any, this.token)
-    return this.token.newAmount(this.number - other.number)
+    return toTokenAmount(
+      this.number - toTokenAmount(any, this.token).number,
+      this.token
+    )
   }
 
   mul(any) {
-    const other = toTokenAmount(any, this.token)
-    return this.token.newAmount(this.number * other.number)
+    return toTokenAmount(
+      this.number * toTokenAmount(any, this.token).number,
+      this.token
+    )
   }
 
   div(any) {
-    const other = toTokenAmount(any, this.token)
-    return this.token.newAmount(this.number / other.number)
+    return toTokenAmount(
+      this.number / toTokenAmount(any, this.token).number,
+      this.token
+    )
   }
 
+  lt(any) {
+    return this.number < toTokenAmount(any, this.token).number
+  }
+  lte(any) {
+    return this.number <= toTokenAmount(any, this.token).number
+  }
   eq(any) {
     return this.number === toTokenAmount(any, this.token).number
   }