UNPKG

@kya-os/mcp-i

Version:

COMING SOON:Production-ready MCP Identity with automatic registration, key rotation, and optimized performance

164 lines (163 loc) 5.57 kB
"use strict"; var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) { if (k2 === undefined) k2 = k; var desc = Object.getOwnPropertyDescriptor(m, k); if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) { desc = { enumerable: true, get: function() { return m[k]; } }; } Object.defineProperty(o, k2, desc); }) : (function(o, m, k, k2) { if (k2 === undefined) k2 = k; o[k2] = m[k]; })); var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) { Object.defineProperty(o, "default", { enumerable: true, value: v }); }) : function(o, v) { o["default"] = v; }); var __importStar = (this && this.__importStar) || (function () { var ownKeys = function(o) { ownKeys = Object.getOwnPropertyNames || function (o) { var ar = []; for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k; return ar; }; return ownKeys(o); }; return function (mod) { if (mod && mod.__esModule) return mod; var result = {}; if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]); __setModuleDefault(result, mod); return result; }; })(); Object.defineProperty(exports, "__esModule", { value: true }); exports.StorageFactory = exports.CombinedStorage = exports.EnvironmentStorage = exports.MemoryStorage = exports.FileStorage = void 0; const fs = __importStar(require("fs")); const path = __importStar(require("path")); class FileStorage { constructor(customPath) { this.filePath = customPath || path.join(process.cwd(), '.mcp-identity.json'); } async load() { try { if (fs.existsSync(this.filePath)) { const content = fs.readFileSync(this.filePath, 'utf-8'); return JSON.parse(content); } } catch { } return null; } async save(identity) { fs.writeFileSync(this.filePath, JSON.stringify(identity, null, 2)); } async exists() { return fs.existsSync(this.filePath); } } exports.FileStorage = FileStorage; class MemoryStorage { constructor(key) { this.key = key || 'default'; } async load() { return MemoryStorage.instances.get(this.key) || null; } async save(identity) { MemoryStorage.instances.set(this.key, identity); } async exists() { return MemoryStorage.instances.has(this.key); } static clear() { MemoryStorage.instances.clear(); } } exports.MemoryStorage = MemoryStorage; MemoryStorage.instances = new Map(); class EnvironmentStorage { static load() { if (process.env.AGENT_DID && process.env.AGENT_PUBLIC_KEY && process.env.AGENT_PRIVATE_KEY) { return { did: process.env.AGENT_DID, publicKey: process.env.AGENT_PUBLIC_KEY, privateKey: process.env.AGENT_PRIVATE_KEY, agentId: process.env.AGENT_ID || '', agentSlug: process.env.AGENT_SLUG || '', registeredAt: process.env.AGENT_REGISTERED_AT || new Date().toISOString() }; } return null; } } exports.EnvironmentStorage = EnvironmentStorage; class CombinedStorage { constructor(providers) { if (providers.length === 0) { throw new Error('At least one storage provider is required'); } this.providers = providers; this.primaryProvider = providers[0]; } async load() { const envIdentity = EnvironmentStorage.load(); if (envIdentity) { return envIdentity; } for (const provider of this.providers) { const identity = await provider.load(); if (identity) { return identity; } } return null; } async save(identity) { await this.primaryProvider.save(identity); } async exists() { for (const provider of this.providers) { if (await provider.exists()) { return true; } } return false; } } exports.CombinedStorage = CombinedStorage; class StorageFactory { static create(options) { const storageType = options?.storage || 'auto'; let baseStorage; if (storageType === 'memory') { baseStorage = new MemoryStorage(options?.memoryKey); } else if (storageType === 'file') { baseStorage = new FileStorage(options?.customPath); } else if (storageType === 'auto') { if (process.env.AWS_LAMBDA_FUNCTION_NAME || process.env.VERCEL || process.env.NETLIFY || !fs.existsSync || typeof fs.writeFileSync !== 'function') { baseStorage = new MemoryStorage(options?.memoryKey); } else { baseStorage = new FileStorage(options?.customPath); } } else { throw new Error(`Unknown storage type: ${storageType}`); } if (options?.encryptionPassword) { const { createEncryptedStorage } = require('./encrypted-storage'); return createEncryptedStorage(baseStorage, options.encryptionPassword); } return baseStorage; } } exports.StorageFactory = StorageFactory;