@zap-socket/server 0.0.1

package/dist/events.d.ts

@@ -0,0 +1,16 @@
+import { z } from "zod";
+export type EventInput = z.ZodTypeAny | undefined;
+export type ZapEvent<T extends EventInput, R = any> = T extends z.ZodTypeAny ? {
+    input: T;
+    process: (input: z.infer<T>) => R;
+} : {
+    input: z.ZodVoid;
+    process: () => R;
+};
+export type EventMap = Record<string, ZapEvent<any, any>>;
+export declare const createZapEvent: <T extends EventInput, R>(eventObj: T extends z.ZodTypeAny ? {
+    input: T;
+    process: (input: z.infer<T>) => R;
+} : {
+    process: () => R;
+}) => ZapEvent<T, R>;

package/dist/events.js

@@ -0,0 +1,10 @@
+import { z } from "zod";
+export const createZapEvent = (eventObj) => {
+    if ("input" in eventObj) {
+        return eventObj;
+    }
+    return {
+        input: z.void(),
+        process: eventObj.process
+    };
+};

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export * from "./events";
+export * from "./server";
+export * from "./utils";

package/dist/index.js

@@ -0,0 +1,3 @@
+export * from "./events";
+export * from "./server";
+export * from "./utils";

package/dist/server.d.ts

@@ -0,0 +1,17 @@
+import type { EventMap } from "./events";
+interface ZapServerConstructorT {
+    port: number;
+}
+export declare class ZapServer {
+    private wss;
+    private wsToId;
+    private idToWs;
+    private events;
+    constructor({ port }: ZapServerConstructorT);
+    private removeClient;
+    attachEvents<T extends EventMap>(events: T): void;
+}
+export declare const createZapServer: ({ port }: {
+    port: number;
+}) => ZapServer;
+export {};

package/dist/server.js

@@ -0,0 +1,53 @@
+import { WebSocketServer } from "ws";
+import { generateId } from "./utils";
+export class ZapServer {
+    wss;
+    wsToId;
+    idToWs;
+    events = {};
+    constructor({ port }) {
+        this.wss = new WebSocketServer({ port });
+        this.wsToId = new Map();
+        this.idToWs = new Map();
+        this.wss.on("connection", (ws) => {
+            const clientId = generateId();
+            console.log(`client ${clientId} connected`);
+            this.wsToId.set(ws, clientId);
+            this.idToWs.set(clientId, ws);
+            ws.on("message", (message) => {
+                if (!this.wsToId.get(ws)) {
+                    const id = generateId();
+                    this.wsToId.set(ws, id);
+                    this.idToWs.set(id, ws);
+                    ws.send("ID " + id);
+                }
+                for (const [event, { process }] of Object.entries(this.events)) {
+                    const parsedMessage = JSON.parse(message.toString());
+                    if (parsedMessage["type"] === event) {
+                        const { input } = parsedMessage;
+                        process(input);
+                    }
+                }
+            });
+            ws.on("close", () => {
+                this.removeClient(ws);
+            });
+            ws.on("error", (err) => {
+                console.error(`WebSocket error for ${clientId}:`, err);
+            });
+        });
+    }
+    removeClient(ws) {
+        const clientId = this.wsToId.get(ws);
+        if (clientId) {
+            this.wsToId.delete(ws);
+            this.idToWs.delete(clientId);
+        }
+    }
+    attachEvents(events) {
+        this.events = events;
+    }
+}
+export const createZapServer = ({ port }) => {
+    return new ZapServer({ port });
+};

package/dist/utils.d.ts

@@ -0,0 +1 @@
+export declare const generateId: (length?: number) => string;

package/dist/utils.js

@@ -0,0 +1,10 @@
+export const generateId = (length = 6) => {
+    const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789";
+    let id = "";
+    const randomValues = new Uint8Array(length);
+    crypto.getRandomValues(randomValues);
+    randomValues.forEach((num) => {
+        id += chars[num % chars.length];
+    });
+    return id;
+};

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "@zap-socket/server",
+  "version": "0.0.1",
+  "description": "A fully typesafe tRPC-inspired WebSocket library with Zod validation, req-res model, and native subscriptions.",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "repository": "https://github.com/atharvparlikar/zap-socket",
+  "author": "Atharv Parlikar",
+  "license": "MIT",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "dev": "tsup src/index.ts --watch",
+    "build": "tsc",
+    "clean": "rm -rf dist"
+  },
+  "devDependencies": {
+    "@types/ws": "^8.18.0",
+    "eslint": "^9.22.0",
+    "tsup": "^8.4.0",
+    "typescript": "^5.8.2",
+    "vitest": "^3.0.9"
+  },
+  "dependencies": {
+    "zod": "^3.24.2"
+  }
+}