dpml-prompt
Version:
DPML-powered AI prompt framework - Revolutionary AI-First CLI system based on Deepractice Prompt Markup Language. Build sophisticated AI agents with structured prompts, memory systems, and execution frameworks.
325 lines (286 loc) • 9.06 kB
JavaScript
const fs = require('fs-extra')
const path = require('path')
const ResourceData = require('./ResourceData')
/**
* 注册表数据管理器 v2.0
* 基于ResourceData数组的全新架构，严格区分资源来源(source)和资源种类(protocol)
*/
class RegistryData {
/**
* @param {string} source - 注册表来源 ('package' | 'project' | 'user')
* @param {string} filePath - 注册表文件路径
* @param {Array<ResourceData>} resources - 资源数据数组
* @param {Object} metadata - 注册表元数据
*/
constructor(source, filePath, resources = [], metadata = {}) {
this.source = source
this.filePath = filePath
this.resources = resources.map(r => r instanceof ResourceData ? r : ResourceData.fromRawData(r))
this.metadata = {
version: "2.0.0",
description: `${source} 级资源注册表`,
createdAt: new Date().toISOString(),
updatedAt: new Date().toISOString(),
...metadata
}
this.cache = new Map()
}
/**
* 从文件加载注册表数据
* @param {string} source - 注册表来源
* @param {string} filePath - 文件路径
* @returns {Promise<RegistryData>} 注册表数据实例
*/
static async fromFile(source, filePath) {
try {
const data = await fs.readJSON(filePath)
// 处理新格式（v2.0）
if (data.version === "2.0.0" && Array.isArray(data.resources)) {
return new RegistryData(source, filePath, data.resources, data.metadata)
}
// 处理旧格式（v1.0）- 自动转换
if (data.resources && typeof data.resources === 'object') {
const resources = []
for (const [protocol, resourcesOfType] of Object.entries(data.resources)) {
if (resourcesOfType && typeof resourcesOfType === 'object') {
for (const [id, reference] of Object.entries(resourcesOfType)) {
resources.push(ResourceData.fromFilePath(
reference.replace(/^@\w+:\/\//, ''),
source,
protocol,
reference
))
}
}
}
return new RegistryData(source, filePath, resources, {
migratedFrom: "v1.0.0",
originalTimestamp: data.timestamp
})
}
throw new Error(`Unsupported registry format in ${filePath}`)
} catch (error) {
throw new Error(`Failed to load ${source} registry from ${filePath}: ${error.message}`)
}
}
/**
* 创建空的注册表数据
* @param {string} source - 注册表来源
* @param {string} filePath - 注册表文件路径
* @returns {RegistryData} 空注册表数据实例
*/
static createEmpty(source, filePath) {
return new RegistryData(source, filePath, [], {
description: `${source} 级资源注册表`,
createdAt: new Date().toISOString()
})
}
/**
* 添加资源
* @param {ResourceData|Object} resource - 资源数据
*/
addResource(resource) {
const resourceData = resource instanceof ResourceData ? resource : ResourceData.fromRawData(resource)
// 对于merged类型的注册表，保持原始来源信息
// 只有在非merged注册表中才强制统一来源
if (this.source !== 'merged' && resourceData.source !== this.source) {
resourceData.source = this.source
}
// 检查是否已存在相同ID的资源
const existingIndex = this.resources.findIndex(r => r.id === resourceData.id && r.protocol === resourceData.protocol)
if (existingIndex >= 0) {
// 更新现有资源
this.resources[existingIndex] = resourceData
} else {
// 添加新资源
this.resources.push(resourceData)
}
this._updateMetadata()
this.cache.clear()
}
/**
* 移除资源
* @param {string} id - 资源ID
* @param {string} protocol - 资源协议
* @returns {boolean} 是否成功移除
*/
removeResource(id, protocol) {
const initialLength = this.resources.length
this.resources = this.resources.filter(r => !(r.id === id && r.protocol === protocol))
const removed = this.resources.length < initialLength
if (removed) {
this._updateMetadata()
this.cache.clear()
}
return removed
}
/**
* 查找资源
* @param {Object} filters - 过滤条件
* @returns {Array<ResourceData>} 匹配的资源数组
*/
findResources(filters = {}) {
return this.resources.filter(resource => resource.matches(filters))
}
/**
* 根据ID查找资源
* @param {string} id - 资源ID
* @param {string} protocol - 资源协议（可选）
* @returns {ResourceData|null} 找到的资源
*/
findResourceById(id, protocol = null) {
return this.resources.find(r => {
if (protocol) {
return r.id === id && r.protocol === protocol
}
return r.id === id
}) || null
}
/**
* 获取指定协议类型的所有资源
* @param {string} protocol - 资源协议
* @returns {Array<ResourceData>} 资源数组
*/
getResourcesByProtocol(protocol) {
return this.resources.filter(r => r.protocol === protocol)
}
/**
* 获取资源Map（兼容旧接口）
* @param {boolean} includeSourcePrefix - 是否包含源前缀
* @returns {Map<string, string>} 资源ID到引用的映射
*/
getResourceMap(includeSourcePrefix = true) {
const cacheKey = `resourceMap_${includeSourcePrefix}`
if (this.cache.has(cacheKey)) {
return this.cache.get(cacheKey)
}
const registry = new Map()
for (const resource of this.resources) {
if (includeSourcePrefix) {
// 包含源前缀的完整ID
registry.set(resource.getFullId(), resource.reference)
// 同时也注册基础ID（用于向后兼容）
registry.set(resource.getBaseId(), resource.reference)
} else {
// 仅使用基础ID
registry.set(resource.getBaseId(), resource.reference)
}
}
this.cache.set(cacheKey, registry)
return registry
}
/**
* 获取所有资源数据
* @returns {Array<ResourceData>} 所有资源数组
*/
getAllResources() {
return [...this.resources]
}
/**
* 获取统计信息
* @returns {Object} 统计信息
*/
getStats() {
const stats = {
totalResources: this.resources.length,
byProtocol: {},
bySource: {}
}
for (const resource of this.resources) {
// 按协议统计
stats.byProtocol[resource.protocol] = (stats.byProtocol[resource.protocol] || 0) + 1
// 按来源统计
stats.bySource[resource.source] = (stats.bySource[resource.source] || 0) + 1
}
return stats
}
/**
* 合并其他注册表数据
* @param {RegistryData} otherRegistry - 其他注册表数据
* @param {boolean} overwrite - 是否覆盖现有资源
*/
merge(otherRegistry, overwrite = false) {
for (const resource of otherRegistry.resources) {
const existing = this.findResourceById(resource.id, resource.protocol)
if (!existing || overwrite) {
this.addResource(resource.clone())
}
}
}
/**
* 保存注册表到文件
* @returns {Promise<void>}
*/
async save() {
try {
// 确保目录存在
await fs.ensureDir(path.dirname(this.filePath))
// 更新元数据
this._updateMetadata()
// 构建保存数据
const saveData = {
version: this.metadata.version,
source: this.source,
metadata: this.metadata,
resources: this.resources.map(r => r.toJSON()),
stats: this.getStats()
}
// 保存文件
await fs.writeJSON(this.filePath, saveData, { spaces: 2 })
} catch (error) {
throw new Error(`Failed to save ${this.source} registry to ${this.filePath}: ${error.message}`)
}
}
/**
* 更新元数据
* @private
*/
_updateMetadata() {
this.metadata.updatedAt = new Date().toISOString()
this.metadata.resourceCount = this.resources.length
}
/**
* 获取注册表大小
* @returns {number} 资源数量
*/
get size() {
return this.resources.length
}
/**
* 检查注册表是否为空
* @returns {boolean} 是否为空
*/
isEmpty() {
return this.resources.length === 0
}
/**
* 清空所有资源
*/
clear() {
this.resources = []
this._updateMetadata()
this.cache.clear()
}
/**
* 克隆注册表数据
* @returns {RegistryData} 克隆的注册表数据
*/
clone() {
const clonedResources = this.resources.map(r => r.clone())
return new RegistryData(this.source, this.filePath, clonedResources, { ...this.metadata })
}
/**
* 转换为JSON对象
* @returns {Object} JSON对象
*/
toJSON() {
return {
version: this.metadata.version,
source: this.source,
metadata: this.metadata,
resources: this.resources.map(r => r.toJSON()),
stats: this.getStats()
}
}
}
module.exports = RegistryData