keys file

Author: humanagent

examples/gated-group/src/keys.ts

@@ -3,15 +3,21 @@ import * as fs from "node:fs";
 import path from "path";
 import { toHex } from "viem";
 
-export function generateKeys(suffix: string) {
-  let encryptionKey =
+export function generateKeys(
+  walletKey?: string,
+  encryptionKey?: string,
+  suffix: string = "",
+) {
+  encryptionKey =
+    encryptionKey ??
     process.env["ENCRYPTION_KEY" + suffix] ??
     toHex(getRandomValues(new Uint8Array(32)));
 
   if (!encryptionKey.startsWith("0x")) {
     encryptionKey = "0x" + encryptionKey;
   }
-  let walletKey =
+  walletKey =
+    walletKey ??
     process.env["WALLET_KEY" + suffix] ??
     toHex(getRandomValues(new Uint8Array(32)));
 
@@ -22,9 +28,9 @@ export function generateKeys(suffix: string) {
 }
 
 export function saveKeys(
-  suffix: string,
   walletKey: string,
   encryptionKey: string,
+  suffix: string = "",
 ) {
   const envFilePath = path.resolve(process.cwd(), ".env");
   const envContent = `\nENCRYPTION_KEY${suffix}=${encryptionKey}\nWALLET_KEY${suffix}=${walletKey}`;

examples/gated-group/src/xmtp.ts

@@ -13,17 +13,22 @@ import { createSigner, createUser } from "./viem.js";
 dotenv.config();
 
 export async function createClient({
-  suffix = "",
+  suffix,
+  walletKey,
+  encryptionKey,
   options,
   streamMessageCallback,
 }: {
   suffix?: string;
+  walletKey?: string;
+  encryptionKey?: string;
   options?: ClientOptions;
   streamMessageCallback?: (message: DecodedMessage) => Promise<void>;
 }): Promise<Client> {
-  const { walletKey, encryptionKey } = generateKeys(suffix);
+  const { walletKey: clientWalletKey, encryptionKey: clientEncryptionKey } =
+    generateKeys(walletKey, encryptionKey, suffix);
 
-  const user = createUser(walletKey);
+  const user = createUser(clientWalletKey);
 
   const env = options?.env ?? "production";
 
@@ -47,7 +52,7 @@ export async function createClient({
   if (streamMessageCallback) {
     void streamMessages(streamMessageCallback, client);
   }
-  saveKeys(suffix, walletKey, encryptionKey);
+  saveKeys(clientWalletKey, clientEncryptionKey, suffix);
   return client;
 }
 

examples/gm/src/keys.ts

@@ -3,15 +3,21 @@ import * as fs from "node:fs";
 import path from "path";
 import { toHex } from "viem";
 
-export function generateKeys(suffix: string) {
-  let encryptionKey =
+export function generateKeys(
+  walletKey?: string,
+  encryptionKey?: string,
+  suffix: string = "",
+) {
+  encryptionKey =
+    encryptionKey ??
     process.env["ENCRYPTION_KEY" + suffix] ??
     toHex(getRandomValues(new Uint8Array(32)));
 
   if (!encryptionKey.startsWith("0x")) {
     encryptionKey = "0x" + encryptionKey;
   }
-  let walletKey =
+  walletKey =
+    walletKey ??
     process.env["WALLET_KEY" + suffix] ??
     toHex(getRandomValues(new Uint8Array(32)));
 
@@ -22,9 +28,9 @@ export function generateKeys(suffix: string) {
 }
 
 export function saveKeys(
-  suffix: string,
   walletKey: string,
   encryptionKey: string,
+  suffix: string = "",
 ) {
   const envFilePath = path.resolve(process.cwd(), ".env");
   const envContent = `\nENCRYPTION_KEY${suffix}=${encryptionKey}\nWALLET_KEY${suffix}=${walletKey}`;

examples/gm/src/xmtp.ts

@@ -13,17 +13,22 @@ import { createSigner, createUser } from "./viem.js";
 dotenv.config();
 
 export async function createClient({
-  suffix = "",
+  suffix,
+  walletKey,
+  encryptionKey,
   options,
   streamMessageCallback,
 }: {
   suffix?: string;
+  walletKey?: string;
+  encryptionKey?: string;
   options?: ClientOptions;
   streamMessageCallback?: (message: DecodedMessage) => Promise<void>;
 }): Promise<Client> {
-  const { walletKey, encryptionKey } = generateKeys(suffix);
+  const { walletKey: clientWalletKey, encryptionKey: clientEncryptionKey } =
+    generateKeys(walletKey, encryptionKey, suffix);
 
-  const user = createUser(walletKey);
+  const user = createUser(clientWalletKey);
 
   const env = options?.env ?? "production";
 
@@ -47,7 +52,7 @@ export async function createClient({
   if (streamMessageCallback) {
     void streamMessages(streamMessageCallback, client);
   }
-  saveKeys(suffix, walletKey, encryptionKey);
+  saveKeys(clientWalletKey, clientEncryptionKey, suffix);
   return client;
 }