@vibe-forge/core 0.7.4 → 0.8.0

package/src/utils/definition-loader.ts

@@ -1,530 +0,0 @@
-import { existsSync } from 'node:fs'
-import { readFile } from 'node:fs/promises'
-import { basename, dirname, relative, resolve } from 'node:path'
-import process from 'node:process'
-
-import { glob } from 'fast-glob'
-import fm from 'front-matter'
-
-export interface Filter {
-  include?: string[]
-  exclude?: string[]
-}
-
-export interface Rule {
-  name?: string
-  description?: string
-  globs?: string | string[]
-  /**
-   * 是否默认加载至系统上下文
-   */
-  always?: boolean
-}
-
-export interface Spec {
-  name?: string
-  always?: boolean
-  description?: string
-  tags?: string[]
-  params?: {
-    name: string
-    description?: string
-  }[]
-  rules?: string[]
-  skills?: string[]
-  mcpServers?: Filter
-  tools?: Filter
-}
-
-export interface LocalRuleReference {
-  type?: 'local'
-  path: string
-  desc?: string
-}
-
-export interface RemoteRuleReference {
-  type: 'remote'
-  tags?: string[]
-  desc?: string
-}
-
-export type RuleReference = string | LocalRuleReference | RemoteRuleReference
-
-export interface SkillSelection {
-  type: 'include' | 'exclude'
-  list: string[]
-}
-
-export interface Entity {
-  name?: string
-  always?: boolean
-  description?: string
-  tags?: string[]
-  prompt?: string
-  promptPath?: string
-  rules?: RuleReference[]
-  skills?: string[] | SkillSelection
-  mcpServers?: Filter
-  tools?: Filter
-}
-
-export interface Skill {
-  name?: string
-  description?: string
-  always?: boolean
-}
-
-export interface Definition<T> {
-  path: string
-  body: string
-  attributes: T
-}
-
-/**
- * 以结构化的方式加载本地文档数据
- */
-export const loadLocalDocuments = async <Attrs extends object>(
-  paths: string[]
-): Promise<Definition<Attrs>[]> => {
-  const promises = paths.map(async (path) => {
-    const content = await readFile(path, 'utf-8')
-    const { body, attributes } = fm<Attrs>(content)
-    return {
-      path,
-      body,
-      attributes
-    }
-  })
-  return Promise.all(promises)
-}
-
-const normalizePath = (path: string) => path.split('\\').join('/')
-
-const stripExtension = (fileName: string) => fileName.replace(/\.[^/.]+$/, '')
-
-const getFirstNonEmptyLine = (text: string) =>
-  text
-    .split('\n')
-    .map(line => line.trim())
-    .find(Boolean)
-
-const toPromptPath = (cwd: string, path: string) => {
-  const relPath = normalizePath(relative(cwd, path))
-  return relPath.startsWith('..') ? normalizePath(path) : relPath
-}
-
-const resolveDocumentName = (
-  path: string,
-  explicitName?: string,
-  indexFileNames: string[] = []
-) => {
-  const trimmedName = explicitName?.trim()
-  if (trimmedName) return trimmedName
-
-  const fileName = basename(path).toLowerCase()
-  if (indexFileNames.includes(fileName)) {
-    return basename(dirname(path))
-  }
-
-  return stripExtension(basename(path))
-}
-
-const resolveDocumentDescription = (
-  body: string,
-  explicitDescription?: string,
-  fallbackName?: string
-) => {
-  const trimmedDescription = explicitDescription?.trim()
-  return trimmedDescription || getFirstNonEmptyLine(body) || fallbackName || ''
-}
-
-const resolveSpecIdentifier = (path: string, explicitName?: string) => {
-  return resolveDocumentName(path, explicitName, ['index.md'])
-}
-
-const toNonEmptyStringArray = (values: unknown): string[] => {
-  if (!Array.isArray(values)) return []
-  return values
-    .filter((value): value is string => typeof value === 'string')
-    .map(value => value.trim())
-    .filter(Boolean)
-}
-
-const isLocalRuleReference = (value: RuleReference): value is LocalRuleReference => {
-  return (
-    value != null &&
-    typeof value === 'object' &&
-    typeof value.path === 'string' &&
-    (value.type == null || value.type === 'local')
-  )
-}
-
-const isRemoteRuleReference = (value: RuleReference): value is RemoteRuleReference => {
-  return (
-    value != null &&
-    typeof value === 'object' &&
-    value.type === 'remote'
-  )
-}
-
-const resolveRulePattern = (pattern: string, baseDir: string) => {
-  const trimmed = pattern.trim()
-  if (!trimmed) return undefined
-  if (trimmed.startsWith('./') || trimmed.startsWith('../')) {
-    return normalizePath(resolve(baseDir, trimmed))
-  }
-  return trimmed
-}
-
-const createRemoteRuleDefinition = (
-  rule: RemoteRuleReference,
-  index: number
-): Definition<Rule> => {
-  const tags = toNonEmptyStringArray(rule.tags)
-  const desc = rule.desc?.trim() || (
-    tags.length > 0
-      ? `远程知识库标签：${tags.join(', ')}`
-      : '远程知识库规则引用'
-  )
-  const bodyParts = [
-    desc,
-    tags.length > 0 ? `知识库标签：${tags.join(', ')}` : undefined,
-    '该规则来自远程知识库引用，不对应本地文件。'
-  ].filter((value): value is string => Boolean(value))
-
-  return {
-    path: `remote-rule-${index + 1}.md`,
-    body: bodyParts.join('\n'),
-    attributes: {
-      name: tags.length > 0 ? `remote:${tags.join(',')}` : `remote-rule-${index + 1}`,
-      description: desc
-    }
-  }
-}
-
-export class DefinitionLoader {
-  private readonly cwd: string
-
-  constructor(cwd: string = process.cwd()) {
-    this.cwd = cwd
-  }
-
-  private async scan(
-    patterns: string[],
-    cwd: string = this.cwd
-  ): Promise<string[]> {
-    const paths = await glob(patterns, { cwd, absolute: true })
-    return paths.sort((a, b) => {
-      const aKey = relative(cwd, a).split('\\').join('/')
-      const bKey = relative(cwd, b).split('\\').join('/')
-      return aKey.localeCompare(bKey)
-    })
-  }
-
-  async loadRules(
-    rules: RuleReference[],
-    options?: {
-      baseDir?: string
-    }
-  ) {
-    const baseDir = options?.baseDir ?? this.cwd
-    const definitions: Definition<Rule>[] = []
-
-    for (const [index, rule] of rules.entries()) {
-      if (typeof rule === 'string') {
-        const pattern = resolveRulePattern(rule, baseDir)
-        if (!pattern) continue
-        definitions.push(
-          ...await loadLocalDocuments<Rule>(
-            await this.scan([pattern])
-          )
-        )
-        continue
-      }
-
-      if (isRemoteRuleReference(rule)) {
-        definitions.push(createRemoteRuleDefinition(rule, index))
-        continue
-      }
-
-      if (!isLocalRuleReference(rule)) continue
-
-      const pattern = resolveRulePattern(rule.path, baseDir)
-      if (!pattern) continue
-
-      const docs = await loadLocalDocuments<Rule>(
-        await this.scan([pattern])
-      )
-
-      definitions.push(
-        ...docs.map((doc) => ({
-          ...doc,
-          attributes: {
-            ...doc.attributes,
-            description: rule.desc?.trim() || doc.attributes.description
-          }
-        }))
-      )
-    }
-
-    return definitions
-  }
-  async loadDefaultRules(): Promise<Definition<Rule>[]> {
-    return this.loadRules([
-      '.ai/rules/*.md',
-      '.ai/plugins/*/rules/*.md'
-    ])
-  }
-  generateRulesPrompt(rules: Definition<Rule>[]): string {
-    const rulesPrompt = rules
-      .map((rule) => {
-        const { path, body, attributes } = rule
-        const name = resolveDocumentName(path, attributes.name)
-        const desc = resolveDocumentDescription(body, attributes.description, name)
-        const content = attributes.always && body.trim()
-          ? `<rule-content>\n${body.trim()}\n</rule-content>\n`
-          : ''
-        return `  - ${name}：${desc}\n${content}--------------------\n`
-      })
-      .filter(Boolean)
-      .join('\n')
-
-    return (
-      '<system-prompt>\n' +
-      '项目系统规则如下：\n' +
-      `${rulesPrompt}\n` +
-      '</system-prompt>\n'
-    )
-  }
-
-  async loadSkills(skills?: string[]): Promise<Definition<Skill>[]> {
-    // 1. Scan for skills in standard locations
-    // Project root skills: .ai/skills/{name}/SKILL.md
-    // Plugin skills: .ai/plugins/{plugin}/skills/{name}/SKILL.md
-
-    // Note: The user code uses readdir to iterate plugins.
-    // We can use glob for easier path finding.
-
-    const patterns = [
-      '.ai/skills/*/SKILL.md',
-      '.ai/plugins/*/skills/*/SKILL.md'
-    ]
-
-    let paths = await this.scan(patterns)
-
-    // Filter by directory name (skill name)
-    if (skills) {
-      paths = paths.filter(path => {
-        return skills.includes(basename(dirname(path)))
-      })
-    }
-
-    return loadLocalDocuments<Skill>(paths)
-  }
-  async loadDefaultSkills(): Promise<Definition<Skill>[]> {
-    return this.loadSkills()
-  }
-  generateSkillsPrompt(skills: Definition<Skill>[]): string {
-    return skills
-      .map((skill) => {
-        const { path, body, attributes } = skill
-        const name = resolveDocumentName(path, attributes.name, ['skill.md'])
-        const desc = resolveDocumentDescription(body, attributes.description, name)
-        return (
-          '技能相关信息如下，通过阅读以下内容了解技能的详细信息：\n' +
-          `- 技能名称：${name}\n` +
-          `- 技能介绍：${desc}\n` +
-          `- 技能文件资源路径：${toPromptPath(this.cwd, dirname(path))}\n` +
-          '- 资源内容：\n' +
-          '<skill-content>\n' +
-          `${body.trim()}\n` +
-          '</skill-content>\n' +
-          '资源内容中的文件路径相对「技能文件资源路径」路径，通过读取相关工具按照实际需要进行阅读。\n'
-        )
-      })
-      .filter(Boolean)
-      .join('\n')
-  }
-  generateSkillsRoutePrompt(skills: Definition<Skill>[]): string {
-    return (
-      '<skills>\n' +
-      `${
-        skills
-          .filter(({ attributes: { always } }) => always !== false)
-          .map(
-            ({ path, body, attributes }) => {
-              const name = resolveDocumentName(path, attributes.name, ['skill.md'])
-              const desc = resolveDocumentDescription(body, attributes.description, name)
-              return `  - ${name}：${desc}\n`
-            }
-          )
-          .join('')
-      }\n` +
-      '</skills>\n'
-    )
-  }
-
-  async loadSpec(name: string): Promise<Definition<Spec> | undefined> {
-    const patterns = [
-      `.ai/specs/${name}.md`,
-      `.ai/specs/${name}/index.md`,
-      `.ai/plugins/*/specs/${name}.md`,
-      `.ai/plugins/*/specs/${name}/index.md`
-    ]
-    const paths = await this.scan(patterns)
-    if (paths.length === 0) return undefined
-
-    const projectPath = paths.find(p => p.includes('/.ai/specs/'))
-    const targetPath = projectPath || paths[0]
-
-    const [doc] = await loadLocalDocuments<Spec>([targetPath])
-    return doc
-  }
-  async loadDefaultSpecs(): Promise<Definition<Spec>[]> {
-    const patterns = [
-      '.ai/specs/*.md',
-      '.ai/specs/*/index.md',
-      '.ai/plugins/*/specs/*.md',
-      '.ai/plugins/*/specs/*/index.md'
-    ]
-    const paths = await this.scan(patterns)
-    return loadLocalDocuments<Spec>(paths)
-  }
-  generateSpecRoutePrompt(specsDocuments: Definition<Spec>[]): string {
-    const specsRouteStr = specsDocuments
-      .filter(({ attributes }) => attributes.always !== false)
-      .map(({ path, body, attributes }) => {
-        const name = resolveDocumentName(path, attributes.name, ['index.md'])
-        const desc = resolveDocumentDescription(body, attributes.description, name)
-        const identifier = resolveSpecIdentifier(path, attributes.name)
-        const params = attributes.params ?? []
-        const paramsPrompt = params.length > 0
-          ? params
-            .map(({ name, description }) => `    - ${name}：${description ?? '无'}\n`)
-            .join('')
-          : '    - 无\n'
-
-        return (
-          `- 流程名称：${name}\n` +
-          `  - 介绍：${desc}\n` +
-          `  - 标识：${identifier}\n` +
-          '  - 参数：\n' +
-          `${paramsPrompt}`
-        )
-      })
-      .join('\n')
-    return (
-      '<system-prompt>\n' +
-      '你是一个专业的项目推进管理大师，能够熟练指导其他实体来为你的目标工作。对你的预期是：\n' +
-      '\n' +
-      '- 永远不要单独完成代码开发工作\n' +
-      '- 必须要协调其他的开发人员来完成任务\n' +
-      '- 必须让他们按照目标进行完成，不要偏离目标，检查他们任务完成后的汇报内容是否符合要求\n' +
-      '\n' +
-      '根据用户需要以及实际的开发目标来决定使用不同的工作流程，调用 `load-spec` mcp tool 完成工作流程的加载。\n' +
-      '- 根据实际需求传入标识，这不是路径，只能使用工具进行加载\n' +
-      '- 通过参数的描述以及实际应用场景决定怎么传入参数\n' +
-      '项目存在如下工作流程：\n' +
-      `${specsRouteStr}\n` +
-      '</system-prompt>\n'
-    )
-  }
-
-  async loadEntity(name: string): Promise<Definition<Entity> | undefined> {
-    // 1. Try to load from index.json (Directory based entity)
-    const jsonPatterns = [
-      `.ai/entities/${name}/index.json`,
-      `.ai/plugins/*/entities/${name}/index.json`
-    ]
-    const jsonPaths = await this.scan(jsonPatterns)
-
-    if (jsonPaths.length > 0) {
-      const projectPath = jsonPaths.find(p => p.includes('/.ai/entities/'))
-      const targetPath = projectPath || jsonPaths[0]
-      return this.loadEntityFromJson(targetPath)
-    }
-
-    // 2. Fallback to Markdown file
-    const patterns = [
-      `.ai/entities/${name}.md`,
-      `.ai/entities/${name}/README.md`,
-      `.ai/plugins/*/entities/${name}.md`,
-      `.ai/plugins/*/entities/${name}/README.md`
-    ]
-    const paths = await this.scan(patterns)
-    if (paths.length === 0) return undefined
-
-    const projectPath = paths.find(p => p.includes('/.ai/entities/'))
-    const targetPath = projectPath || paths[0]
-
-    const [doc] = await loadLocalDocuments<Entity>([targetPath])
-    return doc
-  }
-  private async loadEntityFromJson(jsonPath: string): Promise<Definition<Entity>> {
-    const entityDir = dirname(jsonPath)
-    const jsonVariables: Record<string, string> = {
-      workspaceFolder: process.env.WORKSPACE_FOLDER || this.cwd
-    }
-
-    const content = await readFile(jsonPath, 'utf-8')
-    const entityJSONContent = content.replace(/\$\{([^}]+)\}/g, (_, key) => jsonVariables[key] ?? `$\{${key}}`)
-
-    const entityData = JSON.parse(entityJSONContent) as Entity
-
-    let prompt = entityData.prompt
-    if (!prompt) {
-      const promptPath = entityData.promptPath || 'AGENTS.md'
-      const resolvedPromptPath = resolve(entityDir, promptPath)
-      if (existsSync(resolvedPromptPath)) {
-        prompt = await readFile(resolvedPromptPath, 'utf-8')
-      }
-    }
-
-    return {
-      path: jsonPath,
-      body: prompt || '',
-      attributes: entityData
-    }
-  }
-  async loadDefaultEntities(): Promise<Definition<Entity>[]> {
-    // List both .md and index.json entities
-    const mdPatterns = [
-      '.ai/entities/*.md',
-      '.ai/entities/*/README.md',
-      '.ai/plugins/*/entities/*.md',
-      '.ai/plugins/*/entities/*/README.md'
-    ]
-    const jsonPatterns = [
-      '.ai/entities/*/index.json',
-      '.ai/plugins/*/entities/*/index.json'
-    ]
-
-    const [mdPaths, jsonPaths] = await Promise.all([
-      this.scan(mdPatterns),
-      this.scan(jsonPatterns)
-    ])
-
-    const mdDocs = await loadLocalDocuments<Entity>(mdPaths)
-    const jsonDocs = await Promise.all(jsonPaths.map(p => this.loadEntityFromJson(p)))
-
-    return [...mdDocs, ...jsonDocs]
-  }
-  generateEntitiesRoutePrompt(entities: Definition<Entity>[]): string {
-    return (
-      '<system-prompt>\n' +
-      '项目存在如下实体：\n' +
-      `${
-        entities
-          .filter(({ attributes }) => attributes.always !== false)
-          .map(({ path, attributes, body }) => {
-            const name = resolveDocumentName(path, attributes.name, ['readme.md', 'index.json'])
-            const desc = resolveDocumentDescription(body, attributes.description, name)
-            return `  - ${name}：${desc}\n`
-          })
-          .join('')
-      }\n` +
-      '解决用户问题时，需根据用户需求可以通过 run-tasks 工具指定为实体后，自行调度多个不同类型的实体来完成工作。\n' +
-      '</system-prompt>\n'
-    )
-  }
-}

