@proto-kit/module 0.1.1-develop.1665 → 0.1.1-develop.1688

package/dist/factories/MethodIdFactory.js

@@ -8,5 +8,4 @@ export class MethodIdFactory {
         };
     }
 }
-MethodIdFactory;
 //# sourceMappingURL=MethodIdFactory.js.map

package/dist/factories/MethodIdFactory.js.map

@@ -1 +1 @@
-{"version":3,"file":"MethodIdFactory.js","sourceRoot":"","sources":["../../src/factories/MethodIdFactory.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAE/D,MAAM,OAAO,eAAe;IACnB,MAAM,CAAC,YAAY;QACxB,OAAO;YACL,gBAAgB,EAAE;gBAChB,QAAQ,EAAE,gBAAgB;aAC3B;SACyB,CAAC;IAC/B,CAAC;CACF;AAED,eAA2C,CAAC"}
+{"version":3,"file":"MethodIdFactory.js","sourceRoot":"","sources":["../../src/factories/MethodIdFactory.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAE/D,MAAM,OAAO,eAAe;IACnB,MAAM,CAAC,YAAY;QACxB,OAAO;YACL,gBAAgB,EAAE;gBAChB,QAAQ,EAAE,gBAAgB;aAC3B;SACyB,CAAC;IAC/B,CAAC;CACF"}

package/dist/messages/OutgoingMessage.d.ts

@@ -0,0 +1,96 @@
+import { Field, FlexibleProvablePure, InferProvable } from "o1js";
+import { state, WithPath, WithStateServiceProvider } from "@proto-kit/protocol";
+import { StringKeyOf } from "@proto-kit/common";
+declare const OutgoingMessageKeyStruct_base: (new (value: {
+    index: import("o1js/dist/node/lib/provable/field").Field;
+    tokenId: import("o1js/dist/node/lib/provable/field").Field;
+}) => {
+    index: import("o1js/dist/node/lib/provable/field").Field;
+    tokenId: import("o1js/dist/node/lib/provable/field").Field;
+}) & {
+    _isStruct: true;
+} & Omit<import("o1js/dist/node/lib/provable/types/provable-intf").Provable<{
+    index: import("o1js/dist/node/lib/provable/field").Field;
+    tokenId: import("o1js/dist/node/lib/provable/field").Field;
+}, {
+    index: bigint;
+    tokenId: bigint;
+}>, "fromFields"> & {
+    fromFields: (fields: import("o1js/dist/node/lib/provable/field").Field[]) => {
+        index: import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: import("o1js/dist/node/lib/provable/field").Field;
+    };
+} & {
+    fromValue: (value: {
+        index: string | number | bigint | import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: string | number | bigint | import("o1js/dist/node/lib/provable/field").Field;
+    }) => {
+        index: import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: import("o1js/dist/node/lib/provable/field").Field;
+    };
+    toInput: (x: {
+        index: import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: import("o1js/dist/node/lib/provable/field").Field;
+    }) => {
+        fields?: import("o1js/dist/node/lib/provable/field").Field[] | undefined;
+        packed?: [import("o1js/dist/node/lib/provable/field").Field, number][] | undefined;
+    };
+    toJSON: (x: {
+        index: import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: import("o1js/dist/node/lib/provable/field").Field;
+    }) => {
+        index: string;
+        tokenId: string;
+    };
+    fromJSON: (x: {
+        index: string;
+        tokenId: string;
+    }) => {
+        index: import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: import("o1js/dist/node/lib/provable/field").Field;
+    };
+    empty: () => {
+        index: import("o1js/dist/node/lib/provable/field").Field;
+        tokenId: import("o1js/dist/node/lib/provable/field").Field;
+    };
+};
+export declare class OutgoingMessageKeyStruct extends OutgoingMessageKeyStruct_base {
+}
+export type OutgoingMessage<T> = {
+    key: OutgoingMessageKeyStruct;
+    value: T;
+    type: Field;
+};
+export type OutgoingMessagesRecord = Record<string, FlexibleProvablePure<any>>;
+export declare const OUTGOING_MESSAGE_BASE_PATH: import("o1js/dist/node/lib/provable/field").Field;
+export declare const OUTGOING_MESSAGE_COUNTER_PATH: import("o1js/dist/node/lib/provable/field").Field;
+export declare const outgoingMessage: typeof state;
+declare const OutgoingMessages_base: import("ts-mixer/dist/types/types").Class<any[], WithPath & WithStateServiceProvider, typeof WithPath & typeof WithStateServiceProvider>;
+export declare class OutgoingMessages<Messages extends OutgoingMessagesRecord> extends OutgoingMessages_base {
+    private readonly messages;
+    readonly eventTypes: Record<string, {
+        valueType: FlexibleProvablePure<any>;
+        completeType: FlexibleProvablePure<{
+            key: OutgoingMessageKeyStruct;
+            value: any;
+            type: Field;
+        }>;
+    }>;
+    constructor(messages: Messages);
+    private counterState;
+    static getEventName(key: string): string;
+    computeEventTypes(): {
+        [k: string]: {
+            valueType: FlexibleProvablePure<any>;
+            completeType: FlexibleProvablePure<{
+                key: OutgoingMessageKeyStruct;
+                value: any;
+                type: Field;
+            }>;
+        };
+    };
+    private emitEvent;
+    emitMessage<Key extends StringKeyOf<Messages>>(key: Key, value: InferProvable<Messages[Key]>, tokenId?: Field): Promise<void>;
+}
+export {};
+//# sourceMappingURL=OutgoingMessage.d.ts.map

package/dist/messages/OutgoingMessage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"OutgoingMessage.d.ts","sourceRoot":"","sources":["../../src/messages/OutgoingMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EACL,oBAAoB,EACpB,aAAa,EAGd,MAAM,MAAM,CAAC;AACd,OAAO,EAEL,KAAK,EAEL,QAAQ,EACR,wBAAwB,EACzB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAiB,WAAW,EAAE,MAAM,mBAAmB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAG/D,qBAAa,wBAAyB,SAAQ,6BAG5C;CAAG;AAEL,MAAM,MAAM,eAAe,CAAC,CAAC,IAAI;IAC/B,GAAG,EAAE,wBAAwB,CAAC;IAC9B,KAAK,EAAE,CAAC,CAAC;IACT,IAAI,EAAE,KAAK,CAAC;CACb,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG,MAAM,CAAC,MAAM,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC;AAE/E,eAAO,MAAM,0BAA0B,mDAEtC,CAAC;AAEF,eAAO,MAAM,6BAA6B,mDAEzC,CAAC;AAEF,eAAO,MAAM,eAAe,cAAQ,CAAC;;AAErC,qBAAa,gBAAgB,CAC3B,QAAQ,SAAS,sBAAsB,CACvC,SAAQ,qBAAyC;IAa9B,OAAO,CAAC,QAAQ,CAAC,QAAQ;IAZ5C,SAAgB,UAAU,EAAE,MAAM,CAChC,MAAM,EACN;QACE,SAAS,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC;QACrC,YAAY,EAAE,oBAAoB,CAAC;YACjC,GAAG,EAAE,wBAAwB,CAAC;YAC9B,KAAK,EAAE,GAAG,CAAC;YACX,IAAI,EAAE,KAAK,CAAC;SACb,CAAC,CAAC;KACJ,CACF,CAAC;gBAEkC,QAAQ,EAAE,QAAQ;IAKtD,OAAO,CAAC,YAAY;WAON,YAAY,CAAC,GAAG,EAAE,MAAM;IAI/B,iBAAiB;;;;qBAcP,wBAAwB;uBACtB,GAAG;sBACJ,KAAK;;;;IAQvB,OAAO,CAAC,SAAS;IAeJ,WAAW,CAAC,GAAG,SAAS,WAAW,CAAC,QAAQ,CAAC,EACxD,GAAG,EAAE,GAAG,EACR,KAAK,EAAE,aAAa,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,EACnC,OAAO,GAAE,KAAuB;CAmBnC"}

package/dist/messages/OutgoingMessage.js

@@ -0,0 +1,68 @@
+import { Field, Struct, TokenId, } from "o1js";
+import { RuntimeMethodExecutionContext, state, StateMap, WithPath, WithStateServiceProvider, } from "@proto-kit/protocol";
+// eslint-disable-next-line import/no-extraneous-dependencies
+import { Mixin } from "ts-mixer";
+import { prefixToField } from "@proto-kit/common";
+import { container } from "tsyringe";
+export class OutgoingMessageKeyStruct extends Struct({
+    index: Field,
+    tokenId: Field,
+}) {
+}
+export const OUTGOING_MESSAGE_BASE_PATH = prefixToField("outgoing-message-base-path");
+export const OUTGOING_MESSAGE_COUNTER_PATH = prefixToField("outgoing-message-base-path");
+export const outgoingMessage = state;
+export class OutgoingMessages extends Mixin(WithPath, WithStateServiceProvider) {
+    constructor(messages) {
+        super();
+        this.messages = messages;
+        this.eventTypes = this.computeEventTypes();
+    }
+    counterState() {
+        const state = new StateMap(Field, Field);
+        state.path = OUTGOING_MESSAGE_COUNTER_PATH;
+        state.stateServiceProvider = this.stateServiceProvider;
+        return state;
+    }
+    static getEventName(key) {
+        return `outgoing-${key}`;
+    }
+    computeEventTypes() {
+        return Object.fromEntries(Object.entries(this.messages).map(([key, type]) => {
+            class OutgoingMessageStruct extends Struct({
+                key: OutgoingMessageKeyStruct,
+                value: type,
+                type: Field,
+            }) {
+            }
+            return [
+                OutgoingMessages.getEventName(key),
+                {
+                    valueType: type,
+                    completeType: OutgoingMessageStruct,
+                },
+            ];
+        }));
+    }
+    emitEvent(key, value) {
+        const eventName = OutgoingMessages.getEventName(key);
+        const outgoingMessageType = this.eventTypes[eventName].completeType;
+        return container
+            .resolve(RuntimeMethodExecutionContext)
+            .addEvent(outgoingMessageType, value, eventName);
+    }
+    async emitMessage(key, value, tokenId = TokenId.default) {
+        const stateMap = new StateMap(OutgoingMessageKeyStruct, this.messages[key]);
+        stateMap.path = OUTGOING_MESSAGE_BASE_PATH;
+        stateMap.stateServiceProvider = this.stateServiceProvider;
+        const counterState = this.counterState();
+        const counterOption = await counterState.get(tokenId);
+        const counter = counterOption.orElse(Field(0));
+        const messageKey = { index: counter, tokenId };
+        await stateMap.set(messageKey, value);
+        await counterState.set(tokenId, counter.add(1));
+        const type = prefixToField(key);
+        this.emitEvent(key, { key: messageKey, value, type });
+    }
+}
+//# sourceMappingURL=OutgoingMessage.js.map

package/dist/messages/OutgoingMessage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"OutgoingMessage.js","sourceRoot":"","sources":["../../src/messages/OutgoingMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,EAGL,MAAM,EACN,OAAO,GACR,MAAM,MAAM,CAAC;AACd,OAAO,EACL,6BAA6B,EAC7B,KAAK,EACL,QAAQ,EACR,QAAQ,EACR,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAC7B,6DAA6D;AAC7D,OAAO,EAAE,KAAK,EAAE,MAAM,UAAU,CAAC;AACjC,OAAO,EAAE,aAAa,EAAe,MAAM,mBAAmB,CAAC;AAC/D,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC,MAAM,OAAO,wBAAyB,SAAQ,MAAM,CAAC;IACnD,KAAK,EAAE,KAAK;IACZ,OAAO,EAAE,KAAK;CACf,CAAC;CAAG;AAUL,MAAM,CAAC,MAAM,0BAA0B,GAAG,aAAa,CACrD,4BAA4B,CAC7B,CAAC;AAEF,MAAM,CAAC,MAAM,6BAA6B,GAAG,aAAa,CACxD,4BAA4B,CAC7B,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,KAAK,CAAC;AAErC,MAAM,OAAO,gBAEX,SAAQ,KAAK,CAAC,QAAQ,EAAE,wBAAwB,CAAC;IAajD,YAAoC,QAAkB;QACpD,KAAK,EAAE,CAAC;QAD0B,aAAQ,GAAR,QAAQ,CAAU;QAEpD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC7C,CAAC;IAEO,YAAY;QAClB,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QACzC,KAAK,CAAC,IAAI,GAAG,6BAA6B,CAAC;QAC3C,KAAK,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB,CAAC;QACvD,OAAO,KAAK,CAAC;IACf,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,GAAW;QACpC,OAAO,YAAY,GAAG,EAAE,CAAC;IAC3B,CAAC;IAEM,iBAAiB;QACtB,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YAChD,MAAM,qBAAsB,SAAQ,MAAM,CAAC;gBACzC,GAAG,EAAE,wBAAwB;gBAC7B,KAAK,EAAE,IAAI;gBACX,IAAI,EAAE,KAAK;aACZ,CAAC;aAAG;YAEL,OAAO;gBACL,gBAAgB,CAAC,YAAY,CAAC,GAAG,CAAC;gBAClC;oBACE,SAAS,EAAE,IAAI;oBACf,YAAY,EAAE,qBAIZ;iBACH;aACF,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,SAAS,CACf,GAAQ,EACR,KAAoD;QAEpD,MAAM,SAAS,GAAG,gBAAgB,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;QAErD,MAAM,mBAAmB,GAAG,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC;QAEpE,OAAO,SAAS;aACb,OAAO,CAAC,6BAA6B,CAAC;aACtC,QAAQ,CAEP,mBAAmB,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IAC7C,CAAC;IAEM,KAAK,CAAC,WAAW,CACtB,GAAQ,EACR,KAAmC,EACnC,UAAiB,OAAO,CAAC,OAAO;QAEhC,MAAM,QAAQ,GAAG,IAAI,QAAQ,CAAC,wBAAwB,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5E,QAAQ,CAAC,IAAI,GAAG,0BAA0B,CAAC;QAC3C,QAAQ,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB,CAAC;QAE1D,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACzC,MAAM,aAAa,GAAG,MAAM,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/C,MAAM,UAAU,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;QAE/C,MAAM,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACtC,MAAM,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAEhD,MAAM,IAAI,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;QAEhC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,GAAG,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC;CACF"}

package/dist/state/decorator.d.ts

@@ -0,0 +1,7 @@
+import type { RuntimeModule } from "../runtime/RuntimeModule.js";
+/**
+ * Decorates a runtime module property as state, passing down some
+ * underlying values to improve developer experience.
+ */
+export declare function state(): <TargetRuntimeModule extends RuntimeModule<unknown>>(target: TargetRuntimeModule, propertyKey: string) => void;
+//# sourceMappingURL=decorator.d.ts.map

package/dist/state/decorator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"decorator.d.ts","sourceRoot":"","sources":["../../src/state/decorator.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAgBjE;;;GAGG;AACH,wBAAgB,KAAK,mGAGJ,MAAM,UAmCtB"}

package/dist/state/decorator.js

@@ -0,0 +1,40 @@
+import { Path } from "@proto-kit/protocol";
+const errors = {
+    missingName: (className) => new Error(`Unable to provide a unique identifier for state, ${className} is missing a name. 
+      Did you forget to extend your runtime module with 'extends RuntimeModule'?`),
+    missingRuntime: (className) => new Error(`Unable to provide 'runtime' for state, ${className} is missing a name. 
+      Did you forget to extend your runtime module with 'extends RuntimeModule'?`),
+};
+/**
+ * Decorates a runtime module property as state, passing down some
+ * underlying values to improve developer experience.
+ */
+export function state() {
+    return (target, propertyKey) => {
+        let value;
+        Object.defineProperty(target, propertyKey, {
+            enumerable: true,
+            get: function get() {
+                // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+                const self = this;
+                if (self.name === undefined) {
+                    throw errors.missingName(self.constructor.name);
+                }
+                if (!self.runtime) {
+                    throw errors.missingRuntime(self.constructor.name);
+                }
+                const path = Path.fromProperty(self.name, propertyKey);
+                if (value) {
+                    value.path = path;
+                    // TODO: why is this complaining about `any`?
+                    value.stateServiceProvider = self.runtime.stateServiceProvider;
+                }
+                return value;
+            },
+            set: (newValue) => {
+                value = newValue;
+            },
+        });
+    };
+}
+//# sourceMappingURL=decorator.js.map

package/dist/state/decorator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"decorator.js","sourceRoot":"","sources":["../../src/state/decorator.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAS,MAAM,qBAAqB,CAAC;AAIlD,MAAM,MAAM,GAAG;IACb,WAAW,EAAE,CAAC,SAAiB,EAAE,EAAE,CACjC,IAAI,KAAK,CACP,oDAAoD,SAAS;iFACc,CAC5E;IAEH,cAAc,EAAE,CAAC,SAAiB,EAAE,EAAE,CACpC,IAAI,KAAK,CACP,0CAA0C,SAAS;iFACwB,CAC5E;CACJ,CAAC;AAEF;;;GAGG;AACH,MAAM,UAAU,KAAK;IACnB,OAAO,CACL,MAA2B,EAC3B,WAAmB,EACnB,EAAE;QACF,IAAI,KAAiC,CAAC;QAEtC,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,WAAW,EAAE;YACzC,UAAU,EAAE,IAAI;YAEhB,GAAG,EAAE,SAAS,GAAG;gBACf,yEAAyE;gBACzE,MAAM,IAAI,GAAG,IAA2B,CAAC;gBAEzC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE;oBAC3B,MAAM,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;iBACjD;gBAED,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE;oBACjB,MAAM,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;iBACpD;gBAED,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBACvD,IAAI,KAAK,EAAE;oBACT,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC;oBAElB,6CAA6C;oBAE7C,KAAK,CAAC,oBAAoB,GAAG,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC;iBAChE;gBACD,OAAO,KAAK,CAAC;YACf,CAAC;YAED,GAAG,EAAE,CAAC,QAAwB,EAAE,EAAE;gBAChC,KAAK,GAAG,QAAQ,CAAC;YACnB,CAAC;SACF,CAAC,CAAC;IACL,CAAC,CAAC;AACJ,CAAC"}

package/dist/testing/TestingRuntime.d.ts

@@ -0,0 +1,8 @@
+import { ModulesConfig } from "@proto-kit/common";
+import { Runtime, RuntimeModulesRecord } from "../runtime/Runtime";
+import { InMemoryStateService } from "../state/InMemoryStateService";
+export declare function createTestingRuntime<Modules extends RuntimeModulesRecord>(modules: Modules, config: ModulesConfig<Modules>): {
+    runtime: Runtime<Modules>;
+    state: InMemoryStateService;
+};
+//# sourceMappingURL=TestingRuntime.d.ts.map

package/dist/testing/TestingRuntime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestingRuntime.d.ts","sourceRoot":"","sources":["../../src/testing/TestingRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAIlD,OAAO,EAAE,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AAErE,wBAAgB,oBAAoB,CAAC,OAAO,SAAS,oBAAoB,EACvE,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,GAC7B;IACD,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;IAC1B,KAAK,EAAE,oBAAoB,CAAC;CAC7B,CAgCA"}

package/dist/testing/TestingRuntime.js

@@ -0,0 +1,31 @@
+import { StateServiceProvider } from "@proto-kit/protocol";
+import { container } from "tsyringe";
+import { Runtime } from "../runtime/Runtime";
+import { InMemoryStateService } from "../state/InMemoryStateService";
+export function createTestingRuntime(modules, config) {
+    const state = new InMemoryStateService();
+    const Runtimeclass = Runtime.from({
+        modules,
+    });
+    const runtime = new Runtimeclass();
+    runtime.configure(config);
+    runtime.create(() => container.createChildContainer());
+    runtime.dependencyContainer.register("AreProofsEnabled", {
+        useValue: {
+            areProofsEnabled: false,
+            setProofsEnabled(areProofsEnabled) {
+                this.areProofsEnabled = areProofsEnabled;
+            },
+        },
+    });
+    runtime.registerValue({
+        StateServiceProvider: new StateServiceProvider(),
+        Runtime: runtime,
+    });
+    runtime.stateServiceProvider.setCurrentStateService(state);
+    return {
+        runtime,
+        state,
+    };
+}
+//# sourceMappingURL=TestingRuntime.js.map

package/dist/testing/TestingRuntime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TestingRuntime.js","sourceRoot":"","sources":["../../src/testing/TestingRuntime.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,SAAS,EAAE,MAAM,UAAU,CAAC;AAErC,OAAO,EAAE,OAAO,EAAwB,MAAM,oBAAoB,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,+BAA+B,CAAC;AAErE,MAAM,UAAU,oBAAoB,CAClC,OAAgB,EAChB,MAA8B;IAK9B,MAAM,KAAK,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAEzC,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC;QAChC,OAAO;KACR,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;IAEnC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAE1B,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,oBAAoB,EAAE,CAAC,CAAC;IAEvD,OAAO,CAAC,mBAAmB,CAAC,QAAQ,CAAC,kBAAkB,EAAE;QACvD,QAAQ,EAAE;YACR,gBAAgB,EAAE,KAAK;YAEvB,gBAAgB,CAAC,gBAAyB;gBACxC,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;YAC3C,CAAC;SACF;KACF,CAAC,CAAC;IACH,OAAO,CAAC,aAAa,CAAC;QACpB,oBAAoB,EAAE,IAAI,oBAAoB,EAAE;QAChD,OAAO,EAAE,OAAO;KACjB,CAAC,CAAC;IAEH,OAAO,CAAC,oBAAoB,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAE3D,OAAO;QACL,OAAO;QACP,KAAK;KACN,CAAC;AACJ,CAAC"}

package/package.json

@@ -2,7 +2,7 @@
   "name": "@proto-kit/module",
   "license": "MIT",
   "private": false,
-  "version": "0.1.1-develop.1665+66550e65",
+  "version": "0.1.1-develop.1688+a8257e42",
   "type": "module",
   "scripts": {
     "build": "tsc -p tsconfig.json",
@@ -28,8 +28,8 @@
   "peerDependencies": {
     "@proto-kit/common": "*",
     "@proto-kit/protocol": "*",
-    "o1js": "^1.6.0",
+    "o1js": "^2.10.0",
     "tsyringe": "^4.10.0"
   },
-  "gitHead": "66550e6515f3194fc9495f4474f3a08bb24a6130"
+  "gitHead": "a8257e428f8cda2b7252ead17e29482d2fdb0575"
 }

package/src/method/MethodParameterEncoder.ts

@@ -5,6 +5,7 @@ import {
   Provable,
   DynamicProof,
   FlexibleProvablePure,
+  FlexibleProvable,
 } from "o1js";
 import {
   ArgumentTypes,
@@ -29,7 +30,7 @@ const errors = {
 };
 
 type ArgumentType =
-  | FlexibleProvablePure<any>
+  | FlexibleProvable<any>
   | typeof Proof<unknown, unknown>
   | typeof DynamicProof<unknown, unknown>;
 
@@ -122,7 +123,7 @@ export class MethodParameterEncoder {
       );
     }
     // as any, since we shouldn't be using this workaround in the first place
-    return (type as FlexibleProvablePure<unknown>).sizeInFields();
+    return (type as FlexibleProvable<unknown>).sizeInFields();
   }
 
   public constructor(private readonly types: ArgTypeArray) {}
@@ -178,8 +179,9 @@ export class MethodParameterEncoder {
           }
         }
 
