@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
JavaScript
;
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;