From e30008b36765f1dc86c6fe7d7b70b14bd773efb2 Mon Sep 17 00:00:00 2001
From: "Hong Jing (Jingles)" <jinglescode@gmail.com>
Date: Fri, 15 Nov 2024 18:29:21 +0800
Subject: [PATCH] update guide

---
 .../apis/wallets/browserwallet/sign-data.tsx   |  1 -
 apps/playground/src/pages/governance/index.mdx |  4 ----
 .../guides/prove-wallet-ownership/demo.tsx     | 17 +++++++++++++++--
 .../guides/prove-wallet-ownership/index.mdx    | 18 +++++++-----------
 4 files changed, 22 insertions(+), 18 deletions(-)

diff --git a/apps/playground/src/pages/apis/wallets/browserwallet/sign-data.tsx b/apps/playground/src/pages/apis/wallets/browserwallet/sign-data.tsx
index 52fd07428..0372b1c07 100644
--- a/apps/playground/src/pages/apis/wallets/browserwallet/sign-data.tsx
+++ b/apps/playground/src/pages/apis/wallets/browserwallet/sign-data.tsx
@@ -1,6 +1,5 @@
 import { useState } from "react";
 
-import { checkSignature } from "@meshsdk/core";
 import { useWallet } from "@meshsdk/react";
 
 import Input from "~/components/form/input";
diff --git a/apps/playground/src/pages/governance/index.mdx b/apps/playground/src/pages/governance/index.mdx
index cb10eaafd..755461ad5 100644
--- a/apps/playground/src/pages/governance/index.mdx
+++ b/apps/playground/src/pages/governance/index.mdx
@@ -1,8 +1,4 @@
 import { useState } from "react";
-
-import { checkSignature, generateNonce } from "@meshsdk/core";
-import { CardanoWallet, useWallet } from "@meshsdk/react";
-
 import Button from "~/components/button/button";
 import LayoutImageHeaderAndBody from "~/components/layouts/image-header-and-body";
 
diff --git a/apps/playground/src/pages/guides/prove-wallet-ownership/demo.tsx b/apps/playground/src/pages/guides/prove-wallet-ownership/demo.tsx
index 43da0d2aa..7b0a32fdb 100644
--- a/apps/playground/src/pages/guides/prove-wallet-ownership/demo.tsx
+++ b/apps/playground/src/pages/guides/prove-wallet-ownership/demo.tsx
@@ -5,6 +5,7 @@ import { CardanoWallet, useWallet } from "@meshsdk/react";
 
 import Button from "~/components/button/button";
 import DemoResult from "~/components/sections/demo-result";
