Merge pull request #572 from vitaliystoliarovcc/refactor-examples

Refactor examples
bitfinexcom · Jul 19, 2021 · 52ae9c7 · 52ae9c7
2 parents 41334ab + 5838910
commit 52ae9c7
Show file tree

Hide file tree

Showing 53 changed files with 727 additions and 857 deletions.
diff --git a/examples/rest2/claim_positions.js b/examples/rest2/claim_positions.js
@@ -1,18 +1,17 @@
 'use strict'
 
 const Promise = require('bluebird')
+const { RESTv2 } = require('../../index')
 const _isEmpty = require('lodash/isEmpty')
-const runExample = require('../util/run_example')
-
-module.exports = runExample({
-  name: 'rest-claim-positions',
-  rest: { env: true, transform: true },
-  readline: true,
-  params: {
-    filterByMarket: false
-  }
-}, async ({ debug, debugTable, rest, readline, params }) => {
-  const { filterByMarket } = params
+const { args: { apiKey, apiSecret }, debug, debugTable, readline } = require('../util/setup')
+
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const filterByMarket = false
   const allPositions = await rest.positions()
   const positions = _isEmpty(filterByMarket)
     ? allPositions
@@ -49,4 +48,7 @@ module.exports = runExample({
   await Promise.all(positions.map(p => p.claim(rest)))
 
   debug('done!')
-})
+  readline.close()
+}
+
+execute()
diff --git a/examples/rest2/close_positions.js b/examples/rest2/close_positions.js
@@ -2,24 +2,30 @@
 
 const PI = require('p-iteration')
 const _isEmpty = require('lodash/isEmpty')
-const runExample = require('../util/run_example')
-
-module.exports = runExample({
-  name: 'rest-close-positions',
-  rest: { env: true, transform: true },
-  ws: { env: true, transform: true, connect: true, auth: true },
-  readline: true,
-  params: {
-    filterByMarket: null
-  }
-}, async ({
-  debug, debugTable, rest, ws, params, readline
-}) => {
-  const { filterByMarket } = params
+const WSv2 = require('../../lib/transports/ws2')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable, readline } = require('../util/setup')
+
+async function execute () {
+  const ws = new WSv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const filterByMarket = null
   const allPositions = await rest.positions()
+  await ws.open()
+  await ws.auth()
 
   if (allPositions.length === 0) {
     debug('no open positions')
+    await ws.close()
+    readline.close()
     return
   }
 
@@ -34,6 +40,8 @@ module.exports = runExample({
 
   if (positions.length === 0) {
     debug('no positions match filter')
+    await ws.close()
+    readline.close()
     return
   }
 
@@ -57,6 +65,8 @@ module.exports = runExample({
   )
 
   if (confirm.toLowerCase()[0] !== 'y') {
+    await ws.close()
+    readline.close()
     return
   }
 
@@ -70,4 +80,9 @@ module.exports = runExample({
 
   debug('')
   debug('closed %d positions', positions.length)
-})
+
+  await ws.close()
+  readline.close()
+}
+
+execute()
diff --git a/examples/rest2/currencies.js b/examples/rest2/currencies.js
@@ -1,12 +1,11 @@
 'use strict'
 
 const _chunk = require('lodash/chunk')
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { debug } = require('../util/setup')
 
-module.exports = runExample({
-  name: 'rest-get-currencies',
-  rest: true
-}, async ({ debug, rest }) => {
+async function execute () {
+  const rest = new RESTv2()
   debug('fetching currency list...')
 
   const currencies = await rest.currencies()
@@ -18,4 +17,6 @@ module.exports = runExample({
     debug('%s', currencyChunk.join(', '))
   })
   debug('')
-})
+}
+
+execute()
diff --git a/examples/rest2/funding_credits.js b/examples/rest2/funding_credits.js
@@ -2,16 +2,16 @@
 
 const { prepareAmount } = require('bfx-api-node-util')
 const argFromCLI = require('../util/arg_from_cli')
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
 
-module.exports = runExample({
-  name: 'rest-get-funding-credits',
-  rest: { env: true },
-  params: {
-    symbol: argFromCLI(0, 'fUSD')
-  }
-}, async ({ rest, debug, debugTable, params }) => {
-  const { symbol } = params
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const symbol = argFromCLI(0, 'fUSD')
 
   debug('fetching funding credits for %s', symbol)
 
@@ -28,4 +28,6 @@ module.exports = runExample({
       ])
     })
   }
-})
+}
+
+execute()
diff --git a/examples/rest2/funding_info.js b/examples/rest2/funding_info.js
@@ -2,16 +2,16 @@
 
 const { prepareAmount } = require('bfx-api-node-util')
 const argFromCLI = require('../util/arg_from_cli')
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
 
-module.exports = runExample({
-  name: 'rest-get-funding-info',
-  rest: { env: true },
-  params: {
-    symbol: argFromCLI(0, 'fUSD')
-  }
-}, async ({ rest, debug, debugTable, params }) => {
-  const { symbol } = params
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const symbol = argFromCLI(0, 'fUSD')
 
   debug('fetching funding info for %s', symbol)
 
@@ -27,4 +27,6 @@ module.exports = runExample({
       durationLend
     ]]
   })
-})
+}
+
+execute()
diff --git a/examples/rest2/funding_loans.js b/examples/rest2/funding_loans.js
@@ -2,16 +2,16 @@
 
 const { prepareAmount } = require('bfx-api-node-util')
 const argFromCLI = require('../util/arg_from_cli')
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
 
