@ar.io/sdk 1.2.0-alpha.9 → 1.2.1

package/lib/cjs/utils/graphql/processes.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getANTProcessesOwnedByWallet = void 0;
+exports.ArNSEventEmitter = exports.getANTProcessesOwnedByWallet = void 0;
 /**
  * Copyright (C) 2022-2024 Permanent Data Solutions, Inc. All Rights Reserved.
  *
@@ -17,15 +17,15 @@ exports.getANTProcessesOwnedByWallet = void 0;
  * You should have received a copy of the GNU Affero General Public License
  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
+const eventemitter3_1 = require("eventemitter3");
 const plimit_lit_1 = require("plimit-lit");
 const ant_js_1 = require("../../common/ant.js");
 const io_js_1 = require("../../common/io.js");
 const constants_js_1 = require("../../constants.js");
-// throttle the requests to avoid rate limiting
-const throttle = (0, plimit_lit_1.pLimit)(50);
 const getANTProcessesOwnedByWallet = async ({ address, contract = io_js_1.IO.init({
-    processId: constants_js_1.ioDevnetProcessId,
+    processId: constants_js_1.IO_TESTNET_PROCESS_ID,
 }), }) => {
+    const throttle = (0, plimit_lit_1.pLimit)(50);
     // get the record names of the registry - TODO: this may need to be paginated
     const uniqueContractProcessIds = await contract
         .getArNSRecords()
@@ -33,15 +33,95 @@ const getANTProcessesOwnedByWallet = async ({ address, contract = io_js_1.IO.ini
         .filter((record) => record.processId !== undefined)
         .map((record) => record.processId));
     // check the contract owner and controllers
-    const ownedOrControlledByWallet = await Promise.all(uniqueContractProcessIds.filter(async (processId) => throttle(async () => {
+    const ownedOrControlledByWallet = await Promise.all(uniqueContractProcessIds.map(async (processId) => throttle(async () => {
         const ant = ant_js_1.ANT.init({
             processId,
         });
-        const owner = await ant.getOwner();
-        const controllers = await ant.getControllers();
-        return owner === address || controllers.includes(address);
+        const { Owner, Controllers } = await ant.getState();
+        if (Owner === address || Controllers.includes(address)) {
+            return processId;
+        }
+        return;
     })));
+    if (ownedOrControlledByWallet.length === 0) {
+        return [];
+    }
     // TODO: insert gql query to find ANT processes owned by wallet given wallet not currently in the registry
     return [...new Set(ownedOrControlledByWallet)];
 };
 exports.getANTProcessesOwnedByWallet = getANTProcessesOwnedByWallet;
+function timeout(ms, promise) {
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(() => {
+            reject(new Error('Timeout'));
+        }, ms);
+        promise
+            .then((value) => {
+            clearTimeout(timer);
+            resolve(value);
+        })
+            .catch((err) => {
+            clearTimeout(timer);
+            reject(err);
+        });
+    });
+}
+class ArNSEventEmitter extends eventemitter3_1.EventEmitter {
+    contract;
+    timeoutMs; // timeout for each request to 3 seconds
+    throttle;
+    constructor({ contract = io_js_1.IO.init({
+        processId: constants_js_1.IO_TESTNET_PROCESS_ID,
+    }), timeoutMs = 60_000, concurrency = 30, }) {
+        super();
+        this.contract = contract;
+        this.timeoutMs = timeoutMs;
+        this.throttle = (0, plimit_lit_1.pLimit)(concurrency);
+    }
+    async fetchProcessesOwnedByWallet({ address }) {
+        const uniqueContractProcessIds = {};
+        await timeout(this.timeoutMs, this.contract.getArNSRecords().catch((e) => {
+            this.emit('error', `Error getting ArNS records: ${e}`);
+            return {};
+        })).then((records) => {
+            if (!records)
+                return;
+            Object.entries(records).forEach(([name, record]) => {
+                if (record.processId === undefined) {
+                    return;
+                }
+                if (uniqueContractProcessIds[record.processId] === undefined) {
+                    uniqueContractProcessIds[record.processId] = {
+                        state: undefined,
+                        names: {},
+                    };
+                }
+                uniqueContractProcessIds[record.processId].names[name] = record;
+            });
+        });
+        const idCount = Object.keys(uniqueContractProcessIds).length;
+        // check the contract owner and controllers
+        this.emit('progress', 0, idCount);
+        await Promise.all(Object.keys(uniqueContractProcessIds).map(async (processId, i) => this.throttle(async () => {
+            if (uniqueContractProcessIds[processId].state !== undefined) {
+                this.emit('progress', i + 1, idCount);
+                return;
+            }
+            const ant = ant_js_1.ANT.init({
+                processId,
+            });
+            const state = (await timeout(this.timeoutMs, ant.getState()).catch((e) => {
+                this.emit('error', `Error getting state for process ${processId}: ${e}`);
+                return undefined;
+            }));
+            if (state?.Owner === address ||
+                state?.Controllers.includes(address)) {
+                uniqueContractProcessIds[processId].state = state;
+                this.emit('process', processId, uniqueContractProcessIds[processId]);
+            }
+            this.emit('progress', i + 1, idCount);
+        })));
+        this.emit('end', uniqueContractProcessIds);
+    }
+}
+exports.ArNSEventEmitter = ArNSEventEmitter;

package/lib/cjs/utils/index.js

@@ -35,3 +35,4 @@ __exportStar(require("./http-client.js"), exports);
 __exportStar(require("./smartweave.js"), exports);
 __exportStar(require("./graphql/index.js"), exports);
 __exportStar(require("./ao.js"), exports);
+__exportStar(require("./json.js"), exports);

package/lib/cjs/utils/json.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.safeDecode = void 0;
+/**
+ * Copyright (C) 2022-2024 Permanent Data Solutions, Inc. All Rights Reserved.
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+function safeDecode(data) {
+    try {
+        return JSON.parse(data);
+    }
+    catch (e) {
+        return data;
+    }
+}
+exports.safeDecode = safeDecode;

package/lib/cjs/version.js

@@ -18,4 +18,4 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.version = void 0;
 // AUTOMATICALLY GENERATED FILE - DO NOT TOUCH
-exports.version = '1.2.0-alpha.9';
+exports.version = '1.2.1';

package/lib/esm/common/ant-ao.js

@@ -16,6 +16,13 @@ export class AoANTReadable {
             throw new InvalidContractConfigurationError();
         }
     }
+    async getState() {
+        const tags = [{ name: 'Action', value: 'State' }];
+        const res = await this.process.read({
+            tags,
+        });
+        return res;
+    }
     async getInfo() {
         const tags = [{ name: 'Action', value: 'Info' }];
         const info = await this.process.read({
@@ -35,7 +42,7 @@ export class AoANTReadable {
     async getRecord({ undername }) {
         const tags = [
             { name: 'Sub-Domain', value: undername },
-            { name: 'Action', value: 'Get-Record' },
+            { name: 'Action', value: 'Record' },
         ];
         const record = await this.process.read({
             tags,
@@ -51,7 +58,7 @@ export class AoANTReadable {
      * ````
      */
     async getRecords() {
-        const tags = [{ name: 'Action', value: 'Get-Records' }];
+        const tags = [{ name: 'Action', value: 'Records' }];
         const records = await this.process.read({
             tags,
         });
@@ -78,7 +85,7 @@ export class AoANTReadable {
      * ```
      */
     async getControllers() {
-        const tags = [{ name: 'Action', value: 'Get-Controllers' }];
+        const tags = [{ name: 'Action', value: 'Controllers' }];
         const controllers = await this.process.read({
             tags,
         });
@@ -178,7 +185,7 @@ export class AoANTWriteable extends AoANTReadable {
      */
     async addController({ controller, }) {
         const tags = [
-            { name: 'Action', value: 'Set-Controller' },
+            { name: 'Action', value: 'Add-Controller' },
             { name: 'Controller', value: controller },
         ];
         return this.process.send({

package/lib/esm/common/contracts/ao-process.js

@@ -16,6 +16,7 @@
  */
 import { connect } from '@permaweb/aoconnect';
 import { createData } from 'arbundles';
+import { safeDecode } from '../../utils/json.js';
 import { version } from '../../version.js';
 import { WriteInteractionError } from '../error.js';
 import { DefaultLogger } from '../logger.js';
@@ -57,18 +58,22 @@ export class AOProcess {
                     process: this.processId,
                     tags,
                 });
+                if (result.Messages.length === 0) {
+                    throw new Error(`Process ${this.processId} does not support provided action.`);
+                }
                 const tagsOutput = result.Messages[0].Tags;
                 const error = tagsOutput.find((tag) => tag.name === 'Error');
                 if (error) {
                     throw new Error(`${error.Value}: ${result.Messages[0].Data}`);
                 }
-                if (result.Messages.length === 0) {
-                    throw new Error('Process does not support provided action.');
-                }
                 this.logger.debug(`Read interaction result`, {
                     result: result.Messages[0].Data,
                 });
-                const response = JSON.parse(result.Messages[0].Data);
+                // return empty object if no data is returned
+                if (result.Messages[0].Data === undefined) {
+                    return {};
+                }
+                const response = safeDecode(result.Messages[0].Data);
                 return response;
             }
             catch (e) {
@@ -128,7 +133,13 @@ export class AOProcess {
                     const result = output.Messages[0].Data;
                     throw new WriteInteractionError(`${error.Value}: ${result}`);
                 }
-                const resultData = JSON.parse(output.Messages[0].Data);
+                if (output.Messages.length === 0) {
+                    throw new Error(`Process ${this.processId} does not support provided action.`);
+                }
+                if (output.Messages[0].Data === undefined) {
+                    return { id: messageId };
+                }
+                const resultData = safeDecode(output.Messages[0].Data);
                 this.logger.debug('Message result data', {
                     resultData,
                     messageId,