Add a dynamic resource for memories (#1)

Author: murrayju

src/httpServer.ts

@@ -4,6 +4,7 @@ import { apiFactories } from './apis/index.js';
 import { context, serverInfo } from './serverInfo.js';
 import { log } from './shared/boilerplate/src/logger.js';
 import { runMigrations } from './migrate.js';
+import { additionalSetup } from './resources.js';
 
 try {
   log.info('Running database migrations...');
@@ -18,4 +19,5 @@ export const { registerCleanupFn } = httpServerFactory({
   ...serverInfo,
   context,
   apiFactories,
+  additionalSetup,
 });

src/resources.ts

@@ -0,0 +1,41 @@
+import {
+  McpServer,
+  ResourceTemplate,
+} from '@modelcontextprotocol/sdk/server/mcp.js';
+import { AdditionalSetupArgs } from './shared/boilerplate/src/mcpServer.js';
+import { Memory, ServerContext } from './types.js';
+
+export const additionalSetup = ({
+  context: { pgPool, schema },
+  server,
+}: AdditionalSetupArgs<ServerContext>) => {
+  server.registerResource(
+    'memories',
+    new ResourceTemplate('memory://{key}', { list: undefined }),
+    {
+      title: 'Memories',
+      description: 'A collection of memories',
+    },
+    async (uri, { key }) => {
+      const result = await pgPool.query<Memory>(
+        /* sql */ `
+SELECT id, content, created_at, updated_at
+FROM ${schema}.memory
+WHERE key = $1 AND deleted_at IS NULL
+`,
+        [key],
+      );
+
+      return {
+        contents: [
+          {
+            uri: uri.href,
+            text: `Memories: key=\`${key}\`, count=${result.rows.length}
+${result.rows.map((m) => `- (${m.id}) ${m.content}`).join('\n')}
+`,
+          },
+        ],
+      };
+    },
+  );
+};

src/shared/boilerplate

@@ -2,9 +2,11 @@
 import { stdioServerFactory } from './shared/boilerplate/src/stdio.js';
 import { apiFactories } from './apis/index.js';
 import { context, serverInfo } from './serverInfo.js';
+import { additionalSetup } from './resources.js';
 
 stdioServerFactory({
   ...serverInfo,
   context,
   apiFactories,
+  additionalSetup,
 });