UNPKG

rivetkit

Version:

Lightweight libraries for building stateful actors on edge platforms

239 lines (208 loc) 8.4 kB
"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } var _chunkI3FB346Icjs = require('./chunk-I3FB346I.cjs'); // src/inspector/protocol/common.ts var _v4 = require('zod/v4'); var _v42 = _interopRequireDefault(_v4); // src/manager/protocol/query.ts var _zod = require('zod'); var MAX_ACTOR_KEY_SIZE = 128; var ActorKeySchema = _zod.z.array(_zod.z.string().max(MAX_ACTOR_KEY_SIZE)); var CreateRequestSchema = _zod.z.object({ name: _zod.z.string(), key: ActorKeySchema, input: _zod.z.unknown().optional(), region: _zod.z.string().optional() }); var GetForKeyRequestSchema = _zod.z.object({ name: _zod.z.string(), key: ActorKeySchema }); var GetOrCreateRequestSchema = _zod.z.object({ name: _zod.z.string(), key: ActorKeySchema, input: _zod.z.unknown().optional(), region: _zod.z.string().optional() }); var ActorQuerySchema = _zod.z.union([ _zod.z.object({ getForId: _zod.z.object({ name: _zod.z.string(), actorId: _zod.z.string() }) }), _zod.z.object({ getForKey: GetForKeyRequestSchema }), _zod.z.object({ getOrCreateForKey: GetOrCreateRequestSchema }), _zod.z.object({ create: CreateRequestSchema }) ]); var ConnectRequestSchema = _zod.z.object({ query: ActorQuerySchema.describe(_chunkI3FB346Icjs.HEADER_ACTOR_QUERY), encoding: _chunkI3FB346Icjs.EncodingSchema.describe(_chunkI3FB346Icjs.HEADER_ENCODING), connParams: _zod.z.string().optional().describe(_chunkI3FB346Icjs.HEADER_CONN_PARAMS) }); var ConnectWebSocketRequestSchema = _zod.z.object({ query: ActorQuerySchema.describe("query"), encoding: _chunkI3FB346Icjs.EncodingSchema.describe("encoding"), connParams: _zod.z.unknown().optional().describe("conn_params") }); var ConnMessageRequestSchema = _zod.z.object({ actorId: _zod.z.string().describe(_chunkI3FB346Icjs.HEADER_ACTOR_ID), connId: _zod.z.string().describe(_chunkI3FB346Icjs.HEADER_CONN_ID), encoding: _chunkI3FB346Icjs.EncodingSchema.describe(_chunkI3FB346Icjs.HEADER_ENCODING), connToken: _zod.z.string().describe(_chunkI3FB346Icjs.HEADER_CONN_TOKEN) }); var ResolveRequestSchema = _zod.z.object({ query: ActorQuerySchema.describe(_chunkI3FB346Icjs.HEADER_ACTOR_QUERY), connParams: _zod.z.string().optional().describe(_chunkI3FB346Icjs.HEADER_CONN_PARAMS) }); // src/inspector/protocol/common.ts var ActorId = _v42.default.string().brand("ActorId"); var ActorFeature = /* @__PURE__ */ ((ActorFeature2) => { ActorFeature2["Logs"] = "logs"; ActorFeature2["Config"] = "config"; ActorFeature2["Connections"] = "connections"; ActorFeature2["State"] = "state"; ActorFeature2["Console"] = "console"; ActorFeature2["Runtime"] = "runtime"; ActorFeature2["Metrics"] = "metrics"; ActorFeature2["EventsMonitoring"] = "events-monitoring"; ActorFeature2["Database"] = "database"; return ActorFeature2; })(ActorFeature || {}); var ActorLogEntry = _v42.default.object({ level: _v42.default.string(), message: _v42.default.string(), timestamp: _v42.default.string(), metadata: _v42.default.record(_v42.default.string(), _v42.default.any()).optional() }); var ActorSchema = _v42.default.object({ id: ActorId, name: _v42.default.string(), key: _v42.default.array(_v42.default.string()), tags: _v42.default.record(_v42.default.string(), _v42.default.string()).optional(), region: _v42.default.string().optional(), createdAt: _v42.default.string().optional(), startedAt: _v42.default.string().optional(), destroyedAt: _v42.default.string().optional(), features: _v42.default.array(_v42.default.enum(ActorFeature)).optional() }); var OperationSchema = _v42.default.discriminatedUnion("op", [ _v42.default.object({ op: _v42.default.literal("remove"), path: _v42.default.string() }), _v42.default.object({ op: _v42.default.literal("add"), path: _v42.default.string(), value: _v42.default.unknown() }), _v42.default.object({ op: _v42.default.literal("replace"), path: _v42.default.string(), value: _v42.default.unknown() }), _v42.default.object({ op: _v42.default.literal("move"), path: _v42.default.string(), from: _v42.default.string() }), _v42.default.object({ op: _v42.default.literal("copy"), path: _v42.default.string(), from: _v42.default.string() }), _v42.default.object({ op: _v42.default.literal("test"), path: _v42.default.string(), value: _v42.default.unknown() }) ]); var PatchSchema = _v42.default.array(OperationSchema); var ConnectionSchema = _v42.default.object({ params: _v42.default.record(_v42.default.string(), _v42.default.any()).optional(), id: _v42.default.string(), stateEnabled: _v42.default.boolean().optional(), state: _v42.default.any().optional(), auth: _v42.default.record(_v42.default.string(), _v42.default.any()).optional() }); var RealtimeEventSchema = _v42.default.discriminatedUnion("type", [ _v42.default.object({ type: _v42.default.literal("action"), name: _v42.default.string(), args: _v42.default.array(_v42.default.any()), connId: _v42.default.string() }), _v42.default.object({ type: _v42.default.literal("broadcast"), eventName: _v42.default.string(), args: _v42.default.array(_v42.default.any()) }), _v42.default.object({ type: _v42.default.literal("subscribe"), eventName: _v42.default.string(), connId: _v42.default.string() }), _v42.default.object({ type: _v42.default.literal("unsubscribe"), eventName: _v42.default.string(), connId: _v42.default.string() }), _v42.default.object({ type: _v42.default.literal("event"), eventName: _v42.default.string(), args: _v42.default.array(_v42.default.any()), connId: _v42.default.string() }) ]); var RecordedRealtimeEventSchema = RealtimeEventSchema.and( _v42.default.object({ id: _v42.default.string(), timestamp: _v42.default.number() }) ); var DatabaseQuerySchema = _v42.default.object({ sql: _v42.default.string(), args: _v42.default.array(_v42.default.string().or(_v42.default.number())) }); var TableSchema = _v42.default.object({ schema: _v42.default.string(), name: _v42.default.string(), type: _v42.default.enum(["table", "view"]) }); var TablesSchema = _v42.default.array(TableSchema); var ColumnSchema = _v42.default.object({ cid: _v42.default.number(), name: _v42.default.string(), type: _v42.default.string().toLowerCase().transform((val) => { return _v42.default.enum(["integer", "text", "real", "blob", "numeric", "serial"]).parse(val); }), notnull: _v42.default.coerce.boolean(), dflt_value: _v42.default.string().nullable(), pk: _v42.default.coerce.boolean().nullable() }); var ColumnsSchema = _v42.default.array(ColumnSchema); var ForeignKeySchema = _v42.default.object({ id: _v42.default.number(), table: _v42.default.string(), from: _v42.default.string(), to: _v42.default.string() }); var ForeignKeysSchema = _v42.default.array(ForeignKeySchema); var BuildSchema = _v42.default.object({ name: _v42.default.string(), createdAt: _v42.default.string().optional(), tags: _v42.default.record(_v42.default.string(), _v42.default.string()).optional() }); var BuildsSchema = _v42.default.array(BuildSchema); var CreateActorSchema = _v42.default.object({ name: _v42.default.string(), // FIXME: Replace with ActorKeySchema when ready key: _v42.default.array(_v42.default.string().max(MAX_ACTOR_KEY_SIZE)), input: _v42.default.any() }); exports.ActorId = ActorId; exports.ActorFeature = ActorFeature; exports.ActorLogEntry = ActorLogEntry; exports.ActorSchema = ActorSchema; exports.OperationSchema = OperationSchema; exports.PatchSchema = PatchSchema; exports.ConnectionSchema = ConnectionSchema; exports.RealtimeEventSchema = RealtimeEventSchema; exports.RecordedRealtimeEventSchema = RecordedRealtimeEventSchema; exports.DatabaseQuerySchema = DatabaseQuerySchema; exports.TableSchema = TableSchema; exports.TablesSchema = TablesSchema; exports.ColumnSchema = ColumnSchema; exports.ColumnsSchema = ColumnsSchema; exports.ForeignKeySchema = ForeignKeySchema; exports.ForeignKeysSchema = ForeignKeysSchema; exports.BuildSchema = BuildSchema; exports.BuildsSchema = BuildsSchema; exports.CreateActorSchema = CreateActorSchema; //# sourceMappingURL=chunk-DVPXSB4B.cjs.map