-module.exports = runExample({
-  name: 'rest-get-funding-loans',
-  rest: { env: true },
-  params: {
-    symbol: argFromCLI(0, 'fUSD')
-  }
-}, async ({ rest, debug, debugTable, params }) => {
-  const { symbol } = params
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const symbol = argFromCLI(0, 'fUSD')
 
   debug('fetching funding loans for %s', symbol)
 
@@ -28,4 +28,6 @@ module.exports = runExample({
       ])
     })
   }
-})
+}
+
+execute()
diff --git a/examples/rest2/funding_offers.js b/examples/rest2/funding_offers.js
@@ -2,16 +2,16 @@
 
 const { prepareAmount } = require('bfx-api-node-util')
 const argFromCLI = require('../util/arg_from_cli')
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
 
-module.exports = runExample({
-  name: 'rest-get-funding-offers',
-  rest: { env: true },
-  params: {
-    symbol: argFromCLI(0, 'fUSD')
-  }
-}, async ({ rest, debug, debugTable, params }) => {
-  const { symbol } = params
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const symbol = argFromCLI(0, 'fUSD')
 
   debug('fetching funding offers for %s', symbol)
 
@@ -28,4 +28,6 @@ module.exports = runExample({
       ])
     })
   }
-})
+}
+
+execute()
diff --git a/examples/rest2/key_permissions.js b/examples/rest2/key_permissions.js
@@ -1,14 +1,14 @@
 'use strict'
 
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
 
-module.exports = runExample({
-  name: 'rest-key-permissions',
-  rest: {
-    env: true,
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
     transform: true
-  }
-}, async ({ debug, debugTable, rest }) => {
+  })
   debug('fetching permissions')
 
   const perms = await rest.keyPermissions()
@@ -21,4 +21,6 @@ module.exports = runExample({
     rows,
     headers: ['Scope', 'Read', 'Write']
   })
-})
+}
+
+execute()
diff --git a/examples/rest2/ledgers.js b/examples/rest2/ledgers.js
@@ -1,16 +1,19 @@
 'use strict'
 
 const { prepareAmount } = require('bfx-api-node-util')
-const runExample = require('../util/run_example')
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
 const argFromCLI = require('../util/arg_from_cli')
 
-module.exports = runExample({
-  name: 'rest-get-ledgers',
-  rest: { env: true, transform: true },
-  params: {
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const params = {
     ccy: argFromCLI(0, 'all')
   }
-}, async ({ debug, debugTable, rest, params }) => {
   const ccy = params.ccy === 'all' ? null : params.ccy
 
   debug('fetching ledger entries for %s...', ccy || 'all currencies')
@@ -27,4 +30,6 @@ module.exports = runExample({
       'Entry ID', 'Currency', 'Timestamp', 'Amount', 'Balance', 'Description'
     ]
   })
-})
+}
+
+execute()
diff --git a/examples/rest2/list_open_orders.js b/examples/rest2/list_open_orders.js
@@ -3,16 +3,16 @@
 const _capitalize = require('lodash/capitalize')
 const _isEmpty = require('lodash/isEmpty')
 const { prepareAmount, preparePrice } = require('bfx-api-node-util')
-const runExample = require('../util/run_example')
-
-module.exports = runExample({
-  name: 'rest-list-open-orders',
-  rest: { env: true, transform: true },
-  params: {
-    filterByMarket: null
-  }
-}, async ({ rest, debug, debugTable, params }) => {
-  const { filterByMarket } = params
+const { RESTv2 } = require('../../index')
+const { args: { apiKey, apiSecret }, debug, debugTable } = require('../util/setup')
+
+async function execute () {
+  const rest = new RESTv2({
+    apiKey,
+    apiSecret,
+    transform: true
+  })
+  const filterByMarket = null
 
   debug('fetching open orders...')
   const allOrders = await rest.activeOrders()
@@ -40,4 +40,6 @@ module.exports = runExample({
       new Date(o.mtsUpdate).toLocaleString()
     ])
   })
-})
+}
+
+execute()