config.ts•1.89 kB
/**
* Configuration module for the ExploitDB MCP server
* Loads configuration from environment variables with sensible defaults
*/
import * as path from 'path';
import * as fs from 'fs-extra';
import * as dotenv from 'dotenv';
// Load environment variables from .env file
dotenv.config();
// Default configuration
const defaultConfig = {
// Repository configuration
cloneRepository: false,
repositoryUrl: 'https://gitlab.com/exploit-database/exploitdb.git',
csvUrl: 'https://gitlab.com/exploit-database/exploitdb/-/raw/main/files_exploits.csv',
// Local storage paths
dataDir: path.join(process.cwd(), 'data'),
dbPath: path.join(process.cwd(), 'data', 'exploitdb.sqlite'),
// Update frequency in hours (0 to disable automatic updates)
updateInterval: 24,
// Maximum number of results to return per query
maxResults: 10
};
// Load configuration from environment variables
export const config = {
cloneRepository: process.env.CLONE_REPOSITORY === 'true' ? true : defaultConfig.cloneRepository,
repositoryUrl: process.env.REPOSITORY_URL || defaultConfig.repositoryUrl,
csvUrl: process.env.CSV_URL || defaultConfig.csvUrl,
dataDir: process.env.DATA_DIR ? path.resolve(process.env.DATA_DIR) : defaultConfig.dataDir,
dbPath: process.env.DB_PATH ? path.resolve(process.env.DB_PATH) : defaultConfig.dbPath,
updateInterval: process.env.UPDATE_INTERVAL ? parseInt(process.env.UPDATE_INTERVAL, 10) : defaultConfig.updateInterval,
maxResults: process.env.MAX_RESULTS ? parseInt(process.env.MAX_RESULTS, 10) : defaultConfig.maxResults
};
// Ensure data directory exists
export const ensureDataDir = async (): Promise<void> => {
await fs.ensureDir(config.dataDir);
// Create subdirectories if needed
if (config.cloneRepository) {
await fs.ensureDir(path.join(config.dataDir, 'exploitdb'));
}
};
// Export configuration
export default config;