@effect/platform-node-shared 0.57.1 → 4.0.0-beta.0

package/src/NodeClusterSocket.ts

@@ -1,17 +1,16 @@
 /**
  * @since 1.0.0
  */
-import * as Runners from "@effect/cluster/Runners"
-import * as ShardingConfig from "@effect/cluster/ShardingConfig"
-import { Socket } from "@effect/platform/Socket"
-import type * as SocketServer from "@effect/platform/SocketServer"
-import * as RpcClient from "@effect/rpc/RpcClient"
-import * as RpcSerialization from "@effect/rpc/RpcSerialization"
 import * as Effect from "effect/Effect"
 import * as Layer from "effect/Layer"
-import * as Option from "effect/Option"
-import * as NodeSocket from "./NodeSocket.js"
-import * as NodeSocketServer from "./NodeSocketServer.js"
+import * as Runners from "effect/unstable/cluster/Runners"
+import * as ShardingConfig from "effect/unstable/cluster/ShardingConfig"
+import * as RpcClient from "effect/unstable/rpc/RpcClient"
+import * as RpcSerialization from "effect/unstable/rpc/RpcSerialization"
+import { Socket } from "effect/unstable/socket/Socket"
+import type * as SocketServer from "effect/unstable/socket/SocketServer"
+import * as NodeSocket from "./NodeSocket.ts"
+import * as NodeSocketServer from "./NodeSocketServer.ts"
 
 /**
  * @since 1.0.0
@@ -49,9 +48,9 @@ export const layerSocketServer: Layer.Layer<
   ShardingConfig.ShardingConfig
 > = Effect.gen(function*() {
   const config = yield* ShardingConfig.ShardingConfig
-  const listenAddress = Option.orElse(config.runnerListenAddress, () => config.runnerAddress)
-  if (listenAddress._tag === "None") {
+  const listenAddress = config.runnerListenAddress ?? config.runnerAddress
+  if (listenAddress === undefined) {
     return yield* Effect.die("layerSocketServer: ShardingConfig.runnerListenAddress is None")
   }
-  return NodeSocketServer.layer(listenAddress.value)
-}).pipe(Layer.unwrapEffect)
+  return NodeSocketServer.layer(listenAddress)
+}).pipe(Layer.unwrap)

package/src/NodeFileSystem.ts

@@ -1,13 +1,640 @@
 /**
  * @since 1.0.0
  */
+import * as Cause from "effect/Cause"
+import * as Effect from "effect/Effect"
+import { effectify } from "effect/Effect"
+import * as FileSystem from "effect/FileSystem"
+import { pipe } from "effect/Function"
+import * as Layer from "effect/Layer"
+import * as Option from "effect/Option"
+import * as Error from "effect/PlatformError"
+import * as Queue from "effect/Queue"
+import * as Stream from "effect/Stream"
+import * as Crypto from "node:crypto"
+import * as NFS from "node:fs"
+import * as OS from "node:os"
+import * as Path from "node:path"
+import { handleErrnoException } from "./internal/utils.ts"
 
