@ar.io/sdk 3.2.0 → 3.3.0-alpha.10

package/lib/cjs/cli/cli.js

@@ -64,6 +64,12 @@ const utils_js_1 = require("./utils.js");
     options: options_js_1.paginationOptions,
     action: readCommands_js_1.listGateways,
 });
+(0, utils_js_1.makeCommand)({
+    name: 'list-all-delegates',
+    description: 'List all paginated delegates from all gateways',
+    options: options_js_1.paginationOptions,
+    action: readCommands_js_1.listAllDelegatesCLICommand,
+});
 (0, utils_js_1.makeCommand)({
     name: 'get-gateway-delegates',
     description: 'Get the delegates of a gateway',
@@ -145,13 +151,17 @@ const utils_js_1 = require("./utils.js");
     name: 'get-observations',
     description: 'Get observations for an epoch',
     options: options_js_1.epochOptions,
-    action: (o) => (0, utils_js_1.readARIOFromOptions)(o).getObservations((0, utils_js_1.epochInputFromOptions)(o)),
+    action: (o) => (0, utils_js_1.readARIOFromOptions)(o)
+        .getObservations((0, utils_js_1.epochInputFromOptions)(o))
+        .then((result) => result ?? { message: 'No observations found for epoch' }),
 });
 (0, utils_js_1.makeCommand)({
     name: 'get-distributions',
     description: 'Get distributions for an epoch',
     options: options_js_1.epochOptions,
-    action: (o) => (0, utils_js_1.readARIOFromOptions)(o).getDistributions((0, utils_js_1.epochInputFromOptions)(o)),
+    action: (o) => (0, utils_js_1.readARIOFromOptions)(o)
+        .getDistributions((0, utils_js_1.epochInputFromOptions)(o))
+        .then((result) => result ?? { message: 'No distributions found for epoch' }),
 });
 (0, utils_js_1.makeCommand)({
     name: 'get-token-cost',
@@ -654,6 +664,20 @@ const utils_js_1 = require("./utils.js");
         }, (0, utils_js_1.writeActionTagsFromOptions)(options));
     },
 });
