@proto-kit/module 0.1.1-develop.298 → 0.1.1-develop.334

package/dist/index.d.ts

@@ -6,4 +6,5 @@ export * from "./state/InMemoryStateService";
 export * from "./state/decorator";
 export * from "./method/MethodParameterDecoder";
 export * from "./runtime/MethodIdResolver";
+export * from "./factories/MethodIdFactory";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,oBAAoB,CAAC;AACnC,cAAc,yBAAyB,CAAC;AACxC,cAAc,mBAAmB,CAAC;AAClC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,mBAAmB,CAAC;AAClC,cAAc,iCAAiC,CAAC;AAChD,cAAc,4BAA4B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,wBAAwB,CAAC;AACvC,cAAc,oBAAoB,CAAC;AACnC,cAAc,yBAAyB,CAAC;AACxC,cAAc,mBAAmB,CAAC;AAClC,cAAc,8BAA8B,CAAC;AAC7C,cAAc,mBAAmB,CAAC;AAClC,cAAc,iCAAiC,CAAC;AAChD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,6BAA6B,CAAC"}

package/dist/index.js

@@ -6,3 +6,4 @@ export * from "./state/InMemoryStateService";
 export * from "./state/decorator";
 export * from "./method/MethodParameterDecoder";
 export * from "./runtime/MethodIdResolver";
+export * from "./factories/MethodIdFactory";

package/dist/method/MethodParameterDecoder.d.ts

@@ -1,9 +1,11 @@
-import { FlexibleProvable } from "o1js";
+import { FlexibleProvable, ProvableExtended } from "o1js";
 import { RuntimeModule } from "../runtime/RuntimeModule";
 export declare class MethodParameterDecoder {
     private readonly types;
     static fromMethod(target: RuntimeModule<unknown>, methodName: string): MethodParameterDecoder;
+    static fieldSize(type: ProvableExtended<unknown>): number | undefined;
     private constructor();
     fromJSON(argsJSON: string[]): FlexibleProvable<unknown>[];
+    get fieldSize(): number;
 }
 //# sourceMappingURL=MethodParameterDecoder.d.ts.map

package/dist/method/MethodParameterDecoder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MethodParameterDecoder.d.ts","sourceRoot":"","sources":["../../src/method/MethodParameterDecoder.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAoB,MAAM,MAAM,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AASzD,qBAAa,sBAAsB;IAYb,OAAO,CAAC,QAAQ,CAAC,KAAK;WAX5B,UAAU,CAAC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,MAAM;IAW3E,OAAO;IAEA,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,EAAE;CAkBjE"}
+{"version":3,"file":"MethodParameterDecoder.d.ts","sourceRoot":"","sources":["../../src/method/MethodParameterDecoder.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,MAAM,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AASzD,qBAAa,sBAAsB;IAuBb,OAAO,CAAC,QAAQ,CAAC,KAAK;WAtB5B,UAAU,CAAC,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,EAAE,UAAU,EAAE,MAAM;WAiB7D,SAAS,CAAC,IAAI,EAAE,gBAAgB,CAAC,OAAO,CAAC,GAAG,MAAM,GAAG,SAAS;IAK5E,OAAO;IAEA,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,EAAE;IAmBhE,IAAW,SAAS,IAAI,MAAM,CAI7B;CACF"}

package/dist/method/MethodParameterDecoder.js

@@ -5,8 +5,15 @@ export class MethodParameterDecoder {
     static fromMethod(target, methodName) {
         // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
         const paramtypes = Reflect.getMetadata("design:paramtypes", target, methodName);
+        if (paramtypes === undefined) {
+            throw new Error(`Method with name ${methodName} doesn't exist on this module`);
+        }
         return new MethodParameterDecoder(paramtypes);
     }
+    static fieldSize(type) {
+        // as any, since we shouldn't be using this workaround in the first place
+        return type.prototype._fields?.length ?? type.sizeInFields?.();
+    }
     constructor(types) {
         this.types = types;
     }
@@ -25,4 +32,9 @@ export class MethodParameterDecoder {
             return value;
         });
     }
+    get fieldSize() {
+        return this.types
+            .map((type) => MethodParameterDecoder.fieldSize(type) ?? 0)
+            .reduce((a, b) => a + b, 0);
+    }
 }

package/dist/method/runtimeMethod.d.ts

@@ -1,9 +1,9 @@
 import { StateTransition, MethodPublicOutput } from "@proto-kit/protocol";
