trustgraph/ts/packages/base/src/processor/program.ts

/**
 * Scoped Effect runtime helpers for legacy processor classes.
 *
 * These helpers make `Context.Service`/Layer composition the canonical
 * executable path while the processor internals remain Promise-based.
 */

import { Effect, Scope } from "effect";
import { processorLifecycleError, type ProcessorLifecycleError } from "../errors.js";
import { NatsBackend } from "../backend/nats.js";
import { makePubSubService, PubSub } from "../backend/pubsub.js";
import {
  ConsumerFactory,
  FlowRuntime,
  ProducerFactory,
  RequestResponseFactory,
  makeConsumerFactoryService,
  makeProducerFactoryService,
  makeRequestResponseFactoryService,
  runFlowRuntimeScoped,
} from "../messaging/runtime.js";
import {
  loadProcessorRuntimeConfig,
  type ProcessorRuntimeConfigOptions,
} from "../runtime/config.js";
import { loadMessagingRuntimeConfig } from "../runtime/messaging-config.js";
import type { AsyncProcessor, ProcessorConfig } from "./async-processor.js";

type ProcessorRunError<Processor> = Processor extends AsyncProcessor<infer Error, unknown> ? Error : never;
type ProcessorRunRequirements<Processor> = Processor extends AsyncProcessor<unknown, infer Requirements> ? Requirements : never;

export interface ProcessorProgramOptions<
  Config extends ProcessorConfig,
  Error,
  Requirements,
  Processor extends AsyncProcessor<unknown, unknown>,
> {
  readonly id: string;
  readonly make: (config: Config) => Processor;
  readonly loadConfig?: Effect.Effect<Config, Error, Requirements>;
}

export function runProcessorScoped<
  Config extends ProcessorConfig,
  Processor extends AsyncProcessor<unknown, unknown>,
>(
  config: Config,
  make: (config: Config) => Processor,
): Effect.Effect<
  void,
  ProcessorRunError<Processor> | ProcessorLifecycleError,
  PubSub | Scope.Scope | ProcessorRunRequirements<Processor>
> {
  return Effect.gen(function* () {
    const pubsub = yield* PubSub;
    const runtimeConfig = {
      ...config,
      manageProcessSignals: false,
      pubsub: pubsub.backend,
    } as Config;
    const processor = make(runtimeConfig);

    yield* Effect.addFinalizer(() =>
      Effect.tryPromise({
        try: () => processor.stop(),
        catch: (error) => processorLifecycleError(config.id, "stop", error),
      }).pipe(
        Effect.catch((error) =>
          Effect.logError("[Processor] Failed to stop processor", {
            error: error.message,
            operation: error.operation,
            processorId: error.processorId,
          }),
        ),
      ),
    );

    const typedProcessor = processor as unknown as AsyncProcessor<
      ProcessorRunError<Processor>,
      ProcessorRunRequirements<Processor>
    >;
    yield* typedProcessor.startEffect();
  });
}

export function makeProcessorProgram<
  Config extends ProcessorConfig,
  Error = never,
  Requirements = never,
  Processor extends AsyncProcessor<unknown, unknown> = AsyncProcessor,
>(
  options: ProcessorProgramOptions<Config, Error, Requirements, Processor>,
) {
  return Effect.scoped(
    Effect.gen(function* () {
      const config = yield* (
        options.loadConfig ??
        loadProcessorRuntimeConfig(options.id, {
          manageProcessSignals: false,
        } satisfies ProcessorRuntimeConfigOptions)
      );

      const runtimeConfig = {
        ...config,
        manageProcessSignals: false,
      } as Config;

      const pubsub = makePubSubService(new NatsBackend(runtimeConfig.pubsubUrl ?? "nats://localhost:4222"));
      const messagingConfig = yield* loadMessagingRuntimeConfig();
      yield* Effect.addFinalizer(() =>
        pubsub.close.pipe(
          Effect.catch((error) =>
            Effect.logError("[PubSub] Failed to close processor backend", {
              error: error.message,
              operation: error.operation,
            }),
          ),
        ),
      );
      const processorEffect = runProcessorScoped<Config, Processor>(
        runtimeConfig,
        options.make,
      );
      yield* processorEffect.pipe(
        Effect.provideService(PubSub, pubsub),
        Effect.provideService(ProducerFactory, ProducerFactory.of(makeProducerFactoryService(pubsub))),
        Effect.provideService(ConsumerFactory, ConsumerFactory.of(makeConsumerFactoryService(pubsub, messagingConfig))),
        Effect.provideService(
          RequestResponseFactory,
          RequestResponseFactory.of(makeRequestResponseFactoryService(pubsub, messagingConfig)),
        ),
        Effect.provideService(FlowRuntime, FlowRuntime.of({ run: runFlowRuntimeScoped })),
      );
    }),
  );
}

export const makeAsyncProcessorProgram = makeProcessorProgram;
export const makeFlowProcessorProgram = makeProcessorProgram;