@wener/mcps 1.0.2 → 1.0.5

package/package.json

@@ -1,86 +1,146 @@
 {
-  "name": "@wener/mcps",
-  "version": "1.0.2",
-  "description": "",
-  "author": "",
-  "bin": {
-    "mcps": "./dist/mcps-cli.mjs"
-  },
-  "main": "./dist/index.mjs",
-  "imports": {
-    "#/*": "./*"
-  },
-  "exports": {
-    ".": {
-      "types": "./src/index.ts",
-      "default": "./lib/index.js"
-    },
-    "./server": {
-      "types": "./src/server/index.ts",
-      "default": "./lib/server/index.js"
-    }
-  },
-  "files": [
-    "dist",
-    "lib",
-    "src",
-    "README.md"
-  ],
-  "keywords": [],
-  "dependencies": {
-    "better-sqlite3": "^12.6.2"
-  },
-  "devDependencies": {
-    "@ai-sdk/mcp": "^1.0.18",
-    "@ai-sdk/openai-compatible": "^2.0.26",
-    "@ai-sdk/react": "^3.0.71",
-    "@base-ui/react": "^1.1.0",
-    "@hono/mcp": "^0.2.3",
-    "@hono/node-server": "^1.19.9",
-    "@hono/vite-dev-server": "^0.24.1",
-    "@mikro-orm/core": "^7.0.0-rc.0",
-    "@mikro-orm/decorators": "^7.0.0-rc.0",
-    "@mikro-orm/sqlite": "^7.0.0-rc.0",
-    "@modelcontextprotocol/sdk": "^1.25.3",
-    "@orpc/client": "^1.13.4",
-    "@orpc/contract": "^1.13.4",
-    "@orpc/json-schema": "^1.13.4",
-    "@orpc/openapi": "^1.13.4",
-    "@orpc/server": "^1.13.4",
-    "@orpc/zod": "^1.13.4",
-    "@streamdown/cjk": "^1.0.1",
-    "@streamdown/code": "^1.0.1",
-    "@streamdown/math": "^1.0.1",
-    "@streamdown/mermaid": "^1.0.1",
-    "@wener/client": "^1.0.26",
-    "@wener/common": "^2.0.3",
-    "ai": "^6.0.69",
-    "commander": "^14.0.3",
-    "consola": "^3.4.2",
-    "hono": "^4.11.7",
-    "kysely": "^0.28.11",
-    "lru-cache": "^11.2.5",
-    "lucide-react": "^0.562.0",
-    "mysql2": "^3.16.3",
-    "openai": "^6.17.0",
-    "pg-query-stream": "^4.12.0",
-    "react": "^19.2.4",
-    "react-dom": "^19.2.4",
-    "react-markdown": "^10.1.0",
-    "react-router-dom": "^7.13.0",
-    "std-env": "^3.10.0",
-    "streamdown": "^2.1.0",
-    "tarn": "^3.0.2",
-    "tedious": "^19.2.0",
-    "yaml": "^2.8.2",
-    "zod": "^4.3.6",
-    "@wener/ai": "0.1.0"
-  },
-  "scripts": {
-    "build": "bun run ./src/scripts/bundle.ts",
-    "dev": "pnpm tsx --watch src/mcps-cli.ts",
-    "dev:web": "vite",
-    "test": "vitest run",
-    "typecheck": "tsc --noEmit"
-  }
-}
+	"name": "@wener/mcps",
+	"version": "1.0.5",
+	"type": "module",
+	"description": "",
+	"author": "",
+	"bin": {
+		"mcps": "./dist/mcps-cli.mjs"
+	},
+	"main": "./dist/index.mjs",
+	"imports": {
+		"#/*": "./*"
+	},
+	"exports": {
+		".": "./src/index.ts",
+		"./server": "./src/server/index.ts",
+		"./entities": "./src/audit/entities/index.ts",
+		"./contracts": "./src/contracts/index.ts",
+		"./chat": "./src/chat/index.ts",
+		"./providers": "./src/providers/McpServerHandlerDef.ts",
+		"./cli": "./src/cli.ts",
+		"./audit/server": "./src/audit/server/index.ts"
+	},
+	"files": [
+		"dist",
+		"lib",
+		"src",
+		"README.md"
+	],
+	"keywords": [],
+	"scripts": {
+		"build": "bun run ./src/scripts/bundle.ts",
+		"dev": "pnpm tsx --watch src/mcps-cli.ts",
+		"dev:web": "vite",
+		"prepublishOnly": "make build",
+		"test": "vitest run",
+		"typecheck": "tsc --noEmit"
+	},
+	"peerDependencies": {
+		"@mikro-orm/core": "*",
+		"@mikro-orm/decorators": "*",
+		"@mikro-orm/sql": "*",
+		"@wener/server": "*"
+	},
+	"peerDependenciesMeta": {
+		"@mikro-orm/core": {
+			"optional": true
+		},
+		"@mikro-orm/decorators": {
+			"optional": true
+		},
+		"@mikro-orm/sql": {
+			"optional": true
+		},
+		"@wener/server": {
+			"optional": true
+		}
+	},
+	"dependencies": {
+		"emittery": "^1.2.0"
+	},
+	"devDependencies": {
+		"@ai-sdk/mcp": "^1.0.18",
+		"@ai-sdk/openai-compatible": "^2.0.26",
+		"@ai-sdk/react": "^3.0.71",
+		"@base-ui/react": "^1.1.0",
+		"@hono/mcp": "^0.2.3",
+		"@larksuiteoapi/node-sdk": "^1.59.0",
+		"@hono/node-server": "^1.19.9",
+		"@hono/vite-dev-server": "^0.24.1",
+		"@mikro-orm/core": "^7.0.0-rc.2",
+		"@mikro-orm/decorators": "^7.0.0-rc.2",
+		"@mikro-orm/sql": "^7.0.0-rc.2",
+		"@modelcontextprotocol/sdk": "^1.25.3",
+		"@orpc/client": "^1.13.4",
+		"@orpc/contract": "^1.13.4",
+		"@orpc/json-schema": "^1.13.4",
+		"@orpc/openapi": "^1.13.4",
+		"@orpc/server": "^1.13.4",
+		"@orpc/zod": "^1.13.4",
+		"@streamdown/cjk": "^1.0.1",
+		"@streamdown/code": "^1.0.1",
+		"@streamdown/math": "^1.0.1",
+		"@streamdown/mermaid": "^1.0.1",
+		"@wener/ai": "workspace:*",
+		"@wener/client": "workspace:^1.0.26",
+		"@wener/common": "^2.0.3",
+		"@wener/server": "workspace:*",
+		"ai": "^6.0.69",
+		"commander": "^14.0.3",
+		"consola": "^3.4.2",
+		"hono": "^4.11.7",
+		"kysely": "^0.28.11",
+		"lru-cache": "^11.2.5",
+		"lucide-react": "^0.562.0",
+		"mysql2": "^3.16.3",
+		"openai": "^6.17.0",
+		"pg-query-stream": "^4.12.0",
+		"react": "^19.2.4",
+		"react-dom": "^19.2.4",
+		"react-markdown": "^10.1.0",
+		"react-router-dom": "^7.13.0",
+		"std-env": "^3.10.0",
+		"streamdown": "^2.1.0",
+		"tarn": "^3.0.2",
+		"tedious": "^19.2.0",
+		"yaml": "^2.8.2",
+		"zod": "^4.3.6"
+	},
+	"publishConfig": {
+		"exports": {
+			".": {
+				"types": "./src/index.ts",
+				"default": "./lib/index.js"
+			},
+			"./server": {
+				"types": "./src/server/index.ts",
+				"default": "./lib/server/index.js"
+			},
+			"./entities": {
+				"types": "./src/audit/entities/index.ts",
+				"default": "./lib/audit/entities/index.js"
+			},
+			"./contracts": {
+				"types": "./src/contracts/index.ts",
+				"default": "./lib/contracts/index.js"
+			},
+			"./chat": {
+				"types": "./src/chat/index.ts",
+				"default": "./lib/chat/index.js"
+			},
+			"./providers": {
+				"types": "./src/providers/McpServerHandlerDef.ts",
+				"default": "./lib/providers/McpServerHandlerDef.js"
+			},
+			"./cli": {
+				"types": "./src/cli.ts",
+				"default": "./lib/cli.js"
+			},
+			"./audit/server": {
+				"types": "./src/audit/server/index.ts",
+				"default": "./lib/audit/server/index.js"
+			}
+		}
+	}
+}