-import { ToFieldable } from "@proto-kit/common";
+import { ArgumentTypes } from "@proto-kit/common";
 import type { RuntimeModule } from "../runtime/RuntimeModule.js";
 export declare function toStateTransitionsHash(stateTransitions: StateTransition<any>[]): import("o1js/dist/node/lib/field.js").Field;
-export type WrappedMethod = (...args: unknown[]) => MethodPublicOutput;
-export declare function toWrappedMethod(this: RuntimeModule<unknown>, methodName: string, moduleMethod: (...args: unknown[]) => unknown, methodArguments: ToFieldable[]): WrappedMethod;
+export type WrappedMethod = (...args: ArgumentTypes) => MethodPublicOutput;
+export declare function toWrappedMethod(this: RuntimeModule<unknown>, methodName: string, moduleMethod: (...args: ArgumentTypes) => unknown, methodArguments: ArgumentTypes): WrappedMethod;
 export declare function combineMethodName(runtimeModuleName: string, methodName: string): string;
 export declare const runtimeMethodMetadataKey = "yab-method";
 export declare const runtimeMethodNamesMetadataKey = "proto-kit-runtime-methods";

package/dist/method/runtimeMethod.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"runtimeMethod.d.ts","sourceRoot":"","sources":["../../src/method/runtimeMethod.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,eAAe,EAGf,kBAAkB,EAEnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAIL,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAE3B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAoBjE,wBAAgB,sBAAsB,CAEpC,gBAAgB,EAAE,eAAe,CAAC,GAAG,CAAC,EAAE,+CAczC;AAGD,MAAM,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,kBAAkB,CAAC;AAEvE,wBAAgB,eAAe,CAC7B,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,EAC5B,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,EAC7C,eAAe,EAAE,WAAW,EAAE,iBA0F/B;AAED,wBAAgB,iBAAiB,CAC/B,iBAAiB,EAAE,MAAM,EACzB,UAAU,EAAE,MAAM,UAGnB;AAED,eAAO,MAAM,wBAAwB,eAAe,CAAC;AACrD,eAAO,MAAM,6BAA6B,8BAA8B,CAAC;AAEzE;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAC7B,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,EAC9B,WAAW,EAAE,MAAM,WAKpB;AAED,wBAAgB,aAAa,aAEjB,cAAc,OAAO,CAAC,cAClB,MAAM,cACN,kBAAkB,UAwFjC"}
+{"version":3,"file":"runtimeMethod.d.ts","sourceRoot":"","sources":["../../src/method/runtimeMethod.ts"],"names":[],"mappings":"AASA,OAAO,EACL,eAAe,EAGf,kBAAkB,EAEnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAOL,aAAa,EAGd,MAAM,mBAAmB,CAAC;AAE3B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6BAA6B,CAAC;AAoBjE,wBAAgB,sBAAsB,CAEpC,gBAAgB,EAAE,eAAe,CAAC,GAAG,CAAC,EAAE,+CAczC;AAGD,MAAM,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,EAAE,aAAa,KAAK,kBAAkB,CAAC;AAE3E,wBAAgB,eAAe,CAC7B,IAAI,EAAE,aAAa,CAAC,OAAO,CAAC,EAC5B,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,CAAC,GAAG,IAAI,EAAE,aAAa,KAAK,OAAO,EACjD,eAAe,EAAE,aAAa,iBAqG/B;AAED,wBAAgB,iBAAiB,CAC/B,iBAAiB,EAAE,MAAM,EACzB,UAAU,EAAE,MAAM,UAGnB;AAED,eAAO,MAAM,wBAAwB,eAAe,CAAC;AACrD,eAAO,MAAM,6BAA6B,8BAA8B,CAAC;AAEzE;;;;;;;GAOG;AACH,wBAAgB,eAAe,CAC7B,MAAM,EAAE,aAAa,CAAC,OAAO,CAAC,EAC9B,WAAW,EAAE,MAAM,WAKpB;AAED,wBAAgB,aAAa,aAEjB,cAAc,OAAO,CAAC,cAClB,MAAM,cACN,kBAAkB,UAwFjC"}

package/dist/method/runtimeMethod.js

@@ -1,4 +1,5 @@
-import { Field, Poseidon, Proof } from "o1js";
+/* eslint-disable max-statements */
+import { Field, Poseidon, Proof, } from "o1js";
 import { container } from "tsyringe";
 import { DefaultProvableHashList, ProvableStateTransition, MethodPublicOutput, RuntimeMethodExecutionContext, } from "@proto-kit/protocol";
 import { toProver, } from "@proto-kit/common";