+(0, utils_js_1.makeCommand)({
+    name: 'write-action',
+    description: 'Send a write action to an AO Process',
+    options: [...options_js_1.writeActionOptions, options_js_1.optionMap.processId],
+    action: async (options) => {
+        const process = new index_js_1.AOProcess({
+            processId: (0, utils_js_1.requiredProcessIdFromOptions)(options),
+        });
+        return process.send({
+            tags: (0, utils_js_1.writeActionTagsFromOptions)(options).tags ?? [],
+            signer: (0, utils_js_1.requiredAoSignerFromOptions)(options),
+        });
+    },
+});
 if (process.argv[1].includes('bin/ar.io') || // Running from global .bin
     process.argv[1].includes('cli/cli') // Running from source
 ) {

package/lib/cjs/cli/commands/readCommands.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getVault = exports.getGatewayVaults = exports.getPrimaryName = exports.getCostDetails = exports.getTokenCost = exports.getPrescribedNames = exports.getPrescribedObservers = exports.getEpoch = exports.listArNSReturnedNames = exports.getArNSReturnedName = exports.listArNSReservedNames = exports.getArNSReservedName = exports.listArNSRecords = exports.getArNSRecord = exports.getAllowedDelegates = exports.getDelegations = exports.getGatewayDelegates = exports.listGateways = exports.getGateway = void 0;
+exports.getVault = exports.getGatewayVaults = exports.getPrimaryName = exports.getCostDetails = exports.getTokenCost = exports.getPrescribedNames = exports.getPrescribedObservers = exports.getEpoch = exports.listArNSReturnedNames = exports.getArNSReturnedName = exports.listArNSReservedNames = exports.getArNSReservedName = exports.listArNSRecords = exports.getArNSRecord = exports.getAllowedDelegates = exports.getDelegations = exports.getGatewayDelegates = exports.listAllDelegatesCLICommand = exports.listGateways = exports.getGateway = void 0;
 const token_js_1 = require("../../types/token.js");
 const utils_js_1 = require("../utils.js");
 async function getGateway(o) {
@@ -16,6 +16,11 @@ async function listGateways(o) {
     return gateways.items.length ? gateways : { message: 'No gateways found' };
 }
 exports.listGateways = listGateways;
+async function listAllDelegatesCLICommand(o) {
+    const delegates = await (0, utils_js_1.readARIOFromOptions)(o).getAllDelegates((0, utils_js_1.paginationParamsFromOptions)(o));
+    return delegates.items.length ? delegates : { message: 'No delegates found' };
+}
+exports.listAllDelegatesCLICommand = listAllDelegatesCLICommand;
 async function getGatewayDelegates(o) {
     const address = (0, utils_js_1.requiredAddressFromOptions)(o);
     const result = await (0, utils_js_1.readARIOFromOptions)(o).getGatewayDelegates({

package/lib/cjs/common/ant-registry.js

@@ -22,12 +22,8 @@ const ao_js_1 = require("../utils/ao.js");
 const index_js_2 = require("./index.js");
 class ANTRegistry {
     static init(config) {
-        if (config && config.signer) {
-            const { signer, ...rest } = config;
-            return new AoANTRegistryWriteable({
-                ...rest,
-                signer,
-            });
+        if (config !== undefined && 'signer' in config) {
+            return new AoANTRegistryWriteable(config);
         }
         return new AoANTRegistryReadable(config);
     }
@@ -36,25 +32,22 @@ exports.ANTRegistry = ANTRegistry;
 class AoANTRegistryReadable {
     process;
     constructor(config) {
-        if (config &&
-            ((0, index_js_1.isProcessIdConfiguration)(config) || (0, index_js_1.isProcessConfiguration)(config))) {
-            if ((0, index_js_1.isProcessConfiguration)(config)) {
-                this.process = config.process;
-            }
-            else if ((0, index_js_1.isProcessIdConfiguration)(config)) {
-                this.process = new index_js_2.AOProcess({
-                    processId: config.processId,
-                });
-            }
-            else {
-                throw new index_js_2.InvalidContractConfigurationError();
-            }
-        }
-        else {
+        if (config === undefined || Object.keys(config).length === 0) {
             this.process = new index_js_2.AOProcess({
                 processId: constants_js_1.ANT_REGISTRY_ID,
             });
         }
+        else if ((0, index_js_1.isProcessConfiguration)(config)) {
+            this.process = config.process;
+        }
+        else if ((0, index_js_1.isProcessIdConfiguration)(config)) {
+            this.process = new index_js_2.AOProcess({
+                processId: config.processId,
+            });
+        }
+        else {
+            throw new index_js_2.InvalidContractConfigurationError();
+        }
     }
     // Should we rename this to "getANTsByAddress"? seems more clear, though not same as handler name
     async accessControlList({ address, }) {

package/lib/cjs/common/ant.js

@@ -23,15 +23,12 @@ const ao_js_1 = require("../utils/ao.js");
 const schema_js_1 = require("../utils/schema.js");
 const index_js_2 = require("./index.js");
 class ANT {
-    static init({ signer, strict = false, ...config }) {
-        // ao supported implementation
-        if ((0, index_js_1.isProcessConfiguration)(config) || (0, index_js_1.isProcessIdConfiguration)(config)) {
-            if (!signer) {
-                return new AoANTReadable({ strict, ...config });
-            }
-            return new AoANTWriteable({ signer, strict, ...config });
+    // implementation
+    static init(config) {
+        if (config !== undefined && 'signer' in config) {
+            return new AoANTWriteable(config);
         }
-        throw new index_js_2.InvalidContractConfigurationError();
+        return new AoANTReadable(config);
     }
 }
 exports.ANT = ANT;

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

@@ -24,8 +24,8 @@ const error_js_1 = require("../error.js");
 const logger_js_1 = require("../logger.js");
 class AOProcess {
     logger;
-    processId;
     ao;
+    processId;
     constructor({ processId, ao = (0, aoconnect_1.connect)(), logger = logger_js_1.Logger.default, }) {
         this.processId = processId;
         this.logger = logger;
@@ -42,8 +42,9 @@ class AOProcess {
         let lastError;
         while (attempts < retries) {
             try {
-                this.logger.debug(`Evaluating read interaction on contract`, {
+                this.logger.debug(`Evaluating read interaction on process`, {
                     tags,
+                    processId: this.processId,
                 });
                 // map tags to inputs
                 const dryRunInput = {
@@ -56,13 +57,18 @@ class AOProcess {
                 const result = await this.ao.dryrun(dryRunInput);
                 this.logger.debug(`Read interaction result`, {
                     result,
+                    processId: this.processId,
                 });
                 const error = errorMessageFromOutput(result);
                 if (error !== undefined) {
                     throw new Error(error);
                 }
                 if (result.Messages === undefined || result.Messages.length === 0) {
-                    this.logger.debug(`Process ${this.processId} does not support provided action.`, result, tags);
+                    this.logger.debug(`Process ${this.processId} does not support provided action.`, {
+                        result,
+                        tags,
+                        processId: this.processId,
+                    });
                     throw new Error(`Process ${this.processId} does not support provided action.`);
                 }
                 const messageData = result.Messages?.[0]?.Data;
@@ -70,16 +76,18 @@ class AOProcess {
                 if (this.isMessageDataEmpty(messageData)) {
                     return undefined;
                 }
-                const response = (0, json_js_1.safeDecode)(result.Messages[0].Data);
+                const response = (0, json_js_1.safeDecode)(messageData);
                 return response;
             }
-            catch (e) {
+            catch (error) {
                 attempts++;
                 this.logger.debug(`Read attempt ${attempts} failed`, {
-                    error: e instanceof Error ? e.message : e,
+                    error: error?.message,
+                    stack: error?.stack,
                     tags,
+                    processId: this.processId,
                 });
-                lastError = e;
+                lastError = error;
                 // exponential backoff
                 await new Promise((resolve) => setTimeout(resolve, 2 ** attempts * 1000));
             }
@@ -147,16 +155,17 @@ class AOProcess {
             }
             catch (error) {
                 this.logger.error('Error sending message to process', {
-                    error: error.message,
+                    error: error?.message,
+                    stack: error?.stack,
                     processId: this.processId,
                     tags,
                 });
-                // throw on write interaction errors. No point retrying wr ite interactions, waste of gas.
+                // throw on write interaction errors. No point retrying write interactions, waste of gas.
                 if (error.message.includes('500')) {
                     this.logger.debug('Retrying send interaction', {
                         attempts,
                         retries,
-                        error: error.message,
+                        error: error?.message,
                         processId: this.processId,
                     });
                     // exponential backoff
@@ -174,18 +183,16 @@ class AOProcess {
 exports.AOProcess = AOProcess;
 function errorMessageFromOutput(output) {
     const errorData = output.Error;
-    if (errorData !== undefined) {
-        // TODO: Could clean this one up too, current error is verbose, but not always deterministic for parsing
-        // Throw the whole raw error if AO process level error
-        return errorData;
-    }
-    const error = output.Messages?.[0]?.Tags?.find((tag) => tag.name === 'Error')?.value;
+    // Attempt to extract error details from Messages.Tags if Error is undefined
+    const error = errorData ??
+        output.Messages?.[0]?.Tags?.find((tag) => tag.name === 'Error')?.value;
     if (error !== undefined) {
-        // from [string "aos"]:6846: Name is already registered
-        const lineNumber = error.match(/\d+/)?.[0];
-        const message = error.replace(/\[string "aos"\]:\d+:/, '');
-        // to more user friendly: Name is already registered (line 6846)
-        return `${message} (line ${lineNumber})`.trim();
+        // Consolidated regex to match and extract line number and AO error message or Error Tags
+        const match = error.match(/\[string "aos"]:(\d+):\s*(.+)/);
+        if (match) {
+            const [, lineNumber, errorMessage] = match;
+            return `${errorMessage.trim()} (line ${lineNumber.trim()})`.trim();
+        }
     }
     return undefined;
 }

package/lib/cjs/common/io.js

@@ -1,35 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ARIOWriteable = exports.ARIOReadable = exports.ARIO = void 0;
-/**
- * Copyright (C) 2022-2024 Permanent Data Solutions, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
 const constants_js_1 = require("../constants.js");
 const io_js_1 = require("../types/io.js");
 const ao_js_1 = require("../utils/ao.js");
 const arweave_js_1 = require("../utils/arweave.js");
+const arweave_js_2 = require("./arweave.js");
 const ao_process_js_1 = require("./contracts/ao-process.js");
 const error_js_1 = require("./error.js");
 class ARIO {
+    // Implementation
     static init(config) {
-        if (config && config.signer) {
-            const { signer, ...rest } = config;
-            return new ARIOWriteable({
-                ...rest,
-                signer,
-            });
+        if (config !== undefined && 'signer' in config) {
+            return new ARIOWriteable(config);
         }
         return new ARIOReadable(config);
     }
@@ -38,8 +21,10 @@ exports.ARIO = ARIO;
 class ARIOReadable {
     process;
     epochSettings;
+    arweave;
     constructor(config) {
-        if (!config) {
+        this.arweave = config?.arweave ?? arweave_js_2.defaultArweave;
+        if (config === undefined || Object.keys(config).length === 0) {
             this.process = new ao_process_js_1.AOProcess({
                 processId: constants_js_1.ARIO_TESTNET_PROCESS_ID,
             });
@@ -72,6 +57,13 @@ class ARIOReadable {
         const epochLengthMs = epochSettings.durationMs;
         return Math.floor((timestamp - epochZeroStartTimestamp) / epochLengthMs);
     }
+    async computeCurrentEpochIndex() {
+        const epochSettings = await this.getEpochSettings();
+        const epochZeroStartTimestamp = epochSettings.epochZeroStartTimestamp;
+        const epochLengthMs = epochSettings.durationMs;
+        const currentTimestamp = Date.now();
+        return Math.floor((currentTimestamp - epochZeroStartTimestamp) / epochLengthMs);
+    }
     async computeEpochIndex(params) {
         const epochIndex = params?.epochIndex;
         if (epochIndex !== undefined) {
@@ -89,6 +81,17 @@ class ARIOReadable {
         }));
     }
     async getEpoch(epoch) {
+        const epochIndex = await this.computeEpochIndex(epoch);
+        const currentIndex = await this.computeCurrentEpochIndex();
+        if (epochIndex !== undefined && +epochIndex < currentIndex) {
+            const epochData = await (0, arweave_js_1.getEpochDataFromGql)({
+                arweave: this.arweave,
+                epochIndex: +epochIndex,
+                processId: this.process.processId,
+            });
+            return epochData;
+        }
+        // go to the process epoch and fetch the epoch data
         const allTags = [
             { name: 'Action', value: 'Epoch' },
             { name: 'Epoch-Index', value: await this.computeEpochIndex(epoch) },
@@ -219,7 +222,19 @@ class ARIOReadable {
             tags: (0, arweave_js_1.pruneTags)(allTags),
         });
     }
+    // we need to find the epoch index for the epoch that is currently being distributed and fetch it from gql
     async getObservations(epoch) {
+        const epochIndex = await this.computeEpochIndex(epoch);
+        const currentIndex = await this.computeCurrentEpochIndex();
+        if (epochIndex !== undefined && +epochIndex < currentIndex) {
+            const epochData = await (0, arweave_js_1.getEpochDataFromGql)({
+                arweave: this.arweave,
+                epochIndex: +epochIndex,
+                processId: this.process.processId,
+            });
+            return epochData?.observations;
+        }
+        // go to the process epoch and fetch the observations
         const allTags = [
             { name: 'Action', value: 'Epoch-Observations' },
             { name: 'Epoch-Index', value: await this.computeEpochIndex(epoch) },
@@ -229,6 +244,17 @@ class ARIOReadable {
         });
     }
     async getDistributions(epoch) {
+        const epochIndex = await this.computeEpochIndex(epoch);
+        const currentIndex = await this.computeCurrentEpochIndex();
+        if (epochIndex !== undefined && +epochIndex < currentIndex) {
+            const epochData = await (0, arweave_js_1.getEpochDataFromGql)({
+                arweave: this.arweave,
+                epochIndex: +epochIndex,
+                processId: this.process.processId,
+            });
+            return epochData?.distributions;
+        }
+        // go to the process epoch and fetch the distributions
         const allTags = [
             { name: 'Action', value: 'Epoch-Distributions' },
             { name: 'Epoch-Index', value: await this.computeEpochIndex(epoch) },
@@ -415,34 +441,30 @@ class ARIOReadable {
             tags: [{ name: 'Action', value: 'Gateway-Registry-Settings' }],
         });
     }
+    async getAllDelegates(params) {
+        return this.process.read({
+            tags: [
+                { name: 'Action', value: 'All-Paginated-Delegates' },
+                ...(0, arweave_js_1.paginationParamsToTags)(params),
+            ],
+        });
+    }
 }
 exports.ARIOReadable = ARIOReadable;
 class ARIOWriteable extends ARIOReadable {
     signer;
     constructor({ signer, ...config }) {
-        if (Object.keys(config).length === 0) {
+        if (config === undefined) {
             super({
                 process: new ao_process_js_1.AOProcess({
                     processId: constants_js_1.ARIO_TESTNET_PROCESS_ID,
                 }),
             });
-            this.signer = (0, ao_js_1.createAoSigner)(signer);
-        }
-        else if ((0, io_js_1.isProcessConfiguration)(config)) {
-            super({ process: config.process });
-            this.signer = (0, ao_js_1.createAoSigner)(signer);
-        }
-        else if ((0, io_js_1.isProcessIdConfiguration)(config)) {
-            super({
-                process: new ao_process_js_1.AOProcess({
-                    processId: config.processId,
-                }),
-            });
-            this.signer = (0, ao_js_1.createAoSigner)(signer);
         }
         else {
-            throw new error_js_1.InvalidContractConfigurationError();
+            super(config);
         }
+        this.signer = (0, ao_js_1.createAoSigner)(signer);
     }
     async transfer({ target, qty, }, options) {
         const { tags = [] } = options || {};

package/lib/cjs/constants.js

@@ -29,6 +29,6 @@ exports.arioDevnetProcessId = exports.ARIO_DEVNET_PROCESS_ID;
 exports.ARIO_TESTNET_PROCESS_ID = 'agYcCFJtrMG6cqMuZfskIkFTGvUPddICmtQSBIoPdiA';
 exports.ANT_REGISTRY_ID = 'i_le_yKKPVstLTDSmkHRqf-wYphMnwB9OhleiTgMkWc';
 exports.MARIO_PER_ARIO = 1_000_000;
-exports.AOS_MODULE_ID = 'cbn0KKrBZH7hdNkNokuXLtGryrWM--PjSTBqIzw9Kkk';
-exports.ANT_LUA_ID = '16_FyX-V2QU0RPSh1GIaEETSaUjNb0oVjCFpVbAfQq4';
+exports.AOS_MODULE_ID = 'BRyNPIJWZaQ4IudfNnsfvMrZBO2YDPjgKqg_wCYT2U8';
+exports.ANT_LUA_ID = 'k9tQkbnFYZOGp6ist1yFuaqk_wOkzM5KUSNDtWzCLtg';
 exports.DEFAULT_SCHEDULER_ID = '_GQ33BkPtZrqxA84vM8Zk-N2aO0toNNu_C-l-rawrBA';

package/lib/cjs/utils/ao.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.initANTStateForAddress = exports.defaultTargetManifestId = exports.createAoSigner = exports.isAoSigner = exports.evolveANT = exports.spawnANT = void 0;
+exports.parseAoEpochData = exports.initANTStateForAddress = exports.defaultTargetManifestId = exports.createAoSigner = exports.isAoSigner = exports.evolveANT = exports.spawnANT = void 0;
 /**
  * Copyright (C) 2022-2024 Permanent Data Solutions, Inc.
  *
@@ -22,12 +22,8 @@ const zod_1 = require("zod");
 const arweave_js_1 = require("../common/arweave.js");
 const index_js_1 = require("../common/index.js");
 const constants_js_1 = require("../constants.js");
-async function spawnANT({ signer, module = constants_js_1.AOS_MODULE_ID, luaCodeTxId = constants_js_1.ANT_LUA_ID, ao = (0, aoconnect_1.connect)(), scheduler = constants_js_1.DEFAULT_SCHEDULER_ID, state, stateContractTxId, antRegistryId = constants_js_1.ANT_REGISTRY_ID, logger = index_js_1.Logger.default, arweave = arweave_js_1.defaultArweave, }) {
-    //TODO: cache locally and only fetch if not cached
-    const luaString = (await arweave.transactions.getData(luaCodeTxId, {
-        decode: true,
-        string: true,
-    }));
+async function spawnANT({ signer, module = constants_js_1.AOS_MODULE_ID, ao = (0, aoconnect_1.connect)(), scheduler = constants_js_1.DEFAULT_SCHEDULER_ID, state, stateContractTxId, antRegistryId = constants_js_1.ANT_REGISTRY_ID, logger = index_js_1.Logger.default, }) {
+    // TODO: use On-Boot data handler for bootstrapping state instead of initialize-state
     const processId = await ao.spawn({
         module,
         scheduler,
@@ -37,10 +33,6 @@ async function spawnANT({ signer, module = constants_js_1.AOS_MODULE_ID, luaCode
                 name: 'ANT-Registry-Id',
                 value: antRegistryId,
             },
-            {
-                name: 'Source-Code-TX-ID', // utility for understanding what the original source id of the lua code was
-                value: luaCodeTxId,
-            },
         ],
     });
     const aosClient = new index_js_1.AOProcess({
@@ -48,21 +40,10 @@ async function spawnANT({ signer, module = constants_js_1.AOS_MODULE_ID, luaCode
         ao,
         logger,
     });
-    const { id: evalId } = await aosClient.send({
-        tags: [
-            { name: 'Action', value: 'Eval' },
-            { name: 'App-Name', value: 'ArNS-ANT' },
-            { name: 'Source-Code-TX-ID', value: luaCodeTxId },
-        ],
-        data: luaString,
-        signer,
-    });
     logger.debug(`Spawned ANT`, {
         processId,
         module,
         scheduler,
-        luaCodeTxId,
-        evalId,
     });
     if (state) {
         const { id: initializeMsgId } = await aosClient.send({
@@ -82,6 +63,7 @@ async function spawnANT({ signer, module = constants_js_1.AOS_MODULE_ID, luaCode
             initializeMsgId,
         });
     }
+    // This could be done by the ANT in On-Boot to self-register with its tagged ANT registry
     const registryClient = index_js_1.ANTRegistry.init({
         process: new index_js_1.AOProcess({
             processId: antRegistryId,
@@ -171,6 +153,20 @@ function createAoSigner(signer) {
             typeof signer.setPublicKey === 'function') {
             await signer.setPublicKey();
         }
+        if (signer instanceof arbundles_1.ArconnectSigner) {
+            // Sign using Arconnect signDataItem API
+            const signedDataItem = await signer['signer'].signDataItem({
+                data,
+                tags,
+                target,
+                anchor,
+            });
+            const dataItem = new arbundles_1.DataItem(Buffer.from(signedDataItem));
+            return {
+                id: await dataItem.id,
+                raw: await dataItem.getRaw(),
+            };
+        }
         const dataItem = (0, arbundles_1.createData)(data, signer, { tags, target, anchor });
         const signedData = dataItem.sign(signer).then(async () => ({
             id: await dataItem.id,
@@ -200,3 +196,21 @@ function initANTStateForAddress({ owner, targetId, ttlSeconds = 3600, keywords =
     };
 }
 exports.initANTStateForAddress = initANTStateForAddress;
+/**
+ * Uses zod schema to parse the epoch data
+ */
+function parseAoEpochData(value) {
+    const epochDataSchema = zod_1.z.object({
+        startTimestamp: zod_1.z.number(),
+        startHeight: zod_1.z.number(),
+        distributions: zod_1.z.any(),
+        endTimestamp: zod_1.z.number(),
+        prescribedObservers: zod_1.z.any(),
+        prescribedNames: zod_1.z.array(zod_1.z.string()),
+        observations: zod_1.z.any(),
+        distributionTimestamp: zod_1.z.number(),
+        epochIndex: zod_1.z.number(),
+    });
+    return epochDataSchema.parse(value);
+}
+exports.parseAoEpochData = parseAoEpochData;

package/lib/cjs/utils/arweave.js

@@ -1,22 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.paginationParamsToTags = exports.pruneTags = exports.isBlockHeight = exports.validateArweaveId = void 0;
-/**
- * Copyright (C) 2022-2024 Permanent Data Solutions, Inc.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
+exports.epochDistributionNoticeGqlQuery = exports.getEpochDataFromGql = exports.paginationParamsToTags = exports.pruneTags = exports.isBlockHeight = exports.validateArweaveId = void 0;
 const constants_js_1 = require("../constants.js");
+const ao_js_1 = require("./ao.js");
 const validateArweaveId = (id) => {
     return constants_js_1.ARWEAVE_TX_REGEX.test(id);
 };
@@ -44,3 +30,52 @@ const paginationParamsToTags = (params) => {
     return (0, exports.pruneTags)(tags);
 };
 exports.paginationParamsToTags = paginationParamsToTags;
+/**
+ * Get the epoch with distribution data for the current epoch
+ * @param arweave - The Arweave instance
+ * @returns The epoch with distribution data
+ */
+const getEpochDataFromGql = async ({ arweave, epochIndex, processId = constants_js_1.ARIO_TESTNET_PROCESS_ID, }) => {
+    // fetch from gql
+    const query = (0, exports.epochDistributionNoticeGqlQuery)({ epochIndex, processId });
+    const response = await arweave.api.post('graphql', query);
+    // parse the nodes to get the id
+    if (response.data.data.transactions?.edges?.length === 0) {
+        return undefined;
+    }
+    const id = response.data.data.transactions.edges[0].node.id;
+    // fetch the transaction from arweave
+    const transaction = await arweave.api.get(id);
+    const data = transaction.data;
+    // assert it is the correct type
+    return (0, ao_js_1.parseAoEpochData)(data);
+};
+exports.getEpochDataFromGql = getEpochDataFromGql;
+const epochDistributionNoticeGqlQuery = ({ epochIndex, processId = constants_js_1.ARIO_TESTNET_PROCESS_ID, }) => {
+    // write the query
+    const gqlQuery = JSON.stringify({
+        query: `
+      query {
+        transactions(
+          tags: [
+            { name: "From-Process", values: ["${processId}"] }
+            { name: "Action", values: ["Epoch-Distribution-Notice"] }
+            { name: "Epoch-Index", values: ["${epochIndex}"] }
+            { name: "Data-Protocol", values: ["ao"] }
+          ],
+          owners: ["fcoN_xJeisVsPXA-trzVAuIiqO3ydLQxM-L4XbrQKzY"],
+          first: 1,
+          sort: HEIGHT_DESC
+        ) {
+          edges {
+            node {
+              id
+            }
+          }
+        }
+      }
+    `,
+    });
+    return gqlQuery;
+};
+exports.epochDistributionNoticeGqlQuery = epochDistributionNoticeGqlQuery;

package/lib/cjs/version.js

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