From 5b3177563ab776e7afdcf0edc747e451d14f7f34 Mon Sep 17 00:00:00 2001
From: h0ngcha0 <lbs.lhc@gmail.com>
Date: Tue, 13 Feb 2024 22:45:38 +0100
Subject: [PATCH] Fix devnet.ts script

---
 scripts/devnet.ts | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/scripts/devnet.ts b/scripts/devnet.ts
index 30ba7c8..2a015fd 100644
--- a/scripts/devnet.ts
+++ b/scripts/devnet.ts
@@ -164,14 +164,18 @@ export async function getCreatedContractId(provider: NodeProvider, txId: string,
   return binToHex(contractIdFromAddress(address))
 }
 
-function getSigner(privateKeys: string[], index: number) {
-  if (privateKeys.length === 0) {
-    throw new Error('No private key specified')
-  }
-  if (index < 0 || index >= privateKeys.length) {
-    throw new Error(`Invalid private key index: ${index}`)
+function getSigner(privateKeys: string[] | string, index: number) {
+  if (Array.isArray(privateKeys)) {
+    if (privateKeys.length === 0) {
+      throw new Error('No private key specified')
+    }
+    if (index < 0 || index >= privateKeys.length) {
+      throw new Error(`Invalid private key index: ${index}`)
+    }
+    return new PrivateKeyWallet({ privateKey: privateKeys[index] })
+  } else {
+    return new PrivateKeyWallet({ privateKey: privateKeys })
   }
-  return new PrivateKeyWallet({ privateKey: privateKeys[index] })
 }
 
 program