@dignetwork/dig-sdk 0.0.1-alpha.16 → 0.0.1-alpha.162

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (83) hide show
  1. package/dist/DataIntegrityTree/DataIntegrityTree.d.ts +11 -2
  2. package/dist/DataIntegrityTree/DataIntegrityTree.d.ts.map +1 -1
  3. package/dist/DataIntegrityTree/DataIntegrityTree.js +94 -13
  4. package/dist/DigNetwork/ContentServer.d.ts +24 -5
  5. package/dist/DigNetwork/ContentServer.d.ts.map +1 -1
  6. package/dist/DigNetwork/ContentServer.js +239 -27
  7. package/dist/DigNetwork/DigNetwork.d.ts +15 -9
  8. package/dist/DigNetwork/DigNetwork.d.ts.map +1 -1
  9. package/dist/DigNetwork/DigNetwork.js +129 -189
  10. package/dist/DigNetwork/DigPeer.d.ts +8 -3
  11. package/dist/DigNetwork/DigPeer.d.ts.map +1 -1
  12. package/dist/DigNetwork/DigPeer.js +68 -115
  13. package/dist/DigNetwork/IncentiveServer.d.ts.map +1 -1
  14. package/dist/DigNetwork/IncentiveServer.js +6 -5
  15. package/dist/DigNetwork/PropagationServer.d.ts +88 -33
  16. package/dist/DigNetwork/PropagationServer.d.ts.map +1 -1
  17. package/dist/DigNetwork/PropagationServer.js +636 -357
  18. package/dist/blockchain/DataStore.d.ts +25 -7
  19. package/dist/blockchain/DataStore.d.ts.map +1 -1
  20. package/dist/blockchain/DataStore.js +115 -159
  21. package/dist/blockchain/DataStoreSerializer.d.ts +1 -1
  22. package/dist/blockchain/DataStoreSerializer.d.ts.map +1 -1
  23. package/dist/blockchain/FullNodePeer.d.ts +94 -10
  24. package/dist/blockchain/FullNodePeer.d.ts.map +1 -1
  25. package/dist/blockchain/FullNodePeer.js +318 -104
  26. package/dist/blockchain/ServerCoin.d.ts +13 -5
  27. package/dist/blockchain/ServerCoin.d.ts.map +1 -1
  28. package/dist/blockchain/ServerCoin.js +90 -28
  29. package/dist/blockchain/StoreMonitorRegistry.d.ts +85 -0
  30. package/dist/blockchain/StoreMonitorRegistry.d.ts.map +1 -0
  31. package/dist/blockchain/StoreMonitorRegistry.js +241 -0
  32. package/dist/blockchain/Wallet.d.ts +2 -2
  33. package/dist/blockchain/Wallet.d.ts.map +1 -1
  34. package/dist/blockchain/Wallet.js +50 -18
  35. package/dist/blockchain/coins.d.ts +1 -2
  36. package/dist/blockchain/coins.d.ts.map +1 -1
  37. package/dist/blockchain/coins.js +1 -55
  38. package/dist/blockchain/index.d.ts +1 -0
  39. package/dist/blockchain/index.d.ts.map +1 -1
  40. package/dist/blockchain/index.js +1 -0
  41. package/dist/types.d.ts +2 -0
  42. package/dist/types.d.ts.map +1 -1
  43. package/dist/utils/ContentScanner.d.ts +63 -0
  44. package/dist/utils/ContentScanner.d.ts.map +1 -0
  45. package/dist/utils/ContentScanner.js +175 -0
  46. package/dist/utils/Environment.d.ts +21 -0
  47. package/dist/utils/Environment.d.ts.map +1 -0
  48. package/dist/utils/Environment.js +93 -0
  49. package/dist/utils/FileCache.d.ts +1 -1
  50. package/dist/utils/FileCache.d.ts.map +1 -1
  51. package/dist/utils/FileCache.js +2 -2
  52. package/dist/utils/FileTransfer.d.ts +47 -0
  53. package/dist/utils/FileTransfer.d.ts.map +1 -0
  54. package/dist/utils/FileTransfer.js +209 -0
  55. package/dist/utils/NconfManager.js +2 -1
  56. package/dist/utils/PeerRanker.d.ts +55 -0
  57. package/dist/utils/PeerRanker.d.ts.map +1 -0
  58. package/dist/utils/PeerRanker.js +186 -0
  59. package/dist/utils/StoreArchiveManager.d.ts +45 -0
  60. package/dist/utils/StoreArchiveManager.d.ts.map +1 -0
  61. package/dist/utils/StoreArchiveManager.js +153 -0
  62. package/dist/utils/config.d.ts +4 -3
  63. package/dist/utils/config.d.ts.map +1 -1
  64. package/dist/utils/config.js +12 -7
  65. package/dist/utils/directoryUtils.d.ts.map +1 -1
  66. package/dist/utils/directoryUtils.js +9 -12
  67. package/dist/utils/index.d.ts +3 -1
  68. package/dist/utils/index.d.ts.map +1 -1
  69. package/dist/utils/index.js +3 -1
  70. package/dist/utils/merkle.d.ts +2 -0
  71. package/dist/utils/merkle.d.ts.map +1 -0
  72. package/dist/utils/merkle.js +28 -0
  73. package/dist/utils/network.d.ts +2 -1
  74. package/dist/utils/network.d.ts.map +1 -1
  75. package/dist/utils/network.js +54 -30
  76. package/dist/utils/promiseUtils.d.ts +19 -0
  77. package/dist/utils/promiseUtils.d.ts.map +1 -0
  78. package/dist/utils/promiseUtils.js +43 -0
  79. package/dist/utils/ssl.js +2 -2
  80. package/package.json +18 -5
  81. package/dist/utils/deltaUtils.d.ts +0 -2
  82. package/dist/utils/deltaUtils.d.ts.map +0 -1
  83. package/dist/utils/deltaUtils.js +0 -83