package/src/utils/filter.ts

@@ -1,26 +0,0 @@
-export const filterObject = <T extends Record<string, any>>(
-  obj: T,
-  rules: { include?: string[]; exclude?: string[] }
-): Partial<T> => {
-  const { include = [], exclude = [] } = rules
-
-  if (include.length === 0 && exclude.length === 0) {
-    return { ...obj }
-  }
-
-  const result: Partial<T> = {}
-
-  Object.keys(obj).forEach((key) => {
-    // If include list is provided, key MUST be in it
-    if (include.length > 0 && !include.includes(key)) {
-      return
-    }
-    // If key is in exclude list, it MUST NOT be in it
-    if (exclude.includes(key)) {
-      return
-    }
-    result[key as keyof T] = obj[key]
-  })
-
-  return result
-}

package/src/utils/string-transform.ts

@@ -1,37 +0,0 @@
-export function kebabCase(str: string) {
-  return str.replace(/([a-z])([A-Z])/g, '$1-$2').toLowerCase()
-}
-
-export function camelCase(str: string) {
-  return str.replace(/[-_]([a-z])/g, (_, letter) => letter.toUpperCase())
-}
-
-export function transformAllObjectKeys(
-  transformFn: (str: string) => string,
-  obj: unknown
-): unknown {
-  const boundTransformFn = transformAllObjectKeys.bind(null, transformFn)
-  if (typeof obj !== 'object') return obj
-  if (Array.isArray(obj)) {
-    return obj.map(boundTransformFn)
-  }
-  if (obj === null) {
-    return null
-  }
-  const newObj: Record<string, unknown> = {}
-  for (const key in obj) {
-    newObj[transformFn(key)] = boundTransformFn(
-      // @ts-ignore
-      obj[key]
-    )
-  }
-  return newObj
-}
-
-export function transformKebabKey<T>(obj: unknown) {
-  return transformAllObjectKeys(kebabCase, obj) as T
-}
-
-export function transformCamelKey<T>(obj: unknown) {
-  return transformAllObjectKeys(camelCase, obj) as T
-}

package/src/utils/uuid.ts

@@ -1,6 +0,0 @@
-export const uuid = () =>
-  'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, (c) => {
-    const r = (Math.random() * 16) | 0
-    const v = c === 'x' ? r : (r & 0x3) | 0x8
-    return v.toString(16)
-  })