lint and sync upon agent recover

decentralized-identity · Sep 6, 2024 · f6776ec · f6776ec
1 parent 6b51057
commit f6776ec
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 20 deletions.
diff --git a/packages/agent/src/index.ts b/packages/agent/src/index.ts
@@ -19,6 +19,7 @@ export * from './local-key-manager.js';
 export * from './permissions-api.js';
 export * from './rpc-client.js';
 export * from './store-data.js';
+export * from './store-data-protocols.js';
 export * from './store-did.js';
 export * from './store-identity.js';
 export * from './store-key.js';

diff --git a/packages/agent/src/sync-engine-level.ts b/packages/agent/src/sync-engine-level.ts
@@ -306,7 +306,7 @@ export class SyncEngineLevel implements SyncEngine {
               protocol,
               cached       : true
             });
-  
+
             permissionGrantId = messagesReadGrant.grant.id;
             granteeDid = delegateDid;
           } catch(error:any) {
@@ -316,11 +316,11 @@ export class SyncEngineLevel implements SyncEngine {
         }
 
         const { reply: { status, subscription } } = await this.agent.processDwnRequest({
-          author: did,
-          target: did,
-          messageType: DwnInterface.MessagesSubscribe,
+          author        : did,
+          target        : did,
+          messageType   : DwnInterface.MessagesSubscribe,
           granteeDid,
-          messageParams: {
+          messageParams : {
             permissionGrantId
           },
           subscriptionHandler: async (event) => {
@@ -371,12 +371,12 @@ export class SyncEngineLevel implements SyncEngine {
                 protocol,
                 cached       : true
               });
-    
+
               permissionGrantId = messagesReadGrant.grant.id;
               granteeDid = delegateDid;
             } catch(error:any) {
               console.error('SyncEngineLevel: pull - Error fetching MessagesRead permission grant for delegate DID', error);
-              return
+              return;
             }
           }
 
@@ -391,9 +391,9 @@ export class SyncEngineLevel implements SyncEngine {
               permissionGrantId
             }
           });
-    
+
           let reply: MessagesReadReply;
-    
+
           try {
             reply = await this.agent.rpc.sendDwnRequest({
               dwnUrl,
@@ -413,7 +413,7 @@ export class SyncEngineLevel implements SyncEngine {
           if (processStatus.code === 202 || processStatus.code === 204 || processStatus.code === 409) {
             await this.addMessage(did, messageCid);
           }
-        }
+        };
         let permissionGrantId: string | undefined;
         let granteeDid: string | undefined;
         if (delegateDid) {
@@ -425,7 +425,7 @@ export class SyncEngineLevel implements SyncEngine {
               protocol,
               cached       : true
             });
-  
+
             permissionGrantId = messagesReadGrant.grant.id;
             granteeDid = delegateDid;
           } catch(error:any) {
@@ -434,21 +434,21 @@ export class SyncEngineLevel implements SyncEngine {
           }
         }
         const messagesSubscribeMessage = await this.agent.sendDwnRequest({
-          store: false,
-          author: did,
-          target: did,
-          messageType: DwnInterface.MessagesSubscribe,
+          store         : false,
+          author        : did,
+          target        : did,
+          messageType   : DwnInterface.MessagesSubscribe,
           granteeDid,
-          messageParams: {
+          messageParams : {
             permissionGrantId,
           },
           subscriptionHandler
         });
 
         const { status: sendStatus, subscription } = await this.agent.rpc.sendDwnRequest({
           dwnUrl,
-          targetDid: did,
-          message: messagesSubscribeMessage.message,
+          targetDid : did,
+          message   : messagesSubscribeMessage.message,
           subscriptionHandler
         });
 

diff --git a/packages/api/src/web5.ts b/packages/api/src/web5.ts
@@ -17,7 +17,7 @@ import type {
 } from '@web5/agent';
 
 import { Web5UserAgent } from '@web5/user-agent';
-import { DwnRegistrar, WalletConnect } from '@web5/agent';
+import { DwnRegistrar, IdentityProtocolDefinition, JwkProtocolDefinition, WalletConnect } from '@web5/agent';
 
 import { DidApi } from './did-api.js';
 import { DwnApi } from './dwn-api.js';
@@ -280,10 +280,28 @@ export class Web5 {
       const serviceEndpointNodes = techPreview?.dwnEndpoints ?? didCreateOptions?.dwnEndpoints ?? ['https://dwn.tbddev.org/beta'];
 
       // Initialize, if necessary, and start the agent.
-      if (await userAgent.firstLaunch()) {
+      const firstLaunch = await userAgent.firstLaunch();
+      if (firstLaunch) {
         recoveryPhrase = await userAgent.initialize({ password, recoveryPhrase, dwnEndpoints: serviceEndpointNodes });
       }
       await userAgent.start({ password });
+      if (firstLaunch && sync !== 'off') {
+
+        // register only the identity-specific protocols for agent sync
+        await userAgent.sync.registerIdentity({
+          did     : userAgent.agentDid.uri,
+          options : {
+            protocols: [
+              IdentityProtocolDefinition.protocol,
+              JwkProtocolDefinition.protocol
+            ]
+          }
+        });
+
+        // attempt to pull the latest messages from the DWN so that the identities can be in sync
+        await userAgent.sync.sync('pull');
+      }
+
       // Attempt to retrieve the connected Identity if it exists.
       const connectedIdentity: BearerIdentity = await userAgent.identity.connectedIdentity();
       let identity: BearerIdentity;