import { makeAggregateState } from "./agentActivityAggregate.ts";export { makeAggregateState, TERMINAL_AGENT_ACTIVITY_DISPLAY_TTL_MS,} from "./agentActivityAggregate.ts";import type { RelayAgentActivityState, RelayDeliveryResult, RelayPublishResponse,} from "@t3tools/contracts/relay";import * as Context from "effect/Context";import * as DateTime from "effect/DateTime";import * as Effect from "effect/Effect";import * as Layer from "effect/Layer"; import { isTerminalPhase } from "./agentActivityPayloads.ts"; export { isExpiredAgentActivityState } from "./agentActivityPayloads.ts";import * as AgentActivityRows from "./AgentActivityRows.ts";import * as EnvironmentLinks from "../environments/EnvironmentLinks.ts";import * as LiveActivities from "./LiveActivities.ts";import * as ApnsDeliveries from "./ApnsDeliveries.ts";import * as FcmDeliveries from "./FcmDeliveries.ts"; export type AgentActivityPublishError = | FcmDeliveries.FcmDeliveryError | AgentActivityRows.AgentActivityRowUpsertPersistenceError | AgentActivityRows.AgentActivityRowDeletePersistenceError | AgentActivityRows.AgentActivityRowListPersistenceError | EnvironmentLinks.EnvironmentLinkUserListPersistenceError | LiveActivities.LiveActivityTargetListPersistenceError | ApnsDeliveries.ApnsDeliveryError; export class AgentActivityPublisher extends Context.Service< AgentActivityPublisher, { readonly publish: (input: { readonly environmentId: string; readonly environmentPublicKey: string; readonly threadId: string; readonly state: RelayAgentActivityState | null; }) => Effect.Effect<RelayPublishResponse, AgentActivityPublishError>; readonly replayForLiveActivityRegistration: (input: { readonly userId: string; readonly deviceId: string; }) => Effect.Effect<RelayDeliveryResult | null, AgentActivityPublishError>; }>()("t3code-relay/agentActivity/AgentActivityPublisher") {} export const make = Effect.gen(function* () { const rows = yield* AgentActivityRows.AgentActivityRows; const links = yield* EnvironmentLinks.EnvironmentLinks; const liveActivities = yield* LiveActivities.LiveActivities; const apnsDeliveries = yield* ApnsDeliveries.ApnsDeliveries; const fcmDeliveries = yield* FcmDeliveries.FcmDeliveries; const publishForDeliveryUser = Effect.fnUntraced(function* (input: { readonly deliveryUser: EnvironmentLinks.AgentAwarenessDeliveryUserRecord; readonly state: RelayAgentActivityState | null; readonly nowMs: number; }) { const activeStates = input.deliveryUser.liveActivitiesEnabled ? yield* rows.listForUser({ userId: input.deliveryUser.userId }) : []; const liveActivityAggregate = input.deliveryUser.liveActivitiesEnabled ? makeAggregateState({ activeStates, terminalState: input.state && isTerminalPhase(input.state) ? input.state : null, nowMs: input.nowMs, }) : null; const notificationOnlyAggregate = input.deliveryUser.notificationsEnabled && !input.deliveryUser.liveActivitiesEnabled && input.state !== null ? makeAggregateState({ activeStates: isTerminalPhase(input.state) ? [] : [input.state], terminalState: isTerminalPhase(input.state) ? input.state : null, nowMs: input.nowMs, }) : null; const targets = yield* liveActivities.listTargets({ userId: input.deliveryUser.userId }); const deliveriesByTarget = yield* Effect.forEach( targets, Effect.fnUntraced(function* (target) { if (target.platform === "android") { return [yield* fcmDeliveries.enqueue({ target, state: input.state })]; } return yield* Effect.all( [ apnsDeliveries.sendForTarget({ target, aggregate: liveActivityAggregate, nowMs: input.nowMs, }), notificationOnlyAggregate === null ? Effect.succeed(null) : apnsDeliveries.sendPushNotificationForTarget({ target, aggregate: notificationOnlyAggregate, }), ], { concurrency: 2 }, ); }), { concurrency: 4 }, ); return deliveriesByTarget.flat(); }); return AgentActivityPublisher.of({ replayForLiveActivityRegistration: Effect.fn( "relay.agent_activity_publisher.replay_for_live_activity_registration", )(function* (input) { yield* Effect.annotateCurrentSpan({ "relay.mobile.device_id": input.deviceId, "relay.operation": "replayForLiveActivityRegistration", }); const { activeStates, targets } = yield* Effect.all( { activeStates: rows.listForUser({ userId: input.userId }), targets: liveActivities.listTargets({ userId: input.userId }), }, { concurrency: 2 }, ); const target = targets.find((row) => row.device_id === input.deviceId) ?? null; if (target === null) { return null; } if (target.platform === "android") { return yield* fcmDeliveries.enqueue({ target, state: null, replay: true }); } const now = yield* DateTime.now; const aggregate = makeAggregateState({ activeStates, terminalState: null, nowMs: now.epochMilliseconds, }); return yield* apnsDeliveries.sendForTarget({ target, aggregate, nowMs: now.epochMilliseconds, replay: true, }); }), publish: Effect.fn("relay.agent_activity_publisher.publish")(function* (input) { yield* Effect.annotateCurrentSpan({ "relay.environment_id": input.environmentId, "relay.thread_id": input.threadId, "relay.agent_activity.phase": input.state?.phase ?? "deleted", }); if (input.state) { // Terminal states are persisted too (pruned by the cron after they // age out) so a thread that finishes while other agents are active // stays visible as Done/Failed in subsequent aggregates instead of // silently vanishing from the Live Activity. yield* rows.upsert({ environmentPublicKey: input.environmentPublicKey, state: input.state, }); } else { yield* rows.remove({ environmentId: input.environmentId, environmentPublicKey: input.environmentPublicKey, threadId: input.threadId, }); } const deliveryUsers = yield* links.listDeliveryUsersForEnvironment({ environmentId: input.environmentId, environmentPublicKey: input.environmentPublicKey, }); const now = yield* DateTime.now; const deliveriesByUser = yield* Effect.forEach( deliveryUsers, (deliveryUser) => publishForDeliveryUser({ deliveryUser, state: input.state, nowMs: now.epochMilliseconds, }), { concurrency: 4 }, ); const deliveries = deliveriesByUser.flat(); return { ok: true, deliveries: deliveries.filter( (delivery): delivery is RelayDeliveryResult => delivery !== null, ), }; }), });}); export const layer = Layer.effect(AgentActivityPublisher, make);
import { makeAggregateState } from "./agentActivityAggregate.ts";
export {
makeAggregateState,
TERMINAL_AGENT_ACTIVITY_DISPLAY_TTL_MS,
} from "./agentActivityAggregate.ts";
import type {
RelayAgentActivityState,
RelayDeliveryResult,
RelayPublishResponse,
} from "@t3tools/contracts/relay";
import * as Context from "effect/Context";
import * as DateTime from "effect/DateTime";
import * as Effect from "effect/Effect";
import * as Layer from "effect/Layer";
import { isTerminalPhase } from "./agentActivityPayloads.ts";
export { isExpiredAgentActivityState } from "./agentActivityPayloads.ts";
import * as AgentActivityRows from "./AgentActivityRows.ts";
import * as EnvironmentLinks from "../environments/EnvironmentLinks.ts";
import * as LiveActivities from "./LiveActivities.ts";
import * as ApnsDeliveries from "./ApnsDeliveries.ts";
import * as FcmDeliveries from "./FcmDeliveries.ts";
export type AgentActivityPublishError =
| FcmDeliveries.FcmDeliveryError
| AgentActivityRows.AgentActivityRowUpsertPersistenceError
| AgentActivityRows.AgentActivityRowDeletePersistenceError
| AgentActivityRows.AgentActivityRowListPersistenceError
| EnvironmentLinks.EnvironmentLinkUserListPersistenceError
| LiveActivities.LiveActivityTargetListPersistenceError
| ApnsDeliveries.ApnsDeliveryError;
export class AgentActivityPublisher extends Context.Service<
AgentActivityPublisher,
{
readonly publish: (input: {
readonly environmentId: string;
readonly environmentPublicKey: string;
readonly threadId: string;
readonly state: RelayAgentActivityState | null;
}) => Effect.Effect<RelayPublishResponse, AgentActivityPublishError>;
readonly replayForLiveActivityRegistration: (input: {
readonly userId: string;
readonly deviceId: string;
}) => Effect.Effect<RelayDeliveryResult | null, AgentActivityPublishError>;
}
>()("t3code-relay/agentActivity/AgentActivityPublisher") {}
export const make = Effect.gen(function* () {
const rows = yield* AgentActivityRows.AgentActivityRows;
const links = yield* EnvironmentLinks.EnvironmentLinks;
const liveActivities = yield* LiveActivities.LiveActivities;
const apnsDeliveries = yield* ApnsDeliveries.ApnsDeliveries;
const fcmDeliveries = yield* FcmDeliveries.FcmDeliveries;
const publishForDeliveryUser = Effect.fnUntraced(function* (input: {
readonly deliveryUser: EnvironmentLinks.AgentAwarenessDeliveryUserRecord;
readonly state: RelayAgentActivityState | null;
readonly nowMs: number;
}) {
const activeStates = input.deliveryUser.liveActivitiesEnabled
? yield* rows.listForUser({ userId: input.deliveryUser.userId })
: [];
const liveActivityAggregate = input.deliveryUser.liveActivitiesEnabled
? makeAggregateState({
activeStates,
terminalState: input.state && isTerminalPhase(input.state) ? input.state : null,
nowMs: input.nowMs,
})
: null;
const notificationOnlyAggregate =
input.deliveryUser.notificationsEnabled &&
!input.deliveryUser.liveActivitiesEnabled &&
input.state !== null
? makeAggregateState({
activeStates: isTerminalPhase(input.state) ? [] : [input.state],
terminalState: isTerminalPhase(input.state) ? input.state : null,
nowMs: input.nowMs,
})
: null;
const targets = yield* liveActivities.listTargets({ userId: input.deliveryUser.userId });
const deliveriesByTarget = yield* Effect.forEach(
targets,
Effect.fnUntraced(function* (target) {
if (target.platform === "android") {
return [yield* fcmDeliveries.enqueue({ target, state: input.state })];
}
return yield* Effect.all(
[
apnsDeliveries.sendForTarget({
target,
aggregate: liveActivityAggregate,
nowMs: input.nowMs,
}),
notificationOnlyAggregate === null
? Effect.succeed(null)
: apnsDeliveries.sendPushNotificationForTarget({
target,
aggregate: notificationOnlyAggregate,
}),
],
{ concurrency: 2 },
);
}),
{ concurrency: 4 },
);
return deliveriesByTarget.flat();
});
return AgentActivityPublisher.of({
replayForLiveActivityRegistration: Effect.fn(
"relay.agent_activity_publisher.replay_for_live_activity_registration",
)(function* (input) {
yield* Effect.annotateCurrentSpan({
"relay.mobile.device_id": input.deviceId,
"relay.operation": "replayForLiveActivityRegistration",
});
const { activeStates, targets } = yield* Effect.all(
{
activeStates: rows.listForUser({ userId: input.userId }),
targets: liveActivities.listTargets({ userId: input.userId }),
},
{ concurrency: 2 },
);
const target = targets.find((row) => row.device_id === input.deviceId) ?? null;
if (target === null) {
return null;
}
if (target.platform === "android") {
return yield* fcmDeliveries.enqueue({ target, state: null, replay: true });
}
const now = yield* DateTime.now;
const aggregate = makeAggregateState({
activeStates,
terminalState: null,
nowMs: now.epochMilliseconds,
});
return yield* apnsDeliveries.sendForTarget({
target,
aggregate,
nowMs: now.epochMilliseconds,
replay: true,
});
}),
publish: Effect.fn("relay.agent_activity_publisher.publish")(function* (input) {
yield* Effect.annotateCurrentSpan({
"relay.environment_id": input.environmentId,
"relay.thread_id": input.threadId,
"relay.agent_activity.phase": input.state?.phase ?? "deleted",
});
if (input.state) {
// Terminal states are persisted too (pruned by the cron after they
// age out) so a thread that finishes while other agents are active
// stays visible as Done/Failed in subsequent aggregates instead of
// silently vanishing from the Live Activity.
yield* rows.upsert({
environmentPublicKey: input.environmentPublicKey,
state: input.state,
});
} else {
yield* rows.remove({
environmentId: input.environmentId,
environmentPublicKey: input.environmentPublicKey,
threadId: input.threadId,
});
}
const deliveryUsers = yield* links.listDeliveryUsersForEnvironment({
environmentId: input.environmentId,
environmentPublicKey: input.environmentPublicKey,
});
const now = yield* DateTime.now;
const deliveriesByUser = yield* Effect.forEach(
deliveryUsers,
(deliveryUser) =>
publishForDeliveryUser({
deliveryUser,
state: input.state,
nowMs: now.epochMilliseconds,
}),
{ concurrency: 4 },
);
const deliveries = deliveriesByUser.flat();
return {
ok: true,
deliveries: deliveries.filter(
(delivery): delivery is RelayDeliveryResult => delivery !== null,
),
};
}),
});
});
export const layer = Layer.effect(AgentActivityPublisher, make);