-        return (type as FlexibleProvablePure<unknown>).fromFields(
-          structFields
+        return (type as FlexibleProvable<unknown>).fromFields(
+          structFields,
+          []
         ) as any;
       })
     );

package/src/runtime/Runtime.ts

@@ -222,10 +222,23 @@ export class RuntimeZkProgrammable<
 
     return splitRuntimeMethods().map((bucket, index) => {
       const name = `RuntimeProgram-${index}`;
+      const wrappedBucket = Object.fromEntries(
+        Object.entries(bucket).map(([methodName, methodDef]) => [
+          methodName,
+          {
+            privateInputs: methodDef.privateInputs,
+            method: async (...args: any[]) => {
+              const publicOutput = await methodDef.method(...args);
+              return { publicOutput };
+            },
+          },
+        ])
+      );
+
       const program = ZkProgram({
         name,
         publicOutput: MethodPublicOutput,
-        methods: bucket,
+        methods: wrappedBucket,
       });
 
       const SelfProof = ZkProgram.Proof(program);

package/test/method/MethodParameterEncoder.test.ts

@@ -25,8 +25,10 @@ const TestProgram = ZkProgram({
       privateInputs: [],
       method: async (input: PublicKey) => {
         return {
-          a: Field(input.x),
-          b: Bool(input.isOdd),
+          publicOutput: {
+            a: Field(input.x),
+            b: Bool(input.isOdd),
+          },
         };
       },
     },
@@ -117,5 +119,5 @@ describe("MethodParameterEncoder", () => {
     expect(decoded1.publicOutput.b.equals(output.b).toBoolean()).toStrictEqual(
       true
     );
-  }, 30000);
+  }, 60_000);
 });

package/test/method/runtimeMethod-fail.test.ts

@@ -17,8 +17,10 @@ const TestProgram = ZkProgram({
       privateInputs: [],
       method: async (input: PublicKey) => {
         return {
-          a: Field(input.x),
-          b: Bool(input.isOdd),
+          publicOutput: {
+            a: Field(input.x),
+            b: Bool(input.isOdd),
+          },
         };
       },
     },