From be6e0c00ddd9f63dca976fe3c847938e5625c9cc Mon Sep 17 00:00:00 2001
From: Denis Fadeev <denis@fadeev.org>
Date: Fri, 8 Nov 2024 16:22:36 +0300
Subject: [PATCH] Use `localnet.json` containing both PID and addresses (#59)

---
 packages/tasks/src/check.ts    |  9 +++++----
 packages/tasks/src/localnet.ts | 12 ++++++++----
 packages/tasks/src/stop.ts     | 15 ++++++++-------
 3 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/packages/tasks/src/check.ts b/packages/tasks/src/check.ts
index 4b9d568..ca05aa1 100644
--- a/packages/tasks/src/check.ts
+++ b/packages/tasks/src/check.ts
@@ -2,17 +2,18 @@ import { task, types } from "hardhat/config";
 import fs from "fs";
 import ansis from "ansis";
 
-const LOCALNET_PID_FILE = "./localnet.pid";
+const LOCALNET_JSON_FILE = "./localnet.json";
 
 const localnetCheck = async (args: any) => {
   await new Promise((resolve) => setTimeout(resolve, args.delay * 1000));
 
-  if (!fs.existsSync(LOCALNET_PID_FILE)) {
-    console.log(ansis.red("Localnet is not running (PID file missing)."));
+  if (!fs.existsSync(LOCALNET_JSON_FILE)) {
+    console.log(ansis.red("Localnet is not running (JSON file missing)."));
     process.exit(1);
   }
 
-  const pid = fs.readFileSync(LOCALNET_PID_FILE, "utf-8").trim();
+  const jsonData = JSON.parse(fs.readFileSync(LOCALNET_JSON_FILE, "utf-8"));
+  const pid = jsonData.pid;
 
   try {
     process.kill(Number(pid), 0);
diff --git a/packages/tasks/src/localnet.ts b/packages/tasks/src/localnet.ts
index 3490674..de893eb 100644
--- a/packages/tasks/src/localnet.ts
+++ b/packages/tasks/src/localnet.ts
@@ -6,7 +6,7 @@ import ansis from "ansis";
 import fs from "fs";
 import { confirm } from "@inquirer/prompts";
 
-const LOCALNET_PID_FILE = "./localnet.pid";
+const LOCALNET_JSON_FILE = "./localnet.json";
 
 const killProcessOnPort = async (port: number, forceKill: boolean) => {
   try {
@@ -83,8 +83,8 @@ const localnet = async (args: any) => {
     if (anvilProcess) {
       anvilProcess.kill();
     }
-    if (fs.existsSync(LOCALNET_PID_FILE)) {
-      fs.unlinkSync(LOCALNET_PID_FILE);
+    if (fs.existsSync(LOCALNET_JSON_FILE)) {
+      fs.unlinkSync(LOCALNET_JSON_FILE);
     }
   };
 
@@ -130,7 +130,11 @@ const localnet = async (args: any) => {
 
     console.table(zetaAddresses);
 
-    fs.writeFileSync(LOCALNET_PID_FILE, process.pid.toString(), "utf-8");
+    fs.writeFileSync(
+      LOCALNET_JSON_FILE,
+      JSON.stringify({ pid: process.pid, addresses: addr }, null, 2),
+      "utf-8"
+    );
   } catch (error: any) {
     console.error(ansis.red`Error initializing localnet: ${error}`);
     cleanup();
diff --git a/packages/tasks/src/stop.ts b/packages/tasks/src/stop.ts
index 4d7091f..de58235 100644
--- a/packages/tasks/src/stop.ts
+++ b/packages/tasks/src/stop.ts
@@ -2,15 +2,16 @@ import { task } from "hardhat/config";
 import fs from "fs";
 import ansis from "ansis";
 
-const LOCALNET_PID_FILE = "./localnet.pid";
+const LOCALNET_JSON_FILE = "./localnet.json";
 
 const localnetStop = async (args: any) => {
-  if (!fs.existsSync(LOCALNET_PID_FILE)) {
-    console.log(ansis.red("Localnet is not running or PID file is missing."));
+  if (!fs.existsSync(LOCALNET_JSON_FILE)) {
+    console.log(ansis.red("Localnet is not running or JSON file is missing."));
     return;
   }
 
-  const pid = fs.readFileSync(LOCALNET_PID_FILE, "utf-8").trim();
+  const jsonData = JSON.parse(fs.readFileSync(LOCALNET_JSON_FILE, "utf-8"));
+  const pid = jsonData.pid;
 
   try {
     process.kill(Number(pid), 0); // check that the process is running
@@ -23,10 +24,10 @@ const localnetStop = async (args: any) => {
   } catch (err) {
     console.log(ansis.yellow(`Localnet process (PID: ${pid}) is not running.`));
     try {
-      fs.unlinkSync(LOCALNET_PID_FILE);
-      console.log(ansis.green("Localnet PID file deleted."));
+      fs.unlinkSync(LOCALNET_JSON_FILE);
+      console.log(ansis.green("Localnet JSON file deleted."));
     } catch (err) {
-      console.error(ansis.red(`Failed to delete PID file: ${err}`));
+      console.error(ansis.red(`Failed to delete JSON file: ${err}`));
     }
   }
 };