update keys

Author: humanagent

README.md

@@ -26,9 +26,12 @@ ENCRYPTION_KEY= # encryption key for the local database
 You can generate random keys with the following command:
 
 ```bash
-yarn gen:keys
+yarn gen:keys <name>
 ```
 
+> [!TIP]
+> Running the `gen:keys` or `gen:keys <name>` command will write to the existing `.env` file.
+
 ### Fetching messages
 
 There are to ways to fetch messages from a conversation, one is by starting a stream

scripts/generateKeys.ts

@@ -1,23 +1,39 @@
 import { writeFile } from "node:fs/promises";
 import { join } from "node:path";
-import { generatePrivateKey } from "viem/accounts";
+import { generatePrivateKey, privateKeyToAccount } from "viem/accounts";
 import { generateEncryptionKeyHex } from "@/helpers";
 
 console.log("Generating keys...");
 
+const person = process.argv[2];
 const walletKey = generatePrivateKey();
+const account = privateKeyToAccount(walletKey);
 const encryptionKeyHex = generateEncryptionKeyHex();
+const publicKey = account.address;
 
 const filePath = join(process.cwd(), ".env");
 
-await writeFile(
-  filePath,
-  `WALLET_KEY=${walletKey}
+// Format the environment variables based on whether a person name was provided
+let envContent;
+if (person) {
+  envContent = `# ${person.toLowerCase()}
+WALLET_KEY_${person.toUpperCase()}=${walletKey}
+ENCRYPTION_KEY_${person.toUpperCase()}=${encryptionKeyHex}
+# public key is ${publicKey}
+`;
+} else {
+  envContent = `# generic keys
+WALLET_KEY=${walletKey}
 ENCRYPTION_KEY=${encryptionKeyHex}
-`,
-  {
-    flag: "a",
-  },
-);
+# public key is ${publicKey}
+`;
+}
 
-console.log(`Keys written to ${filePath}`);
+await writeFile(filePath, envContent, { flag: "a" });
+
+// Log appropriate message based on whether a person name was provided
+if (person) {
+  console.log(`Keys for ${person} written to ${filePath}`);
+} else {
+  console.log(`Generic keys written to ${filePath}`);
+}