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

package/lib/cjs/cli/cli.js

@@ -145,13 +145,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',

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;

package/lib/cjs/common/io.js

@@ -1,4 +1,7 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ARIOWriteable = exports.ARIOReadable = exports.ARIO = void 0;
 /**
@@ -16,6 +19,7 @@ exports.ARIOWriteable = exports.ARIOReadable = exports.ARIO = void 0;
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+const arweave_1 = __importDefault(require("arweave"));
 const constants_js_1 = require("../constants.js");
 const io_js_1 = require("../types/io.js");
 const ao_js_1 = require("../utils/ao.js");
@@ -23,23 +27,32 @@ const arweave_js_1 = require("../utils/arweave.js");
 const ao_process_js_1 = require("./contracts/ao-process.js");
 const error_js_1 = require("./error.js");
 class ARIO {
-    static init(config) {
-        if (config && config.signer) {
+    static init({ arweave, ...config } = {}) {
+        if (config !== undefined && config.signer) {
             const { signer, ...rest } = config;
             return new ARIOWriteable({
                 ...rest,
                 signer,
+                arweave,
             });
         }
-        return new ARIOReadable(config);
+        return new ARIOReadable({
+            arweave,
+            ...config,
+        });
     }
 }
 exports.ARIO = ARIO;
 class ARIOReadable {
     process;
     epochSettings;
-    constructor(config) {
-        if (!config) {
+    arweave;
+    constructor({ arweave = arweave_1.default.init({
+        host: 'arweave.net',
+        port: 443,
+        protocol: 'https',
+    }), ...config }) {
+        if (config === undefined) {
             this.process = new ao_process_js_1.AOProcess({
                 processId: constants_js_1.ARIO_TESTNET_PROCESS_ID,
             });
@@ -55,6 +68,7 @@ class ARIOReadable {
         else {
             throw new error_js_1.InvalidContractConfigurationError();
         }
+        this.arweave = arweave;
     }
     async getInfo() {
         return this.process.read({
@@ -72,6 +86,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 +110,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 +251,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 +273,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) },
@@ -419,12 +474,13 @@ class ARIOReadable {
 exports.ARIOReadable = ARIOReadable;
 class ARIOWriteable extends ARIOReadable {
     signer;
-    constructor({ signer, ...config }) {
+    constructor({ signer, arweave, ...config }) {
         if (Object.keys(config).length === 0) {
             super({
                 process: new ao_process_js_1.AOProcess({
                     processId: constants_js_1.ARIO_TESTNET_PROCESS_ID,
                 }),
+                arweave: arweave,
             });
             this.signer = (0, ao_js_1.createAoSigner)(signer);
         }
@@ -437,6 +493,7 @@ class ARIOWriteable extends ARIOReadable {
                 process: new ao_process_js_1.AOProcess({
                     processId: config.processId,
                 }),
+                arweave: arweave,
             });
             this.signer = (0, ao_js_1.createAoSigner)(signer);
         }

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.AOS_MODULE_ID = 'tyojTpJ9fIva_7BjTA9ruGM4uk2vyTfhVUulnbVxR8I';
 exports.ANT_LUA_ID = '16_FyX-V2QU0RPSh1GIaEETSaUjNb0oVjCFpVbAfQq4';
 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,
@@ -200,3 +182,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.1';

package/lib/esm/cli/cli.js

@@ -143,13 +143,17 @@ makeCommand({
     name: 'get-observations',
     description: 'Get observations for an epoch',
     options: epochOptions,
-    action: (o) => readARIOFromOptions(o).getObservations(epochInputFromOptions(o)),
+    action: (o) => readARIOFromOptions(o)
+        .getObservations(epochInputFromOptions(o))
+        .then((result) => result ?? { message: 'No observations found for epoch' }),
 });
 makeCommand({
     name: 'get-distributions',
     description: 'Get distributions for an epoch',
     options: epochOptions,
-    action: (o) => readARIOFromOptions(o).getDistributions(epochInputFromOptions(o)),
+    action: (o) => readARIOFromOptions(o)
+        .getDistributions(epochInputFromOptions(o))
+        .then((result) => result ?? { message: 'No distributions found for epoch' }),
 });
 makeCommand({
     name: 'get-token-cost',

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

@@ -21,8 +21,8 @@ import { WriteInteractionError } from '../error.js';
 import { Logger } from '../logger.js';
 export class AOProcess {
     logger;
-    processId;
     ao;
+    processId;
     constructor({ processId, ao = connect(), logger = Logger.default, }) {
         this.processId = processId;
         this.logger = logger;

package/lib/esm/common/io.js

@@ -13,29 +13,39 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+import Arweave from 'arweave';
 import { ARIO_TESTNET_PROCESS_ID } from '../constants.js';
 import { isProcessConfiguration, isProcessIdConfiguration, } from '../types/io.js';
 import { createAoSigner } from '../utils/ao.js';
-import { paginationParamsToTags, pruneTags } from '../utils/arweave.js';
+import { getEpochDataFromGql, paginationParamsToTags, pruneTags, } from '../utils/arweave.js';
 import { AOProcess } from './contracts/ao-process.js';
 import { InvalidContractConfigurationError } from './error.js';
 export class ARIO {
-    static init(config) {
-        if (config && config.signer) {
+    static init({ arweave, ...config } = {}) {
+        if (config !== undefined && config.signer) {
             const { signer, ...rest } = config;
             return new ARIOWriteable({
                 ...rest,
                 signer,
+                arweave,
             });
         }
-        return new ARIOReadable(config);
+        return new ARIOReadable({
+            arweave,
+            ...config,
+        });
     }
 }
 export class ARIOReadable {
     process;
     epochSettings;
-    constructor(config) {
-        if (!config) {
+    arweave;
+    constructor({ arweave = Arweave.init({
+        host: 'arweave.net',
+        port: 443,
+        protocol: 'https',
+    }), ...config }) {
+        if (config === undefined) {
             this.process = new AOProcess({
                 processId: ARIO_TESTNET_PROCESS_ID,
             });
@@ -51,6 +61,7 @@ export class ARIOReadable {
         else {
             throw new InvalidContractConfigurationError();
         }
+        this.arweave = arweave;
     }
     async getInfo() {
         return this.process.read({
@@ -68,6 +79,13 @@ export 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) {
@@ -85,6 +103,17 @@ export class ARIOReadable {
         }));
     }
     async getEpoch(epoch) {
+        const epochIndex = await this.computeEpochIndex(epoch);
+        const currentIndex = await this.computeCurrentEpochIndex();
+        if (epochIndex !== undefined && +epochIndex < currentIndex) {
+            const epochData = await 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) },
@@ -215,7 +244,19 @@ export class ARIOReadable {
             tags: 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 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) },
@@ -225,6 +266,17 @@ export class ARIOReadable {
         });
     }
     async getDistributions(epoch) {
+        const epochIndex = await this.computeEpochIndex(epoch);
+        const currentIndex = await this.computeCurrentEpochIndex();
+        if (epochIndex !== undefined && +epochIndex < currentIndex) {
+            const epochData = await 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) },
@@ -414,12 +466,13 @@ export class ARIOReadable {
 }
 export class ARIOWriteable extends ARIOReadable {
     signer;
-    constructor({ signer, ...config }) {
+    constructor({ signer, arweave, ...config }) {
         if (Object.keys(config).length === 0) {
             super({
                 process: new AOProcess({
                     processId: ARIO_TESTNET_PROCESS_ID,
                 }),
+                arweave: arweave,
             });
             this.signer = createAoSigner(signer);
         }
@@ -432,6 +485,7 @@ export class ARIOWriteable extends ARIOReadable {
                 process: new AOProcess({
                     processId: config.processId,
                 }),
+                arweave: arweave,
             });
             this.signer = createAoSigner(signer);
         }

package/lib/esm/constants.js

@@ -26,6 +26,6 @@ export const arioDevnetProcessId = ARIO_DEVNET_PROCESS_ID;
 export const ARIO_TESTNET_PROCESS_ID = 'agYcCFJtrMG6cqMuZfskIkFTGvUPddICmtQSBIoPdiA';
 export const ANT_REGISTRY_ID = 'i_le_yKKPVstLTDSmkHRqf-wYphMnwB9OhleiTgMkWc';
 export const MARIO_PER_ARIO = 1_000_000;
-export const AOS_MODULE_ID = 'cbn0KKrBZH7hdNkNokuXLtGryrWM--PjSTBqIzw9Kkk';
+export const AOS_MODULE_ID = 'tyojTpJ9fIva_7BjTA9ruGM4uk2vyTfhVUulnbVxR8I';
 export const ANT_LUA_ID = '16_FyX-V2QU0RPSh1GIaEETSaUjNb0oVjCFpVbAfQq4';
 export const DEFAULT_SCHEDULER_ID = '_GQ33BkPtZrqxA84vM8Zk-N2aO0toNNu_C-l-rawrBA';

package/lib/esm/utils/ao.js

@@ -19,12 +19,8 @@ import { z } from 'zod';
 import { defaultArweave } from '../common/arweave.js';
 import { ANTRegistry, AOProcess, Logger } from '../common/index.js';
 import { ANT_LUA_ID, ANT_REGISTRY_ID, AOS_MODULE_ID, DEFAULT_SCHEDULER_ID, } from '../constants.js';
-export async function spawnANT({ signer, module = AOS_MODULE_ID, luaCodeTxId = ANT_LUA_ID, ao = connect(), scheduler = DEFAULT_SCHEDULER_ID, state, stateContractTxId, antRegistryId = ANT_REGISTRY_ID, logger = Logger.default, arweave = defaultArweave, }) {
-    //TODO: cache locally and only fetch if not cached
-    const luaString = (await arweave.transactions.getData(luaCodeTxId, {
-        decode: true,
-        string: true,
-    }));
+export async function spawnANT({ signer, module = AOS_MODULE_ID, ao = connect(), scheduler = DEFAULT_SCHEDULER_ID, state, stateContractTxId, antRegistryId = ANT_REGISTRY_ID, logger = Logger.default, }) {
+    // TODO: use On-Boot data handler for bootstrapping state instead of initialize-state
     const processId = await ao.spawn({
         module,
         scheduler,
@@ -34,10 +30,6 @@ export async function spawnANT({ signer, module = AOS_MODULE_ID, luaCodeTxId = A
                 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 AOProcess({
@@ -45,21 +37,10 @@ export async function spawnANT({ signer, module = AOS_MODULE_ID, luaCodeTxId = A
         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({
@@ -79,6 +60,7 @@ export async function spawnANT({ signer, module = AOS_MODULE_ID, luaCodeTxId = A
             initializeMsgId,
         });
     }
+    // This could be done by the ANT in On-Boot to self-register with its tagged ANT registry
     const registryClient = ANTRegistry.init({
         process: new AOProcess({
             processId: antRegistryId,
@@ -192,3 +174,20 @@ export function initANTStateForAddress({ owner, targetId, ttlSeconds = 3600, key
         },
     };
 }
+/**
+ * Uses zod schema to parse the epoch data
+ */
+export function parseAoEpochData(value) {
+    const epochDataSchema = z.object({
+        startTimestamp: z.number(),
+        startHeight: z.number(),
+        distributions: z.any(),
+        endTimestamp: z.number(),
+        prescribedObservers: z.any(),
+        prescribedNames: z.array(z.string()),
+        observations: z.any(),
+        distributionTimestamp: z.number(),
+        epochIndex: z.number(),
+    });
+    return epochDataSchema.parse(value);
+}