@effect-agent/platform-cloudflare 0.1.0-beta.42 → 0.1.0-beta.44

package/src/scheduling.ts

@@ -119,6 +119,7 @@ const ScheduleOwnerRequest = Schema.Union([
   ScheduleListRequest,
   ScheduleControlRequest,
 ]);
+
 type ScheduleOwnerRequest = typeof ScheduleOwnerRequest.Type;
 
 const ScheduleOwnerFailure = Schema.Union([
@@ -131,6 +132,7 @@ const ScheduleOwnerFailure = Schema.Union([
   ScheduleFailpointError,
   ScheduleOwnerProtocolError,
 ]);
+
 type ScheduleOwnerFailure = typeof ScheduleOwnerFailure.Type;
 
 const ScheduleOwnerResponse = Schema.Union([
@@ -138,6 +140,7 @@ const ScheduleOwnerResponse = Schema.Union([
   Schema.TaggedStruct("Page", { value: ScheduleSnapshotPageSchema }),
   Schema.TaggedStruct("Failed", { failure: ScheduleOwnerFailure }),
 ]);
+
 type ScheduleOwnerResponse = typeof ScheduleOwnerResponse.Type;
 
 const decodeScheduleOwnerRequest = Schema.decodeUnknownEffect(ScheduleOwnerRequest);
@@ -181,6 +184,7 @@ export class CloudflareSchedulingClient {
             ScheduleStorageError.make({ operation: "Schedule Owner protocol", reason: "corrupt" }),
           ),
         );
+
         const raw = yield* Effect.tryPromise({
           try: () => namespace.get(namespace.idFromName(scheduleOwnerKey(owner))).schedule(encoded),
           catch: () =>
@@ -189,12 +193,15 @@ export class CloudflareSchedulingClient {
               reason: "unavailable",
             }),
         });
+
         const response = yield* decodeScheduleOwnerResponse(raw).pipe(
           Effect.mapError(() =>
             ScheduleStorageError.make({ operation: "Schedule Owner protocol", reason: "corrupt" }),
           ),
         );
+
         if (response._tag !== "Failed") return response;
+
         return yield* response.failure._tag === "ScheduleOwnerProtocolError"
           ? ScheduleStorageError.make({ operation: "Schedule Owner protocol", reason: "corrupt" })
           : response.failure;
@@ -213,6 +220,7 @@ export class CloudflareSchedulingClient {
             }),
           ),
         );