package/src/{chat/audit.ts → audit/chat.ts}

@@ -4,7 +4,7 @@
  */
 import type { Context } from 'hono';
 import consola from 'consola';
-import type { ChatProtocolType, RequestStatus as RequestStatusType, ChatAuditStats } from '../entities/types';
+import type { ChatProtocolType, RequestStatus as RequestStatusType, ChatAuditStats } from './types';
 
 const log = consola.withTag('chat-audit');
 
@@ -77,7 +77,7 @@ export interface ChatAuditStore {
 	/**
 	 * Get aggregate statistics
 	 */
-	getStats(options: { from?: Date; to?: Date }): Promise<import('../entities/types').ChatAuditStats>;
+	getStats(options: { from?: Date; to?: Date }): Promise<ChatAuditStats>;
 }
 
 export interface ChatAuditQueryOptions {
@@ -93,7 +93,7 @@ export interface ChatAuditQueryOptions {
 }
 
 // Re-export ChatAuditStats from entities
-export type { ChatAuditStats } from '../entities/types';
+export type { ChatAuditStats } from './types';
 
 /**
  * In-memory audit store implementation

package/src/audit/entities/index.ts

@@ -0,0 +1,6 @@
+export * from '../types';
+
+export { ChatRequestEntity, ChatProtocolType, RequestStatus } from './ChatRequestEntity';
+export { McpRequestEntity, McpServerType, McpRequestType } from './McpRequestEntity';
+export { RequestLogEntity } from './RequestLogEntity';
+export { ResponseEntity } from './ResponseEntity';

package/src/audit/server/db.ts

@@ -0,0 +1,65 @@
+import { MikroORM, type Options } from '@mikro-orm/core';
+import { SqliteDriver } from '@mikro-orm/sql';
+import { createSqliteDialect } from '@wener/server/mikro-orm';
+import type { DbConfig } from '../../server/schema';
+import { ChatRequestEntity } from '../entities/ChatRequestEntity';
+import { McpRequestEntity } from '../entities/McpRequestEntity';
+import { RequestLogEntity } from '../entities/RequestLogEntity';
+import { ResponseEntity } from '../entities/ResponseEntity';
+
+export { RequestLogEntity };
+
+let orm: MikroORM<SqliteDriver> | null = null;
+let initPromise: Promise<MikroORM<SqliteDriver>> | null = null;
+
+async function getOrmConfig(dbConfig?: DbConfig): Promise<Options<SqliteDriver>> {
+	const dbPath = dbConfig?.path || '.mcps.db';
+	return {
+		driver: SqliteDriver,
+		dbName: dbPath,
+		entities: [ChatRequestEntity, McpRequestEntity, RequestLogEntity, ResponseEntity],
+		driverOptions: await createSqliteDialect(dbPath),
+		debug: process.env.NODE_ENV === 'development',
+		allowGlobalContext: true,
+	};
+}
+
+export async function ensureDbInitialized(dbConfig?: DbConfig): Promise<MikroORM<SqliteDriver>> {
+	if (orm) return orm;
+	if (initPromise) return initPromise;
+
+	initPromise = (async () => {
+		const config = await getOrmConfig(dbConfig);
+		orm = await MikroORM.init(config);
+		await orm.schema.update();
+		return orm;
+	})();
+
+	try {
+		return await initPromise;
+	} catch (e) {
+		initPromise = null;
+		throw e;
+	}
+}
+
+export function getOrm(): MikroORM<SqliteDriver> {
+	if (!orm) throw new Error('Database not initialized');
+	return orm;
+}
+
+export function getEntityManager() {
+	return getOrm().em;
+}
+
+export async function closeDb(): Promise<void> {
+	if (orm) {
+		await orm.close();
+		orm = null;
+		initPromise = null;
+	}
+}
+
+export function isDbInitialized(): boolean {
+	return orm !== null;
+}

package/src/audit/server/index.ts

@@ -0,0 +1,8 @@
+export {
+	setupAudit,
+	AuditRouter,
+	addAuditEvent,
+	queryAuditEvents,
+	getAuditStats,
+	clearAuditEvents,
+} from './plugin';