From df7145eccd695723cf5206723271c23c2e08c438 Mon Sep 17 00:00:00 2001
From: mikera <mike@mikera.net>
Date: Wed, 25 Dec 2024 11:05:57 +0000
Subject: [PATCH] Add Protonet warning before running stress test

---
 convex-core/src/main/java/convex/core/Networks.java   |  8 ++++++++
 convex-core/src/main/java/convex/core/cvm/Peer.java   |  8 ++++++++
 .../src/main/java/convex/gui/server/StressPanel.java  | 11 ++++++++++-
 3 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 convex-core/src/main/java/convex/core/Networks.java

diff --git a/convex-core/src/main/java/convex/core/Networks.java b/convex-core/src/main/java/convex/core/Networks.java
new file mode 100644
index 000000000..c25542314
--- /dev/null
+++ b/convex-core/src/main/java/convex/core/Networks.java
@@ -0,0 +1,8 @@
+package convex.core;
+
+import convex.core.data.Hash;
+
+public class Networks {
+
+	public static final Hash PRONONET_GENESIS=Hash.parse("0xb0e44f2a645abfa539f5b96b7a0eabb0f902866feaff0f7c12d1213e02333f13");
+}
diff --git a/convex-core/src/main/java/convex/core/cvm/Peer.java b/convex-core/src/main/java/convex/core/cvm/Peer.java
index 5bbc25120..770b6fac3 100644
--- a/convex-core/src/main/java/convex/core/cvm/Peer.java
+++ b/convex-core/src/main/java/convex/core/cvm/Peer.java
@@ -677,4 +677,12 @@ public boolean isReadyToPublish() {
 		if (ps.getBalance()<CPoSConstants.MINIMUM_EFFECTIVE_STAKE) return false; 
 		return true;
 	}
+
+	/**
+	 * Gets the genesis state hash for this peer
+	 * @return
+	 */
+	public Hash getGenesisHash() {
+		return getGenesisState().getHash();
+	}
 }
diff --git a/convex-gui/src/main/java/convex/gui/server/StressPanel.java b/convex-gui/src/main/java/convex/gui/server/StressPanel.java
index 2856266d6..127d33941 100644
--- a/convex-gui/src/main/java/convex/gui/server/StressPanel.java
+++ b/convex-gui/src/main/java/convex/gui/server/StressPanel.java
@@ -19,6 +19,7 @@
 import javax.swing.JCheckBox;
 import javax.swing.JComboBox;
 import javax.swing.JLabel;
+import javax.swing.JOptionPane;
 import javax.swing.JPanel;
 import javax.swing.JSpinner;
 import javax.swing.JSplitPane;
@@ -30,6 +31,7 @@
 import org.slf4j.LoggerFactory;
 
 import convex.api.Convex;
+import convex.core.Networks;
 import convex.core.Result;
 import convex.core.crypto.AKeyPair;
 import convex.core.cvm.transactions.ATransaction;
@@ -38,6 +40,7 @@
 import convex.core.cvm.transactions.Transfer;
 import convex.core.data.ACell;
 import convex.core.data.AVector;
+import convex.core.data.Hash;
 import convex.core.cvm.Address;
 import convex.core.data.Strings;
 import convex.core.lang.Reader;
@@ -83,6 +86,12 @@ public StressPanel(Convex peerView) {
 		btnRun = new JButton("Run Test");
 		actionPanel.add(btnRun);
 		btnRun.addActionListener(e -> {
+			Hash network=peerView.getLocalServer().getPeer().getGenesisHash();
+			if (network.equals(Networks.PRONONET_GENESIS)) {
+				int confirm=JOptionPane.showConfirmDialog(this, "This is the live network. Running a stress test is likley to be expensive! Are you really sure you want to do this?", "Run test on Live network?", JOptionPane.WARNING_MESSAGE);
+				if (confirm!=JOptionPane.OK_OPTION) return;
+			}
+			
 			btnRun.setEnabled(false);
 			Address address=peerConvex.getAddress();
 			AKeyPair kp=peerConvex.getKeyPair();
@@ -238,7 +247,7 @@ protected String doStressRun() throws Exception {
 		
 			// Generate client accounts
 			StringBuilder cmdsb=new StringBuilder();
-			cmdsb.append("(let [f (fn [k] (let [a (create-account k)] (transfer a 1000000000) a))] ");
+			cmdsb.append("(let [f (fn [k] (let [a (deploy `(do (set-key ~k) (set-controller #13))] (transfer a 1000000000) a))] ");
 			cmdsb.append("  (mapv f [");
 			for (int i=0; i<clientCount; i++) {
 				AKeyPair kp=AKeyPair.generate();