import type {  RelayClientEnvironmentRecord,  RelayEnvironmentStatusResponse,} from "@t3tools/contracts/relay";import { decodeRelayJwt } from "@t3tools/shared/relayJwt";import {  RelayEnvironmentConnectScope,  RelayEnvironmentStatusScope,} from "@t3tools/contracts/relay";import * as Context from "effect/Context";import * as Effect from "effect/Effect";import * as Layer from "effect/Layer";import * as Option from "effect/Option";import * as Ref from "effect/Ref";import * as Semaphore from "effect/Semaphore";import * as Stream from "effect/Stream";import * as SubscriptionRef from "effect/SubscriptionRef"; import * as ManagedRelay from "./managedRelay.ts";import * as ClientCapabilities from "../platform/capabilities.ts";import * as Connectivity from "../connection/connectivity.ts";import { mapManagedRelayError } from "../connection/errors.ts";import { ConnectionBlockedError, type ConnectionAttemptError } from "../connection/model.ts";import * as ConnectionWakeups from "../connection/wakeups.ts"; export type RelayEnvironmentAvailability = "checking" | "online" | "offline" | "error"; export interface RelayDiscoveredEnvironment {  readonly environment: RelayClientEnvironmentRecord;  readonly availability: RelayEnvironmentAvailability;  readonly status: Option.Option<RelayEnvironmentStatusResponse>;  readonly error: Option.Option<ConnectionAttemptError>;} export interface RelayEnvironmentDiscoveryState {  readonly environments: ReadonlyMap<string, RelayDiscoveredEnvironment>;  readonly refreshing: boolean;  readonly offline: boolean;  readonly error: Option.Option<ConnectionAttemptError>;} export class RelayEnvironmentDiscovery extends Context.Service<  RelayEnvironmentDiscovery,  {    readonly state: SubscriptionRef.SubscriptionRef<RelayEnvironmentDiscoveryState>;    readonly refresh: Effect.Effect<void>;  }>()("@t3tools/client-runtime/relay/discovery/RelayEnvironmentDiscovery") {} export const EMPTY_RELAY_ENVIRONMENT_DISCOVERY_STATE: RelayEnvironmentDiscoveryState = {  environments: new Map(),  refreshing: false,  offline: false,  error: Option.none(),}; function validateStatus(  environment: RelayClientEnvironmentRecord,  status: RelayEnvironmentStatusResponse,): Effect.Effect<RelayEnvironmentStatusResponse, ConnectionAttemptError> {  if (status.environmentId !== environment.environmentId) {    return Effect.fail(      new ConnectionBlockedError({        reason: "configuration",        detail: "Relay returned status for a different environment.",      }),    );  }  if (    status.endpoint.httpBaseUrl !== environment.endpoint.httpBaseUrl ||    status.endpoint.wsBaseUrl !== environment.endpoint.wsBaseUrl ||    status.endpoint.providerKind !== environment.endpoint.providerKind  ) {    return Effect.fail(      new ConnectionBlockedError({        reason: "configuration",        detail: "Relay returned status for a different environment endpoint.",      }),    );  }  if (    status.descriptor !== undefined &&    status.descriptor.environmentId !== environment.environmentId  ) {    return Effect.fail(      new ConnectionBlockedError({        reason: "configuration",        detail: "Relay returned a descriptor for a different environment.",      }),    );  }  return Effect.succeed(status);} function relayAccountId(clerkToken: string): Option.Option<string> {  try {    return Option.fromNullishOr(decodeRelayJwt(clerkToken).sub).pipe(      Option.filter((subject) => subject.length > 0),    );  } catch {    return Option.none();  }} export const make = Effect.fn("RelayEnvironmentDiscovery.make")(function* () {  const relay = yield* ManagedRelay.ManagedRelayClient;  const session = yield* ClientCapabilities.CloudSession;  const connectivity = yield* Connectivity.Connectivity;  const wakeups = yield* ConnectionWakeups.ConnectionWakeups;  const state = yield* SubscriptionRef.make(EMPTY_RELAY_ENVIRONMENT_DISCOVERY_STATE);  const refreshLock = yield* Semaphore.make(1);  const hasRefreshed = yield* Ref.make(false);  const accountGeneration = yield* Ref.make(0);  const activeAccountId = yield* Ref.make<Option.Option<string>>(Option.none());  const refreshGeneration = yield* Ref.make(0);  const offlineReportFingerprints = yield* Ref.make<ReadonlyMap<string, string>>(new Map());   const clearOfflineReport = Effect.fn("RelayEnvironmentDiscovery.clearOfflineReport")(function* (    environmentId: string,  ) {    yield* Ref.update(offlineReportFingerprints, (current) => {      if (!current.has(environmentId)) {        return current;      }      const next = new Map(current);      next.delete(environmentId);      return next;    });  });   const updateEnvironment = Effect.fn("RelayEnvironmentDiscovery.updateEnvironment")(function* (    generation: number,    environmentId: string,    update: (current: RelayDiscoveredEnvironment) => RelayDiscoveredEnvironment,  ) {    if ((yield* Ref.get(accountGeneration)) !== generation) {      return;    }    yield* SubscriptionRef.update(state, (current) => {      const entry = current.environments.get(environmentId);      if (entry === undefined) {        return current;      }      const environments = new Map(current.environments);      environments.set(environmentId, update(entry));      return { ...current, environments };    });  });   const refreshStatus = Effect.fn("RelayEnvironmentDiscovery.refreshStatus")(function* (    generation: number,    clerkToken: string,    environment: RelayClientEnvironmentRecord,  ) {    const result = yield* relay      .getEnvironmentStatus({        clerkToken,        scopes: [RelayEnvironmentStatusScope, RelayEnvironmentConnectScope],        environmentId: environment.environmentId,      })      .pipe(        Effect.mapError(mapManagedRelayError),        Effect.flatMap((status) => validateStatus(environment, status)),        Effect.result,      );     if (result._tag === "Success") {      if (result.success.status === "offline") {        const fingerprint = `${result.success.endpoint.httpBaseUrl}\n${result.success.error ?? ""}`;        const shouldReport = yield* Ref.modify(offlineReportFingerprints, (current) => {          if (current.get(environment.environmentId) === fingerprint) {            return [false, current];          }          return [true, new Map(current).set(environment.environmentId, fingerprint)];        });        if (shouldReport) {          yield* Effect.logWarning("Relay environment health check reported offline", {            environmentId: result.success.environmentId,            endpoint: result.success.endpoint.httpBaseUrl,            message: result.success.error,            traceId: result.success.traceId,          });        }      } else {        yield* clearOfflineReport(environment.environmentId);      }      yield* updateEnvironment(generation, environment.environmentId, (current) => ({        ...current,        availability: result.success.status,        status: Option.some(result.success),        error: Option.none(),      }));      return;    }     yield* clearOfflineReport(environment.environmentId);    yield* updateEnvironment(generation, environment.environmentId, (current) => ({      ...current,      availability: "error",      error: Option.some(result.failure),    }));  });   const refresh = refreshLock.withPermits(1)(    Effect.gen(function* () {      yield* Ref.set(hasRefreshed, true);      if ((yield* connectivity.status) === "offline") {        yield* SubscriptionRef.update(state, (current) => ({          ...current,          refreshing: false,          offline: true,        }));        return;      }       let generation = yield* Ref.get(accountGeneration);      yield* Ref.set(refreshGeneration, generation);      yield* SubscriptionRef.set(state, {        environments: new Map(),        refreshing: true,        offline: false,        error: Option.none(),      });       const clerkToken = yield* session.clerkToken;      if ((yield* Ref.get(accountGeneration)) !== generation) {        return;      }      const accountId = relayAccountId(clerkToken);      const previousAccountId = yield* Ref.get(activeAccountId);      if (        Option.isSome(previousAccountId) &&        (!Option.isSome(accountId) || previousAccountId.value !== accountId.value)      ) {        generation = yield* Ref.updateAndGet(accountGeneration, (current) => current + 1);        yield* Ref.set(refreshGeneration, generation);      }      yield* Ref.set(activeAccountId, accountId);       const environments = yield* relay        .listEnvironments({ clerkToken })        .pipe(Effect.mapError(mapManagedRelayError));      if ((yield* Ref.get(accountGeneration)) !== generation) {        return;      }      const next = new Map<string, RelayDiscoveredEnvironment>();      for (const environment of environments) {        next.set(environment.environmentId, {          environment,          availability: "checking",          status: Option.none(),          error: Option.none(),        });      }      yield* SubscriptionRef.update(state, (current) => ({        ...current,        environments: next,      }));       yield* Effect.forEach(        environments,        (environment) => refreshStatus(generation, clerkToken, environment),        {          concurrency: "unbounded",          discard: true,        },      );      if ((yield* Ref.get(accountGeneration)) !== generation) {        return;      }      yield* SubscriptionRef.update(state, (current) => ({        ...current,        refreshing: false,      }));    }).pipe(      Effect.catch((error) =>        Effect.gen(function* () {          const generation = yield* Ref.get(refreshGeneration);          if ((yield* Ref.get(accountGeneration)) !== generation) {            return;          }          yield* SubscriptionRef.update(state, (current) => ({            ...current,            refreshing: false,            error: Option.some(error),          }));        }),      ),    ),  );   yield* connectivity.changes.pipe(    Stream.changes,    Stream.runForEach((networkStatus) =>      networkStatus === "offline"        ? SubscriptionRef.update(state, (current) => ({            ...current,            refreshing: false,            offline: true,          }))        : Ref.get(hasRefreshed).pipe(            Effect.flatMap((shouldRefresh) => (shouldRefresh ? refresh : Effect.void)),          ),    ),    Effect.forkScoped,  );  yield* wakeups.changes.pipe(    Stream.runForEach((reason) =>      reason === "credentials-changed"        ? Effect.gen(function* () {            yield* Ref.update(accountGeneration, (current) => current + 1);            yield* Ref.set(activeAccountId, Option.none());            yield* Ref.set(offlineReportFingerprints, new Map());            const shouldRefresh = yield* Ref.get(hasRefreshed);            yield* SubscriptionRef.set(state, EMPTY_RELAY_ENVIRONMENT_DISCOVERY_STATE);            if (shouldRefresh) {              yield* refresh.pipe(Effect.forkScoped);            }          })        : Effect.void,    ),    Effect.forkScoped,  );   return RelayEnvironmentDiscovery.of({ state, refresh });}); export const layer = Layer.effect(RelayEnvironmentDiscovery, make());