@@ -40,6 +41,7 @@ export function toWrappedMethod(methodName, moduleMethod, methodArguments) {
             throw errors.fieldNotConstant("methodId");
         }
         input.transaction.methodId.assertEquals(thisMethodId, "Runtimemethod called with wrong methodId on the transaction object");
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
         const parameterTypes = Reflect.getMetadata("design:paramtypes", this, methodName);
         /**
          * Use the type info obtained previously to convert
@@ -47,14 +49,18 @@ export function toWrappedMethod(methodName, moduleMethod, methodArguments) {
          */
         const argsFields = args.flatMap((argument, index) => {
             if (argument instanceof Proof) {
-                return [
-                    ...argument.publicInput?.toFields(),
-                    ...argument.publicOutput?.toFields(),
-                ];
-            }
-            else {
-                return parameterTypes[index].toFields(argument);
+                // eslint-disable-next-line max-len
+                // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+                const argumentType = parameterTypes[index];
+                const publicOutputType = argumentType?.publicOutputType;
+                const publicInputType = argumentType?.publicInputType;
+                const inputFields = publicInputType?.toFields(argument.publicInput) ?? [];
+                const outputFields = publicOutputType?.toFields(argument.publicOutput) ?? [];
+                return [...inputFields, ...outputFields];
             }
+            // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+            const argumentType = parameterTypes[index];
+            return argumentType.toFields(argument);
         });
         // Assert that the argsHash that has been signed matches the given arguments
         // We can use js-if here, because methodArguments is statically sizes

package/dist/runtime/MethodIdResolver.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MethodIdResolver.d.ts","sourceRoot":"","sources":["../../src/runtime/MethodIdResolver.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,OAAO,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AAE/D;;;GAGG;AACH,qBAAa,gBAAgB;IAMzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAN1B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAEpB;gBAGY,OAAO,EAAE,OAAO,CAAC,oBAAoB,CAAC,EACtC,OAAO,EAAE,oBAAoB;IAkBzC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS;IAcnE,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM;CAQnE"}
+{"version":3,"file":"MethodIdResolver.d.ts","sourceRoot":"","sources":["../../src/runtime/MethodIdResolver.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,OAAO,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AAE/D;;;GAGG;AACH,qBACa,gBAAgB;IAMzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAN1B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAEpB;gBAGY,OAAO,EAAE,OAAO,CAAC,oBAAoB,CAAC,EACtC,OAAO,EAAE,oBAAoB;IAkBzC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS;IAcnE,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM;CAQnE"}

package/dist/runtime/MethodIdResolver.js

@@ -1,10 +1,20 @@
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
 import { stringToField } from "@proto-kit/protocol";
 import { Poseidon } from "o1js";
+import { injectable } from "tsyringe";
 /**
  * Please see `getMethodId` to learn more about
  * methodId encoding
  */
-export class MethodIdResolver {
+let MethodIdResolver = class MethodIdResolver {
     constructor(runtime, modules) {
         this.runtime = runtime;
         this.modules = modules;
@@ -21,12 +31,11 @@ export class MethodIdResolver {
         }, {});
     }
     getMethodNameFromId(methodId) {
-        const { moduleName, methodName } = this.dictionary[methodId.toString()];
-        // eslint-disable-next-line no-warning-comments
-        // TODO Replace by throwing exception?
-        if (moduleName === undefined || methodName === undefined) {
+        const methodPath = this.dictionary[methodId.toString()];
+        if (methodPath === undefined) {
             return undefined;
         }
+        const { moduleName, methodName } = methodPath;
         this.runtime.assertIsValidModuleName(this.modules, moduleName);
         return [moduleName, methodName];
     }
@@ -37,4 +46,9 @@ export class MethodIdResolver {
             stringToField(methodName),
         ]).toBigInt();
     }
-}
+};
+MethodIdResolver = __decorate([
+    injectable(),
+    __metadata("design:paramtypes", [Function, Object])
+], MethodIdResolver);
+export { MethodIdResolver };

package/dist/runtime/Runtime.d.ts

@@ -16,10 +16,6 @@ export type RuntimeModulesRecord = ModulesRecord<TypedClass<RuntimeModule<unknow
  * Definition / required arguments for the Runtime class
  */
 export interface RuntimeDefinition<Modules extends RuntimeModulesRecord> {
-    /**
-     * @deprecated
-     */
-    state?: StateService;
     modules: Modules;
     config?: ModulesConfig<Modules>;
 }
