Merge commit '36c66dd290d3ce6eb1ccd310d0c658d4a32bb8eb' as 'repos/effect'

2026-07-19 03:25:18 +05:30
parent dd1071cc10
commit 2daf979036
2214 changed files with 673090 additions and 0 deletions
@@ -0,0 +1,72 @@
+/**
+ * Node.js socket constructors and layers for Effect sockets.
+ *
+ * This module re-exports the shared Node socket support for TCP connections,
+ * Unix domain socket connections, and Node `Duplex` streams. It also provides
+ * WebSocket constructor layers: one that uses `globalThis.WebSocket` when
+ * present and falls back to `ws`, one that always uses `ws`, and one that
+ * creates a `Socket.Socket` layer for a WebSocket URL.
+ *
+ * @since 4.0.0
+ */
+import { NodeWS as WS } from "@effect/platform-node-shared/NodeSocket"
+import type * as Duration from "effect/Duration"
+import type * as Effect from "effect/Effect"
+import { flow } from "effect/Function"
+import * as Layer from "effect/Layer"
+import * as Socket from "effect/unstable/socket/Socket"
+
+/**
+ * @since 4.0.0
+ */
+export * from "@effect/platform-node-shared/NodeSocket"
+
+/**
+ * Provides a `Socket.WebSocketConstructor`, using `globalThis.WebSocket` when
+ * available and falling back to the `ws` package otherwise.
+ *
+ * @category layers
+ * @since 4.0.0
+ */
+export const layerWebSocketConstructor: Layer.Layer<
+  Socket.WebSocketConstructor
+> = Layer.sync(Socket.WebSocketConstructor)(() => {
+  if ("WebSocket" in globalThis) {
+    return (url, protocols) => new globalThis.WebSocket(url, protocols)
+  }
+  return (url, protocols) => new WS.WebSocket(url, protocols) as unknown as globalThis.WebSocket
+})
+
+/**
+ * Provides a `Socket.WebSocketConstructor` backed explicitly by the `ws`
+ * package.
+ *
+ * @category layers
+ * @since 4.0.0
+ */
+export const layerWebSocketConstructorWS: Layer.Layer<
+  Socket.WebSocketConstructor
+> = Layer.succeed(Socket.WebSocketConstructor)(
+  (url, protocols) => new WS.WebSocket(url, protocols) as unknown as globalThis.WebSocket
+)
+
+/**
+ * Creates a `Socket.Socket` layer for a WebSocket URL using the Node WebSocket
+ * constructor layer, honoring protocol, open-timeout, and close-code error
+ * options.
+ *
+ * @category layers
+ * @since 4.0.0
+ */
+export const layerWebSocket: (
+  url: string | Effect.Effect<string>,
+  options?: {
+    readonly closeCodeIsError?: ((code: number) => boolean) | undefined
+    readonly openTimeout?: Duration.Input | undefined
+    readonly protocols?: string | Array<string> | undefined
+  } | undefined
+) => Layer.Layer<Socket.Socket, never, never> = flow(
+  Socket.makeWebSocket,
+  Layer.effect(Socket.Socket),
+  Layer.provide(layerWebSocketConstructor)
+)