Some cleanup

packages/safe-chain/src/shell-integration/supported-shells/fish.js

@@ -3,6 +3,7 @@ import {
   doesExecutableExistOnSystem,
   removeLinesMatchingPattern,
   getScriptsDir,
+  getSafeChainDir,
 } from "../helpers.js";
 import { execSync } from "child_process";
 import path from "path";
@@ -40,12 +41,27 @@ function teardown(tools) {
     eol
   );
 
+  removeLinesMatchingPattern(
+    startupFile,
+    /^set\s+-gx\s+SAFE_CHAIN_DIR\s+.*#\s*Safe-chain/,
+    eol
+  );
+
   return true;
 }
 
 function setup() {
   const startupFile = getStartupFile();
 
+  const customDir = getSafeChainDir();
+  if (customDir) {
+    addLineToFile(
+      startupFile,
+      `set -gx SAFE_CHAIN_DIR "${customDir}" # Safe-chain installation directory`,
+      eol
+    );
+  }
+
   addLineToFile(
     startupFile,
     `source ${path.join(getScriptsDir(), "init-fish.fish")} # Safe-chain Fish initialization script`,