@prmichaelsen/remember-mcp 3.18.1 → 3.19.0

package/src/server.ts

@@ -8,11 +8,13 @@ import {
   ErrorCode,
   McpError,
 } from '@modelcontextprotocol/sdk/types.js';
-import { config, validateConfig } from './config.js';
+import { config, validateConfig, loadAuthSchemeConfig } from './config.js';
 import { initWeaviateClient, testWeaviateConnection, getWeaviateClient } from './weaviate/client.js';
 import { initFirestore, testFirestoreConnection } from './firestore/init.js';
 import { logger } from './utils/logger.js';
 import type { AuthContext } from './types/auth.js';
+import { bootstrapOAuth } from './auth/oauth-bootstrap.js';
+import { createServer } from './server-factory.js';
 
 // Import memory tools
 import { createMemoryTool, handleCreateMemory } from './tools/create-memory.js';
@@ -261,8 +263,20 @@ async function main(): Promise<void> {
   try {
     logger.info('Starting remember-mcp server...');
 
-    // Initialize server
-    const server = await initServer();
+    const authConfig = loadAuthSchemeConfig();
+    let server: Server;
+
+    if (authConfig.scheme === 'oauth') {
+      // OAuth mode: exchange API token for JWT, then create server via factory
+      logger.info('Auth scheme: oauth');
+      const { accessToken, userId } = await bootstrapOAuth();
+      logger.info(`Authenticated as userId: ${userId}`);
+      server = await createServer(accessToken, userId);
+    } else {
+      // Service mode: standalone server with no auth (used behind mcp-auth)
+      logger.info('Auth scheme: service');
+      server = await initServer();
+    }
 
     // Start with stdio transport
     const transport = new StdioServerTransport();