Allow signing transactions for a custom chain id

package.json

 {
   "name": "@hiveio/metamask-snap",
-  "version": "1.2.1",
+  "version": "1.3.1",
   "description": "Hive wallet extension allowing you to sign transactions using keys derived from your Metamask wallet",
   "main": "./dist/bundle.js",
   "files": [

snap.manifest.json

 {
-  "version": "1.2.1",
+  "version": "1.3.1",
   "description": "Hive wallet extension allowing you to sign transactions using keys derived from your Metamask wallet",
   "proposedName": "Hive Wallet",
   "repository": {
@@ -7,7 +7,7 @@
     "url": "git+https://gitlab.syncad.com/hive/metamask-snap.git"
   },
   "source": {
-    "shasum": "VxtYK0n+vdiIHkFdTswCQng05NPFRb1wZ78mBBP5dbk=",
+    "shasum": "8OiCG07i8jcpwdDezZyvzyBWJftYgWno1hcNL/gUWjc=",
     "location": {
       "npm": {
         "filePath": "dist/bundle.js",

src/hive/wax.ts

-import { createWaxFoundation, type IWaxBaseInterface } from "@hiveio/wax";
+import { createWaxFoundation, DEFAULT_WAX_OPTIONS, type IWaxBaseInterface } from "@hiveio/wax";
 
-let _wax: undefined | IWaxBaseInterface;
-export const getWax = async (): Promise<IWaxBaseInterface> => {
-  if (!_wax)
-    _wax = await createWaxFoundation();
+const waxInstances: Record<string, IWaxBaseInterface> = {};
+export const getWax = async (chainId: string = DEFAULT_WAX_OPTIONS.chainId): Promise<IWaxBaseInterface> => {
+  if (!waxInstances[chainId])
+    waxInstances[chainId] = await createWaxFoundation(chainId !== undefined ? { chainId } : undefined);
 
-  return _wax;
+  return waxInstances[chainId];
 };

src/index.ts

@@ -27,7 +27,7 @@ export const onRpcRequest = async ({
 
     case 'hive_signTransaction':
       return {
-        signatures: await signTransaction(origin, request.params.transaction, request.params.keys)
+        signatures: await signTransaction(origin, request.params.transaction, request.params.keys, request.params.chainId)
       };
 
     case 'hive_decrypt':

src/rpc.ts

@@ -20,6 +20,7 @@ export type SignTransactionRequest = {
   method: 'hive_signTransaction';
   params: {
     transaction: string;
+    chainId?: string;
     keys: KeyIndex[];
   };
 }

src/snap/dialogs/ConfirmTransactionSign.tsx

-import { Bold, Copyable, Text, Box } from "@metamask/snaps-sdk/jsx";
+import { Bold, Copyable, Text, Box, Banner, Italic } from "@metamask/snaps-sdk/jsx";
 import { KeyIndex } from "../../rpc";
 import { KeyTypeNotice } from "./components/KeyTypeNotice";
+import { DEFAULT_WAX_OPTIONS } from "@hiveio/wax";
 
-export const ConfirmTransactionSign = (origin: string, transaction: string, keys: KeyIndex[]) => snap.request({
+export const ConfirmTransactionSign = (origin: string, transaction: string, keys: KeyIndex[], chainId?: string) => snap.request({
   method: 'snap_dialog',
   params: {
     type: 'confirmation',
     content: (
       <Box>
         <Text>
-          <Bold>{ origin }</Bold> asked to sign a transaction:
+          <Bold>{origin}</Bold> asked to sign a transaction:
         </Text>
-        <Copyable value={ transaction } />
+        <Copyable value={transaction} />
+        {chainId && chainId !== DEFAULT_WAX_OPTIONS.chainId && <Banner title="Custom chain signing" severity="warning">
+          <Text>
+            <Bold>Warning:</Bold> You are signing this transaction for a custom chain. Make sure you trust this chain: <Italic>{chainId}</Italic>
+          </Text>
+        </Banner>}
         <Text>
           Confirm if you want to sign it using your:
         </Text>

src/snap/encodeBuffer.ts

@@ -8,7 +8,7 @@ export const encodeBuffer = async (origin: string, buffer: string, firstKey: Key
   const confirmDecode = await ConfirmBufferSign(origin, buffer, firstKey, secondKey);
 
   if(!confirmDecode)
-    throw new Error('User denied the buffer decode');
+    throw new Error('User denied the buffer encode');
 
   // The order is important: First create wax, then create wallet
   const wax = await getWax();

src/snap/signTransaction.ts

@@ -5,17 +5,17 @@ import { getTempWallet } from "../hive/beekeeper";
 import { ConfirmTransactionSign } from "./dialogs/ConfirmTransactionSign";
 import type { THexString } from "@hiveio/wax";
 
-export const signTransaction = async (origin: string, transaction: string, keys: KeyIndex[]): Promise<THexString[]> => {
+export const signTransaction = async (origin: string, transaction: string, keys: KeyIndex[], chainId?: string): Promise<THexString[]> => {
   if (keys.length < 1)
     throw new Error('No keys provided');
 
-  const confirmSign = await ConfirmTransactionSign(origin, transaction, keys);
+  const confirmSign = await ConfirmTransactionSign(origin, transaction, keys, chainId);
 
   if(!confirmSign)
-    throw new Error('User denied the transaction');
+    throw new Error('User denied the transaction signing');
 
   // The order is important: First create wax, then transaction and if all success then create wallet
-  const wax = await getWax();
+  const wax = await getWax(chainId);
   const tx = wax.createTransactionFromJson(transaction);
   const wallet = await getTempWallet();