@@ -7,119 +7,319 @@ exports.FullNodePeer = void 0;
7
7
  const path_1 = __importDefault(require("path"));
8
8
  const os_1 = __importDefault(require("os"));
9
9
  const fs_1 = __importDefault(require("fs"));
10
- const datalayer_driver_1 = require("datalayer-driver");
11
- const chia_server_coin_1 = require("chia-server-coin");
10
+ const datalayer_driver_1 = require("@dignetwork/datalayer-driver");
12
11
  const promises_1 = require("dns/promises");
13
12
  const net_1 = __importDefault(require("net"));
14
- const lodash_1 = require("lodash");
15
13
  const nanospinner_1 = require("nanospinner");
16
14
  const config_1 = require("../utils/config");
15
+ const Environment_1 = require("../utils/Environment");
16
+ const node_cache_1 = __importDefault(require("node-cache"));
17
+ const bottleneck_1 = __importDefault(require("bottleneck"));
17
18
  const FULLNODE_PORT = 8444;
18
19
  const LOCALHOST = "127.0.0.1";
20
+ const CHIA_NODES_HOST = "chia-nodes";
19
21
  const DNS_HOSTS = [
20
22
  "dns-introducer.chia.net",
21
23
  "chia.ctrlaltdel.ch",
22
24
  "seeder.dexie.space",
23
25
  "chia.hoffmang.com",
24
26
  ];
