mirror of
https://github.com/Memo-2023/mana-monorepo.git
synced 2026-05-15 15:59:40 +02:00
7766ea5021
All 5 milestones landed today in one continuous session: registry, health cache, fallback router, observability, and consumer migration. 115 service-side tests, validator covers 2538 files.
108 lines
3.6 KiB
TypeScript
108 lines
3.6 KiB
TypeScript
/**
|
|
* Sync database connection for MCP tool execution.
|
|
*
|
|
* MCP tools read and write via the mana_sync database (the shared sync
|
|
* event log), not mana_platform. This matches the local-first pattern:
|
|
* writes go to sync_changes, clients pick them up on next sync.
|
|
*
|
|
* The connection uses the same env var as mana-ai: SYNC_DATABASE_URL.
|
|
*/
|
|
|
|
import postgres from 'postgres';
|
|
|
|
const SYNC_DATABASE_URL =
|
|
process.env.SYNC_DATABASE_URL ?? 'postgresql://mana:devpassword@localhost:5432/mana_sync';
|
|
|
|
let syncPool: ReturnType<typeof postgres> | null = null;
|
|
|
|
/** Returns the shared sync database connection pool. */
|
|
export function getSyncConnection() {
|
|
if (!syncPool) {
|
|
syncPool = postgres(SYNC_DATABASE_URL, { max: 5, idle_timeout: 30 });
|
|
}
|
|
return syncPool;
|
|
}
|
|
|
|
export type SyncSql = ReturnType<typeof postgres>;
|
|
|
|
/**
|
|
* Run a callback within an RLS-scoped transaction for the given user.
|
|
* Sets `app.current_user_id` so the sync_changes RLS policy allows
|
|
* reads and writes only for that user's data.
|
|
*/
|
|
export async function withUser<T>(
|
|
userId: string,
|
|
fn: (tx: postgres.TransactionSql<Record<string, never>>) => Promise<T>
|
|
): Promise<T> {
|
|
if (!userId) throw new Error('withUser: empty userId');
|
|
const sql = getSyncConnection();
|
|
return sql.begin(async (tx) => {
|
|
await tx`SELECT set_config('app.current_user_id', ${userId}, true)`;
|
|
return fn(tx);
|
|
}) as Promise<T>;
|
|
}
|
|
|
|
/**
|
|
* Read the latest version of all records for a user + app + table from
|
|
* sync_changes. Applies field-level LWW to reconstruct current state.
|
|
*
|
|
* This is the server-side equivalent of the Dexie liveQuery: it replays
|
|
* the sync_changes log to build the latest record versions. For small
|
|
* datasets this is fine; for large tables we'll need materialized
|
|
* snapshots (like mana-ai's mission_snapshots).
|
|
*/
|
|
export async function readLatestRecords(
|
|
userId: string,
|
|
appId: string,
|
|
tableName: string
|
|
): Promise<Record<string, unknown>[]> {
|
|
const sql = getSyncConnection();
|
|
// Get the latest change per record_id (by created_at desc), then
|
|
// reconstruct the record. Only include non-deleted records.
|
|
const rows = await sql<{ record_id: string; data: Record<string, unknown>; op: string }[]>`
|
|
SELECT DISTINCT ON (record_id)
|
|
record_id, data, op
|
|
FROM sync_changes
|
|
WHERE user_id = ${userId}
|
|
AND app_id = ${appId}
|
|
AND table_name = ${tableName}
|
|
ORDER BY record_id, created_at DESC
|
|
`;
|
|
|
|
// Filter out deleted records and records with delete ops
|
|
return rows
|
|
.filter((r) => r.op !== 'delete' && r.data && !(r.data as Record<string, unknown>).deletedAt)
|
|
.map((r) => r.data);
|
|
}
|
|
|
|
/**
|
|
* Write a new record via sync_changes INSERT. The record will appear
|
|
* on the user's devices on their next sync cycle.
|
|
*
|
|
* MCP-Tool calls always carry `origin='agent'` because the pipeline
|
|
* that produced the value is an AI agent invoking a tool — the
|
|
* actor's `kind` may be `system` (the MCP server itself) but the
|
|
* write semantics are agent-driven for conflict-detection purposes.
|
|
*/
|
|
export async function writeRecord(
|
|
userId: string,
|
|
appId: string,
|
|
tableName: string,
|
|
recordId: string,
|
|
op: 'insert' | 'update' | 'delete',
|
|
data: Record<string, unknown>,
|
|
fieldMeta: Record<string, string>
|
|
): Promise<void> {
|
|
await withUser(userId, async (tx) => {
|
|
await tx`
|
|
INSERT INTO sync_changes
|
|
(app_id, table_name, record_id, user_id, op, data, field_meta, client_id, schema_version, actor, origin)
|
|
VALUES
|
|
(${appId}, ${tableName}, ${recordId}, ${userId}, ${op},
|
|
${tx.json(data as never)}, ${tx.json(fieldMeta as never)},
|
|
'mcp-server', 1,
|
|
${tx.json({ kind: 'system', principalId: 'system:mcp', displayName: 'MCP Server' } as never)},
|
|
'agent')
|
|
`;
|
|
});
|
|
}
|