v2.0.0-beta.10

bboynam99 · Feb 14, 2020 · d10e687 · d10e687
1 parent fd5d1e4
commit d10e687
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@uniswap/sdk",
   "license": "GPL-3.0-or-later",
-  "version": "2.0.0-beta.9",
+  "version": "2.0.0-beta.10",
   "main": "dist/index.js",
   "typings": "dist/index.d.ts",
   "files": [

diff --git a/src/entities/exchange.ts b/src/entities/exchange.ts
@@ -76,6 +76,11 @@ export class Exchange {
     return this.tokenAmounts[1].token
   }
 
+  public reserveOf(token: Token): TokenAmount {
+    invariant(token.equals(this.token0) || token.equals(this.token1), 'TOKEN')
+    return token.equals(this.token0) ? this.reserve0 : this.reserve1
+  }
+
   public getOutputAmount(inputAmount: TokenAmount): [TokenAmount, Exchange] {
     invariant(inputAmount.token.equals(this.token0) || inputAmount.token.equals(this.token1), 'TOKEN')
     invariant(JSBI.greaterThan(inputAmount.raw, ZERO), 'ZERO')

diff --git a/src/entities/trade.ts b/src/entities/trade.ts
@@ -44,6 +44,9 @@ export class Trade {
 
   constructor(route: Route, amount: TokenAmount, tradeType: TradeType) {
     invariant(amount.token.equals(tradeType === TradeType.EXACT_INPUT ? route.input : route.output), 'TOKEN')
+    const firstExchange = route.exchanges[tradeType === TradeType.EXACT_INPUT ? 0 : route.exchanges.length - 1]
+    // ensure that the amount is strictly less that the exchange's balance
+    invariant(JSBI.lessThan(amount.raw, firstExchange.reserveOf(amount.token).raw), 'RESERVE')
     const amounts: TokenAmount[] = new Array(route.path.length)
     const nextExchanges: Exchange[] = new Array(route.exchanges.length)
     if (tradeType === TradeType.EXACT_INPUT) {