25
- const CONNECTION_TIMEOUT = 2000;
26
- const CACHE_DURATION = 30000; // Cache duration in milliseconds
27
+ const CONNECTION_TIMEOUT = 2000; // in milliseconds
28
+ const CACHE_DURATION = 30000; // in milliseconds
29
+ const COOLDOWN_DURATION = 600000; // 10 minutes in milliseconds
30
+ const MAX_PEERS_TO_FETCH = 5; // Maximum number of peers to fetch from DNS
31
+ const MAX_REQUESTS_PER_MINUTE = 100; // Per-peer rate limit
32
+ /**
33
+ * FullNodePeer manages connections to full nodes, prioritizing certain peers and handling reliability.
34
+ */
27
35
  class FullNodePeer {
36
+ // Private constructor for singleton pattern
28
37
  constructor(peer) {
29
38
  this.peer = peer;
30
39
  }
40
+ /**
41
+ * Retrieves the singleton instance of FullNodePeer.
42
+ * @returns {FullNodePeer} The singleton instance.
43
+ */
44
+ static getInstance() {
45
+ if (!FullNodePeer.instance) {
46
+ FullNodePeer.instance = new FullNodePeer(null); // Temporarily set to null
47
+ }
48
+ return FullNodePeer.instance;
49
+ }
50
+ /**
51
+ * Initializes the singleton instance by connecting to the best peer.
52
+ */
53
+ async initialize() {
54
+ if (this.peer)
55
+ return; // Already initialized
56
+ try {
57
+ const bestPeer = await this.getBestPeer();
58
+ this.peer = bestPeer;
59
+ FullNodePeer.instance = this; // Assign the initialized instance
60
+ }
61
+ catch (error) {
62
+ console.error(`Fullnode Initialization failed: ${error.message}`);
63
+ throw error;
64
+ }
65
+ }
66
+ /**
67
+ * Connects and returns the best available peer.
68
+ * Implements singleton behavior.
69
+ * @returns {Promise<Peer>} The connected Peer instance.
70
+ */
31
71
  static async connect() {
32
- const peer = await FullNodePeer.getBestPeer();
33
- return new FullNodePeer(peer).peer;
72
+ const instance = FullNodePeer.getInstance();
73
+ // Remove cached peer to ensure a new connection each time
74
+ await instance.initialize();
75
+ return instance.peer;
34
76
  }
77
+ /**
78
+ * Checks if a given port on a host is reachable.
79
+ * @param {string} host - The host IP address.
80
+ * @param {number} port - The port number.
81
+ * @param {number} timeout - Connection timeout in milliseconds.
82
+ * @returns {Promise<boolean>} Whether the port is reachable.
83
+ */
35
84
  static isPortReachable(host, port, timeout = CONNECTION_TIMEOUT) {
36
85
  return new Promise((resolve) => {
37
86
  const socket = new net_1.default.Socket()
38
87
  .setTimeout(timeout)
39
- .once("error", () => resolve(false))
40
- .once("timeout", () => resolve(false))
88
+ .once("error", () => {
89
+ socket.destroy();
90
+ resolve(false);
91
+ })
92
+ .once("timeout", () => {
93
+ socket.destroy();
94
+ resolve(false);
95
+ })
41
96
  .connect(port, host, () => {
42
97
  socket.end();
43
98
  resolve(true);
44
99
  });
45
100
  });
46
101
  }
102
+ /**
103
+ * Validates an IPv4 address.
104
+ * @param {string} ip - The IP address to validate.
105
+ * @returns {boolean} Whether the IP address is valid.
106
+ */
47
107
  static isValidIpAddress(ip) {
48
- const ipv4Regex = /^(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$/;
108
+ const ipv4Regex = /^(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)(\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)){3}$/;
49
109
  return ipv4Regex.test(ip);
50
110
  }
51
111
  /**
52
- * Retrieves the TRUSTED_FULLNODE IP from the environment
53
- * and verifies if it is a valid IP address.
54
- *
55
- * @returns {string | null} The valid IP address or null if invalid
112
+ * Retrieves the TRUSTED_FULLNODE IP from the environment and verifies its validity.
113
+ * @returns {string | null} The trusted full node IP or null if invalid.
56
114
  */
57
115
  static getTrustedFullNode() {
58
- const trustedNodeIp = process.env.TRUSTED_FULLNODE || null;
116
+ const trustedNodeIp = Environment_1.Environment.TRUSTED_FULLNODE || null;
59
117
  if (trustedNodeIp && FullNodePeer.isValidIpAddress(trustedNodeIp)) {
60
118
  return trustedNodeIp;
61
119
  }
62
120
  return null;
63
121
  }
122
+ /**
123
+ * Fetches new peer IPs from DNS introducers and prioritized hosts.
124
+ * Utilizes caching to avoid redundant DNS resolutions.
125
+ * @returns {Promise<string[]>} An array of reachable peer IPs.
126
+ */
64
127
  static async fetchNewPeerIPs() {
65
128
  const trustedNodeIp = FullNodePeer.getTrustedFullNode();
66
129
  const priorityIps = [];
130
+ // Define prioritized peers
131
+ FullNodePeer.prioritizedPeers = [CHIA_NODES_HOST, LOCALHOST];
132
+ // Add trustedNodeIp if available
133
+ if (trustedNodeIp) {
134
+ FullNodePeer.prioritizedPeers.unshift(trustedNodeIp);
135
+ }
67
136
  // Prioritize trustedNodeIp
68
137
  if (trustedNodeIp &&
138
+ !FullNodePeer.cooldownCache.has(trustedNodeIp) &&
69
139
  (await FullNodePeer.isPortReachable(trustedNodeIp, FULLNODE_PORT))) {
70
140
  priorityIps.push(trustedNodeIp);
71
141
  }
72
142
  // Prioritize LOCALHOST
73
- if (await FullNodePeer.isPortReachable(LOCALHOST, FULLNODE_PORT)) {
143
+ if (!FullNodePeer.cooldownCache.has(LOCALHOST) &&
144
+ (await FullNodePeer.isPortReachable(LOCALHOST, FULLNODE_PORT))) {
74
145
  priorityIps.push(LOCALHOST);
75
146
  }
147
+ // Prioritize CHIA_NODES_HOST
148
+ if (!FullNodePeer.cooldownCache.has(CHIA_NODES_HOST) &&
149
+ (await FullNodePeer.isPortReachable(CHIA_NODES_HOST, FULLNODE_PORT))) {
150
+ priorityIps.push(CHIA_NODES_HOST);
151
+ }
76
152
  if (priorityIps.length > 0) {
77
153
  return priorityIps;
78
154
  }
155
+ // Check if cached peer IPs exist
156
+ const cachedPeerIPs = FullNodePeer.peerIPCache.get("peerIPs");
157
+ if (cachedPeerIPs) {
158
+ return cachedPeerIPs;
159
+ }
79
160
  // Fetch peers from DNS introducers
161
+ const fetchedPeers = [];
80
162
  for (const DNS_HOST of DNS_HOSTS) {
163
+ // Check if DNS_HOST is in failedDNSCache
164
+ if (FullNodePeer.failedDNSCache.has(DNS_HOST)) {
165
+ console.warn(`Skipping DNS host ${DNS_HOST} due to recent failure.`);
166
+ continue;
167
+ }
81
168
  try {
82
169
  const ips = await (0, promises_1.resolve4)(DNS_HOST);
83
170
  if (ips && ips.length > 0) {
84
- const shuffledIps = ips.sort(() => 0.5 - Math.random());
171
+ const shuffledIps = FullNodePeer.shuffleArray(ips);
85
172
  const reachableIps = [];
86
173
  for (const ip of shuffledIps) {
87
- if (await FullNodePeer.isPortReachable(ip, FULLNODE_PORT)) {
174
+ if (!FullNodePeer.cooldownCache.has(ip) &&
175
+ (await FullNodePeer.isPortReachable(ip, FULLNODE_PORT))) {
88
176
  reachableIps.push(ip);
89
177
  }
90
- if (reachableIps.length === 5)
91
- break;
178
+ if (reachableIps.length === MAX_PEERS_TO_FETCH)
179
+ break; // Limit to MAX_PEERS_TO_FETCH peers
92
180
  }
93
181
  if (reachableIps.length > 0) {
94
- return reachableIps;
182
+ fetchedPeers.push(...reachableIps);
95
183
  }
96
184
  }
97
185
  }
98
186
  catch (error) {
99
187
  console.error(`Failed to resolve IPs from ${DNS_HOST}: ${error.message}`);
188
+ // Add DNS_HOST to failedDNSCache for cooldown
189
+ FullNodePeer.failedDNSCache.set(DNS_HOST, true);
100
190
  }
101
191
  }
192
+ // Cache the fetched peer IPs
193
+ if (fetchedPeers.length > 0) {
194
+ FullNodePeer.peerIPCache.set("peerIPs", fetchedPeers);
195
+ return fetchedPeers;
196
+ }
102
197
  throw new Error("No reachable IPs found in any DNS records.");
103
198
  }
104
- static async getPeerIPs() {
105
- const ips = await FullNodePeer.memoizedFetchNewPeerIPs();
106
- const reachableIps = await Promise.all(ips.map(async (ip) => {
107
- if (ip && (await FullNodePeer.isPortReachable(ip, FULLNODE_PORT))) {
108
- return ip;
199
+ /**
200
+ * Shuffles an array using the Fisher-Yates algorithm.
201
+ * @param {T[]} array - The array to shuffle.
202
+ * @returns {T[]} The shuffled array.
203
+ */
204
+ static shuffleArray(array) {
205
+ const shuffled = [...array];
206
+ for (let i = shuffled.length - 1; i > 0; i--) {
207
+ const j = Math.floor(Math.random() * (i + 1));
208
+ [shuffled[i], shuffled[j]] = [shuffled[j], shuffled[i]];
209
+ }
210
+ return shuffled;
211
+ }
212
+ /**
213
+ * Initializes the peer weights map.
214
+ * Assigns higher initial weights to prioritized peers.
215
+ * @param {string[]} peerIPs - An array of peer IPs.
216
+ */
217
+ static initializePeerWeights(peerIPs) {
218
+ for (const ip of peerIPs) {
219
+ if (!FullNodePeer.peerWeights.has(ip)) {
220
+ if (ip === CHIA_NODES_HOST ||
221
+ ip === LOCALHOST ||
222
+ ip === FullNodePeer.getTrustedFullNode()) {
223
+ FullNodePeer.peerWeights.set(ip, 5); // Higher weight for prioritized peers
224
+ }
225
+ else {
226
+ FullNodePeer.peerWeights.set(ip, 1); // Default weight
227
+ }
109
228
  }
110
- return null;
111
- })).then((results) => results.filter((ip) => ip !== null));
112
- if (reachableIps.length > 0) {
113
- return reachableIps;
114
229
  }
115
- // @ts-ignore
116
- if (FullNodePeer.memoizedFetchNewPeerIPs?.cache?.clear) {
117
- // @ts-ignore
118
- FullNodePeer.memoizedFetchNewPeerIPs.cache.clear();
230
+ }
231
+ /**
232
+ * Selects the next peer based on round-robin selection.
233
+ * @returns {string} The selected peer IP.
234
+ */
235
+ static selectPeerRoundRobin() {
236
+ const peers = Array.from(FullNodePeer.peerWeights.keys()).filter((ip) => !FullNodePeer.cooldownCache.has(ip));
237
+ if (peers.length === 0) {
238
+ throw new Error("All fullnode peers are in cooldown.");
239
+ }
240
+ // Round-robin selection
241
+ const selectedPeer = peers[FullNodePeer.roundRobinIndex % peers.length];
242
+ FullNodePeer.roundRobinIndex += 1;
243
+ return selectedPeer;
244
+ }
245
+ /**
246
+ * Retrieves all reachable peer IPs, excluding those in cooldown.
247
+ * @returns {Promise<string[]>} An array of reachable peer IPs.
248
+ */
249
+ static async getPeerIPs() {
250
+ const peerIPs = await FullNodePeer.fetchNewPeerIPs();
251
+ return peerIPs;
252
+ }
253
+ /**
254
+ * Initializes the peer weights based on prioritization and reliability.
255
+ * @param {string[]} peerIPs - An array of peer IPs.
256
+ */
257
+ static setupPeers(peerIPs) {
258
+ FullNodePeer.initializePeerWeights(peerIPs);
259
+ }
260
+ /**
261
+ * Connects to the best available peer based on round-robin selection and reliability.
262
+ * @returns {Promise<Peer>} The connected Peer instance.
263
+ */
264
+ async getBestPeer() {
265
+ const now = Date.now();
266
+ // Removed cachedPeer logic to ensure a new connection each time
267
+ // Fetch peer IPs
268
+ const peerIPs = await FullNodePeer.getPeerIPs();
269
+ // Setup peer weights with prioritization
270
+ FullNodePeer.setupPeers(peerIPs);
271
+ // Round-robin selection
272
+ let selectedPeerIP;
273
+ try {
274
+ selectedPeerIP = FullNodePeer.selectPeerRoundRobin();
275
+ }
276
+ catch (error) {
277
+ throw new Error(`Failed to select a fullnode peer: ${error.message}`);
278
+ }
279
+ // Attempt to create a peer connection
280
+ const sslFolder = path_1.default.resolve(os_1.default.homedir(), ".dig", "ssl");
281
+ const certFile = path_1.default.join(sslFolder, "public_dig.crt");
282
+ const keyFile = path_1.default.join(sslFolder, "public_dig.key");
283
+ if (!fs_1.default.existsSync(sslFolder)) {
284
+ fs_1.default.mkdirSync(sslFolder, { recursive: true });
285
+ }
286
+ const tls = new datalayer_driver_1.Tls(certFile, keyFile);
287
+ let peer;
288
+ try {
289
+ peer = await datalayer_driver_1.Peer.new(`${selectedPeerIP}:${FULLNODE_PORT}`, false, tls);
290
+ }
291
+ catch (error) {
292
+ console.error(`Failed to create fullnode peer for IP ${selectedPeerIP}: ${error.message}`);
293
+ // Add to cooldown
294
+ FullNodePeer.cooldownCache.set(selectedPeerIP, true);
295
+ // Decrease weight or remove peer
296
+ const currentWeight = FullNodePeer.peerWeights.get(selectedPeerIP) || 1;
297
+ if (currentWeight > 1) {
298
+ FullNodePeer.peerWeights.set(selectedPeerIP, currentWeight - 1);
299
+ }
300
+ else {
301
+ FullNodePeer.peerWeights.delete(selectedPeerIP);
302
+ }
303
+ throw new Error(`Unable to connect to fullnode peer ${selectedPeerIP}`);
119
304
  }
120
- return FullNodePeer.memoizedFetchNewPeerIPs();
305
+ // Create a Bottleneck limiter for this peer
306
+ const limiter = new bottleneck_1.default({
307
+ maxConcurrent: 1, // One request at a time per peer
308
+ minTime: 60000 / MAX_REQUESTS_PER_MINUTE, // e.g., 600 ms between requests for 100 requests/min
309
+ });
310
+ // Store PeerInfo
311
+ FullNodePeer.peerInfos.set(selectedPeerIP, {
312
+ peer: peer,
313
+ weight: FullNodePeer.peerWeights.get(selectedPeerIP) || 1,
314
+ address: selectedPeerIP,
315
+ });
316
+ // Initialize rate limiter for this peer
317
+ FullNodePeer.peerLimiters.set(selectedPeerIP, limiter);
318
+ const proxiedPeer = this.createPeerProxy(peer, selectedPeerIP);
319
+ console.log(`Using Fullnode Peer: ${selectedPeerIP}`);
320
+ return proxiedPeer;
121
321
  }
122
- static createPeerProxy(peer) {
322
+ createPeerProxy(peer, peerIP) {
123
323
  return new Proxy(peer, {
124
324
  get: (target, prop) => {
125
325
  const originalMethod = target[prop];
@@ -129,8 +329,7 @@ class FullNodePeer {
129
329
  // Start the timeout to forget the peer after 1 minute
130
330
  const timeoutPromise = new Promise((_, reject) => {
131
331
  timeoutId = setTimeout(() => {
132
- FullNodePeer.cachedPeer = null;
133
- reject(new Error("Operation timed out. Reconnecting to a new peer."));
332
+ reject(new Error("Operation timed out. Reconnecting to a new fullnode peer."));
134
333
  }, 60000); // 1 minute
135
334
  });
136
335
  try {
@@ -147,11 +346,10 @@ class FullNodePeer {
147
346
  }
148
347
  catch (error) {
149
348
  // If the error is WebSocket-related or timeout, reset the peer
150
- if (error.message.includes("WebSocket") || error.message.includes("Operation timed out")) {
151
- FullNodePeer.cachedPeer = null;
152
- // @ts-ignore
153
- FullNodePeer.memoizedFetchNewPeerIPs.cache.clear();
154
- const newPeer = await FullNodePeer.getBestPeer();
349
+ if (error.message.includes("WebSocket") ||
350
+ error.message.includes("Operation timed out")) {
351
+ FullNodePeer.handlePeerDisconnection(peerIP);
352
+ const newPeer = await this.getBestPeer();
155
353
  return newPeer[prop](...args);
156
354
  }
157
355
  throw error;
@@ -162,86 +360,102 @@ class FullNodePeer {
162
360
  },
163
361
  });
164
362
  }
165
- static async getBestPeer() {
166
- const now = Date.now();
167
- if (FullNodePeer.cachedPeer &&
168
- now - FullNodePeer.cachedPeer.timestamp < CACHE_DURATION) {
169
- return FullNodePeer.cachedPeer.peer;
363
+ /**
364
+ * Handles peer disconnection by marking it in cooldown and updating internal states.
365
+ * @param {string} peerIP - The IP address of the disconnected peer.
366
+ */
367
+ static handlePeerDisconnection(peerIP) {
368
+ // Add the faulty peer to the cooldown cache
369
+ FullNodePeer.cooldownCache.set(peerIP, true);
370
+ // Decrease weight or remove peer
371
+ const currentWeight = FullNodePeer.peerWeights.get(peerIP) || 1;
372
+ if (currentWeight > 1) {
373
+ FullNodePeer.peerWeights.set(peerIP, currentWeight - 1);
170
374
  }
171
- const sslFolder = path_1.default.resolve(os_1.default.homedir(), ".dig", "ssl");
172
- const certFile = path_1.default.join(sslFolder, "public_dig.crt");
173
- const keyFile = path_1.default.join(sslFolder, "public_dig.key");
174
- if (!fs_1.default.existsSync(sslFolder)) {
175
- fs_1.default.mkdirSync(sslFolder, { recursive: true });
375
+ else {
376
+ FullNodePeer.peerWeights.delete(peerIP);
176
377
  }
177
- new chia_server_coin_1.Tls(certFile, keyFile);
178
- const peerIPs = await FullNodePeer.getPeerIPs();
179
- const trustedNodeIp = process.env.TRUSTED_FULLNODE || null;
180
- const peers = await Promise.all(peerIPs.map(async (ip) => {
181
- if (ip) {
182
- try {
183
- const peer = await datalayer_driver_1.Peer.new(`${ip}:${FULLNODE_PORT}`, false, certFile, keyFile);
184
- return FullNodePeer.createPeerProxy(peer);
185
- }
186
- catch (error) {
187
- console.error(`Failed to create peer for IP ${ip}: ${error.message}`);
188
- return null;
189
- }
378
+ // Remove from peerInfos
379
+ FullNodePeer.peerInfos.delete(peerIP);
380
+ // Remove the limiter
381
+ FullNodePeer.peerLimiters.delete(peerIP);
382
+ console.warn(`Fullnode Peer ${peerIP} has been marked as disconnected and is in cooldown.`);
383
+ }
384
+ /**
385
+ * Extracts the IP address from a Peer instance.
386
+ * @param {Peer} peer - The Peer instance.
387
+ * @returns {string | null} The extracted IP address or null if not found.
388
+ */
389
+ static extractPeerIP(peer) {
390
+ for (const [ip, info] of FullNodePeer.peerInfos.entries()) {
391
+ if (info.peer === peer) {
392
+ return ip;
190
393
  }
191
- return null;
192
- })).then((results) => results.filter((peer) => peer !== null));
193
- if (peers.length === 0) {
194
- throw new Error("No peers available, please try again.");
195
- }
196
- await new Promise((resolve) => setTimeout(resolve, 1000));
197
- const peakHeights = await Promise.all(peers.map((peer) => peer
198
- .getPeak()
199
- .then((height) => height)
200
- .catch((error) => {
201
- console.error(`Failed to get peak for peer: ${error.message}`);
202
- return null;
203
- })));
204
- const validHeights = peakHeights.filter((height) => height !== null);
205
- if (validHeights.length === 0) {
206
- throw new Error("No valid peak heights obtained from any peer.");
207
- }
208
- const highestPeak = Math.max(...validHeights);
209
- // Prioritize LOCALHOST and TRUSTED_NODE_IP if they have the highest peak height
210
- let bestPeerIndex = validHeights.findIndex((height, index) => height === highestPeak &&
211
- (peerIPs[index] === LOCALHOST || peerIPs[index] === trustedNodeIp));
212
- // If LOCALHOST or TRUSTED_NODE_IP don't have the highest peak, select any peer with the highest peak
213
- if (bestPeerIndex === -1) {
214
- bestPeerIndex = validHeights.findIndex((height) => height === highestPeak);
215
- }
216
- const bestPeer = peers[bestPeerIndex];
217
- FullNodePeer.cachedPeer = { peer: bestPeer, timestamp: now };
218
- console.log(`Using Peer: ${peerIPs[bestPeerIndex]}`);
219
- return bestPeer;
220
- }
221
- getPeer() {
222
- return this.peer;
394
+ }
395
+ return null;
223
396
  }
397
+ /**
398
+ * Waits for a coin to be confirmed (spent) on the blockchain.
399
+ * @param {Buffer} parentCoinInfo - The parent coin information.
400
+ * @returns {Promise<boolean>} Whether the coin was confirmed.
401
+ */
224
402
  static async waitForConfirmation(parentCoinInfo) {
225
403
  const spinner = (0, nanospinner_1.createSpinner)("Waiting for confirmation...").start();
226
- const peer = await FullNodePeer.connect();
404
+ let peer;
405
+ try {
406
+ peer = await FullNodePeer.connect();
407
+ }
408
+ catch (error) {
409
+ spinner.error({ text: "Failed to connect to a fullnode peer." });
410
+ console.error(`waitForConfirmation connection error: ${error.message}`);
411
+ throw error;
412
+ }
413
+ // Extract peer IP to access the corresponding limiter
414
+ const peerIP = FullNodePeer.extractPeerIP(peer);
415
+ if (!peerIP) {
416
+ spinner.error({ text: "Failed to extract fullnode peer IP." });
417
+ throw new Error("Failed to extract peer IP.");
418
+ }
419
+ const limiter = FullNodePeer.peerLimiters.get(peerIP);
420
+ if (!limiter) {
421
+ spinner.error({ text: "No rate limiter found for the fullnode peer." });
422
+ throw new Error("No rate limiter found for the peer.");
423
+ }
227
424
  try {
228
425
  while (true) {
229
- const confirmed = await peer.isCoinSpent(parentCoinInfo, config_1.MIN_HEIGHT, Buffer.from(config_1.MIN_HEIGHT_HEADER_HASH, "hex"));
426
+ // Schedule the isCoinSpent method call through the limiter
427
+ const confirmed = await limiter.schedule(() => peer.isCoinSpent(parentCoinInfo, config_1.MIN_HEIGHT, Buffer.from(config_1.MIN_HEIGHT_HEADER_HASH, "hex")));
230
428
  if (confirmed) {
231
429
  spinner.success({ text: "Coin confirmed!" });
232
430
  return true;
233
431
  }
432
+ // Wait for 5 seconds before the next check
234
433
  await new Promise((resolve) => setTimeout(resolve, 5000));
235
434
  }
236
435
  }
237
436
  catch (error) {
238
437
  spinner.error({ text: "Error while waiting for confirmation." });
438
+ console.error(`waitForConfirmation error: ${error.message}`);
239
439
  throw error;
240
440
  }
241
441
  }
242
442
  }
243
443
  exports.FullNodePeer = FullNodePeer;
244
- FullNodePeer.cachedPeer = null;
245
- (() => {
246
- FullNodePeer.memoizedFetchNewPeerIPs = (0, lodash_1.memoize)(FullNodePeer.fetchNewPeerIPs);
247
- })();
444
+ // Singleton instance
445
+ FullNodePeer.instance = null;
446
+ // Cooldown cache to exclude faulty peers temporarily
447
+ FullNodePeer.cooldownCache = new node_cache_1.default({ stdTTL: COOLDOWN_DURATION / 1000 });
448
+ // Failed DNS hosts cooldown cache
449
+ FullNodePeer.failedDNSCache = new node_cache_1.default({ stdTTL: COOLDOWN_DURATION / 1000 });
450
+ // Peer reliability weights
451
+ FullNodePeer.peerWeights = new Map();
452
+ // List of prioritized peers
453
+ FullNodePeer.prioritizedPeers = [];
454
+ // Map to store PeerInfo
455
+ FullNodePeer.peerInfos = new Map();
456
+ // Cache for fetched peer IPs
457
+ FullNodePeer.peerIPCache = new node_cache_1.default({ stdTTL: CACHE_DURATION / 1000 });
458
+ // Map to store rate limiters per peer IP
459
+ FullNodePeer.peerLimiters = new Map();
460
+ // Round-robin index
461
+ FullNodePeer.roundRobinIndex = 0;
@@ -1,21 +1,29 @@
1
- import { ServerCoin as ServerCoinDriver } from "datalayer-driver";
1
+ import { ServerCoin as ServerCoinDriver } from "@dignetwork/datalayer-driver";
2
2
  import { NconfManager } from "../utils/NconfManager";
3
3
  export declare class ServerCoin {
4
4
  private storeId;
5
5
  static readonly serverCoinManager: NconfManager;
6
6
  constructor(storeId: string);
7
- createForEpoch(peerIp: string): Promise<ServerCoinDriver>;
8
- saveServerCoinData(serverCoin: ServerCoinDriver, epoch: number, peerIp: string): Promise<void>;
7
+ createForEpoch(peerIp: string, rootHash: string): Promise<ServerCoinDriver>;
8
+ saveServerCoinData(serverCoin: ServerCoinDriver, epoch: number, peerIp: string, rootHash: string): Promise<void>;
9
9
  melt(epoch: number, peerIp: string): Promise<void>;
10
10
  private removeServerCoinData;
11
+ getAllEpochPeers(epoch: number, blacklist?: string[]): Promise<string[]>;
12
+ getActiveEpochPeers(blacklist?: string[]): Promise<string[]>;
11
13
  sampleCurrentEpoch(sampleSize?: number, blacklist?: string[]): Promise<string[]>;
12
14
  sampleServerCoinsByEpoch(epoch: number, sampleSize?: number, blacklist?: string[]): Promise<string[]>;
13
- static getCurrentEpoch(): number;
15
+ static getCurrentEpoch(): {
16
+ epoch: number;
17
+ round: number;
18
+ };
14
19
  ensureServerCoinExists(peerIp: string): Promise<void>;
15
20
  meltOutdatedEpochs(peerIp: string): Promise<void>;
16
21
  static meltUntrackedStoreCoins(): Promise<void>;
17
22
  private getServerCoinsForStore;
18
23
  hasEpochCoinBeenCreated(currentEpoch: number, peerIp: string): Promise<boolean>;
19
- static calculateEpoch(currentTimestampUTC: Date): number;
24
+ static calculateEpochAndRound(currentTimestampUTC: Date): {
25
+ epoch: number;
26
+ round: number;
27
+ };
20
28
  }
21
29
  //# sourceMappingURL=ServerCoin.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"ServerCoin.d.ts","sourceRoot":"","sources":["../../src/blockchain/ServerCoin.ts"],"names":[],"mappings":"AACA,OAAO,EAML,UAAU,IAAI,gBAAgB,EAE/B,MAAM,kBAAkB,CAAC;AAI1B,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAMrD,qBAAa,UAAU;IACrB,OAAO,CAAC,OAAO,CAAS;IACxB,gBAAuB,iBAAiB,eAAwC;gBAGpE,OAAO,EAAE,MAAM;IAKd,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA2DzD,kBAAkB,CAC7B,UAAU,EAAE,gBAAgB,EAC5B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,IAAI,CAAC;IAqBH,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YAoDjD,oBAAoB;IAgBrB,kBAAkB,CAC7B,UAAU,GAAE,MAAU,EACtB,SAAS,GAAE,MAAM,EAAO,GACvB,OAAO,CAAC,MAAM,EAAE,CAAC;IAMP,wBAAwB,CACnC,KAAK,EAAE,MAAM,EACb,UAAU,GAAE,MAAU,EACtB,SAAS,GAAE,MAAM,EAAO,GACvB,OAAO,CAAC,MAAM,EAAE,CAAC;WAsCN,eAAe,IAAI,MAAM;IAK1B,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiDrD,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;WA+B1C,uBAAuB;YA4D7B,sBAAsB;IAWvB,uBAAuB,CAClC,YAAY,EAAE,MAAM,EACpB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,OAAO,CAAC;WAuBL,cAAc,CAAC,mBAAmB,EAAE,IAAI,GAAG,MAAM;CAiBhE"}
1
+ {"version":3,"file":"ServerCoin.d.ts","sourceRoot":"","sources":["../../src/blockchain/ServerCoin.ts"],"names":[],"mappings":"AACA,OAAO,EAML,UAAU,IAAI,gBAAgB,EAE/B,MAAM,8BAA8B,CAAC;AAGtC,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AAYrD,qBAAa,UAAU;IACrB,OAAO,CAAC,OAAO,CAAS;IACxB,gBAAuB,iBAAiB,eAEtC;gBAEU,OAAO,EAAE,MAAM;IAKd,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA4D3E,kBAAkB,CAC7B,UAAU,EAAE,gBAAgB,EAC5B,KAAK,EAAE,MAAM,EACb,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,MAAM,GACf,OAAO,CAAC,IAAI,CAAC;IAsBH,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YAoDjD,oBAAoB;IAerB,gBAAgB,CAC3B,KAAK,EAAE,MAAM,EACb,SAAS,GAAE,MAAM,EAAO,GACvB,OAAO,CAAC,MAAM,EAAE,CAAC;IAoDP,mBAAmB,CAC9B,SAAS,GAAE,MAAM,EAAO,GACvB,OAAO,CAAC,MAAM,EAAE,CAAC;IAMP,kBAAkB,CAC7B,UAAU,GAAE,MAAU,EACtB,SAAS,GAAE,MAAM,EAAO,GACvB,OAAO,CAAC,MAAM,EAAE,CAAC;IAMP,wBAAwB,CACnC,KAAK,EAAE,MAAM,EACb,UAAU,GAAE,MAAU,EACtB,SAAS,GAAE,MAAM,EAAO,GACvB,OAAO,CAAC,MAAM,EAAE,CAAC;WAeN,eAAe,IAAI;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE;IAKpD,sBAAsB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiFrD,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;WA+B1C,uBAAuB;YAkE7B,sBAAsB;IAWvB,uBAAuB,CAClC,YAAY,EAAE,MAAM,EACpB,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,OAAO,CAAC;WAuBL,sBAAsB,CAAC,mBAAmB,EAAE,IAAI,GAAG;QAC/D,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE,MAAM,CAAC;KACf;CA2BF"}