Add handlers for account import, export, CRUD

iron-fish · Apr 24, 2024 · b4a6847 · b4a6847
1 parent b9c7fcf
commit b4a6847
Show file tree

Hide file tree

Showing 8 changed files with 278 additions and 58 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/mobile-app/app/(tabs)/transact.tsx b/packages/mobile-app/app/(tabs)/transact.tsx
@@ -2,6 +2,8 @@ import { View, Text, ScrollView, TextInput } from "react-native";
 import { useFacade } from "../../data/facades";
 import { Button } from "@ironfish/ui";
 import { useQueryClient } from "@tanstack/react-query";
+import React from "react";
+import { AccountFormat } from "@ironfish/sdk";
 
 export default function Transact() {
   const facade = useFacade();
@@ -15,8 +17,24 @@ export default function Transact() {
       })
     }
   });
+  const importAccount = facade.importAccount.useMutation({
+    onSuccess: () => {
+      qc.invalidateQueries({
+        queryKey: ['getAccounts']
+      })
+    }
+  });
+  const removeAccount = facade.removeAccount.useMutation({
+    onSuccess: () => {
+      qc.invalidateQueries({
+        queryKey: ['getAccounts']
+      })
+    }
+  });
   const exportAccount = facade.exportAccount.useMutation();
 
