rivetkit
Version:
Lightweight libraries for building stateful actors on edge platforms
239 lines (208 loc) • 8.4 kB
JavaScript
"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