stop

Author: fadeev

src/commands/start.ts

@@ -181,8 +181,12 @@ const startLocalnet = async (options: {
 
   const enabledChains = options.chains || [];
 
-  // Initialize the processes array
-  const processes: ProcessInfo[] = [];
+  const processes: ProcessInfo[] = [
+    {
+      command: "localnet",
+      pid: process.pid,
+    },
+  ];
 
   try {
     execSync("which anvil");
@@ -332,7 +336,7 @@ const waitForTonContainerToStop = async () => {
   }
 };
 
-const cleanup = async (options: { chains: string[] }) => {
+export const cleanup = async (options: { chains: string[] }) => {
   logger.info("Shutting down processes and cleaning up...");
 
   // Close readline interface if it exists
@@ -349,6 +353,9 @@ const cleanup = async (options: { chains: string[] }) => {
       const processData = JSON.parse(fs.readFileSync(PROCESS_FILE, "utf-8"));
       if (processData && processData.processes) {
         for (const proc of processData.processes) {
+          if (proc.command === "localnet") {
+            continue;
+          }
           try {
             process.kill(proc.pid, "SIGKILL");
             logger.info(

src/commands/stop.ts

@@ -1,44 +1,63 @@
 import ansis from "ansis";
 import { Command } from "commander";
 import fs from "fs";
+import path from "path";
 
-const LOCALNET_JSON_FILE = "./localnet.json";
+import { LOCALNET_DIR, NetworkID, REGISTRY_FILE } from "../constants";
+import { initLogger } from "../logger";
+import { cleanup } from "./start";
 
-const stopLocalnet = () => {
-  if (!fs.existsSync(LOCALNET_JSON_FILE)) {
-    console.log(ansis.red("Localnet is not running or JSON file is missing."));
-    return;
-  }
+const stopLocalnet = async () => {
+  initLogger("info");
+  const chains: string[] = [];
 
-  const jsonData = JSON.parse(fs.readFileSync(LOCALNET_JSON_FILE, "utf-8"));
-  const pid = jsonData.pid;
+  // Read localnet PID before doing anything
+  const processFile = path.join(LOCALNET_DIR, "process.json");
+  let localnetPid: number | undefined;
+  try {
+    if (fs.existsSync(processFile)) {
+      const data = JSON.parse(fs.readFileSync(processFile, "utf-8"));
+      const processes =
+        (data?.processes as { command: string; pid: number }[]) || [];
+      localnetPid = processes.find((p) => p?.command === "localnet")?.pid;
+    }
+  } catch (err) {
+    // Ignore errors when reading PID; proceed with best-effort cleanup
+  }
 
+  // Derive whether TON was enabled from the registry (only needed for TON Docker shutdown)
   try {
-    process.kill(Number(pid), 0); // check that the process is running
-    try {
-      process.kill(Number(pid));
-      console.log(ansis.green(`Successfully stopped localnet (PID: ${pid})`));
-    } catch (err) {
-      console.error(ansis.red(`Failed to stop localnet: ${err}`));
+    if (fs.existsSync(REGISTRY_FILE)) {
+      const registry = JSON.parse(fs.readFileSync(REGISTRY_FILE, "utf-8"));
+      if (registry && registry[NetworkID.TON]) {
+        chains.push("ton");
+      }
     }
   } catch (err) {
-    console.log(ansis.yellow(`Localnet process (PID: ${pid}) is not running.`));
-    try {
-      fs.unlinkSync(LOCALNET_JSON_FILE);
-      console.log(ansis.green("Localnet JSON file deleted."));
-    } catch (err) {
-      console.error(ansis.red(`Failed to delete JSON file: ${err}`));
+    // If registry is unreadable, proceed with best-effort cleanup
+  }
+
+  try {
+    await cleanup({ chains });
+
+    // After cleanup is done, kill localnet by PID if we captured it
+    if (typeof localnetPid === "number" && Number.isFinite(localnetPid)) {
+      try {
+        process.kill(localnetPid, "SIGKILL");
+      } catch {
+        // Ignore if already stopped or cannot kill
+      }
     }
+
+    console.log(ansis.green("Localnet stopped successfully."));
+  } catch (error) {
+    console.error(ansis.red(`Failed to stop localnet: ${error}`));
+    process.exit(1);
   }
 };
 
 export const stopCommand = new Command("stop")
   .description("Stop localnet")
-  .action(() => {
-    try {
-      stopLocalnet();
-    } catch (error) {
-      console.error(ansis.red(`Error: ${error}`));
-      process.exit(1);
-    }
+  .action(async () => {
+    await stopLocalnet();
   });