diff --git a/common/changes/@cityofzion/bs-neo3/CU-86dunk072-2_2024-09-20-18-24.json b/common/changes/@cityofzion/bs-neo3/CU-86dunk072-2_2024-09-20-18-24.json new file mode 100644 index 0000000..c129cb9 --- /dev/null +++ b/common/changes/@cityofzion/bs-neo3/CU-86dunk072-2_2024-09-20-18-24.json @@ -0,0 +1,10 @@ +{ + "changes": [ + { + "packageName": "@cityofzion/bs-neo3", + "comment": "update route displayed", + "type": "patch" + } + ], + "packageName": "@cityofzion/bs-neo3" +} \ No newline at end of file diff --git a/packages/bs-neo3/src/__tests__/services/swap/handlers/FlamingoSwapRouteHandler.spec.ts b/packages/bs-neo3/src/__tests__/services/swap/handlers/FlamingoSwapRouteHandler.spec.ts index 9bf364b..7b172ab 100644 --- a/packages/bs-neo3/src/__tests__/services/swap/handlers/FlamingoSwapRouteHandler.spec.ts +++ b/packages/bs-neo3/src/__tests__/services/swap/handlers/FlamingoSwapRouteHandler.spec.ts @@ -35,6 +35,25 @@ describe('FlamingoSwapRouteHandler', () => { expect(response).toEqual(expectedResponse) }) + it('should calculate best route for swap - bNEO to NEO', async () => { + const expectedResponse: SwapRoute[] = [ + { + tokenToUse: FlamingoSwapHelper.getFlamingoSwapToken(network, 'bNEO'), + reserveTokenToUse: '0', + tokenToReceive: FlamingoSwapHelper.getFlamingoSwapToken(network, 'NEO'), + reserveTokenToReceive: '0', + }, + ] + + const response = await FlamingoSwapRouteHandler.calculateBestRouteForSwap({ + network, + tokenToReceive: FlamingoSwapHelper.getFlamingoSwapToken(network, 'NEO'), + tokenToUse: FlamingoSwapHelper.getFlamingoSwapToken(network, 'bNEO'), + }) + + expect(response).toEqual(expectedResponse) + }) + it('should calculate best route for swap - NEO to GAS', async () => { const expectedResponse: SwapRoute[] = [ { @@ -60,6 +79,25 @@ describe('FlamingoSwapRouteHandler', () => { expect(response).toEqual(expectedResponse) }) + it('should calculate best route for swap - NEO to bNEO', async () => { + const expectedResponse: SwapRoute[] = [ + { + tokenToUse: FlamingoSwapHelper.getFlamingoSwapToken(network, 'NEO'), + reserveTokenToUse: '0', + tokenToReceive: FlamingoSwapHelper.getFlamingoSwapToken(network, 'bNEO'), + reserveTokenToReceive: '0', + }, + ] + + const response = await FlamingoSwapRouteHandler.calculateBestRouteForSwap({ + network, + tokenToReceive: FlamingoSwapHelper.getFlamingoSwapToken(network, 'bNEO'), + tokenToUse: FlamingoSwapHelper.getFlamingoSwapToken(network, 'NEO'), + }) + + expect(response).toEqual(expectedResponse) + }) + it('should calculate best route for swap - GAS to SWTH', async () => { const expectedResponse: SwapRoute[] = [ { diff --git a/packages/bs-neo3/src/services/swap/handlers/FlamingoSwapRouteHandler.ts b/packages/bs-neo3/src/services/swap/handlers/FlamingoSwapRouteHandler.ts index 432bf9c..1afa5d5 100644 --- a/packages/bs-neo3/src/services/swap/handlers/FlamingoSwapRouteHandler.ts +++ b/packages/bs-neo3/src/services/swap/handlers/FlamingoSwapRouteHandler.ts @@ -60,6 +60,23 @@ export class FlamingoSwapRouteHandler { if (routePath.length === 0) return [] + const isWrapOrUnwrapNeo = + FlamingoSwapHelper.isWrapNeo(network, [tokenToUse, tokenToReceive]) || + FlamingoSwapHelper.isUnwrapNeo(network, [tokenToReceive, tokenToUse]) + + if (isWrapOrUnwrapNeo) { + const swapRoute: SwapRoute[] = [ + { + reserveTokenToReceive: '0', + reserveTokenToUse: '0', + tokenToReceive, + tokenToUse, + }, + ] + + return swapRoute + } + const { stack: reservesStack } = await this.#invokeReserves({ network, routePath }) return this.#createSwapRoute({ network, routePath, reservesStack })