+  const [importAccountText, setImportAccountText] = React.useState('')
+
   return (
     <ScrollView>
       <Text>Accounts</Text>
@@ -25,7 +43,15 @@ export default function Transact() {
           <Text>{account.name}</Text>
           <Button
             onClick={async () => {
-              const otherResult = await exportAccount.mutateAsync({ name: account.name });
+              await removeAccount.mutateAsync({ name: account.name });
+              console.log('Removed Account', account.name)
+            }}
+          >
+          Remove Account
+          </Button>
+          <Button
+            onClick={async () => {
+              const otherResult = await exportAccount.mutateAsync({ name: account.name, format: AccountFormat.Base64Json });
               console.log('Exported Account:', otherResult)
             }}
           >
@@ -41,6 +67,15 @@ export default function Transact() {
       >
         Create Account
       </Button>
+      <TextInput value={importAccountText} onChangeText={setImportAccountText} placeholder="Import account"/>
+      <Button
+        onClick={async () => {
+          const otherResult = await importAccount.mutateAsync({ account: importAccountText });
+          console.log('Imported Account:', otherResult)
+        }}
+      >
+        Import Account
+      </Button>
     </ScrollView>
   );
 }
diff --git a/packages/mobile-app/data/facades/accounts/demoHandlers.ts b/packages/mobile-app/data/facades/accounts/demoHandlers.ts
@@ -1,6 +1,7 @@
 import { f } from "data-facade";
 import { z } from "zod";
 import { AccountsMethods } from "./types";
+import { AccountFormat } from "@ironfish/sdk";
 
 let ACCOUNTS = [
   { id: 0, name: "alice", viewOnlyAccount: "alice" },
@@ -14,11 +15,6 @@ async function getAccounts(limit: number) {
 }
 
 export const accountsHandlers = f.facade<AccountsMethods>({
-  getAccounts: f.handler.query(async () => {
-    const accounts = await getAccounts(ACCOUNTS.length);
-    console.log("getAccounts", accounts);
-    return accounts;
-  }),
   createAccount: f.handler
     .input(
       z.object({
@@ -39,13 +35,73 @@ export const accountsHandlers = f.facade<AccountsMethods>({
     .input(
       z.object({
         name: z.string(),
+        format: z.nativeEnum(AccountFormat),
+        language: z.string().optional(),
+        viewOnly: z.boolean().optional(),
       }),
     )
     .mutation(async ({ name }) => {
       const account = ACCOUNTS.find((a) => a.name === name)
       if (account === undefined) {
-        throw new Error("No accounts found");
+        throw new Error(`No account found with name ${name}`);
       }
       return JSON.stringify(account);
     }),
+  getAccount: f.handler.input(
+    z.object({
+      name: z.string(),
+    }),
+  ).query(async ({ name }) => {
+    const account = ACCOUNTS.find((a) => a.name === name);
+    if (account === undefined) {
+      throw new Error(`No account found with name ${name}`);
+    }
+    return account
+  }),
+  getAccounts: f.handler.query(async () => {
+      const accounts = await getAccounts(ACCOUNTS.length);
+      console.log("getAccounts", accounts);
+      return accounts;
+    }),
+  importAccount: f.handler
+    .input(
+      z.object({
+        account: z.string(),
+        name: z.string(),
+      }),
+    )
+    .mutation(async ({ account, name }) => {
+      const existingId = ACCOUNTS.at(-1)?.id
+      if (existingId === undefined) {
+        throw new Error("No accounts found");
+      }
+      const importedAccount = { id: existingId + 1, name, viewOnlyAccount: account }
+      console.log("createAccount", account);
+      ACCOUNTS.push(importedAccount);
+      return importedAccount;
+    }),
+  removeAccount: f.handler
+    .input(
+      z.object({
+        name: z.string(),
+      }),
+    )
+    .mutation(async ({ name }) => {
+      const accountIndex = ACCOUNTS.findIndex((a) => a.name === name);
+      ACCOUNTS.splice(accountIndex, 1);
+    }),
+  renameAccount: f.handler
+    .input(
+      z.object({
+        name: z.string(),
+        newName: z.string(),
+      }),
+    )
+    .mutation(async ({ name, newName }) => {
+      const account = ACCOUNTS.find((a) => a.name === name);
+      if (!account) {
+        throw new Error("No account found");
+      }
+      account.name = newName;
+    }),
 });
diff --git a/packages/mobile-app/data/facades/accounts/handlers.ts b/packages/mobile-app/data/facades/accounts/handlers.ts
@@ -2,11 +2,9 @@ import { f } from "data-facade";
 import { z } from "zod";
 import { AccountsMethods } from "./types";
 import { wallet } from "../../wallet";
+import { AccountFormat, LanguageKey, LanguageUtils } from "@ironfish/sdk";
 
 export const accountsHandlers = f.facade<AccountsMethods>({
-  getAccounts: f.handler.query(async () => {
-    return await wallet.getAccounts();
-  }),
   createAccount: f.handler
     .input(
       z.object({
@@ -17,14 +15,61 @@ export const accountsHandlers = f.facade<AccountsMethods>({
       const account = await wallet.createAccount(name);
       return account;
     }),
-    exportAccount: f.handler
+  exportAccount: f.handler
     .input(
       z.object({
         name: z.string(),
+        format: z.nativeEnum(AccountFormat),
+        language: z.string().optional(),
+        viewOnly: z.boolean().optional(),
       }),
     )
-    .mutation(async ({ name }) => {
-      const account = await wallet.exportAccount(name);
+    .mutation(async ({ name, format, language, viewOnly }) => {
+      if (language && !Object.hasOwn(LanguageUtils.LANGUAGES, language)) {
+        throw new Error(`Language ${language} is not supported`);
+      }
+      const typedLanguage: LanguageKey = language as LanguageKey;
+
+      const account = await wallet.exportAccount(name, format, { language: typedLanguage, viewOnly });
       return account;
     }),
+  getAccount: f.handler.input(
+    z.object({
+      name: z.string(),
+    }),
+  ).query(async ({ name }) => {
+    return await wallet.getAccount(name);
+  }),
+  getAccounts: f.handler.query(async () => {
+    return await wallet.getAccounts();
+  }),
+  importAccount: f.handler
+    .input(
+      z.object({
+        account: z.string(),
+        name: z.string().optional(),
+      }),
+    )
+    .mutation(async ({ account, name }) => {
+      return await wallet.importAccount(account, name);
+    }),
+  removeAccount: f.handler
+    .input(
+      z.object({
+        name: z.string(),
+      }),
+    )
+    .mutation(async ({ name }) => {
+      await wallet.removeAccount(name);
+    }),
+  renameAccount: f.handler
+    .input(
+      z.object({
+        name: z.string(),
+        newName: z.string(),
+      }),
+    )
+    .mutation(async ({ name, newName }) => {
+      await wallet.renameAccount(name, newName);
+    }),
 });