feat: session restoration for streamable HTTP transport

src/auth/sdkOAuthClientProvider.test.ts

@@ -29,6 +29,9 @@ vi.mock('@src/constants.js', () => ({
         TTL_MS: 30 * 24 * 60 * 60 * 1000, // 30 days
         DEFAULT_SCOPES: [],
       },
+      SESSION: {
+        SUBDIR: 'client',
+      },
     },
   },
   getGlobalConfigDir: vi.fn(() => '/mock/config/dir'),
@@ -127,7 +130,7 @@ describe('SDKOAuthClientProvider', () => {
     it('should initialize with custom session storage path', () => {
       provider = new SDKOAuthClientProvider('test-server', mockConfig, '/custom/path');
 
-      expect(FileStorageService).toHaveBeenCalledWith('/custom/path');
+      expect(FileStorageService).toHaveBeenCalledWith('/custom/path', 'client');
     });
 
     it('should load persisted data on initialization', () => {

src/auth/sdkOAuthClientProvider.ts

@@ -1,5 +1,4 @@
 import { randomUUID } from 'node:crypto';
-import path from 'path';
 
 import { OAuthClientProvider } from '@modelcontextprotocol/sdk/client/auth.js';
 import type {
@@ -8,7 +7,7 @@ import type {
   OAuthTokens,
 } from '@modelcontextprotocol/sdk/shared/auth.js';
 
-import { AUTH_CONFIG, getGlobalConfigDir } from '@src/constants.js';
+import { AUTH_CONFIG } from '@src/constants.js';
 import logger from '@src/logger/logger.js';
 
 import { ClientSessionData } from './sessionTypes.js';
@@ -47,9 +46,8 @@ export class SDKOAuthClientProvider implements OAuthClientProvider {
     this.serverName = serverName;
     this.config = config;
 
-    // Create FileStorageService and ClientSessionRepository
-    const storagePath = sessionStoragePath || path.join(getGlobalConfigDir(), 'clientSessions');
-    const fileStorage = new FileStorageService(storagePath);
+    // Create FileStorageService and ClientSessionRepository with 'client' subdirectory
+    const fileStorage = new FileStorageService(sessionStoragePath, AUTH_CONFIG.CLIENT.SESSION.SUBDIR);
     this.sessionRepository = new ClientSessionRepository(fileStorage);
 
     // Set up client metadata for registration with better defaults

src/auth/sessionTypes.ts

@@ -43,3 +43,15 @@ export interface AuthRequestData extends ExpirableData {
   resource?: string;
   scopes?: string[];
 }
+
+// Streamable HTTP session data for session restoration
+export interface StreamableSessionData extends ExpirableData {
+  tags?: string[];
+  tagExpression?: string; // JSON stringified TagExpression
+  tagQuery?: string; // JSON stringified TagQuery
+  tagFilterMode?: 'simple-or' | 'advanced' | 'preset' | 'none';
+  presetName?: string;
+  enablePagination?: boolean;
+  customTemplate?: string;
+  lastAccessedAt: number;
+}

src/auth/storage/authCodeRepository.test.ts

@@ -273,7 +273,7 @@ describe('AuthCodeRepository', () => {
 
       // Check that file was created with correct prefix
       const expectedFileName = AUTH_CONFIG.SERVER.AUTH_CODE.FILE_PREFIX + code + '.json';
-      const filePath = path.join(tempDir, expectedFileName);
+      const filePath = path.join(storage.getStorageDir(), expectedFileName);
 
       expect(fs.existsSync(filePath)).toBe(true);
     });

src/auth/storage/authRequestRepository.test.ts

@@ -326,7 +326,7 @@ describe('AuthRequestRepository', () => {
 
       // Check that file was created with correct prefix
       const expectedFileName = AUTH_CONFIG.SERVER.AUTH_REQUEST.FILE_PREFIX + authRequestId + '.json';
-      const filePath = path.join(tempDir, expectedFileName);
+      const filePath = path.join(storage.getStorageDir(), expectedFileName);
 
       expect(fs.existsSync(filePath)).toBe(true);
     });

src/auth/storage/clientDataRepository.test.ts

@@ -339,7 +339,7 @@ describe('ClientDataRepository', () => {
 
       // Check that file was created with correct prefix
       const expectedFileName = AUTH_CONFIG.SERVER.SESSION.FILE_PREFIX + clientId + '.json';
-      const filePath = path.join(tempDir, expectedFileName);
+      const filePath = path.join(storage.getStorageDir(), expectedFileName);
 
       expect(fs.existsSync(filePath)).toBe(true);
     });

src/auth/storage/clientSessionRepository.test.ts

@@ -573,7 +573,7 @@ describe('ClientSessionRepository', () => {
       // Check that file was created with correct prefix
       const sessionId = `${AUTH_CONFIG.CLIENT.SESSION.ID_PREFIX}${serverName}`;
       const expectedFileName = AUTH_CONFIG.CLIENT.SESSION.FILE_PREFIX + sessionId + '.json';
-      const filePath = path.join(tempDir, expectedFileName);
+      const filePath = path.join(storage.getStorageDir(), expectedFileName);
 
       expect(fs.existsSync(filePath)).toBe(true);
     });