Versioning storage based on mcp-auth version number to let us iterate on storage format

This commit is contained in:
Glen Maddern 2025-03-31 16:21:34 +11:00
parent a97fd9e5c6
commit 1382827ebd
3 changed files with 28 additions and 35 deletions

View file

@ -2,16 +2,17 @@ import crypto from 'crypto'
import path from 'path'
import os from 'os'
import fs from 'fs/promises'
import { MCP_REMOTE_VERSION } from './utils'
/**
* MCP Remote Authentication Configuration
*
*
* This module handles the storage and retrieval of authentication-related data for MCP Remote.
*
*
* Configuration directory structure:
* - The config directory is determined by MCP_REMOTE_CONFIG_DIR env var or defaults to ~/.mcp-auth
* - Each file is prefixed with a hash of the server URL to separate configurations for different servers
*
*
* Files stored in the config directory:
* - {server_hash}_client_info.json: Contains OAuth client registration information
* - Format: OAuthClientInformation object with client_id and other registration details
@ -19,18 +20,14 @@ import fs from 'fs/promises'
* - Format: OAuthTokens object with access_token, refresh_token, and expiration information
* - {server_hash}_code_verifier.txt: Contains the PKCE code verifier for the current OAuth flow
* - Format: Plain text string used for PKCE verification
*
*
* All JSON files are stored with 2-space indentation for readability.
*/
/**
* Known configuration file names that might need to be cleaned
*/
export const knownConfigFiles = [
'client_info.json',
'tokens.json',
'code_verifier.txt',
];
export const knownConfigFiles = ['client_info.json', 'tokens.json', 'code_verifier.txt']
/**
* Deletes all known configuration files for a specific server
@ -48,7 +45,9 @@ export async function cleanServerConfig(serverUrlHash: string): Promise<void> {
* @returns The path to the configuration directory
*/
export function getConfigDir(): string {
return process.env.MCP_REMOTE_CONFIG_DIR || path.join(os.homedir(), '.mcp-auth')
const baseConfigDir = process.env.MCP_REMOTE_CONFIG_DIR || path.join(os.homedir(), '.mcp-auth')
// Add a version subdirectory so we don't need to worry about backwards/forwards compatibility yet
return path.join(baseConfigDir, `mcp-remote-${MCP_REMOTE_VERSION}`)
}
/**
@ -110,20 +109,20 @@ export async function deleteConfigFile(serverUrlHash: string, filename: string):
* @returns The parsed file content or undefined if the file doesn't exist
*/
export async function readJsonFile<T>(
serverUrlHash: string,
filename: string,
serverUrlHash: string,
filename: string,
schema: any,
clean: boolean = false
clean: boolean = false,
): Promise<T | undefined> {
try {
await ensureConfigDir()
// If clean flag is set, delete the file before trying to read it
if (clean) {
await deleteConfigFile(serverUrlHash, filename)
return undefined
}
const filePath = getConfigFilePath(serverUrlHash, filename)
const content = await fs.readFile(filePath, 'utf-8')
return await schema.parseAsync(JSON.parse(content))
@ -141,11 +140,7 @@ export async function readJsonFile<T>(
* @param filename The name of the file to write
* @param data The data to write
*/
export async function writeJsonFile(
serverUrlHash: string,
filename: string,
data: any
): Promise<void> {
export async function writeJsonFile(serverUrlHash: string, filename: string, data: any): Promise<void> {
try {
await ensureConfigDir()
const filePath = getConfigFilePath(serverUrlHash, filename)
@ -165,20 +160,20 @@ export async function writeJsonFile(
* @returns The file content as a string
*/
export async function readTextFile(
serverUrlHash: string,
serverUrlHash: string,
filename: string,
errorMessage?: string,
clean: boolean = false
clean: boolean = false,
): Promise<string> {
try {
await ensureConfigDir()
// If clean flag is set, delete the file before trying to read it
if (clean) {
await deleteConfigFile(serverUrlHash, filename)
throw new Error('File deleted due to clean flag')
}
const filePath = getConfigFilePath(serverUrlHash, filename)
return await fs.readFile(filePath, 'utf-8')
} catch (error) {
@ -192,11 +187,7 @@ export async function readTextFile(
* @param filename The name of the file to write
* @param text The text to write
*/
export async function writeTextFile(
serverUrlHash: string,
filename: string,
text: string
): Promise<void> {
export async function writeTextFile(serverUrlHash: string, filename: string, text: string): Promise<void> {
try {
await ensureConfigDir()
const filePath = getConfigFilePath(serverUrlHash, filename)
@ -205,4 +196,4 @@ export async function writeTextFile(
console.error(`Error writing ${filename}:`, error)
throw error
}
}
}