+import Codeblock from "~/components/text/codeblock";
 
 export default function Demo() {
   const { wallet, connected } = useWallet();
@@ -16,15 +17,27 @@ export default function Demo() {
 
     setLoading(true);
     const nonce = generateNonce("Sign to login in to Mesh: ");
-    const signature = await wallet.signData(nonce);
+    const address = (await wallet.getUsedAddresses())[0];
+    const signature = await wallet.signData(nonce, address);
 
-    const result = checkSignature(nonce, signature);
+    const result = checkSignature(nonce, signature, address);
     setResponse(result.toString());
     setLoading(false);
   }
 
+  let code = ``;
+  code += `import { checkSignature, generateNonce } from "@meshsdk/core";\n`;
+  code += `\n`;
+  code += `const nonce = generateNonce("Sign to login in to Mesh: ");\n`;
+  code += `\n`;
+  code += `const address = (await wallet.getUsedAddresses())[0];\n`;
+  code += `const signature = await wallet.signData(nonce, address);\n`;
+  code += `\n`;
+  code += `const result = checkSignature(nonce, signature, address);\n`;
+
   return (
     <>
+      <Codeblock data={code} />
       <p>Connect your wallet and click on the button to sign a message.</p>
       <CardanoWallet />
       <Button
diff --git a/apps/playground/src/pages/guides/prove-wallet-ownership/index.mdx b/apps/playground/src/pages/guides/prove-wallet-ownership/index.mdx
index c734e1452..53b426fb2 100644
--- a/apps/playground/src/pages/guides/prove-wallet-ownership/index.mdx
+++ b/apps/playground/src/pages/guides/prove-wallet-ownership/index.mdx
@@ -1,8 +1,4 @@
 import { useState } from "react";
-
-import { checkSignature, generateNonce } from "@meshsdk/core";
-import { CardanoWallet, useWallet } from "@meshsdk/react";
-
 import Button from "~/components/button/button";
 import LayoutImageHeaderAndBody from "~/components/layouts/image-header-and-body";
 import { guideownership } from "~/data/links-guides";
@@ -66,7 +62,7 @@ To ensure the signature is valid, the same cryptographic algorithm is applied to
 
 The User model stored in the database of the backend server must have two compulsory fields: public address and nonce. Furthermore, this address has to be unique. Other details about the user, such as username, Twitter ID, and name fields can be added, but are not essential for this process.
 
-On Cardano, to obtain a user's public address as an identifier, we can use their wallet's staking address. This will be stored in the server side database, so that authorized wallets can be linked. The user never has to worry about manually entering their address, since it can be retrieved using **wallet.getRewardAddresses()**.
+On Cardano, to obtain a user's public address as an identifier, we can use their wallet's staking address. This will be stored in the server side database, so that authorized wallets can be linked. The user never has to worry about manually entering their address, since it can be retrieved using **wallet.getUsedAddresses()**.
 
 With the user's wallet connected, the first step is to get the user's staking address and send it to our backend server.
 
@@ -75,7 +71,7 @@ const { wallet, connected } = useWallet();
 
 async function frontendStartLoginProcess() {
   if (connected) {
-    const userAddress = (await wallet.getRewardAddresses())[0];
+    const userAddress = (await wallet.getUsedAddresses())[0];
 
      // do: send request with 'userAddress' to the backend
   }
@@ -117,7 +113,7 @@ We request the user's authorization and show them the message that is to be sign
 ```
 async function frontendSignMessage(nonce) {
   try {
-    const userAddress = (await wallet.getRewardAddresses())[0];
+    const userAddress = (await wallet.getUsedAddresses())[0];
     const signature = await wallet.signData(nonce, userAddress);
 
     // do: send request with 'signature' and 'userAddress' to the backend
@@ -143,7 +139,7 @@ import { checkSignature } from '@meshsdk/core';
 async function backendVerifySignature(userAddress, signature) {
   // do: get 'nonce' from user (database) using 'userAddress'
 
-  const result = checkSignature(nonce, signature);
+  const result = checkSignature(nonce, signature, userAddress);
 
   // do: update 'nonce' in the database with another random string
 
@@ -177,7 +173,7 @@ export default function Page() {
 
   async function frontendStartLoginProcess() {
     if (connected) {
-      const userAddress = (await wallet.getRewardAddresses())[0];
+      const userAddress = (await wallet.getUsedAddresses())[0];
       const nonce = await backendGetNonce(userAddress);
       await frontendSignMessage(nonce);
     }
@@ -185,7 +181,7 @@ export default function Page() {
 
   async function frontendSignMessage(nonce) {
     try {
-      const userAddress = (await wallet.getRewardAddresses())[0];
+      const userAddress = (await wallet.getUsedAddresses())[0];
       const signature = await wallet.signData(nonce, userAddress);
       await backendVerifySignature(userAddress, signature);
     } catch (error) {
@@ -217,7 +213,7 @@ async function backendGetNonce(userAddress) {
 async function backendVerifySignature(userAddress, signature) {
   // do: get 'nonce' from database
 
-  const result = checkSignature(nonce, signature);
+  const result = checkSignature(nonce, signature, userAddress);
   if(result){
     // create JWT or approve certain process
   }