Merge pull request #117 from DIG-Network/release/v0.0.1-alpha.128

Release/v0.0.1 alpha.128

Author: MichaelTaylor3D

CHANGELOG.md

@@ -2,6 +2,14 @@
 
 All notable changes to this project will be documented in this file. See [standard-version](https://github.com/conventional-changelog/standard-version) for commit guidelines.
 
+### [0.0.1-alpha.128](https://github.com/DIG-Network/dig-chia-sdk/compare/v0.0.1-alpha.127...v0.0.1-alpha.128) (2024-10-04)
+
+
+### Features
+
+* add ping network of update logic ([e210be2](https://github.com/DIG-Network/dig-chia-sdk/commit/e210be2d86a824003d756b8fbd30602d75250651))
+* add ping network of update logic ([dbff9bf](https://github.com/DIG-Network/dig-chia-sdk/commit/dbff9bfc7b416dcaf5ede8fed4d06a148c98572e))
+
 ### [0.0.1-alpha.127](https://github.com/DIG-Network/dig-chia-sdk/compare/v0.0.1-alpha.126...v0.0.1-alpha.127) (2024-10-04)
 
 

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@dignetwork/dig-sdk",
-  "version": "0.0.1-alpha.127",
+  "version": "0.0.1-alpha.128",
   "description": "",
   "type": "commonjs",
   "main": "./dist/index.js",

package.json

@@ -105,7 +105,20 @@ export class DigNetwork {
     fs.unlinkSync(path.join(DIG_FOLDER_PATH, "stores", storeId + ".json"));
   }
 
-  public async syncStoreFromPeers(maxRootsToProcess?: number): Promise<void> {
+  public static async pingNetworkOfUpdate(storeId: string, rootHash: string): Promise<void> {
+    const serverCoin = new ServerCoin(storeId);
+    // When an update is made, ping 10 network peers to pull updates from this store
+    const digPeers = await serverCoin.sampleCurrentEpoch(10);
+    for (const peer of digPeers) {
+      const digPeer = new DigPeer(peer, storeId);
+      await digPeer.propagationServer.pingUpdate(rootHash);
+    }
+  }
+
+  public async syncStoreFromPeers(
+    prioritizedPeer?: DigPeer,
+    maxRootsToProcess?: number
+  ): Promise<void> {
     console.log("Starting file download process...");
     let peerBlackList: string[] = [];
 
@@ -145,12 +158,16 @@ export class DigNetwork {
         while (true) {
           try {
             // Find a peer with the store and root hash
-            selectedPeer = await DigNetwork.findPeerWithStoreKey(
-              this.dataStore.StoreId,
-              rootInfo.root_hash,
-              undefined,
-              peerBlackList
-            );
+            if (prioritizedPeer) {
+              selectedPeer = prioritizedPeer;
+            } else {
+              selectedPeer = await DigNetwork.findPeerWithStoreKey(
+                this.dataStore.StoreId,
+                rootInfo.root_hash,
+                undefined,
+                peerBlackList
+              );
+            }
 
             if (!selectedPeer) {
               console.error(

src/DigNetwork/DigNetwork.ts

@@ -80,6 +80,53 @@ export class PropagationServer {
     });
   }
 
+  /**
+   * Ping the current peer about an update to the store, passing rootHash.
+   * @param rootHash - The root hash for the store update.
+   */
+  async pingUpdate(rootHash: string): Promise<void> {
+    const spinner = createSpinner(`Pinging peer ${this.ipAddress}...`).start();
+
+    try {
+      const httpsAgent = this.createHttpsAgent();
+      const url = `https://${formatHost(this.ipAddress)}:${
+        PropagationServer.port
+      }/update`;
+
+      const config: AxiosRequestConfig = {
+        httpsAgent,
+        headers: {
+          "Content-Type": "application/json",
+        },
+      };
+
+      // Data to send in the request (storeId and rootHash)
+      const data = {
+        storeId: this.storeId,
+        rootHash: rootHash,
+        updateTime: new Date().toISOString(),
+      };
+
+      try {
+        const response = await axios.post(url, data, config);
+        console.log(green(`✔ Successfully pinged peer: ${this.ipAddress}`));
+        spinner.success({
+          text: green(
+            `✔ Successfully pinged peer ${this.ipAddress} with rootHash ${rootHash}`
+          ),
+        });
+        return response.data;
+      } catch (error: any) {
+        console.error(red(`✖ Failed to ping peer: ${this.ipAddress}`));
+        console.error(red(error.message));
+        throw error;
+      }
+    } catch (error: any) {
+      spinner.error({ text: red("✖ Error pinging network peer.") });
+      console.error(red(error.message));
+    }
+  }
+
   /**
    * Adds a custom inactivity timeout for large file transfers.
    */
@@ -253,7 +300,11 @@ export class PropagationServer {
    * Upload a file to the server by sending a PUT request.
    * Logs progress using a local cli-progress bar.
    */
-  async uploadFile(label: string, dataPath: string, uncompress: boolean = false) {
+  async uploadFile(
+    label: string,
+    dataPath: string,
+    uncompress: boolean = false
+  ) {
     const filePath = path.join(STORE_PATH, this.storeId, dataPath);
 
     const { nonce, fileExists } = await this.getFileNonce(dataPath);