@6qat/tcp-connection 0.1.0

package/README.md

@@ -0,0 +1,12 @@
+# TCP Connection
+
+A robust TCP connection library with Effect.js integration.
+
+## Installation
+
+```bash
+bun add @6qat/tcp-connection
+# or
+npm install @6qat/tcp-connection
+# or
+yarn add @6qat/tcp-connection

package/dist/index.d.ts

@@ -0,0 +1 @@
+export * from "./tcp-stream";

package/dist/index.js

@@ -0,0 +1,2 @@
+// Export everything from tcp-stream
+export * from "./tcp-stream";

package/dist/tcp-stream.d.ts

@@ -0,0 +1,13 @@
+import { Effect, Stream, Duration } from "effect";
+interface TcpConnection {
+    readonly stream: Stream.Stream<Uint8Array, Error>;
+    readonly send: (data: Uint8Array) => Effect.Effect<void>;
+    readonly sendText: (data: string) => Effect.Effect<void>;
+    readonly close: Effect.Effect<void>;
+}
+export declare const createTcpConnection: (options: {
+    host: string;
+    port: number;
+    timeout?: Duration.Duration;
+}) => Effect.Effect<TcpConnection, Error>;
+export {};

package/dist/tcp-stream.js

@@ -0,0 +1,56 @@
+import { Effect, Stream, Queue, pipe, Fiber, Duration } from "effect";
+export const createTcpConnection = (options) => {
+    return Effect.gen(function* () {
+        // Create queues for incoming and outgoing data
+        const incomingQueue = yield* Queue.unbounded();
+        const outgoingQueue = yield* Queue.unbounded();
+        // Create deferred for connection cleanup
+        const socketEffect = yield* Effect.tryPromise(() => Bun.connect({
+            port: options.port,
+            hostname: options.host,
+            socket: {
+                data(_socket, data) {
+                    Queue.unsafeOffer(incomingQueue, data);
+                },
+                error(_socket, error) {
+                    //Queue.unsafeOffer(incomingQueue, error)
+                    Queue.shutdown(outgoingQueue);
+                },
+                close(_socket) {
+                    Queue.shutdown(incomingQueue);
+                    Queue.shutdown(outgoingQueue);
+                },
+            },
+        })).pipe(Effect.timeout(options.timeout ?? Duration.millis(3000)), Effect.flatMap((maybeSocket) => maybeSocket
+            ? Effect.succeed(maybeSocket)
+            : Effect.fail(new Error("Connection timeout"))));
+        // Fiber for writing outgoing data
+        const writerFiber = yield* pipe(Effect.iterate(undefined, {
+            while: () => true,
+            body: () => pipe(Queue.take(outgoingQueue), Effect.flatMap((data) => Effect.try({
+                try: () => {
+                    const bytesWritten = socketEffect.write(data);
+                    if (bytesWritten !== data.length) {
+                        throw new Error("Partial write");
+                    }
+                },
+                catch: (error) => new Error(`Write failed: ${error}`),
+            }))),
+        }), Effect.fork);
+        // Cleanup procedure
+        const close = Effect.gen(function* () {
+            console.log("Closing connection");
+            socketEffect.end();
+            yield* Queue.shutdown(incomingQueue);
+            yield* Queue.shutdown(outgoingQueue);
+            yield* Fiber.interrupt(writerFiber);
+        });
+        // returns TCPConnection
+        return {
+            stream: Stream.fromQueue(incomingQueue).pipe(Stream.ensuring(close)),
+            send: (data) => Queue.offer(outgoingQueue, data),
+            sendText: (data) => Queue.offer(outgoingQueue, new TextEncoder().encode(data)),
+            close,
+        };
+    });
+};

package/package.json

@@ -0,0 +1,28 @@
+{
+	"name": "@6qat/tcp-connection",
+	"version": "0.1.0",
+	"description": "TCP connection library with Effect.js integration",
+	"module": "dist/index.js",
+	"main": "dist/index.js",
+	"types": "dist/index.d.ts",
+	"type": "module",
+	"files": ["dist", "README.md"],
+	"scripts": {
+		"build": "tsc",
+		"prepublishOnly": "npm run build",
+		"test": "echo \"Error: no test specified\" && exit 1"
+	},
+	"keywords": ["tcp", "connection", "effect", "bun"],
+	"author": "Your Name",
+	"license": "MIT",
+	"devDependencies": {
+		"@types/bun": "latest",
+		"typescript": "^5"
+	},
+	"peerDependencies": {
+		"effect": "^2.0.0"
+	},
+	"publishConfig": {
+		"access": "public"
+	}
+}