From a517577a8a02fb5bedc77b7d98f36ee594b47cb8 Mon Sep 17 00:00:00 2001
From: Dennor <matysiewicz.kamil@gmail.com>
Date: Tue, 20 Mar 2018 13:16:27 +0100
Subject: [PATCH] Fix for building registry deps (#195)

---
 packages/cli/src/commands/socket-add.js   | 14 +++++++++-----
 packages/cli/src/utils/sockets/sockets.js | 16 +++++++++++-----
 2 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/packages/cli/src/commands/socket-add.js b/packages/cli/src/commands/socket-add.js
index 942e6ce3..dd9e32fe 100644
--- a/packages/cli/src/commands/socket-add.js
+++ b/packages/cli/src/commands/socket-add.js
@@ -26,12 +26,12 @@ export default class SocketAdd {
       if (cmd.socket) {
         // Socket dependency
         this.socket = await this.Socket.get(cmd.socket)
-        return this.socket.addDependency(socketFromRegistry)
+        await this.socket.addDependency(socketFromRegistry)
+      } else {
+        // Project dependency
+        await this.Socket.add(socketFromRegistry)
       }
 
-      // Project dependency
-      await this.Socket.add(socketFromRegistry)
-
       const status = format.grey('socket added:')
       const name = format.cyan(this.socketFromRegistry.name)
       const version = format.dim(`(${this.socketFromRegistry.version})`)
@@ -61,7 +61,11 @@ export default class SocketAdd {
         }
       } else {
         echo()
-        echo(`${format.red(err)}\n`)
+        if(err.message) {
+          echo(`${format.red(err.message)}\n`)
+        } else {
+          echo(`${format.red(err)}\n`)
+        }
         echo()
       }
     }
diff --git a/packages/cli/src/utils/sockets/sockets.js b/packages/cli/src/utils/sockets/sockets.js
index 7a2a8a09..d6f91bb9 100644
--- a/packages/cli/src/utils/sockets/sockets.js
+++ b/packages/cli/src/utils/sockets/sockets.js
@@ -329,11 +329,10 @@ class Socket {
     debug(`Getting Socket: ${socketName}`)
     const socket = Socket.getLocal(socketName)
     const loadedSocket = await socket.loadRemote()
-    await loadedSocket.getDepsRegistrySockets()
-
     if (!socket.existLocally) {
       await socket.loadFromRegistry()
     }
+    await loadedSocket.getDepsRegistrySockets()
     return socket
   }
 
@@ -564,7 +563,7 @@ class Socket {
 
   getSocketNodeModulesChecksum () {
     debug('getSocketNodeModulesChecksum')
-    return hashdirectory.sync(path.join(this.socketPath, '.dist', 'node_modules'))
+    return hashdirectory.sync(path.join(this.getSocketPath(), '.dist', 'node_modules'))
   }
 
   getSocketConfigFile () {
@@ -1021,9 +1020,16 @@ class Socket {
         await new Promise((resolve, reject) => {
           fs.createReadStream(fileName)
             .pipe(unzip.Extract({ path: this.getSocketPath() }))
-            .on('close', () => {
+            .on('close', async () => {
               debug('Unzip finished')
-              resolve()
+
+              // Build registry socket.
+              try {
+                await this.build()
+              } catch(e) {
+                return reject(e)
+              }
+              return resolve()
             })
         })
       }