refactor: convert diag API to async/await (#1157)

License: MIT
Signed-off-by: Alan Shaw <alan.shaw@protocol.ai>

src/diag/cmds.js

@@ -1,17 +1,19 @@
 'use strict'
 
-const promisify = require('promisify-es6')
+const configure = require('../lib/configure')
 
-module.exports = (send) => {
-  return promisify((opts, callback) => {
-    if (typeof (opts) === 'function') {
-      callback = opts
-      opts = {}
-    }
+module.exports = configure(({ ky }) => {
+  return options => {
+    options = options || {}
 
-    send({
-      path: 'diag/cmds',
-      qs: opts
-    }, callback)
-  })
-}
+    const searchParams = new URLSearchParams(options.searchParams)
+    if (options.verbose != null) searchParams.set('verbose', options.verbose)
+
+    return ky.get('diag/cmds', {
+      timeout: options.timeout,
+      signal: options.signal,
+      headers: options.headers,
+      searchParams
+    }).json()
+  }
+})

src/diag/index.js

@@ -1,13 +1,9 @@
 'use strict'
 
-const moduleConfig = require('../utils/module-config')
+const callbackify = require('callbackify')
 
-module.exports = (arg) => {
-  const send = moduleConfig(arg)
-
-  return {
-    net: require('./net')(send),
-    sys: require('./sys')(send),
-    cmds: require('./cmds')(send)
-  }
-}
+module.exports = config => ({
+  net: callbackify.variadic(require('./net')(config)),
+  sys: callbackify.variadic(require('./sys')(config)),
+  cmds: callbackify.variadic(require('./cmds')(config))
+})

src/diag/net.js

@@ -1,17 +1,16 @@
 'use strict'
 
-const promisify = require('promisify-es6')
+const configure = require('../lib/configure')
 
-module.exports = (send) => {
-  return promisify((opts, callback) => {
-    if (typeof (opts) === 'function') {
-      callback = opts
-      opts = {}
-    }
+module.exports = configure(({ ky }) => {
+  return options => {
+    options = options || {}
 
-    send({
-      path: 'diag/net',
-      qs: opts
-    }, callback)
-  })
-}
+    return ky.get('diag/net', {
+      timeout: options.timeout,
+      signal: options.signal,
+      headers: options.headers,
+      searchParams: options.searchParams
+    }).json()
+  }
+})

src/diag/sys.js

@@ -1,17 +1,16 @@
 'use strict'
 
-const promisify = require('promisify-es6')
+const configure = require('../lib/configure')
 
-module.exports = (send) => {
-  return promisify((opts, callback) => {
-    if (typeof (opts) === 'function') {
-      callback = opts
-      opts = {}
-    }
+module.exports = configure(({ ky }) => {
+  return options => {
+    options = options || {}
 
-    send({
-      path: 'diag/sys',
-      qs: opts
-    }, callback)
-  })
-}
+    return ky.get('diag/sys', {
+      timeout: options.timeout,
+      signal: options.signal,
+      headers: options.headers,
+      searchParams: options.searchParams
+    }).json()
+  }
+})

src/utils/load-commands.js

@@ -94,7 +94,8 @@ function requireCommands (send, config) {
     bootstrap: require('../bootstrap')(config),
     config: require('../config')(config),
     dag: require('../dag')(config),
-    dht: require('../dht')(config)
+    dht: require('../dht')(config),
+    diag: require('../diag')(config)
   }
 
   Object.assign(cmds.refs, {
@@ -123,7 +124,6 @@ function requireCommands (send, config) {
 
     // Miscellaneous
     commands: require('../commands'),
-    diag: require('../diag'),
     id: require('../id'),
     key: require('../key'),
     log: require('../log'),