Consolidate connect logic

mantoni · Aug 21, 2018 · 4c70735 · 4c70735
1 parent 6159e90
commit 4c70735
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 37 deletions.
diff --git a/lib/client.js b/lib/client.js
@@ -1,24 +1,7 @@
 'use strict';
 
-const net = require('net');
 const out = require('./out');
-const portfile = require('./portfile');
-
-function connect(callback) {
-  portfile.read((config) => {
-    if (!config) {
-      callback('Not running');
-      return;
-    }
-    const socket = net.connect(config.port, '127.0.0.1', () => {
-      callback(null, socket, config.token);
-    });
-    socket.once('error', () => {
-      process.exitCode = 1;
-      callback('Could not connect');
-    });
-  });
-}
+const connect = require('./connect');
 
 exports.stop = function (callback) {
   connect((err, socket, token) => {

diff --git a/lib/connect.js b/lib/connect.js
@@ -0,0 +1,20 @@
+'use strict';
+
+const net = require('net');
+const portfile = require('./portfile');
+
+module.exports = function (callback) {
+  portfile.read((config) => {
+    if (!config) {
+      callback('Not running');
+      return;
+    }
+    const socket = net.connect(config.port, '127.0.0.1', () => {
+      callback(null, socket, config.token);
+    });
+    socket.once('error', () => {
+      process.exitCode = 1;
+      callback('Could not connect');
+    });
+  });
+};
diff --git a/lib/launcher.js b/lib/launcher.js
@@ -1,26 +1,10 @@
 'use strict';
 
-const net = require('net');
 const out = require('./out');
-const portfile = require('./portfile');
-
-function check(callback) {
-  portfile.read((config) => {
-    if (!config) {
-      callback('Not running');
-      return;
-    }
-    const socket = net.connect(config.port, '127.0.0.1', () => {
-      callback(null, socket, config.token);
-    });
-    socket.once('error', () => {
-      callback('Could not connect');
-    });
-  });
-}
+const connect = require('./connect');
 
 function wait(callback) {
-  check((err, socket, token) => {
+  connect((err, socket, token) => {
     if (err) {
       setTimeout(() => {
         wait(callback);
@@ -54,7 +38,7 @@ function launch(callback) {
 }
 
 exports.launch = function (callback) {
-  check((err, socket) => {
+  connect((err, socket) => {
     if (err) {
       launch(callback);
     } else {