@@ -38,7 +34,6 @@ export declare class Runtime<Modules extends RuntimeModulesRecord> extends Modul
     program?: ReturnType<typeof Experimental.ZkProgram>;
     definition: RuntimeDefinition<Modules>;
     zkProgrammable: ZkProgrammable<undefined, MethodPublicOutput>;
-    private readonly stateServiceProviderInstance;
     /**
      * Creates a new Runtime from the provided config
      *
@@ -47,8 +42,8 @@ export declare class Runtime<Modules extends RuntimeModulesRecord> extends Modul
     constructor(definition: RuntimeDefinition<Modules>);
     create(childContainerProvider: ChildContainerProvider): void;
     get appChain(): AreProofsEnabled | undefined;
-    get stateService(): StateService;
     get stateServiceProvider(): StateServiceProvider;
+    get stateService(): StateService;
     get methodIdResolver(): MethodIdResolver;
     /**
      * @returns The dependency injection container of this runtime

package/dist/runtime/Runtime.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"Runtime.d.ts","sourceRoot":"","sources":["../../src/runtime/Runtime.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,mBAAmB,EAAc,MAAM,UAAU,CAAC;AAC3D,OAAO,EACL,WAAW,EACX,eAAe,EACf,aAAa,EACb,aAAa,EACb,UAAU,EACV,cAAc,EACd,cAAc,EAEd,gBAAgB,EAAE,sBAAsB,EACzC,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,YAAY,EACb,MAAM,qBAAqB,CAAC;AAU7B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D;;;;;GAKG;AACH,MAAM,MAAM,oBAAoB,GAAG,aAAa,CAC9C,UAAU,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CACnC,CAAC;AAOF;;GAEG;AACH,MAAM,WAAW,iBAAiB,CAAC,OAAO,SAAS,oBAAoB;IACrE;;OAEG;IACH,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;CACjC;AAED,qBAAa,qBAAqB,CAChC,OAAO,SAAS,oBAAoB,CACpC,SAAQ,cAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC;IAE3B,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC;gBAAzB,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC;IAInD,IAAW,QAAQ,iCAElB;IAEM,gBAAgB,IAAI,cAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC;CAgHzE;AAED;;;GAGG;AACH,qBACa,OAAO,CAAC,OAAO,SAAS,oBAAoB,CACvD,SAAQ,eAAe,CAAC,OAAO,CAC/B,YAAW,kBAAkB;WAEf,IAAI,CAAC,OAAO,SAAS,oBAAoB,EACrD,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC,GACrC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IASxB,OAAO,CAAC,EAAE,UAAU,CAAC,OAAO,YAAY,CAAC,SAAS,CAAC,CAAC;IAEpD,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAEvC,cAAc,EAAE,cAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;IAErE,OAAO,CAAC,QAAQ,CAAC,4BAA4B,CAG3C;IAEF;;;;OAIG;gBACgB,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC;IAQlD,MAAM,CAAC,sBAAsB,EAAE,sBAAsB;IAM5D,IAAW,QAAQ,IAAI,gBAAgB,GAAG,SAAS,CAElD;IAED,IAAW,YAAY,IAAI,YAAY,CAEtC;IAED,IAAW,oBAAoB,IAAI,oBAAoB,CAEtD;IAED,IAAW,gBAAgB,IAAI,gBAAgB,CAE9C;IAED;;OAEG;IACH,IAAW,mBAAmB,IAAI,mBAAmB,CAEpD;IAED;;;OAGG;IACI,aAAa,CAClB,QAAQ,EAAE,MAAM,GACf,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,GAAG,SAAS;IAwBhD;;;;;;OAMG;IACI,cAAc,CACnB,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,EAChC,eAAe,EAAE,YAAY,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;IAQ9D;;OAEG;IACH,IAAW,kBAAkB,aAE5B;CACF"}
+{"version":3,"file":"Runtime.d.ts","sourceRoot":"","sources":["../../src/runtime/Runtime.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,mBAAmB,EAAc,MAAM,UAAU,CAAC;AAC3D,OAAO,EACL,WAAW,EACX,eAAe,EACf,aAAa,EACb,aAAa,EACb,UAAU,EACV,cAAc,EACd,cAAc,EAEd,gBAAgB,EAChB,sBAAsB,EACvB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,kBAAkB,EAClB,oBAAoB,EACpB,YAAY,EACb,MAAM,qBAAqB,CAAC;AAU7B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D;;;;;GAKG;AACH,MAAM,MAAM,oBAAoB,GAAG,aAAa,CAC9C,UAAU,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CACnC,CAAC;AAOF;;GAEG;AACH,MAAM,WAAW,iBAAiB,CAAC,OAAO,SAAS,oBAAoB;IACrE,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;CACjC;AAED,qBAAa,qBAAqB,CAChC,OAAO,SAAS,oBAAoB,CACpC,SAAQ,cAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC;IAE3B,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC;gBAAzB,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC;IAInD,IAAW,QAAQ,iCAElB;IAEM,gBAAgB,IAAI,cAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC;CAgHzE;AAED;;;GAGG;AACH,qBACa,OAAO,CAAC,OAAO,SAAS,oBAAoB,CACvD,SAAQ,eAAe,CAAC,OAAO,CAC/B,YAAW,kBAAkB;WAEf,IAAI,CAAC,OAAO,SAAS,oBAAoB,EACrD,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC,GACrC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IASxB,OAAO,CAAC,EAAE,UAAU,CAAC,OAAO,YAAY,CAAC,SAAS,CAAC,CAAC;IAEpD,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAEvC,cAAc,EAAE,cAAc,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;IAErE;;;;OAIG;gBACgB,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC;IAQlD,MAAM,CAAC,sBAAsB,EAAE,sBAAsB;IAI5D,IAAW,QAAQ,IAAI,gBAAgB,GAAG,SAAS,CAElD;IAED,IAAW,oBAAoB,IAAI,oBAAoB,CAItD;IAED,IAAW,YAAY,IAAI,YAAY,CAEtC;IAED,IAAW,gBAAgB,IAAI,gBAAgB,CAE9C;IAED;;OAEG;IACH,IAAW,mBAAmB,IAAI,mBAAmB,CAEpD;IAED;;;OAGG;IACI,aAAa,CAClB,QAAQ,EAAE,MAAM,GACf,CAAC,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,GAAG,SAAS;IAwBhD;;;;;;OAMG;IACI,cAAc,CACnB,UAAU,EAAE,WAAW,CAAC,OAAO,CAAC,EAChC,eAAe,EAAE,YAAY,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;IAQ9D;;OAEG;IACH,IAAW,kBAAkB,aAE5B;CACF"}

package/dist/runtime/Runtime.js

@@ -12,10 +12,9 @@ var Runtime_1;
 /* eslint-disable @typescript-eslint/no-explicit-any,@typescript-eslint/no-unsafe-assignment,max-lines */
 import { Experimental } from "o1js";
 import { injectable } from "tsyringe";