+
         return yield* Schema.decodeUnknownEffect(PersistedJson)(encoded).pipe(
           Effect.mapError(() =>
             ScheduleValidationError.make({
@@ -225,6 +233,7 @@ export class CloudflareSchedulingClient {
       const create: Scheduling["Service"]["create"] = (agent, input, options) =>
         Effect.gen(function* () {
           const payload = yield* encodeInput(agent, input);
+
           const response = yield* call(options.scope.owner, {
             _tag: "Create",
             schemaVersion: 1,
@@ -232,6 +241,7 @@ export class CloudflareSchedulingClient {
             input: payload,
             ...options,
           });
+
           return response._tag === "Snapshot"
             ? response.value
             : yield* ScheduleStorageError.make({
@@ -243,6 +253,7 @@ export class CloudflareSchedulingClient {
       const update: Scheduling["Service"]["update"] = (agent, input, options) =>
         Effect.gen(function* () {
           const payload = yield* encodeInput(agent, input);
+
           const response = yield* call(options.scope.owner, {
             _tag: "Update",
             schemaVersion: 1,
@@ -250,6 +261,7 @@ export class CloudflareSchedulingClient {
             input: payload,
             ...options,
           });
+
           return response._tag === "Snapshot"
             ? response.value
             : yield* ScheduleStorageError.make({
@@ -266,6 +278,7 @@ export class CloudflareSchedulingClient {
             scope,
             scheduleId,
           });
+
           return response._tag === "Snapshot"
             ? response.value
             : yield* ScheduleStorageError.make({
@@ -283,6 +296,7 @@ export class CloudflareSchedulingClient {
             ...(options.after === undefined ? {} : { after: options.after }),
             ...(options.limit === undefined ? {} : { limit: options.limit }),
           });
+
           return response._tag === "Page"
             ? response.value
             : yield* ScheduleStorageError.make({
@@ -306,6 +320,7 @@ export class CloudflareSchedulingClient {
             scheduleId,
             expectedRevision,
           });
+
           return response._tag === "Snapshot"
             ? response.value
             : yield* ScheduleStorageError.make({
@@ -340,6 +355,7 @@ const decodeOwnerName = Effect.fn("decodeScheduleOwnerName")(function* (
       message: "Schedule Owner objects require an idFromName identity",
     });
   }
+
   const tuple = yield* Schema.decodeUnknownEffect(
     Schema.fromJsonString(Schema.Tuple([Schema.String, Schema.String])),
   )(name).pipe(
@@ -347,6 +363,7 @@ const decodeOwnerName = Effect.fn("decodeScheduleOwnerName")(function* (
       ScheduleOwnerProtocolError.make({ message: "Schedule Owner object name is malformed" }),
     ),
   );
+
   return yield* Schema.decodeUnknownEffect(ScheduleOwner)({
     tenantId: tuple[0],
     ownerId: tuple[1],
@@ -371,10 +388,12 @@ const transactionLayer: Layer.Layer<
   DoScheduleTransaction,
   Effect.gen(function* () {
     const alarms = yield* DurableObjectAlarm.DurableObjectAlarm;
+
     return DoScheduleTransaction.of({
       run: (body) =>
         Effect.gen(function* () {
           const nowMillis = yield* Clock.currentTimeMillis;
+
           return yield* alarms
             .transaction((transaction) =>
               body((replacement) =>
@@ -434,6 +453,7 @@ const ensureOwner = (
   request: ScheduleOwnerRequest,
 ): Effect.Effect<void, ScheduleOwnerProtocolError> => {
   const observed = requestOwner(request);
+
   return observed.tenantId === expected.tenantId && observed.ownerId === expected.ownerId
     ? Effect.void
     : Effect.fail(
@@ -447,6 +467,7 @@ const handleScheduleRequest = Effect.fn("ScheduleOwner.handleRequest")(function*
   encoded: unknown,
 ): Effect.fn.Return<unknown, never, Scheduling | ScheduleOwnerIdentity> {
   const decoded = yield* decodeScheduleOwnerRequest(encoded).pipe(Effect.result);
+
   if (decoded._tag === "Failure") {
     return yield* encodeScheduleOwnerResponse(
       scheduleProtocolFailure("The Schedule request could not be decoded"),
@@ -455,6 +476,7 @@ const handleScheduleRequest = Effect.fn("ScheduleOwner.handleRequest")(function*
   const request = decoded.success;
   const { owner } = yield* ScheduleOwnerIdentity;
   const scheduling = yield* Scheduling;
+
   const response = yield* Effect.gen(function* () {
     yield* ensureOwner(owner, request);
     switch (request._tag) {
@@ -467,6 +489,7 @@ const handleScheduleRequest = Effect.fn("ScheduleOwner.handleRequest")(function*
           deliveryPrincipal: request.deliveryPrincipal,
           definitions: request.definitions,
         });
+
         return { _tag: "Snapshot" as const, value };
       }
       case "Update": {
@@ -479,6 +502,7 @@ const handleScheduleRequest = Effect.fn("ScheduleOwner.handleRequest")(function*
           definitions: request.definitions,
           expectedRevision: request.expectedRevision,
         });
+
         return { _tag: "Snapshot" as const, value };
       }
       case "Get":
@@ -509,6 +533,7 @@ const handleScheduleRequest = Effect.fn("ScheduleOwner.handleRequest")(function*
                   request.scheduleId,
                   request.expectedRevision,
                 );
+
         return { _tag: "Snapshot" as const, value };
       }
     }
@@ -522,6 +547,7 @@ const handleScheduleRequest = Effect.fn("ScheduleOwner.handleRequest")(function*
           ),
     ),
   );
+
   return yield* encodeScheduleOwnerResponse(response).pipe(Effect.orDie);
 });
 
@@ -545,8 +571,10 @@ const scheduleAlarmHandler = (limits: SchedulingLimits) =>
         const alarmControl = yield* DoScheduleAlarmControl;
         const { owner } = yield* ScheduleOwnerIdentity;
         const nowMillis = yield* Clock.currentTimeMillis;
+
         yield* alarmControl.prearm(nowMillis + limits.recoveryPollMillis);
         const pass = yield* scheduling.runDue(owner);
+
         if (pass.failed > 0) {
           yield* alarmControl.prearm((yield* Clock.currentTimeMillis) + limits.recoveryPollMillis);
         } else {
@@ -585,14 +613,17 @@ export const makeScheduleOwnerObjectClass = <E>(
     ScheduleOwnerIdentity,
     Effect.gen(function* () {
       const state = yield* EffectCfDurableObjectState.DurableObjectState;
+
       return ScheduleOwnerIdentity.of({ owner: yield* decodeOwnerName(state.raw.id.name) });
     }),
   );
+
   const sqlLayer = Layer.unwrap(
     Effect.map(EffectCfDurableObjectState.DurableObjectState, (state) =>
       SqliteClient.layer({ storage: state.raw.storage }),
     ),
   );
+
   const application = Layer.merge(Scheduling.layer(limits), ScheduleDriver.layer(limits)).pipe(
     Layer.provideMerge(
       scheduleStoreLayer.pipe(Layer.provide(transactionLayer), Layer.provide(sqlLayer)),
@@ -609,6 +640,7 @@ export const makeScheduleOwnerObjectClass = <E>(
     Layer.provide(host),
     Layer.provideMerge(ownerLayer),
   );
+
   const runtime: Layer.Layer<
     ScheduleRuntimeServices,
     E | ScheduleStorageError | ScheduleOwnerProtocolError | ScheduleValidationError,
@@ -617,6 +649,7 @@ export const makeScheduleOwnerObjectClass = <E>(
     Effect.gen(function* () {
       const state = yield* EffectCfDurableObjectState.DurableObjectState;
       const scope = yield* Effect.scope;
+
       return yield* state.blockConcurrencyWhile(Layer.buildWithScope(application, scope));
     }),
   );
@@ -636,5 +669,6 @@ export const makeScheduleOwnerObjectClass = <E>(
       return super.alarm?.(alarmInfo);
     }
   }
+
   return ScheduleOwnerObject;
 };

package/src/subscriptions.ts

@@ -70,6 +70,7 @@ export const validateCloudflareSubscriptionLimits = (
 ): Effect.Effect<void, CloudflareSubscriptionConfigError> => {
   const worstCaseMillis =
     4 * Math.ceil(limits.batchSize / limits.concurrency) * limits.operationTimeoutMillis;
+
   return worstCaseMillis <= MAX_ALARM_WALL_MILLIS
     ? Effect.void
     : Effect.fail(
@@ -146,16 +147,19 @@ const SubscriptionPartitionRequest = Schema.Union([
   AcceptRequest,
   StatusRequest,
 ]);
+
 type SubscriptionPartitionRequest = typeof SubscriptionPartitionRequest.Type;
 
 const SubscriptionPage = Schema.Struct({
   items: Schema.Array(SubscriptionSnapshot),
   next: Schema.NullOr(Schema.Natural),
 });
+
 const DeliveryPage = Schema.Struct({
   items: Schema.Array(SubscriptionDeliverySnapshot),
   next: Schema.NullOr(Schema.String),
 });
+
 const IntakeStatus = Schema.Struct({
   ...EventAcknowledgement.fields,
   routingComplete: Schema.Boolean,
@@ -169,6 +173,7 @@ const SubscriptionPartitionFailure = Schema.Union([
   SubscriptionFailpointError,
   SubscriptionPartitionProtocolError,
 ]);
+
 type SubscriptionPartitionFailure = typeof SubscriptionPartitionFailure.Type;
 
 const SubscriptionPartitionResponse = Schema.Union([
@@ -179,6 +184,7 @@ const SubscriptionPartitionResponse = Schema.Union([
   Schema.TaggedStruct("Status", { value: IntakeStatus }),
   Schema.TaggedStruct("Failed", { failure: SubscriptionPartitionFailure }),
 ]);
+
 type SubscriptionPartitionResponse = typeof SubscriptionPartitionResponse.Type;
 
 const decodeRequest = Schema.decodeUnknownEffect(SubscriptionPartitionRequest);
@@ -205,6 +211,7 @@ export class SubscriptionPartitionNamespace extends Context.Service<
 
 const storageFailure = (code: string): SubscriptionError =>
   SubscriptionError.make({ reason: "storage", code });
+
 const corruptFailure = (code: string): SubscriptionError =>
   SubscriptionError.make({ reason: "corrupt", code });
 
@@ -224,9 +231,11 @@ export class CloudflareSubscriptionsClient {
           if (!samePartition(partition, addressedPartition)) {
             return yield* SubscriptionError.make({ reason: "unauthorized", code: "partition" });
           }
+
           const encoded = yield* encodeRequest(request).pipe(
             Effect.mapError(() => corruptFailure("subscription-partition-protocol")),
           );
+
           const raw = yield* Effect.tryPromise({
             try: () =>
               namespace
@@ -234,6 +243,7 @@ export class CloudflareSubscriptionsClient {
                 .subscription(encoded),
             catch: () => storageFailure("call-subscription-partition"),
           });
+
           return yield* decodeResponse(raw).pipe(
             Effect.mapError(() => corruptFailure("subscription-partition-protocol")),
           );
@@ -245,6 +255,7 @@ export class CloudflareSubscriptionsClient {
             : SubscriptionPartitionProtocolError.make({
                 message: "Unexpected subscription response",
               });
+
         const asProtocolError = (): SubscriptionError =>
           corruptFailure("subscription-partition-protocol");
 
@@ -257,14 +268,17 @@ export class CloudflareSubscriptionsClient {
                 scope,
                 options,
               });
+
               if (response._tag === "Snapshot") return response.value;
               const failure = failed(response);
+
               if (
                 failure._tag === "SubscriptionError" ||
                 failure._tag === "SubscriptionSourceError" ||
                 failure._tag === "SubscriptionFailpointError"
               )
                 return yield* failure;
+
               return yield* asProtocolError();
             }),
           listSubscriptions: (scope, after, limit) =>
@@ -276,8 +290,10 @@ export class CloudflareSubscriptionsClient {
                 ...(after === undefined ? {} : { after }),
                 ...(limit === undefined ? {} : { limit }),
               });
+
               if (response._tag === "SubscriptionPage") return response.value;
               const failure = failed(response);
+
               return yield* failure._tag === "SubscriptionError" ? failure : asProtocolError();
             }),
           cancelSubscription: (scope, key) =>
@@ -288,13 +304,16 @@ export class CloudflareSubscriptionsClient {
                 scope,
                 key,
               });
+
               if (response._tag === "Snapshot") return response.value;
               const failure = failed(response);
+
               if (
                 failure._tag === "SubscriptionError" ||
                 failure._tag === "SubscriptionFailpointError"
               )
                 return yield* failure;
+
               return yield* asProtocolError();
             }),
           listDeliveries: (scope, key, after, limit) =>
@@ -307,8 +326,10 @@ export class CloudflareSubscriptionsClient {
                 ...(after === undefined ? {} : { after }),
                 ...(limit === undefined ? {} : { limit }),
               });
+
               if (response._tag === "DeliveryPage") return response.value;
               const failure = failed(response);
+
               return yield* failure._tag === "SubscriptionError" ? failure : asProtocolError();
             }),
         });
@@ -328,15 +349,19 @@ export class CloudflareSubscriptionsClient {
                   SubscriptionError.make({ reason: "validation", code: "event-payload" }),
                 ),
               );
+
               const response = yield* call(request.partition, request);
+
               if (response._tag === "Acknowledgement") return response.value;
               const failure = failed(response);
+
               if (
                 failure._tag === "SubscriptionError" ||
                 failure._tag === "SubscriptionSourceError" ||
                 failure._tag === "SubscriptionFailpointError"
               )
                 return yield* failure;
+
               return yield* asProtocolError();
             }),
           status: (principal, source, eventId) =>
@@ -349,8 +374,10 @@ export class CloudflareSubscriptionsClient {
                 source,
                 eventId,
               });
+
               if (response._tag === "Status") return response.value;
               const failure = failed(response);
+
               return yield* failure._tag === "SubscriptionError" ? failure : asProtocolError();
             }),
         });
@@ -376,6 +403,7 @@ const decodePartitionName = Effect.fn("decodeSubscriptionPartitionName")(functio
       message: "Subscription Partition objects require an idFromName identity",
     });
   }
+
   const tuple = yield* Schema.decodeUnknownEffect(
     Schema.fromJsonString(Schema.Tuple([Schema.String, Schema.String])),
   )(name).pipe(
@@ -385,6 +413,7 @@ const decodePartitionName = Effect.fn("decodeSubscriptionPartitionName")(functio
       }),
     ),
   );
+
   return yield* Schema.decodeUnknownEffect(SourcePartition)({
     tenantId: tuple[0],
     address: tuple[1],
@@ -398,6 +427,7 @@ const decodePartitionName = Effect.fn("decodeSubscriptionPartitionName")(functio
 });
 
 const samePartition = Schema.toEquivalence(SourcePartition);
+
 const requestPartition = (request: SubscriptionPartitionRequest): SourcePartition =>
   request._tag === "Accept" || request._tag === "Status"
     ? request.partition
@@ -407,6 +437,7 @@ const handleRequest = Effect.fn("SubscriptionPartition.handleRequest")(function*
   encoded: unknown,
 ) {
   const decoded = yield* decodeRequest(encoded).pipe(Effect.result);
+
   if (decoded._tag === "Failure") {
     return yield* encodeResponse(
       protocolFailure("The subscription request could not be decoded"),
@@ -414,6 +445,7 @@ const handleRequest = Effect.fn("SubscriptionPartition.handleRequest")(function*
   }
   const request = decoded.success;
   const { partition } = yield* SubscriptionPartitionIdentity;
+
   if (!samePartition(partition, requestPartition(request))) {
     return yield* encodeResponse(
       protocolFailure("The request partition does not match the addressed object"),
@@ -421,6 +453,7 @@ const handleRequest = Effect.fn("SubscriptionPartition.handleRequest")(function*
   }
   const subscriptions = yield* Subscriptions;
   const intake = yield* SubscriptionIntake;
+
   const response = yield* Effect.gen(function* (): Effect.fn.Return<
     SubscriptionPartitionResponse,
     SubscriptionPartitionFailure
@@ -473,6 +506,7 @@ const handleRequest = Effect.fn("SubscriptionPartition.handleRequest")(function*
         : Effect.succeed(protocolFailure("The subscription operation failed outside its contract")),
     ),
   );
+
   return yield* encodeResponse(response).pipe(Effect.orDie);
 });
 
@@ -486,10 +520,12 @@ const transactionLayer: Layer.Layer<
   DoSubscriptionTransaction,
   Effect.gen(function* () {
     const alarms = yield* DurableObjectAlarm.DurableObjectAlarm;
+
     return DoSubscriptionTransaction.of({
       run: (body) =>
         Effect.gen(function* () {
           const nowMillis = yield* Clock.currentTimeMillis;
+
           return yield* alarms
             .transaction((transaction) =>
               body((replacement) =>
@@ -542,7 +578,9 @@ const alarmHandler = (limits: SubscriptionLimits) =>
         );
         const driver = yield* SubscriptionDriver;
         const alarmControl = yield* DoSubscriptionAlarmControl;
+
         yield* alarmControl.prearm((yield* Clock.currentTimeMillis) + limits.retryMillis);
+
         const pass = yield* driver.runDue.pipe(
           Effect.catchCause((cause) =>
             Cause.hasInterrupts(cause)
@@ -553,6 +591,7 @@ const alarmHandler = (limits: SubscriptionLimits) =>
                 ),
           ),
         );
+
         if (pass.failed > 0) {
           yield* alarmControl.prearm((yield* Clock.currentTimeMillis) + limits.retryMillis);
         } else {
@@ -596,20 +635,24 @@ export const makeSubscriptionPartitionObjectClass = <E>(
   limits: SubscriptionLimits = defaultSubscriptionLimits,
 ): SubscriptionPartitionObjectClass => {
   const cloudflareLimitsLayer = Layer.effectDiscard(validateCloudflareSubscriptionLimits(limits));
+
   const identityLayer = Layer.effect(
     SubscriptionPartitionIdentity,
     Effect.gen(function* () {
       const state = yield* EffectCfDurableObjectState.DurableObjectState;
+
       return SubscriptionPartitionIdentity.of({
         partition: yield* decodePartitionName(state.raw.id.name),
       });
     }),
   );
+
   const sqlLayer = Layer.unwrap(
     Effect.map(EffectCfDurableObjectState.DurableObjectState, (state) =>
       SqliteClient.layer({ storage: state.raw.storage }),
     ),
   );
+
   const partitionStore = Layer.unwrap(
     Effect.map(SubscriptionPartitionIdentity, ({ partition }) =>
       doSubscriptionStoreLayer(partition).pipe(
@@ -618,6 +661,7 @@ export const makeSubscriptionPartitionObjectClass = <E>(
       ),
     ),
   );
+
   const application = Layer.mergeAll(
     Subscriptions.layer(limits),
     SubscriptionIntake.layer(limits),
@@ -633,6 +677,7 @@ export const makeSubscriptionPartitionObjectClass = <E>(
     Layer.provide(host),
     Layer.provideMerge(identityLayer),
   );
+
   const runtime: Layer.Layer<
     SubscriptionRuntimeServices,
     E | SubscriptionError | SubscriptionPartitionProtocolError | CloudflareSubscriptionConfigError,
@@ -641,21 +686,26 @@ export const makeSubscriptionPartitionObjectClass = <E>(
     Effect.gen(function* () {
       const state = yield* EffectCfDurableObjectState.DurableObjectState;
       const scope = yield* Effect.scope;
+
       return yield* state.blockConcurrencyWhile(Layer.buildWithScope(application, scope));
     }),
   );
+
   const rpc = {
     subscription: (encoded: unknown) => handleRequest(encoded),
   } satisfies EffectCfDurableObject.DurableObjectRpc<SubscriptionRuntimeServices>;
+
   const Base = EffectCfDurableObject.make(runtime, {
     initialize: Effect.void,
     rpc,
     alarms: alarmHandler(limits),
   });
+
   class SubscriptionPartitionObject extends Base {
     override alarm(alarmInfo?: AlarmInvocationInfo): Promise<void> | void {
       return super.alarm?.(alarmInfo);
     }
   }
+
   return SubscriptionPartitionObject;
 };