From 2e45098033aef77d9495b0e5be5986fac5bba581 Mon Sep 17 00:00:00 2001
From: csrapr <26280794+csrapr@users.noreply.github.com>
Date: Tue, 25 May 2021 18:44:05 +0100
Subject: [PATCH] Fix: lamassu-register with async local store

---
 bin/lamassu-register | 31 +++++++++++++++++--------------
 1 file changed, 17 insertions(+), 14 deletions(-)

diff --git a/bin/lamassu-register b/bin/lamassu-register
index 1e73f714..1fbb98a4 100755
--- a/bin/lamassu-register
+++ b/bin/lamassu-register
@@ -1,5 +1,6 @@
 #!/usr/bin/env node
 
+const { asyncLocalStorage, defaultStore } = require('../lib/async-storage')
 const authentication = require('../lib/new-admin/graphql/modules/authentication')
 const options = require('../lib/options')
 
@@ -29,20 +30,22 @@ if (role !== 'user' && role !== 'superuser') {
   process.exit(2)
 }
 
-authentication.createRegisterToken(name, role).then(token => {
-  if (!token) {
-    console.log(`A user named ${name} already exists!`)
-    process.exit(2)
-  }
+asyncLocalStorage.run(defaultStore(), () => {
+  authentication.createRegisterToken(name, role).then(token => {
+    if (!token) {
+      console.log(`A user named ${name} already exists!`)
+      process.exit(2)
+    }
 
-  if (domain === 'localhost') {
-    console.log(`https://${domain}:3001/register?t=${token.token}`)
-  } else {
-    console.log(`https://${domain}/register?t=${token.token}`)
-  }
+    if (domain === 'localhost') {
+      console.log(`https://${domain}:3001/register?t=${token.token}`)
+    } else {
+      console.log(`https://${domain}/register?t=${token.token}`)
+    }
 
-  process.exit(0)
-}).catch(err => {
-  console.log('Error: %s', err)
-  process.exit(3)
+    process.exit(0)
+  }).catch(err => {
+    console.log('Error: %s', err)
+    process.exit(3)
+  })
 })