-import { ModuleContainer, ZkProgrammable } from "@proto-kit/common";
-import { MethodPublicOutput, StateServiceProvider, } from "@proto-kit/protocol";
+import { ModuleContainer, ZkProgrammable, } from "@proto-kit/common";
+import { MethodPublicOutput, } from "@proto-kit/protocol";
 import { combineMethodName, isRuntimeMethod, toWrappedMethod, } from "../method/runtimeMethod";
-import { MethodIdFactory } from "../factories/MethodIdFactory";
 const errors = {
     methodNotFound: (methodKey) => new Error(`Unable to find method with id ${methodKey}`),
 };
@@ -110,9 +109,6 @@ let Runtime = Runtime_1 = class Runtime extends ModuleContainer {
      */
     constructor(definition) {
         super(definition);
-        this.stateServiceProviderInstance = new StateServiceProvider(
-        // eslint-disable-next-line etc/no-deprecated
-        this.definition.state);
         this.definition = definition;
         this.zkProgrammable = new RuntimeZkProgrammable(this);
     }
@@ -120,16 +116,15 @@ let Runtime = Runtime_1 = class Runtime extends ModuleContainer {
     // TODO Remove after changing DFs to type-based approach
     create(childContainerProvider) {
         super.create(childContainerProvider);
-        this.registerDependencyFactories([MethodIdFactory]);
     }
     get appChain() {
         return this.container.resolve("AreProofsEnabled");
     }
-    get stateService() {
-        return this.stateServiceProviderInstance.stateService;
-    }
     get stateServiceProvider() {
-        return this.stateServiceProviderInstance;
+        return this.dependencyContainer.resolve("StateServiceProvider");
+    }
+    get stateService() {
+        return this.stateServiceProvider.stateService;
     }
     get methodIdResolver() {
         return this.container.resolve("MethodIdResolver");

package/dist/runtime/RuntimeModule.d.ts

@@ -1,15 +1,6 @@
 import { ConfigurableModule, Presets } from "@proto-kit/common";
-import { NetworkState, RuntimeTransaction, StateService } from "@proto-kit/protocol";
-import type { Runtime, RuntimeDefinition, RuntimeModulesRecord } from "./Runtime";
+import { NetworkState, RuntimeTransaction } from "@proto-kit/protocol";
 import { RuntimeEnvironment } from "./RuntimeEnvironment";
-/**
- * This type exists to carry over certain runtime properties
- * to runtime modules, until we can inject them through DI.
- */
-export interface PartialRuntime extends Pick<Runtime<RuntimeModulesRecord>, "zkProgrammable"> {
-    definition: Pick<RuntimeDefinition<RuntimeModulesRecord>, "state">;
-    get stateService(): StateService;
-}
 /**
  * Base class for runtime modules providing the necessary utilities.
  */

package/dist/runtime/RuntimeModule.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RuntimeModule.d.ts","sourceRoot":"","sources":["../../src/runtime/RuntimeModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEhE,OAAO,EACL,YAAY,EACZ,kBAAkB,EAClB,YAAY,EAGb,MAAM,qBAAqB,CAAC;AAI7B,OAAO,KAAK,EACV,OAAO,EACP,iBAAiB,EACjB,oBAAoB,EACrB,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAM1D;;;GAGG;AACH,MAAM,WAAW,cACf,SAAQ,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,EAAE,gBAAgB,CAAC;IAC7D,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,EAAE,OAAO,CAAC,CAAC;IAEnE,IAAI,YAAY,IAAI,YAAY,CAAC;CAClC;AAED;;GAEG;AACH,qBACa,aAAa,CAAC,MAAM,CAAE,SAAQ,kBAAkB,CAAC,MAAM,CAAC;IACnE,OAAc,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAM;IAE7C;;OAEG;IACH,SAAgB,kBAAkB,EAAE,MAAM,EAAE,CAAM;IAElD;;;;OAIG;IACI,eAAe,UAAQ;IAEvB,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,OAAO,CAAC,EAAE,kBAAkB,CAAC;;IAYpC,OAAO,CAAC,SAAS;IAWjB,IAAW,WAAW,IAAI,kBAAkB,CAE3C;IAED,IAAW,OAAO,IAAI,YAAY,CAEjC;CACF"}
+{"version":3,"file":"RuntimeModule.d.ts","sourceRoot":"","sources":["../../src/runtime/RuntimeModule.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,OAAO,EAAE,MAAM,mBAAmB,CAAC;AAEhE,OAAO,EACL,YAAY,EACZ,kBAAkB,EAInB,MAAM,qBAAqB,CAAC;AAS7B,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAM1D;;GAEG;AACH,qBACa,aAAa,CAAC,MAAM,CAAE,SAAQ,kBAAkB,CAAC,MAAM,CAAC;IACnE,OAAc,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAM;IAE7C;;OAEG;IACH,SAAgB,kBAAkB,EAAE,MAAM,EAAE,CAAM;IAElD;;;;OAIG;IACI,eAAe,UAAQ;IAEvB,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,OAAO,CAAC,EAAE,kBAAkB,CAAC;;IAYpC,OAAO,CAAC,SAAS;IAWjB,IAAW,WAAW,IAAI,kBAAkB,CAE3C;IAED,IAAW,OAAO,IAAI,YAAY,CAEjC;CACF"}

package/package.json

@@ -2,7 +2,7 @@
   "name": "@proto-kit/module",
   "license": "MIT",
   "private": false,
-  "version": "0.1.1-develop.298+188a8c8",
+  "version": "0.1.1-develop.334+dbab046",
   "type": "module",
   "scripts": {
     "build": "tsc -p tsconfig.json",
@@ -31,5 +31,5 @@
     "o1js": "0.13.1",
     "tsyringe": "^4.7.0"
   },
-  "gitHead": "188a8c85859be7432b5c196219589d559356171b"
+  "gitHead": "dbab046225087a3fa87542300460bd47e9432dbc"
 }

package/src/index.ts

@@ -6,3 +6,4 @@ export * from "./state/InMemoryStateService";
 export * from "./state/decorator";
 export * from "./method/MethodParameterDecoder";
 export * from "./runtime/MethodIdResolver";
+export * from "./factories/MethodIdFactory";

package/src/method/MethodParameterDecoder.ts

@@ -19,9 +19,20 @@ export class MethodParameterDecoder {
       methodName
     );
 
+    if (paramtypes === undefined) {
+      throw new Error(
+        `Method with name ${methodName} doesn't exist on this module`
+      );
+    }
+
     return new MethodParameterDecoder(paramtypes);
   }
 
+  public static fieldSize(type: ProvableExtended<unknown>): number | undefined {
+    // as any, since we shouldn't be using this workaround in the first place
+    return (type as any).prototype._fields?.length ?? type.sizeInFields?.();
+  }
+
   private constructor(private readonly types: ProvableExtended<unknown>[]) {}
 
   public fromJSON(argsJSON: string[]): FlexibleProvable<unknown>[] {
@@ -42,4 +53,10 @@ export class MethodParameterDecoder {
       return value;
     });
   }
+
+  public get fieldSize(): number {
+    return this.types
+      .map((type) => MethodParameterDecoder.fieldSize(type) ?? 0)
+      .reduce((a, b) => a + b, 0);
+  }
 }

