From f0bfc44e7246c03655c02e9d644cbcceb1172ea2 Mon Sep 17 00:00:00 2001
From: Craig Raw <craigraw@gmail.com>
Date: Thu, 28 Mar 2024 12:01:48 +0200
Subject: [PATCH] avoid saving lock file link for default instance if
 environment variable is set

---
 .../sparrow/instance/Instance.java              | 17 ++++++++++++++---
 .../com/sparrowwallet/sparrow/io/Storage.java   |  8 ++------
 2 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/src/main/java/com/sparrowwallet/sparrow/instance/Instance.java b/src/main/java/com/sparrowwallet/sparrow/instance/Instance.java
index b667952d..c684252f 100644
--- a/src/main/java/com/sparrowwallet/sparrow/instance/Instance.java
+++ b/src/main/java/com/sparrowwallet/sparrow/instance/Instance.java
@@ -4,6 +4,7 @@ import com.sparrowwallet.sparrow.io.Storage;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.io.File;
 import java.io.IOException;
 import java.net.ConnectException;
 import java.net.SocketException;
@@ -23,6 +24,7 @@ import java.util.Set;
 
 public abstract class Instance {
     private static final Logger log = LoggerFactory.getLogger(Instance.class);
+    private static final String LINK_ENV_PROPERTY = "SPARROW_NO_LOCK_FILE_LINK";
 
     public final String applicationId;
     private final boolean autoExit;
@@ -64,7 +66,7 @@ public abstract class Instance {
             serverChannel.register(selector, SelectionKey.OP_ACCEPT);
             lockFile.toFile().deleteOnExit();
         } catch(Exception e) {
-            throw new InstanceException("Could not open UNIX socket at " + lockFile.toAbsolutePath(), e);
+            throw new InstanceException("Could not open UNIX socket lock file for instance at " + lockFile.toAbsolutePath(), e);
         }
 
         Thread thread = new Thread(() -> {
@@ -121,7 +123,7 @@ public abstract class Instance {
             try {
                 Files.deleteIfExists(lockFile);
                 startServer(lockFile);
-            } catch(IOException ex) {
+            } catch(Exception ex) {
                 throw new InstanceException("Could not delete lock file from previous instance", e);
             }
         } catch(Exception e) {
@@ -187,8 +189,17 @@ public abstract class Instance {
     }
 
     private Path getUserLockFilePointer() {
+        if(Boolean.parseBoolean(System.getenv(LINK_ENV_PROPERTY))) {
+            return null;
+        }
+
         try {
-            return Storage.getSparrowDir(true).toPath().resolve(applicationId + ".default");
+            File sparrowHome = Storage.getSparrowHome(true);
+            if(!sparrowHome.exists()) {
+                Storage.createOwnerOnlyDirectory(sparrowHome);
+            }
+
+            return sparrowHome.toPath().resolve(applicationId + ".default");
         } catch(Exception e) {
             return null;
         }
diff --git a/src/main/java/com/sparrowwallet/sparrow/io/Storage.java b/src/main/java/com/sparrowwallet/sparrow/io/Storage.java
index 223434e8..1af64d79 100644
--- a/src/main/java/com/sparrowwallet/sparrow/io/Storage.java
+++ b/src/main/java/com/sparrowwallet/sparrow/io/Storage.java
@@ -536,15 +536,11 @@ public class Storage {
     }
 
     public static File getSparrowDir() {
-        return getSparrowDir(false);
-    }
-
-    public static File getSparrowDir(boolean useDefault) {
         File sparrowDir;
         if(Network.get() != Network.MAINNET) {
-            sparrowDir = new File(getSparrowHome(useDefault), Network.get().getName());
+            sparrowDir = new File(getSparrowHome(), Network.get().getName());
         } else {
-            sparrowDir = getSparrowHome(useDefault);
+            sparrowDir = getSparrowHome();
         }
 
         if(!sparrowDir.exists()) {