Update index.js

index.js

@@ -17,53 +17,71 @@ const bot = new TelegramBot(botToken);
 app.use(express.json());
 app.use(express.urlencoded({ extended: true }));
 
+let isGenerating = false;
+
 app.get('/', (req, res) => {
   res.send(`
-    <form action="/generate" method="POST">
-      <label>Number of addresses to generate:</label>
-      <input type="number" name="count" required>
-      <button type="submit">Generate</button>
-    </form>
+    <h1>Bitcoin Address Generator</h1>
+    <button onclick="startGeneration()" ${isGenerating ? 'disabled' : ''}>Generate</button>
+    <p>Status: ${isGenerating ? 'Generating addresses...' : 'Idle'}</p>
+    <script>
+      async function startGeneration() {
+        const response = await fetch('/generate', { method: 'POST' });
+        const result = await response.text();
+        alert(result);
+        location.reload();
+      }
+    </script>
   `);
 });
 
-app.post('/generate', (req, res) => {
-  const numberOfAddresses = parseInt(req.body.count, 10);
-  if (isNaN(numberOfAddresses) || numberOfAddresses <= 0) {
-    return res.status(400).send('Invalid number of addresses');
+app.post('/generate', async (req, res) => {
+  if (isGenerating) {
+    return res.send('Generation already in progress');
   }
 
+  isGenerating = true;
+
   // Send Telegram notification when generation starts
-  bot.sendMessage(chatId, `Generating ${numberOfAddresses} addresses...`)
+  bot.sendMessage(chatId, 'Address generation started.')
     .catch(err => console.error('Failed to send start notification:', err.message));
 
+  res.send('Generation started. Check Telegram for any matches.');
+
   const addressList = readAddressesFromFile();
 
-  for (let i = 0; i < numberOfAddresses; i++) {
-    const mnemonic = bip39.generateMnemonic(128);
-    const seed = bip39.mnemonicToSeedSync(mnemonic);
-    const network = bitcoin.networks.bitcoin;
-    const root = bip32.fromSeed(seed, network);
-    const account = root.derivePath("m/84'/0'/0'/0");
-    const addressNode = account.derive(0);
-    const publicKey = addressNode.publicKey;
-
-    const { address: bech32Address } = bitcoin.payments.p2wpkh({ pubkey: publicKey, network });
-    const p2pkhAddress = bitcoin.payments.p2pkh({ pubkey: publicKey, network }).address;
-
-    const addressInfo = `
-      Mnemonic: ${mnemonic}
-      Bech32 SegWit Address: ${bech32Address}
-      P2PKH Address: ${p2pkhAddress}
-    `;
-
-    if (addressList.includes(bech32Address) || addressList.includes(p2pkhAddress)) {
-      sendAddressInfoToBot(addressInfo);
+  // Run generation loop until stopped
+  while (isGenerating) {
+    try {
+      const mnemonic = bip39.generateMnemonic(128);
+      const seed = bip39.mnemonicToSeedSync(mnemonic);
+      const network = bitcoin.networks.bitcoin;
+      const root = bip32.fromSeed(seed, network);
+      const account = root.derivePath("m/84'/0'/0'/0");
+      const addressNode = account.derive(0);
+      const publicKey = addressNode.publicKey;
+
+      const { address: bech32Address } = bitcoin.payments.p2wpkh({ pubkey: publicKey, network });
+      const p2pkhAddress = bitcoin.payments.p2pkh({ pubkey: publicKey, network }).address;
+
+      const addressInfo = `
+        Mnemonic: ${mnemonic}
+        Bech32 SegWit Address: ${bech32Address}
+        P2PKH Address: ${p2pkhAddress}
+      `;
+
+      if (addressList.includes(bech32Address) || addressList.includes(p2pkhAddress)) {
+        sendAddressInfoToBot(addressInfo);
+      }
+
+      // Prevent CPU overload
+      await new Promise(resolve => setTimeout(resolve, 0));
+    } catch (error) {
+      console.error('Error during generation:', error.message);
+      bot.sendMessage(chatId, `Error during generation: ${error.message}`)
+        .catch(err => console.error('Failed to send error notification:', err.message));
     }
   }
-
-  // Send confirmation to webpage without address details
-  res.send(`Addresses generated. Check Telegram for any matches.<br><a href="/">Back</a>`);
 });
 
 function readAddressesFromFile() {
@@ -84,6 +102,19 @@ function sendAddressInfoToBot(info) {
     .catch(err => console.error('Failed to send address notification:', err.message));
 }
 
+// Handle server shutdown gracefully
+process.on('SIGTERM', () => {
+  isGenerating = false;
+  console.log('Server shutting down');
+  process.exit(0);
+});
+
+process.on('SIGINT', () => {
+  isGenerating = false;
+  console.log('Server interrupted');
+  process.exit(0);
+});
+
 app.listen(port, () => {
   console.log(`Server started on port ${port}`);
 });