package/src/method/runtimeMethod.ts

@@ -1,4 +1,11 @@
-import { Field, FlexibleProvable, Poseidon, Proof } from "o1js";
+/* eslint-disable max-statements */
+import {
+  Field,
+  FlexibleProvable,
+  Poseidon,
+  Proof,
+  ProvableExtended,
+} from "o1js";
 import { container } from "tsyringe";
 import {
   StateTransition,
@@ -12,6 +19,11 @@ import {
   toProver,
   ZkProgrammable,
   ToFieldable,
+  ToFieldableStatic,
+  ProofTypes,
+  ArgumentTypes,
+  TypedClass,
+  O1JSPrimitive,
 } from "@proto-kit/common";
 
 import type { RuntimeModule } from "../runtime/RuntimeModule.js";
@@ -53,13 +65,13 @@ export function toStateTransitionsHash(
 }
 
 // eslint-disable-next-line etc/prefer-interface
-export type WrappedMethod = (...args: unknown[]) => MethodPublicOutput;
+export type WrappedMethod = (...args: ArgumentTypes) => MethodPublicOutput;
 
 export function toWrappedMethod(
   this: RuntimeModule<unknown>,
   methodName: string,
-  moduleMethod: (...args: unknown[]) => unknown,
-  methodArguments: ToFieldable[]
+  moduleMethod: (...args: ArgumentTypes) => unknown,
+  methodArguments: ArgumentTypes
 ) {
   const executionContext = container.resolve<RuntimeMethodExecutionContext>(
     RuntimeMethodExecutionContext
@@ -99,11 +111,9 @@ export function toWrappedMethod(
       "Runtimemethod called with wrong methodId on the transaction object"
     );
 
-    const parameterTypes: FlexibleProvable<unknown>[] = Reflect.getMetadata(
-      "design:paramtypes",
-      this,
-      methodName
-    );
+    // eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
+    const parameterTypes: ProofTypes[] | ToFieldableStatic[] =
+      Reflect.getMetadata("design:paramtypes", this, methodName);
 
     /**
      * Use the type info obtained previously to convert
@@ -111,13 +121,26 @@ export function toWrappedMethod(
      */
     const argsFields = args.flatMap((argument, index) => {
       if (argument instanceof Proof) {
-        return [
-          ...argument.publicInput?.toFields(),
-          ...argument.publicOutput?.toFields(),
-        ];
-      } else {
-        return parameterTypes[index].toFields(argument as any);
+        // eslint-disable-next-line max-len
+        // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+        const argumentType = parameterTypes[index] as ProofTypes;
+
+        const publicOutputType = argumentType?.publicOutputType;
+
+        const publicInputType = argumentType?.publicInputType;
+
+        const inputFields =
+          publicInputType?.toFields(argument.publicInput) ?? [];
+
+        const outputFields =
+          publicOutputType?.toFields(argument.publicOutput) ?? [];
+
+        return [...inputFields, ...outputFields];
       }
+
+      // eslint-disable-next-line @typescript-eslint/consistent-type-assertions
+      const argumentType = parameterTypes[index] as ToFieldableStatic;
+      return argumentType.toFields(argument);
     });
 
     // Assert that the argsHash that has been signed matches the given arguments
@@ -207,7 +230,7 @@ export function runtimeMethod() {
 
     descriptor.value = function value(
       this: RuntimeModule<unknown>,
-      ...args: ToFieldable[]
+      ...args: ArgumentTypes
     ) {
       const constructorName = this.constructor.name;
 

package/src/runtime/MethodIdResolver.ts

@@ -1,5 +1,6 @@
 import { stringToField } from "@proto-kit/protocol";
 import { Poseidon } from "o1js";
+import { injectable } from "tsyringe";
 
 import type { Runtime, RuntimeModulesRecord } from "./Runtime";
 
@@ -7,6 +8,7 @@ import type { Runtime, RuntimeModulesRecord } from "./Runtime";
  * Please see `getMethodId` to learn more about
  * methodId encoding
  */
+@injectable()
 export class MethodIdResolver {
   private readonly dictionary: {
     [key: string]: { moduleName: string; methodName: string };
@@ -33,14 +35,14 @@ export class MethodIdResolver {
   }
 
   public getMethodNameFromId(methodId: bigint): [string, string] | undefined {
-    const { moduleName, methodName } = this.dictionary[methodId.toString()];
+    const methodPath = this.dictionary[methodId.toString()];
 
-    // eslint-disable-next-line no-warning-comments
-    // TODO Replace by throwing exception?
-    if (moduleName === undefined || methodName === undefined) {
+    if (methodPath === undefined) {
       return undefined;
     }
 
+    const { moduleName, methodName } = methodPath;
+
     this.runtime.assertIsValidModuleName(this.modules, moduleName);
 
     return [moduleName, methodName];

package/src/runtime/Runtime.ts

@@ -11,7 +11,8 @@ import {
   ZkProgrammable,
   PlainZkProgram,
   WithZkProgrammable,
-  AreProofsEnabled, ChildContainerProvider
+  AreProofsEnabled,
+  ChildContainerProvider,
 } from "@proto-kit/common";
 import {
   MethodPublicOutput,
@@ -50,10 +51,6 @@ const errors = {
  * Definition / required arguments for the Runtime class
  */
 export interface RuntimeDefinition<Modules extends RuntimeModulesRecord> {
-  /**
-   * @deprecated
-   */
-  state?: StateService;
   modules: Modules;
   config?: ModulesConfig<Modules>;
 }
@@ -210,11 +207,6 @@ export class Runtime<Modules extends RuntimeModulesRecord>
 
   public zkProgrammable: ZkProgrammable<undefined, MethodPublicOutput>;
 
-  private readonly stateServiceProviderInstance = new StateServiceProvider(
-    // eslint-disable-next-line etc/no-deprecated
-    this.definition.state
-  );
-
   /**
    * Creates a new Runtime from the provided config
    *
@@ -230,20 +222,20 @@ export class Runtime<Modules extends RuntimeModulesRecord>
   // TODO Remove after changing DFs to type-based approach
   public create(childContainerProvider: ChildContainerProvider) {
     super.create(childContainerProvider);
-
-    this.registerDependencyFactories([MethodIdFactory]);
   }
 
   public get appChain(): AreProofsEnabled | undefined {
     return this.container.resolve<AreProofsEnabled>("AreProofsEnabled");
   }
 
-  public get stateService(): StateService {
-    return this.stateServiceProviderInstance.stateService;
+  public get stateServiceProvider(): StateServiceProvider {
+    return this.dependencyContainer.resolve<StateServiceProvider>(
+      "StateServiceProvider"
+    );
   }
 
-  public get stateServiceProvider(): StateServiceProvider {
-    return this.stateServiceProviderInstance;
+  public get stateService(): StateService {
+    return this.stateServiceProvider.stateService;
   }
 
   public get methodIdResolver(): MethodIdResolver {

package/src/runtime/RuntimeModule.ts

@@ -21,17 +21,6 @@ const errors = {
   inputDataNotSet: () => new Error("Input data for runtime execution not set"),
 };
 
-/**
- * This type exists to carry over certain runtime properties
- * to runtime modules, until we can inject them through DI.
- */
-export interface PartialRuntime
-  extends Pick<Runtime<RuntimeModulesRecord>, "zkProgrammable"> {
-  definition: Pick<RuntimeDefinition<RuntimeModulesRecord>, "state">;
-
-  get stateService(): StateService;
-}
-
 /**
  * Base class for runtime modules providing the necessary utilities.
  */