viyv-db-server 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/LICENSE ADDED
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2025 viyv
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,16 @@
1
+ import type { AuthContext, AuthCredentials } from 'viyv-db-core';
2
+ /** resolveAuth コールバックの戻り値型 */
3
+ export interface ResolvedAuth {
4
+ userId: string;
5
+ accessLevel?: 'read' | 'data' | 'schema';
6
+ orgId?: string;
7
+ teamId?: string;
8
+ roles?: string[];
9
+ }
10
+ /** ResolvedAuth → AuthContext 変換 (pure function, no shared state) */
11
+ export declare function toAuthContext(auth: ResolvedAuth): AuthContext;
12
+ /**
13
+ * Request から Bearer token を抽出。
14
+ */
15
+ export declare function extractCredentials(req: Request): AuthCredentials;
16
+ //# sourceMappingURL=auth-adapter.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"auth-adapter.d.ts","sourceRoot":"","sources":["../src/auth-adapter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAiB,MAAM,cAAc,CAAC;AAEhF,8BAA8B;AAC9B,MAAM,WAAW,YAAY;IAC5B,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,QAAQ,CAAC;IACzC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,qEAAqE;AACrE,wBAAgB,aAAa,CAAC,IAAI,EAAE,YAAY,GAAG,WAAW,CAS7D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,OAAO,GAAG,eAAe,CAMhE"}
@@ -0,0 +1,22 @@
1
+ /** ResolvedAuth → AuthContext 変換 (pure function, no shared state) */
2
+ export function toAuthContext(auth) {
3
+ return {
4
+ userId: auth.userId,
5
+ orgId: auth.orgId,
6
+ teamId: auth.teamId,
7
+ roles: auth.roles ?? [],
8
+ accessLevel: auth.accessLevel,
9
+ claims: {},
10
+ };
11
+ }
12
+ /**
13
+ * Request から Bearer token を抽出。
14
+ */
15
+ export function extractCredentials(req) {
16
+ const header = req.headers.get('authorization');
17
+ if (header?.startsWith('Bearer ')) {
18
+ return { type: 'bearer', token: header.slice(7) };
19
+ }
20
+ return { type: 'none' };
21
+ }
22
+ //# sourceMappingURL=auth-adapter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"auth-adapter.js","sourceRoot":"","sources":["../src/auth-adapter.ts"],"names":[],"mappings":"AAWA,qEAAqE;AACrE,MAAM,UAAU,aAAa,CAAC,IAAkB;IAC/C,OAAO;QACN,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;QACvB,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,MAAM,EAAE,EAAE;KACV,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,GAAY;IAC9C,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IAChD,IAAI,MAAM,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QACnC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IACnD,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;AACzB,CAAC"}
@@ -0,0 +1,3 @@
1
+ #!/usr/bin/env node
2
+ export {};
3
+ //# sourceMappingURL=dev-server-cli.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"dev-server-cli.d.ts","sourceRoot":"","sources":["../src/dev-server-cli.ts"],"names":[],"mappings":""}
@@ -0,0 +1,29 @@
1
+ #!/usr/bin/env node
2
+ import { startDevServer } from './dev-server.js';
3
+ const args = process.argv.slice(2);
4
+ let postgresUrl;
5
+ let port;
6
+ let embeddingEnabled = false;
7
+ let embeddingModel;
8
+ for (let i = 0; i < args.length; i++) {
9
+ if (args[i] === '--postgres-url')
10
+ postgresUrl = args[++i];
11
+ if (args[i] === '--port')
12
+ port = Number(args[++i]);
13
+ if (args[i] === '--embedding')
14
+ embeddingEnabled = true;
15
+ if (args[i] === '--embedding-model')
16
+ embeddingModel = args[++i];
17
+ }
18
+ postgresUrl ??= process.env.VIYV_DB_POSTGRES_URL;
19
+ if (!postgresUrl) {
20
+ process.stderr.write('Usage: dev-server --postgres-url <url> [--port <port>] [--embedding] [--embedding-model <model>]\n');
21
+ process.exit(1);
22
+ }
23
+ let embeddingProvider;
24
+ if (embeddingEnabled) {
25
+ const { TransformersEmbeddingProvider } = await import('viyv-db-embedding');
26
+ embeddingProvider = await TransformersEmbeddingProvider.create(embeddingModel);
27
+ }
28
+ startDevServer({ postgresUrl, port, embeddingProvider });
29
+ //# sourceMappingURL=dev-server-cli.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"dev-server-cli.js","sourceRoot":"","sources":["../src/dev-server-cli.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjD,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;AACnC,IAAI,WAA+B,CAAC;AACpC,IAAI,IAAwB,CAAC;AAC7B,IAAI,gBAAgB,GAAG,KAAK,CAAC;AAC7B,IAAI,cAAkC,CAAC;AAEvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;IACtC,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,gBAAgB;QAAE,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1D,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ;QAAE,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACnD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,aAAa;QAAE,gBAAgB,GAAG,IAAI,CAAC;IACvD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,mBAAmB;QAAE,cAAc,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;AACjE,CAAC;AAED,WAAW,KAAK,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;AAEjD,IAAI,CAAC,WAAW,EAAE,CAAC;IAClB,OAAO,CAAC,MAAM,CAAC,KAAK,CACnB,oGAAoG,CACpG,CAAC;IACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;AACjB,CAAC;AAED,IAAI,iBAAiD,CAAC;AACtD,IAAI,gBAAgB,EAAE,CAAC;IACtB,MAAM,EAAE,6BAA6B,EAAE,GAAG,MAAM,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAC5E,iBAAiB,GAAG,MAAM,6BAA6B,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;AAChF,CAAC;AAED,cAAc,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAC"}
@@ -0,0 +1,7 @@
1
+ import type { IEmbeddingProvider } from 'viyv-db-core';
2
+ export declare function startDevServer(config: {
3
+ postgresUrl: string;
4
+ port?: number;
5
+ embeddingProvider?: IEmbeddingProvider;
6
+ }): void;
7
+ //# sourceMappingURL=dev-server.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"dev-server.d.ts","sourceRoot":"","sources":["../src/dev-server.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AA2BvD,wBAAgB,cAAc,CAAC,MAAM,EAAE;IACtC,WAAW,EAAE,MAAM,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,CAAC,EAAE,kBAAkB,CAAC;CACvC,GAAG,IAAI,CA0DP"}
@@ -0,0 +1,76 @@
1
+ import { createServer } from 'node:http';
2
+ import { createViyvHandler } from './index.js';
3
+ function readBody(req) {
4
+ return new Promise((resolve, reject) => {
5
+ const chunks = [];
6
+ req.on('data', (chunk) => chunks.push(chunk));
7
+ req.on('end', () => resolve(Buffer.concat(chunks).toString()));
8
+ req.on('error', reject);
9
+ });
10
+ }
11
+ function toWebRequest(req, body) {
12
+ const url = new URL(req.url ?? '/', `http://${req.headers.host ?? 'localhost'}`);
13
+ const headers = new Headers();
14
+ for (const [key, val] of Object.entries(req.headers)) {
15
+ if (val) {
16
+ headers.set(key, Array.isArray(val) ? val.join(', ') : val);
17
+ }
18
+ }
19
+ return new Request(url.toString(), {
20
+ method: req.method,
21
+ headers,
22
+ body: req.method !== 'GET' && req.method !== 'HEAD' ? body : undefined,
23
+ });
24
+ }
25
+ export function startDevServer(config) {
26
+ const { handler, close } = createViyvHandler({
27
+ postgresUrl: config.postgresUrl,
28
+ resolveAuth: async () => ({ userId: 'dev', accessLevel: 'schema' }),
29
+ embeddingProvider: config.embeddingProvider,
30
+ });
31
+ const port = config.port ?? 3001;
32
+ const server = createServer(async (req, res) => {
33
+ // CORS
34
+ res.setHeader('Access-Control-Allow-Origin', '*');
35
+ res.setHeader('Access-Control-Allow-Methods', 'POST, OPTIONS');
36
+ res.setHeader('Access-Control-Allow-Headers', 'Content-Type, Authorization');
37
+ if (req.method === 'OPTIONS') {
38
+ res.writeHead(204);
39
+ res.end();
40
+ return;
41
+ }
42
+ if (req.method !== 'POST') {
43
+ res.writeHead(405, { 'Content-Type': 'application/json' });
44
+ res.end(JSON.stringify({ error: 'Method not allowed' }));
45
+ return;
46
+ }
47
+ try {
48
+ const body = await readBody(req);
49
+ const webReq = toWebRequest(req, body);
50
+ const webRes = await handler(webReq);
51
+ const resHeaders = {};
52
+ webRes.headers.forEach((value, key) => {
53
+ resHeaders[key] = value;
54
+ });
55
+ res.writeHead(webRes.status, resHeaders);
56
+ res.end(await webRes.text());
57
+ }
58
+ catch (error) {
59
+ res.writeHead(500, { 'Content-Type': 'application/json' });
60
+ res.end(JSON.stringify({ error: error instanceof Error ? error.message : String(error) }));
61
+ }
62
+ });
63
+ server.listen(port, () => {
64
+ process.stderr.write(`viyv-db dev server running on http://localhost:${port}\n`);
65
+ });
66
+ const shutdown = () => {
67
+ server.close(() => {
68
+ close()
69
+ .catch(() => { })
70
+ .finally(() => process.exit(0));
71
+ });
72
+ };
73
+ process.on('SIGINT', shutdown);
74
+ process.on('SIGTERM', shutdown);
75
+ }
76
+ //# sourceMappingURL=dev-server.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"dev-server.js","sourceRoot":"","sources":["../src/dev-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAGzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C,SAAS,QAAQ,CAAC,GAAoB;IACrC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACtC,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtD,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAC/D,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IACzB,CAAC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,GAAoB,EAAE,IAAY;IACvD,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,GAAG,EAAE,UAAU,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,WAAW,EAAE,CAAC,CAAC;IACjF,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;IAC9B,KAAK,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACtD,IAAI,GAAG,EAAE,CAAC;YACT,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC7D,CAAC;IACF,CAAC;IACD,OAAO,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAE;QAClC,MAAM,EAAE,GAAG,CAAC,MAAM;QAClB,OAAO;QACP,IAAI,EAAE,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;KACtE,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,MAI9B;IACA,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,iBAAiB,CAAC;QAC5C,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,WAAW,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,QAAiB,EAAE,CAAC;QAC5E,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;KAC3C,CAAC,CAAC;IAEH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC;IAEjC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;QAC9C,OAAO;QACP,GAAG,CAAC,SAAS,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAC;QAClD,GAAG,CAAC,SAAS,CAAC,8BAA8B,EAAE,eAAe,CAAC,CAAC;QAC/D,GAAG,CAAC,SAAS,CAAC,8BAA8B,EAAE,6BAA6B,CAAC,CAAC;QAE7E,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9B,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACnB,GAAG,CAAC,GAAG,EAAE,CAAC;YACV,OAAO;QACR,CAAC;QAED,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC3B,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;YACzD,OAAO;QACR,CAAC;QAED,IAAI,CAAC;YACJ,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;YACjC,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;YACvC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,MAAM,CAAC,CAAC;YAErC,MAAM,UAAU,GAA2B,EAAE,CAAC;YAC9C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;gBACrC,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;YACzB,CAAC,CAAC,CAAC;YAEH,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACzC,GAAG,CAAC,GAAG,CAAC,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;QACxB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kDAAkD,IAAI,IAAI,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,GAAG,EAAE;QACrB,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;YACjB,KAAK,EAAE;iBACL,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;iBACf,OAAO,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACJ,CAAC,CAAC;IACF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/B,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;AACjC,CAAC"}
@@ -0,0 +1,6 @@
1
+ import type { RpcErrorBody } from 'viyv-db-core/rpc';
2
+ /**
3
+ * Serializes any error into an RPC error body.
4
+ */
5
+ export declare function serializeError(error: unknown): RpcErrorBody;
6
+ //# sourceMappingURL=error-serializer.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"error-serializer.d.ts","sourceRoot":"","sources":["../src/error-serializer.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAErD;;GAEG;AACH,wBAAgB,cAAc,CAAC,KAAK,EAAE,OAAO,GAAG,YAAY,CAsB3D"}
@@ -0,0 +1,26 @@
1
+ import { ViyvDbError } from 'viyv-db-core';
2
+ /**
3
+ * Serializes any error into an RPC error body.
4
+ */
5
+ export function serializeError(error) {
6
+ if (error instanceof ViyvDbError) {
7
+ return {
8
+ code: error.code,
9
+ message: error.message,
10
+ name: error.name,
11
+ };
12
+ }
13
+ if (error instanceof Error) {
14
+ return {
15
+ code: 'INTERNAL_ERROR',
16
+ message: error.message,
17
+ name: error.name,
18
+ };
19
+ }
20
+ return {
21
+ code: 'INTERNAL_ERROR',
22
+ message: String(error),
23
+ name: 'Error',
24
+ };
25
+ }
26
+ //# sourceMappingURL=error-serializer.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"error-serializer.js","sourceRoot":"","sources":["../src/error-serializer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAG3C;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,KAAc;IAC5C,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;QAClC,OAAO;YACN,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,KAAK,CAAC,IAAI;SAChB,CAAC;IACH,CAAC;IAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC5B,OAAO;YACN,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,IAAI,EAAE,KAAK,CAAC,IAAI;SAChB,CAAC;IACH,CAAC;IAED,OAAO;QACN,IAAI,EAAE,gBAAgB;QACtB,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC;QACtB,IAAI,EAAE,OAAO;KACb,CAAC;AACH,CAAC"}
@@ -0,0 +1,7 @@
1
+ import type { IDatabaseService } from 'viyv-db-core';
2
+ import type { RpcRequest, RpcResponse } from 'viyv-db-core/rpc';
3
+ /**
4
+ * Dispatches an RPC request to the appropriate IDatabaseService method.
5
+ */
6
+ export declare function handleRpc(service: IDatabaseService, request: RpcRequest): Promise<RpcResponse>;
7
+ //# sourceMappingURL=handler.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"handler.d.ts","sourceRoot":"","sources":["../src/handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAErD,OAAO,KAAK,EAAa,UAAU,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAI3E;;GAEG;AACH,wBAAsB,SAAS,CAC9B,OAAO,EAAE,gBAAgB,EACzB,OAAO,EAAE,UAAU,GACjB,OAAO,CAAC,WAAW,CAAC,CAkBtB"}
@@ -0,0 +1,66 @@
1
+ import { SemanticSearchNotEnabledError } from 'viyv-db-core';
2
+ import { RPC_VERSION } from 'viyv-db-core/rpc';
3
+ import { serializeError } from './error-serializer.js';
4
+ /**
5
+ * Dispatches an RPC request to the appropriate IDatabaseService method.
6
+ */
7
+ export async function handleRpc(service, request) {
8
+ try {
9
+ if (request.version !== RPC_VERSION) {
10
+ return {
11
+ ok: false,
12
+ error: {
13
+ code: 'VERSION_MISMATCH',
14
+ message: `Expected RPC version "${RPC_VERSION}", got "${request.version}"`,
15
+ name: 'VersionError',
16
+ },
17
+ };
18
+ }
19
+ const result = await dispatch(service, request.method, request.params);
20
+ return { ok: true, result };
21
+ }
22
+ catch (error) {
23
+ return { ok: false, error: serializeError(error) };
24
+ }
25
+ }
26
+ // biome-ignore lint/suspicious/noExplicitAny: RPC dispatch requires flexible types
27
+ async function dispatch(service, method, params) {
28
+ switch (method) {
29
+ case 'getSchema':
30
+ return service.getSchema(params.mode, params.options);
31
+ case 'createTable':
32
+ return service.createTable(params.def, params.reason, params.appliedBy);
33
+ case 'alterTable':
34
+ return service.alterTable(params.table, params.action, params.reason, params.appliedBy);
35
+ case 'dropTable':
36
+ return service.dropTable(params.table, params.reason, params.appliedBy);
37
+ case 'insert':
38
+ return service.insert(params.table, params.rows);
39
+ case 'update':
40
+ return service.update(params.table, params.set, params.where);
41
+ case 'delete':
42
+ return service.delete(params.table, params.where);
43
+ case 'upsert':
44
+ return service.upsert(params.table, params.rows, params.conflictColumns);
45
+ case 'query':
46
+ return service.query(params.sql, params.params);
47
+ case 'executeSql':
48
+ return service.executeSql(params.sql, params.params, params.accessLevel);
49
+ case 'queryTable':
50
+ return service.queryTable(params.table, params.options);
51
+ case 'getMigrationInfo':
52
+ return service.getMigrationInfo();
53
+ case 'rollbackMigration':
54
+ return service.rollbackMigration(params.targetVersion);
55
+ case 'semanticSearch':
56
+ if (!service.semanticSearch) {
57
+ throw new SemanticSearchNotEnabledError();
58
+ }
59
+ return service.semanticSearch(params.options);
60
+ case 'importCsv':
61
+ throw new Error('importCsv is not supported via RPC');
62
+ default:
63
+ throw new Error(`Unknown RPC method: ${method}`);
64
+ }
65
+ }
66
+ //# sourceMappingURL=handler.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"handler.js","sourceRoot":"","sources":["../src/handler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,6BAA6B,EAAE,MAAM,cAAc,CAAC;AAE7D,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAEvD;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,SAAS,CAC9B,OAAyB,EACzB,OAAmB;IAEnB,IAAI,CAAC;QACJ,IAAI,OAAO,CAAC,OAAO,KAAK,WAAW,EAAE,CAAC;YACrC,OAAO;gBACN,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE;oBACN,IAAI,EAAE,kBAAkB;oBACxB,OAAO,EAAE,yBAAyB,WAAW,WAAW,OAAO,CAAC,OAAO,GAAG;oBAC1E,IAAI,EAAE,cAAc;iBACpB;aACD,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;QACvE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;IAC7B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;IACpD,CAAC;AACF,CAAC;AAED,mFAAmF;AACnF,KAAK,UAAU,QAAQ,CAAC,OAAyB,EAAE,MAAiB,EAAE,MAAW;IAChF,QAAQ,MAAM,EAAE,CAAC;QAChB,KAAK,WAAW;YACf,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACvD,KAAK,aAAa;YACjB,OAAO,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QACzE,KAAK,YAAY;YAChB,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QACzF,KAAK,WAAW;YACf,OAAO,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;QACzE,KAAK,QAAQ;YACZ,OAAO,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;QAClD,KAAK,QAAQ;YACZ,OAAO,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/D,KAAK,QAAQ;YACZ,OAAO,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QACnD,KAAK,QAAQ;YACZ,OAAO,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC;QAC1E,KAAK,OAAO;YACX,OAAO,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QACjD,KAAK,YAAY;YAChB,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAC1E,KAAK,YAAY;YAChB,OAAO,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACzD,KAAK,kBAAkB;YACtB,OAAO,OAAO,CAAC,gBAAgB,EAAE,CAAC;QACnC,KAAK,mBAAmB;YACvB,OAAO,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACxD,KAAK,gBAAgB;YACpB,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;gBAC7B,MAAM,IAAI,6BAA6B,EAAE,CAAC;YAC3C,CAAC;YACD,OAAO,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/C,KAAK,WAAW;YACf,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACvD;YACC,MAAM,IAAI,KAAK,CAAC,uBAAuB,MAAM,EAAE,CAAC,CAAC;IACnD,CAAC;AACF,CAAC"}
@@ -0,0 +1,33 @@
1
+ import type { IAccessPolicy, IAuthProvider, IEmbeddingProvider, ITenantResolver } from 'viyv-db-core';
2
+ import type { ResolvedAuth } from './auth-adapter.js';
3
+ export { TenantPool } from './tenant-pool.js';
4
+ export type { TenantPoolConfig } from './tenant-pool.js';
5
+ export { handleRpc } from './handler.js';
6
+ export { serializeError } from './error-serializer.js';
7
+ export { startDevServer } from './dev-server.js';
8
+ export { extractCredentials, toAuthContext } from './auth-adapter.js';
9
+ export type { ResolvedAuth } from './auth-adapter.js';
10
+ export interface ViyvHandlerConfig {
11
+ postgresUrl: string;
12
+ resolveAuth: (req: Request) => Promise<ResolvedAuth>;
13
+ rowLimit?: number;
14
+ queryTimeoutMs?: number;
15
+ maxTenants?: number;
16
+ maxIdleMs?: number;
17
+ authProvider?: IAuthProvider;
18
+ tenantResolver?: ITenantResolver;
19
+ accessPolicy?: IAccessPolicy;
20
+ embeddingProvider?: IEmbeddingProvider;
21
+ }
22
+ export interface ViyvHandlerResult {
23
+ handler: (req: Request) => Promise<Response>;
24
+ close: () => Promise<void>;
25
+ }
26
+ /**
27
+ * Creates a Web API standard handler (Request → Response) for viyv-db RPC.
28
+ * Can be used as a Next.js Route Handler, Cloudflare Worker, or any Web-standard runtime.
29
+ *
30
+ * Returns `{ handler, close }` — call `close()` for graceful shutdown.
31
+ */
32
+ export declare function createViyvHandler(config: ViyvHandlerConfig): ViyvHandlerResult;
33
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAIX,aAAa,EACb,aAAa,EACb,kBAAkB,EAClB,eAAe,EACf,MAAM,cAAc,CAAC;AAStB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAItD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,YAAY,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACzD,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AACtE,YAAY,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAqBtD,MAAM,WAAW,iBAAiB;IACjC,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;IACrD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,cAAc,CAAC,EAAE,eAAe,CAAC;IACjC,YAAY,CAAC,EAAE,aAAa,CAAC;IAC7B,iBAAiB,CAAC,EAAE,kBAAkB,CAAC;CACvC;AAED,MAAM,WAAW,iBAAiB;IACjC,OAAO,EAAE,CAAC,GAAG,EAAE,OAAO,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC7C,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B;AAMD;;;;;GAKG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,iBAAiB,GAAG,iBAAiB,CAwE9E"}
package/dist/index.js ADDED
@@ -0,0 +1,104 @@
1
+ import { AuthenticationError, AuthorizationError, StaticAccessPolicy, UserTenantResolver, } from 'viyv-db-core';
2
+ import { extractCredentials, toAuthContext } from './auth-adapter.js';
3
+ import { handleRpc } from './handler.js';
4
+ import { TenantPool } from './tenant-pool.js';
5
+ export { TenantPool } from './tenant-pool.js';
6
+ export { handleRpc } from './handler.js';
7
+ export { serializeError } from './error-serializer.js';
8
+ export { startDevServer } from './dev-server.js';
9
+ export { extractCredentials, toAuthContext } from './auth-adapter.js';
10
+ /** RPC method → required access level mapping */
11
+ const METHOD_ACCESS = {
12
+ getSchema: 'read',
13
+ query: 'read',
14
+ queryTable: 'read',
15
+ getMigrationInfo: 'read',
16
+ executeSql: 'data',
17
+ insert: 'data',
18
+ update: 'data',
19
+ delete: 'data',
20
+ upsert: 'data',
21
+ createTable: 'schema',
22
+ alterTable: 'schema',
23
+ dropTable: 'schema',
24
+ rollbackMigration: 'schema',
25
+ importCsv: 'schema',
26
+ semanticSearch: 'read',
27
+ };
28
+ function errorResponse(code, message, name, status) {
29
+ return Response.json({ ok: false, error: { code, message, name } }, { status });
30
+ }
31
+ /**
32
+ * Creates a Web API standard handler (Request → Response) for viyv-db RPC.
33
+ * Can be used as a Next.js Route Handler, Cloudflare Worker, or any Web-standard runtime.
34
+ *
35
+ * Returns `{ handler, close }` — call `close()` for graceful shutdown.
36
+ */
37
+ export function createViyvHandler(config) {
38
+ const pool = new TenantPool({
39
+ postgresUrl: config.postgresUrl,
40
+ rowLimit: config.rowLimit,
41
+ queryTimeoutMs: config.queryTimeoutMs,
42
+ maxTenants: config.maxTenants,
43
+ maxIdleMs: config.maxIdleMs,
44
+ embeddingProvider: config.embeddingProvider,
45
+ });
46
+ pool.startCleanup();
47
+ const tenantResolver = config.tenantResolver ?? new UserTenantResolver();
48
+ const accessPolicy = config.accessPolicy ?? new StaticAccessPolicy('schema');
49
+ const handler = async (req) => {
50
+ try {
51
+ // 1. Resolve auth via callback or custom provider
52
+ let authContext;
53
+ if (config.authProvider) {
54
+ const credentials = extractCredentials(req);
55
+ authContext = await config.authProvider.authenticate(credentials);
56
+ }
57
+ else {
58
+ let resolvedAuth;
59
+ try {
60
+ resolvedAuth = await config.resolveAuth(req);
61
+ }
62
+ catch {
63
+ return errorResponse('AUTHENTICATION_FAILED', 'Unauthorized', 'AuthenticationError', 401);
64
+ }
65
+ authContext = toAuthContext(resolvedAuth);
66
+ }
67
+ // 2. Resolve tenant
68
+ const tenantId = tenantResolver.resolve(authContext);
69
+ // 3. Parse RPC request
70
+ let body;
71
+ try {
72
+ body = (await req.json());
73
+ }
74
+ catch {
75
+ return errorResponse('INVALID_REQUEST', 'Invalid JSON body', 'Error', 400);
76
+ }
77
+ // 4. Access policy check
78
+ const requiredAccess = METHOD_ACCESS[body.method] ?? 'schema';
79
+ const decision = accessPolicy.check(authContext, {
80
+ name: body.method,
81
+ requiredAccess,
82
+ });
83
+ if (!decision.allowed) {
84
+ return errorResponse('AUTHORIZATION_DENIED', decision.reason, 'AuthorizationError', 403);
85
+ }
86
+ // 5. Get tenant service and dispatch
87
+ const service = await pool.forTenant(tenantId);
88
+ const result = await handleRpc(service, body);
89
+ return Response.json(result, { status: result.ok ? 200 : 400 });
90
+ }
91
+ catch (error) {
92
+ if (error instanceof AuthenticationError) {
93
+ return errorResponse('AUTHENTICATION_FAILED', error.message, 'AuthenticationError', 401);
94
+ }
95
+ if (error instanceof AuthorizationError) {
96
+ return errorResponse('AUTHORIZATION_DENIED', error.message, 'AuthorizationError', 403);
97
+ }
98
+ const message = error instanceof Error ? error.message : String(error);
99
+ return errorResponse('INTERNAL_ERROR', message, 'Error', 500);
100
+ }
101
+ };
102
+ return { handler, close: () => pool.close() };
103
+ }
104
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AASA,OAAO,EACN,mBAAmB,EACnB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,GAClB,MAAM,cAAc,CAAC;AAEtB,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEtE,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AACzC,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGtE,iDAAiD;AACjD,MAAM,aAAa,GAAgC;IAClD,SAAS,EAAE,MAAM;IACjB,KAAK,EAAE,MAAM;IACb,UAAU,EAAE,MAAM;IAClB,gBAAgB,EAAE,MAAM;IACxB,UAAU,EAAE,MAAM;IAClB,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,MAAM;IACd,MAAM,EAAE,MAAM;IACd,WAAW,EAAE,QAAQ;IACrB,UAAU,EAAE,QAAQ;IACpB,SAAS,EAAE,QAAQ;IACnB,iBAAiB,EAAE,QAAQ;IAC3B,SAAS,EAAE,QAAQ;IACnB,cAAc,EAAE,MAAM;CACtB,CAAC;AAoBF,SAAS,aAAa,CAAC,IAAY,EAAE,OAAe,EAAE,IAAY,EAAE,MAAc;IACjF,OAAO,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AACjF,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAyB;IAC1D,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC;QAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,cAAc,EAAE,MAAM,CAAC,cAAc;QACrC,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;KAC3C,CAAC,CAAC;IAEH,IAAI,CAAC,YAAY,EAAE,CAAC;IAEpB,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,IAAI,kBAAkB,EAAE,CAAC;IACzE,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,IAAI,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IAE7E,MAAM,OAAO,GAAG,KAAK,EAAE,GAAY,EAAqB,EAAE;QACzD,IAAI,CAAC;YACJ,kDAAkD;YAClD,IAAI,WAAwB,CAAC;YAC7B,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;gBACzB,MAAM,WAAW,GAAG,kBAAkB,CAAC,GAAG,CAAC,CAAC;gBAC5C,WAAW,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;YACnE,CAAC;iBAAM,CAAC;gBACP,IAAI,YAA0B,CAAC;gBAC/B,IAAI,CAAC;oBACJ,YAAY,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;gBAC9C,CAAC;gBAAC,MAAM,CAAC;oBACR,OAAO,aAAa,CAAC,uBAAuB,EAAE,cAAc,EAAE,qBAAqB,EAAE,GAAG,CAAC,CAAC;gBAC3F,CAAC;gBACD,WAAW,GAAG,aAAa,CAAC,YAAY,CAAC,CAAC;YAC3C,CAAC;YAED,oBAAoB;YACpB,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YAErD,uBAAuB;YACvB,IAAI,IAAgB,CAAC;YACrB,IAAI,CAAC;gBACJ,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAe,CAAC;YACzC,CAAC;YAAC,MAAM,CAAC;gBACR,OAAO,aAAa,CAAC,iBAAiB,EAAE,mBAAmB,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;YAC5E,CAAC;YAED,yBAAyB;YACzB,MAAM,cAAc,GAAG,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC;YAC9D,MAAM,QAAQ,GAAmB,YAAY,CAAC,KAAK,CAAC,WAAW,EAAE;gBAChE,IAAI,EAAE,IAAI,CAAC,MAAM;gBACjB,cAAc;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACvB,OAAO,aAAa,CAAC,sBAAsB,EAAE,QAAQ,CAAC,MAAM,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC;YAC1F,CAAC;YAED,qCAAqC;YACrC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAE9C,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,mBAAmB,EAAE,CAAC;gBAC1C,OAAO,aAAa,CAAC,uBAAuB,EAAE,KAAK,CAAC,OAAO,EAAE,qBAAqB,EAAE,GAAG,CAAC,CAAC;YAC1F,CAAC;YACD,IAAI,KAAK,YAAY,kBAAkB,EAAE,CAAC;gBACzC,OAAO,aAAa,CAAC,sBAAsB,EAAE,KAAK,CAAC,OAAO,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC;YACxF,CAAC;YACD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO,aAAa,CAAC,gBAAgB,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;QAC/D,CAAC;IACF,CAAC,CAAC;IAEF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;AAC/C,CAAC"}
@@ -0,0 +1,25 @@
1
+ import type { IDatabaseService, IDatabaseServiceFactory, IEmbeddingProvider } from 'viyv-db-core';
2
+ export interface TenantPoolConfig {
3
+ postgresUrl: string;
4
+ rowLimit?: number;
5
+ queryTimeoutMs?: number;
6
+ maxTenants?: number;
7
+ maxIdleMs?: number;
8
+ embeddingProvider?: IEmbeddingProvider;
9
+ }
10
+ export declare class TenantPool implements IDatabaseServiceFactory {
11
+ private readonly config;
12
+ private readonly cache;
13
+ private readonly pending;
14
+ private readonly maxTenants;
15
+ private readonly maxIdleMs;
16
+ private cleanupTimer;
17
+ constructor(config: TenantPoolConfig);
18
+ forTenant(tenantId: string): Promise<IDatabaseService>;
19
+ private createService;
20
+ close(): Promise<void>;
21
+ startCleanup(intervalMs?: number): void;
22
+ private evictLru;
23
+ private evictIdle;
24
+ }
25
+ //# sourceMappingURL=tenant-pool.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tenant-pool.d.ts","sourceRoot":"","sources":["../src/tenant-pool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AASlG,MAAM,WAAW,gBAAgB;IAChC,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,iBAAiB,CAAC,EAAE,kBAAkB,CAAC;CACvC;AAKD,qBAAa,UAAW,YAAW,uBAAuB;IAO7C,OAAO,CAAC,QAAQ,CAAC,MAAM;IANnC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAgC;IACtD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAgD;IACxE,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,YAAY,CAA+C;gBAEtC,MAAM,EAAE,gBAAgB;IAK/C,SAAS,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC;YAwB9C,aAAa;IA4BrB,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAW5B,YAAY,CAAC,UAAU,SAAS,GAAG,IAAI;YAUzB,QAAQ;YAcR,SAAS;CAavB"}
@@ -0,0 +1,103 @@
1
+ import { createPostgresService } from 'viyv-db-postgres';
2
+ const DEFAULT_MAX_TENANTS = 50;
3
+ const DEFAULT_MAX_IDLE_MS = 10 * 60 * 1000; // 10 minutes
4
+ export class TenantPool {
5
+ config;
6
+ cache = new Map();
7
+ pending = new Map();
8
+ maxTenants;
9
+ maxIdleMs;
10
+ cleanupTimer = null;
11
+ constructor(config) {
12
+ this.config = config;
13
+ this.maxTenants = config.maxTenants ?? DEFAULT_MAX_TENANTS;
14
+ this.maxIdleMs = config.maxIdleMs ?? DEFAULT_MAX_IDLE_MS;
15
+ }
16
+ async forTenant(tenantId) {
17
+ // Cache hit
18
+ const existing = this.cache.get(tenantId);
19
+ if (existing) {
20
+ existing.lastUsedAt = Date.now();
21
+ return existing.service;
22
+ }
23
+ // Deduplicate concurrent creation for the same tenantId
24
+ const inflight = this.pending.get(tenantId);
25
+ if (inflight) {
26
+ return inflight;
27
+ }
28
+ const promise = this.createService(tenantId);
29
+ this.pending.set(tenantId, promise);
30
+ try {
31
+ return await promise;
32
+ }
33
+ finally {
34
+ this.pending.delete(tenantId);
35
+ }
36
+ }
37
+ async createService(tenantId) {
38
+ // Evict LRU if at capacity
39
+ if (this.cache.size >= this.maxTenants) {
40
+ await this.evictLru();
41
+ }
42
+ const pgConfig = {
43
+ postgresUrl: this.config.postgresUrl,
44
+ tenantId,
45
+ accessLevel: 'schema',
46
+ rowLimit: this.config.rowLimit ?? 1000,
47
+ queryTimeoutMs: this.config.queryTimeoutMs ?? 30_000,
48
+ autoInitialize: true,
49
+ poolMin: 1,
50
+ poolMax: 3,
51
+ insertBatchSize: 2000,
52
+ };
53
+ const service = await createPostgresService(pgConfig, this.config.embeddingProvider);
54
+ this.cache.set(tenantId, {
55
+ service,
56
+ lastUsedAt: Date.now(),
57
+ });
58
+ return service;
59
+ }
60
+ async close() {
61
+ if (this.cleanupTimer) {
62
+ clearInterval(this.cleanupTimer);
63
+ this.cleanupTimer = null;
64
+ }
65
+ const closing = Array.from(this.cache.values()).map((entry) => entry.service.close());
66
+ this.cache.clear();
67
+ await Promise.all(closing);
68
+ }
69
+ startCleanup(intervalMs = 60_000) {
70
+ this.cleanupTimer = setInterval(() => {
71
+ void this.evictIdle();
72
+ }, intervalMs);
73
+ // Don't block process exit
74
+ if (this.cleanupTimer.unref) {
75
+ this.cleanupTimer.unref();
76
+ }
77
+ }
78
+ async evictLru() {
79
+ let oldest = null;
80
+ for (const [key, entry] of this.cache) {
81
+ if (!oldest || entry.lastUsedAt < oldest.time) {
82
+ oldest = { key, time: entry.lastUsedAt };
83
+ }
84
+ }
85
+ if (oldest) {
86
+ const entry = this.cache.get(oldest.key);
87
+ this.cache.delete(oldest.key);
88
+ await entry?.service.close();
89
+ }
90
+ }
91
+ async evictIdle() {
92
+ const now = Date.now();
93
+ const closing = [];
94
+ for (const [key, entry] of this.cache) {
95
+ if (now - entry.lastUsedAt > this.maxIdleMs) {
96
+ this.cache.delete(key);
97
+ closing.push(entry.service.close());
98
+ }
99
+ }
100
+ await Promise.all(closing);
101
+ }
102
+ }
103
+ //# sourceMappingURL=tenant-pool.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"tenant-pool.js","sourceRoot":"","sources":["../src/tenant-pool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AAiBzD,MAAM,mBAAmB,GAAG,EAAE,CAAC;AAC/B,MAAM,mBAAmB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAEzD,MAAM,OAAO,UAAU;IAOO;IANZ,KAAK,GAAG,IAAI,GAAG,EAAqB,CAAC;IACrC,OAAO,GAAG,IAAI,GAAG,EAAqC,CAAC;IACvD,UAAU,CAAS;IACnB,SAAS,CAAS;IAC3B,YAAY,GAA0C,IAAI,CAAC;IAEnE,YAA6B,MAAwB;QAAxB,WAAM,GAAN,MAAM,CAAkB;QACpD,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,mBAAmB,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,mBAAmB,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,QAAgB;QAC/B,YAAY;QACZ,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,QAAQ,EAAE,CAAC;YACd,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,OAAO,QAAQ,CAAC,OAAO,CAAC;QACzB,CAAC;QAED,wDAAwD;QACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,QAAQ,EAAE,CAAC;YACd,OAAO,QAAQ,CAAC;QACjB,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEpC,IAAI,CAAC;YACJ,OAAO,MAAM,OAAO,CAAC;QACtB,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC/B,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,QAAgB;QAC3C,2BAA2B;QAC3B,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACvB,CAAC;QAED,MAAM,QAAQ,GAAmB;YAChC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,QAAQ;YACR,WAAW,EAAE,QAAQ;YACrB,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI;YACtC,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,IAAI,MAAM;YACpD,cAAc,EAAE,IAAI;YACpB,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,CAAC;YACV,eAAe,EAAE,IAAI;SACrB,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,qBAAqB,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC;QAErF,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE;YACxB,OAAO;YACP,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,KAAK;QACV,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC1B,CAAC;QAED,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QACtF,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IAED,YAAY,CAAC,UAAU,GAAG,MAAM;QAC/B,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE;YACpC,KAAK,IAAI,CAAC,SAAS,EAAE,CAAC;QACvB,CAAC,EAAE,UAAU,CAAC,CAAC;QACf,2BAA2B;QAC3B,IAAI,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC7B,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,QAAQ;QACrB,IAAI,MAAM,GAAyC,IAAI,CAAC;QACxD,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACvC,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC/C,MAAM,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,CAAC;YAC1C,CAAC;QACF,CAAC;QACD,IAAI,MAAM,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACzC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC9B,MAAM,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC;QAC9B,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,SAAS;QACtB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAoB,EAAE,CAAC;QAEpC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACvC,IAAI,GAAG,GAAG,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC7C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACvB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YACrC,CAAC;QACF,CAAC;QAED,MAAM,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;CACD"}
package/package.json ADDED
@@ -0,0 +1,39 @@
1
+ {
2
+ "name": "viyv-db-server",
3
+ "version": "0.1.0",
4
+ "type": "module",
5
+ "description": "Server library for hosting viyv-db as an HTTP endpoint",
6
+ "license": "MIT",
7
+ "author": "viyv",
8
+ "engines": {
9
+ "node": ">=20"
10
+ },
11
+ "main": "dist/index.js",
12
+ "types": "dist/index.d.ts",
13
+ "exports": {
14
+ ".": {
15
+ "types": "./dist/index.d.ts",
16
+ "default": "./dist/index.js"
17
+ }
18
+ },
19
+ "files": [
20
+ "dist",
21
+ "!dist/__tests__"
22
+ ],
23
+ "dependencies": {
24
+ "zod": "^3.23.0",
25
+ "viyv-db-core": "0.1.0",
26
+ "viyv-db-postgres": "0.1.0"
27
+ },
28
+ "devDependencies": {
29
+ "typescript": "^5.7.0",
30
+ "vitest": "^2.1.0",
31
+ "viyv-db-embedding": "0.1.0"
32
+ },
33
+ "scripts": {
34
+ "build": "tsc",
35
+ "typecheck": "tsc --noEmit",
36
+ "test": "vitest run",
37
+ "clean": "rm -rf dist"
38
+ }
39
+ }