-import type { FileSystem } from "@effect/platform/FileSystem"
-import type { Layer } from "effect/Layer"
-import * as internal from "./internal/fileSystem.js"
+const handleBadArgument = (method: string) => (err: unknown) =>
+  Error.badArgument({
+    module: "FileSystem",
+    method,
+    description: (err as Error).message ?? String(err)
+  })
+
+// == access
+
+const access = ((): FileSystem.FileSystem["access"] => {
+  const nodeAccess = effectify(
+    NFS.access,
+    handleErrnoException("FileSystem", "access"),
+    handleBadArgument("access")
+  )
+  return (path, options) => {
+    let mode = NFS.constants.F_OK
+    if (options?.readable) {
+      mode |= NFS.constants.R_OK
+    }
+    if (options?.writable) {
+      mode |= NFS.constants.W_OK
+    }
+    return nodeAccess(path, mode)
+  }
+})()
+
+// == copy
+
+const copy = ((): FileSystem.FileSystem["copy"] => {
+  const nodeCp = effectify(
+    NFS.cp,
+    handleErrnoException("FileSystem", "copy"),
+    handleBadArgument("copy")
+  )
+  return (fromPath, toPath, options) =>
+    nodeCp(fromPath, toPath, {
+      force: options?.overwrite ?? false,
+      preserveTimestamps: options?.preserveTimestamps ?? false,
+      recursive: true
+    })
+})()
+
+// == copyFile
+
+const copyFile = (() => {
+  const nodeCopyFile = effectify(
+    NFS.copyFile,
+    handleErrnoException("FileSystem", "copyFile"),
+    handleBadArgument("copyFile")
+  )
+  return (fromPath: string, toPath: string) => nodeCopyFile(fromPath, toPath)
+})()
+
+// == chmod
+
+const chmod = (() => {
+  const nodeChmod = effectify(
+    NFS.chmod,
+    handleErrnoException("FileSystem", "chmod"),
+    handleBadArgument("chmod")
+  )
+  return (path: string, mode: number) => nodeChmod(path, mode)
+})()
+
+// == chown
+
+const chown = (() => {
+  const nodeChown = effectify(
+    NFS.chown,
+    handleErrnoException("FileSystem", "chown"),
+    handleBadArgument("chown")
+  )
+  return (path: string, uid: number, gid: number) => nodeChown(path, uid, gid)
+})()
+
+// == link
+
+const link = (() => {
+  const nodeLink = effectify(
+    NFS.link,
+    handleErrnoException("FileSystem", "link"),
+    handleBadArgument("link")
+  )
+  return (existingPath: string, newPath: string) => nodeLink(existingPath, newPath)
+})()
+
+// == makeDirectory
+
+const makeDirectory = ((): FileSystem.FileSystem["makeDirectory"] => {
+  const nodeMkdir = effectify(
+    NFS.mkdir,
+    handleErrnoException("FileSystem", "makeDirectory"),
+    handleBadArgument("makeDirectory")
+  )
+  return (path, options) =>
+    nodeMkdir(path, {
+      recursive: options?.recursive ?? false,
+      mode: options?.mode
+    })
+})()
+
+// == makeTempDirectory
+
+const makeTempDirectoryFactory = (method: string): FileSystem.FileSystem["makeTempDirectory"] => {
+  const nodeMkdtemp = effectify(
+    NFS.mkdtemp,
+    handleErrnoException("FileSystem", method),
+    handleBadArgument(method)
+  )
+  return (options) =>
+    Effect.suspend(() => {
+      const prefix = options?.prefix ?? ""
+      const directory = typeof options?.directory === "string"
+        ? Path.join(options.directory, ".")
+        : OS.tmpdir()
+
+      return nodeMkdtemp(prefix ? Path.join(directory, prefix) : directory + "/")
+    })
+}
+const makeTempDirectory = makeTempDirectoryFactory("makeTempDirectory")
+
+// == remove
+
+const removeFactory = (method: string): FileSystem.FileSystem["remove"] => {
+  const nodeRm = effectify(
+    NFS.rm,
+    handleErrnoException("FileSystem", method),
+    handleBadArgument(method)
+  )
+  return (path, options) =>
+    nodeRm(
+      path,
+      { recursive: options?.recursive ?? false, force: options?.force ?? false }
+    )
+}
+const remove = removeFactory("remove")
+
+// == makeTempDirectoryScoped
+
+const makeTempDirectoryScoped = ((): FileSystem.FileSystem["makeTempDirectoryScoped"] => {
+  const makeDirectory = makeTempDirectoryFactory("makeTempDirectoryScoped")
+  const removeDirectory = removeFactory("makeTempDirectoryScoped")
+  return (options) =>
+    Effect.acquireRelease(
+      makeDirectory(options),
+      (directory) => Effect.orDie(removeDirectory(directory, { recursive: true }))
+    )
+})()
+
+// == open
+
+const openFactory = (method: string): FileSystem.FileSystem["open"] => {
+  const nodeOpen = effectify(
+    NFS.open,
+    handleErrnoException("FileSystem", method),
+    handleBadArgument(method)
+  )
+  const nodeClose = effectify(
+    NFS.close,
+    handleErrnoException("FileSystem", method),
+    handleBadArgument(method)
+  )
+
+  return (path, options) =>
+    pipe(
+      Effect.acquireRelease(
+        nodeOpen(path, options?.flag ?? "r", options?.mode),
+        (fd) => Effect.orDie(nodeClose(fd))
+      ),
+      Effect.map((fd) => makeFile(FileSystem.FileDescriptor(fd), options?.flag?.startsWith("a") ?? false))
+    )
+}
+const open = openFactory("open")
+
+const makeFile = (() => {
+  const nodeReadFactory = (method: string) =>
+    effectify(
+      NFS.read,
+      handleErrnoException("FileSystem", method),
+      handleBadArgument(method)
+    )
+  const nodeRead = nodeReadFactory("read")
+  const nodeReadAlloc = nodeReadFactory("readAlloc")
+  const nodeStat = effectify(
+    NFS.fstat,
+    handleErrnoException("FileSystem", "stat"),
+    handleBadArgument("stat")
+  )
+  const nodeTruncate = effectify(
+    NFS.ftruncate,
+    handleErrnoException("FileSystem", "truncate"),
+    handleBadArgument("truncate")
+  )
+
+  const nodeSync = effectify(
+    NFS.fsync,
+    handleErrnoException("FileSystem", "sync"),
+    handleBadArgument("sync")
+  )
+
+  const nodeWriteFactory = (method: string) =>
+    effectify(
+      NFS.write,
+      handleErrnoException("FileSystem", method),
+      handleBadArgument(method)
+    )
+  const nodeWrite = nodeWriteFactory("write")
+  const nodeWriteAll = nodeWriteFactory("writeAll")
+
+  class FileImpl implements FileSystem.File {
+    readonly [FileSystem.FileTypeId]: typeof FileSystem.FileTypeId
+    readonly fd: FileSystem.File.Descriptor
+    private readonly append: boolean
+
+    private position: bigint = 0n
+
+    constructor(
+      fd: FileSystem.File.Descriptor,
+      append: boolean
+    ) {
+      this[FileSystem.FileTypeId] = FileSystem.FileTypeId
+      this.fd = fd
+      this.append = append
+    }
+
+    get stat() {
+      return Effect.map(nodeStat(this.fd), makeFileInfo)
+    }
+
+    get sync() {
+      return nodeSync(this.fd)
+    }
+
+    seek(offset: FileSystem.SizeInput, from: FileSystem.SeekMode) {
+      const offsetSize = FileSystem.Size(offset)
+      return Effect.sync(() => {
+        if (from === "start") {
+          this.position = offsetSize
+        } else if (from === "current") {
+          this.position = this.position + offsetSize
+        }
+
+        return this.position
+      })
+    }
+
+    read(buffer: Uint8Array) {
+      return Effect.suspend(() => {
+        const position = this.position
+        return Effect.map(
+          nodeRead(this.fd, { buffer, position }),
+          (bytesRead) => {
+            const sizeRead = FileSystem.Size(bytesRead)
+            this.position = position + sizeRead
+            return sizeRead
+          }
+        )
+      })
+    }
+
+    readAlloc(size: FileSystem.SizeInput) {
+      const sizeNumber = Number(size)
+      return Effect.suspend(() => {
+        const buffer = Buffer.allocUnsafeSlow(sizeNumber)
+        const position = this.position
+        return Effect.map(
+          nodeReadAlloc(this.fd, { buffer, position }),
+          (bytesRead): Buffer | undefined => {
+            if (bytesRead === 0) {
+              return undefined
+            }
+
+            this.position = position + BigInt(bytesRead)
+            if (bytesRead === sizeNumber) {
+              return buffer
+            }
+
+            const dst = Buffer.allocUnsafeSlow(bytesRead)
+            buffer.copy(dst, 0, 0, bytesRead)
+            return dst
+          }
+        )
+      })
+    }
+
+    truncate(length?: FileSystem.SizeInput) {
+      return Effect.map(nodeTruncate(this.fd, length ? Number(length) : undefined), () => {
+        if (!this.append) {
+          const len = BigInt(length ?? 0)
+          if (this.position > len) {
+            this.position = len
+          }
+        }
+      })
+    }
+
+    write(buffer: Uint8Array) {
+      return Effect.suspend(() => {
+        const position = this.position
+        return Effect.map(
+          nodeWrite(this.fd, buffer, undefined, undefined, this.append ? undefined : Number(position)),
+          (bytesWritten) => {
+            const sizeWritten = FileSystem.Size(bytesWritten)
+            if (!this.append) {
+              this.position = position + sizeWritten
+            }
+            return sizeWritten
+          }
+        )
+      })
+    }
+
+    private writeAllChunk(buffer: Uint8Array): Effect.Effect<void, Error.PlatformError> {
+      return Effect.suspend(() => {
+        const position = this.position
+        return Effect.flatMap(
+          nodeWriteAll(this.fd, buffer, undefined, undefined, this.append ? undefined : Number(position)),
+          (bytesWritten) => {
+            if (bytesWritten === 0) {
+              return Effect.fail(
+                Error.systemError({
+                  module: "FileSystem",
+                  method: "writeAll",
+                  kind: "WriteZero",
+                  pathOrDescriptor: this.fd,
+                  description: "write returned 0 bytes written"
+                })
+              )
+            }
+
+            if (!this.append) {
+              this.position = position + BigInt(bytesWritten)
+            }
+
+            return bytesWritten < buffer.length ? this.writeAllChunk(buffer.subarray(bytesWritten)) : Effect.void
+          }
+        )
+      })
+    }
+
+    writeAll(buffer: Uint8Array) {
+      return this.writeAllChunk(buffer)
+    }
+  }
+
+  return (fd: FileSystem.File.Descriptor, append: boolean): FileSystem.File => new FileImpl(fd, append)
+})()
+
+// == makeTempFile
+
+const makeTempFileFactory = (method: string): FileSystem.FileSystem["makeTempFile"] => {
+  const makeDirectory = makeTempDirectoryFactory(method)
+  return Effect.fnUntraced(function*(options) {
+    const directory = yield* makeDirectory(options)
+    const random = Crypto.randomBytes(6).toString("hex")
+    const name = Path.join(directory, options?.suffix ? `${random}${options.suffix}` : random)
+    yield* writeFile(name, new Uint8Array(0))
+    return name
+  })
+}
+const makeTempFile = makeTempFileFactory("makeTempFile")
+
+// == makeTempFileScoped
+
+const makeTempFileScoped = ((): FileSystem.FileSystem["makeTempFileScoped"] => {
+  const makeFile = makeTempFileFactory("makeTempFileScoped")
+  const removeDirectory = removeFactory("makeTempFileScoped")
+  return (options) =>
+    Effect.acquireRelease(
+      makeFile(options),
+      (file) => Effect.orDie(removeDirectory(Path.dirname(file), { recursive: true }))
+    )
+})()
+
+// == readDirectory
+
+const readDirectory: FileSystem.FileSystem["readDirectory"] = (path, options) =>
+  Effect.tryPromise({
+    try: () => NFS.promises.readdir(path, options),
+    catch: (err) => handleErrnoException("FileSystem", "readDirectory")(err as any, [path])
+  })
+
+// == readFile
+
+const readFile = (path: string) =>
+  Effect.callback<Uint8Array, Error.PlatformError>((resume, signal) => {
+    try {
+      NFS.readFile(path, { signal }, (err, data) => {
+        if (err) {
+          resume(Effect.fail(handleErrnoException("FileSystem", "readFile")(err, [path])))
+        } else {
+          resume(Effect.succeed(data))
+        }
+      })
+    } catch (err) {
+      resume(Effect.fail(handleBadArgument("readFile")(err)))
+    }
+  })
+
+// == readLink
+
+const readLink = (() => {
+  const nodeReadLink = effectify(
+    NFS.readlink,
+    handleErrnoException("FileSystem", "readLink"),
+    handleBadArgument("readLink")
+  )
+  return (path: string) => nodeReadLink(path)
+})()
+
+// == realPath
+
+const realPath = (() => {
+  const nodeRealPath = effectify(
+    NFS.realpath,
+    handleErrnoException("FileSystem", "realPath"),
+    handleBadArgument("realPath")
+  )
+  return (path: string) => nodeRealPath(path)
+})()
+
+// == rename
+
+const rename = (() => {
+  const nodeRename = effectify(
+    NFS.rename,
+    handleErrnoException("FileSystem", "rename"),
+    handleBadArgument("rename")
+  )
+  return (oldPath: string, newPath: string) => nodeRename(oldPath, newPath)
+})()
+
+// == stat
+
+const makeFileInfo = (stat: NFS.Stats): FileSystem.File.Info => ({
+  type: stat.isFile() ?
+    "File" :
+    stat.isDirectory() ?
+    "Directory" :
+    stat.isSymbolicLink() ?
+    "SymbolicLink" :
+    stat.isBlockDevice() ?
+    "BlockDevice" :
+    stat.isCharacterDevice() ?
+    "CharacterDevice" :
+    stat.isFIFO() ?
+    "FIFO" :
+    stat.isSocket() ?
+    "Socket" :
+    "Unknown",
+  mtime: stat.mtime,
+  atime: stat.atime,
+  birthtime: stat.birthtime,
+  dev: stat.dev,
+  rdev: stat.rdev,
+  ino: stat.ino,
+  mode: stat.mode,
+  nlink: stat.nlink,
+  uid: stat.uid,
+  gid: stat.gid,
+  size: FileSystem.Size(stat.size),
+  blksize: FileSystem.Size(stat.blksize),
+  blocks: stat.blocks
+})
+const stat = (() => {
+  const nodeStat = effectify(
+    NFS.stat,
+    handleErrnoException("FileSystem", "stat"),
+    handleBadArgument("stat")
+  )
+  return (path: string) => Effect.map(nodeStat(path), makeFileInfo)
+})()
+
+// == symlink
+
+const symlink = (() => {
+  const nodeSymlink = effectify(
+    NFS.symlink,
+    handleErrnoException("FileSystem", "symlink"),
+    handleBadArgument("symlink")
+  )
+  return (target: string, path: string) => nodeSymlink(target, path)
+})()
+
+// == truncate
+
+const truncate = (() => {
+  const nodeTruncate = effectify(
+    NFS.truncate,
+    handleErrnoException("FileSystem", "truncate"),
+    handleBadArgument("truncate")
+  )
+  return (path: string, length?: FileSystem.SizeInput) =>
+    nodeTruncate(path, length !== undefined ? Number(length) : undefined)
+})()
+
+// == utimes
+
+const utimes = (() => {
+  const nodeUtimes = effectify(
+    NFS.utimes,
+    handleErrnoException("FileSystem", "utime"),
+    handleBadArgument("utime")
+  )
+  return (path: string, atime: number | Date, mtime: number | Date) => nodeUtimes(path, atime, mtime)
+})()
+
+// == watch
+
+const watchNode = (path: string) =>
+  Stream.callback<FileSystem.WatchEvent, Error.PlatformError>((queue) =>
+    Effect.acquireRelease(
+      Effect.sync(() => {
+        const watcher = NFS.watch(path, {}, (event, path) => {
+          if (!path) return
+          switch (event) {
+            case "rename": {
+              Effect.runFork(Effect.matchEffect(stat(path), {
+                onSuccess: (_) => Queue.offer(queue, { _tag: "Create", path }),
+                onFailure: (_) => Queue.offer(queue, { _tag: "Remove", path })
+              }))
+              return
+            }
+            case "change": {
+              Queue.offerUnsafe(queue, { _tag: "Update", path })
+              return
+            }
+          }
+        })
+        watcher.on("error", (error) => {
+          Queue.failCauseUnsafe(
+            queue,
+            Cause.fail(
+              Error.systemError({
+                module: "FileSystem",
+                kind: "Unknown",
+                method: "watch",
+                pathOrDescriptor: path,
+                cause: error
+              })
+            )
+          )
+        })
+        watcher.on("close", () => {
+          Queue.endUnsafe(queue)
+        })
+        return watcher
+      }),
+      (watcher) => Effect.sync(() => watcher.close())
+    )
+  )
+
+const watch = (backend: FileSystem.WatchBackend["Service"] | undefined, path: string) =>
+  stat(path).pipe(
+    Effect.map((stat) => {
+      if (backend) {
+        const stream = backend.register(path, stat)
+        if (stream) return stream
+      }
+      return watchNode(path)
+    }),
+    Stream.unwrap
+  )
+
+// == writeFile
+
+const writeFile: FileSystem.FileSystem["writeFile"] = (path, data, options) =>
+  Effect.callback<void, Error.PlatformError>((resume, signal) => {
+    try {
+      NFS.writeFile(path, data, {
+        signal,
+        flag: options?.flag,
+        mode: options?.mode
+      }, (err) => {
+        if (err) {
+          resume(Effect.fail(handleErrnoException("FileSystem", "writeFile")(err, [path])))
+        } else {
+          resume(Effect.void)
+        }
+      })
+    } catch (err) {
+      resume(Effect.fail(handleBadArgument("writeFile")(err)))
+    }
+  })
+
+const makeFileSystem = Effect.map(Effect.serviceOption(FileSystem.WatchBackend), (backend) =>
+  FileSystem.make({
+    access,
+    chmod,
+    chown,
+    copy,
+    copyFile,
+    link,
+    makeDirectory,
+    makeTempDirectory,
+    makeTempDirectoryScoped,
+    makeTempFile,
+    makeTempFileScoped,
+    open,
+    readDirectory,
+    readFile,
+    readLink,
+    realPath,
+    remove,
+    rename,
+    stat,
+    symlink,
+    truncate,
+    utimes,
+    watch(path) {
+      return watch(Option.getOrUndefined(backend), path)
+    },
+    writeFile
+  }))
 
 /**
  * @since 1.0.0
- * @category layer
+ * @category Layers
  */
-export const layer: Layer<FileSystem> = internal.layer
+export const layer: Layer.Layer<FileSystem.FileSystem> = Layer.effect(FileSystem.FileSystem)(makeFileSystem)