@msafe/sui3-sdk 0.0.8 → 0.0.10

package/src/core/MSafeClient.ts

@@ -1,12 +1,13 @@
 import { MSafeAccountInfo } from '@msafe/sui3-utils';
 
-import { JWTToken } from '@/backend/types';
+import { AddressBookSDK } from '@/core/AddressBookSDK';
 import { CreateHelper } from '@/core/CreateHelper';
 import { MessageHelper } from '@/core/MessageHelper';
 import { MSafeAccount } from '@/core/MSafeAccount';
 import { PublicKeyHelper } from '@/core/PublicKeyHelper';
 import { MSafeConfigOptions, MSafeEnv } from '@/globals/const';
 import { MSafeGlobals } from '@/globals/MSafeGlobals';
+import { JWTToken } from '@/types/backend';
 import { CreateMSafeAccountInfo } from '@/types/creation';
 import { IWallet } from '@/types/wallet';
 
@@ -23,24 +24,33 @@ export class MSafeClient {
   }
 
   async connectWallet(input: { wallet: IWallet; jwtToken?: JWTToken }): Promise<JWTToken> {
+    if (input.jwtToken) {
+      const verified = await this.backend.verifyToken(input.jwtToken);
+      if (verified) {
+        this.globals.wallet = input.wallet;
+        this.backend.setJWTToken(input.jwtToken);
+        return input.jwtToken;
+      }
+    }
+    const jwt = await this.authSign(input.wallet);
     this.globals.wallet = input.wallet;
+    this.backend.setJWTToken(jwt);
+    return jwt;
+  }
+
+  private async authSign(wallet: IWallet): Promise<JWTToken> {
     const messageStr = MessageHelper.welcomeMessage(new Date().toUTCString());
-    const sig = await input.wallet.signPersonalMessage({
+    const sig = await wallet.signPersonalMessage({
       messageStr,
     });
     return this.backend.authSign({
-      address: await input.wallet.address(),
+      address: await wallet.address(),
       message: messageStr,
       signature: sig.signature,
-      walletType: input.wallet.walletType,
+      walletType: wallet.walletType,
     });
   }
 
-  async setWallet(input: { wallet: IWallet; jwtToken: JWTToken }) {
-    this.globals.wallet = input.wallet;
-    this.backend.setJWTToken(input.jwtToken);
-  }
-
   async userInfo() {
     return this.globals.backend.getUserInfo(await this.walletAddress());
   }
@@ -86,6 +96,10 @@ export class MSafeClient {
     return this.globals.wallet;
   }
 
+  get AddressBook(): AddressBookSDK {
+    return new AddressBookSDK(this.globals);
+  }
+
   private async walletAddress() {
     return this.wallet.address();
   }

package/src/core/MessageHelper.ts

@@ -1,4 +1,9 @@
+import { normalizeSuiAddress } from '@mysten/sui.js/utils';
+import { MD5 } from 'crypto-js';
+
 import { IntentionHelper, TxIntention } from '@/transactions/intention';
+import { UpdateAddressBookEntry } from '@/types/address-book';
+import { stringToBuffer } from '@/utils';
 
 export class MessageHelper {
   // Message used for MSafe creation
@@ -50,6 +55,29 @@ export class MessageHelper {
       intention: IntentionHelper.de(intentionData),
     };
   }
+
+  static updateAddressBookMessage(updates: UpdateAddressBookEntry[]) {
+    const normalized = updates.map((update) => ({
+      ...update,
+      address: normalizeSuiAddress(update.address),
+    }));
+    function sortObjectKeys(obj: Record<string, any>): Record<string, any> {
+      return Object.keys(obj)
+        .sort()
+        .reduce((result, key) => ({ ...result, [key]: obj[key] }), {} as Record<string, any>);
+    }
+
+    // Sort keys of each object in the updates array
+    const sortedUpdates = normalized.map((update) => sortObjectKeys(update));
+    const raw = JSON.stringify(sortedUpdates, null, 2);
+    const encoded = stringToBuffer(raw);
+    if (encoded.length <= 1024) {
+      // BCS signing can handle this amount of data, directly return
+      return raw;
+    }
+    const md5 = MD5(raw);
+    return `Bulk address book update: ${md5}`;
+  }
 }
 
 export interface ProposeIntentionData {

package/src/core/index.ts

@@ -2,3 +2,4 @@ export { CreateHelper } from './CreateHelper';
 export { MessageHelper } from './MessageHelper';
 export { MSafeClient } from './MSafeClient';
 export { MSafeAccount } from './MSafeAccount';
+export { AddressBookSDK } from './AddressBookSDK';

package/src/types/address-book.ts

@@ -0,0 +1,32 @@
+import { PagedResult } from '@/types/pagination';
+
+export enum OpAddressBookType {
+  Delete = 'delete',
+  Upsert = 'upsert',
+}
+
+export type UpdateAddressBookEntry = DeleteAddressBookEntry | UpsertAddressBookEntry;
+
+export interface DeleteAddressBookEntry {
+  opType: OpAddressBookType.Delete;
+  address: string;
+}
+
+export interface UpsertAddressBookEntry {
+  opType: OpAddressBookType.Upsert;
+  address: string;
+  addressName: string;
+  remark?: string;
+}
+
+export type GetAddressBookResult = PagedResult<GetAddressBookResultEntry>;
+
+export interface GetAddressBookResultEntry {
+  address: string;
+
+  addressName: string;
+
+  remark?: string;
+
+  updatedAt: Date;
+}

package/src/{backend/types.ts → types/backend.ts}

@@ -2,7 +2,7 @@ export type JWTToken = string;
 
 export interface PaginationOption {
   page?: number;
-  pageSize: number;
+  limit?: number;
 }
 
 export interface PaginationResponse<T> {

package/src/types/index.ts

@@ -1,3 +1,6 @@
 export * from './creation';
 export * from './msafe';
 export * from './wallet';
+export * from './pagination';
+export * from './address-book';
+export * from './backend';

package/src/types/pagination.ts

@@ -0,0 +1,15 @@
+export interface PagedResult<T> {
+  data: T[];
+
+  meta: PagedMeta;
+}
+
+export interface PagedMeta {
+  page: number;
+
+  limit: number;
+
+  total: number;
+
+  hasNext: boolean;
+}