@vibe-forge/core 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025-present Vibe-Forge.ai
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/package.json

@@ -0,0 +1,73 @@
+{
+  "name": "@vibe-forge/core",
+  "version": "0.1.0",
+  "imports": {
+    "#~/*.js": {
+      "__vibe-forge__": {
+        "default": "./src/*.ts"
+      },
+      "default": {
+        "import": "./dist/index.mjs",
+        "require": "./dist/index.js"
+      }
+    }
+  },
+  "exports": {
+    ".": {
+      "__vibe-forge__": {
+        "default": "./src/index.ts"
+      },
+      "default": {
+        "import": "./dist/index.mjs",
+        "require": "./dist/index.js"
+      }
+    },
+    "./utils/*": {
+      "__vibe-forge__": {
+        "default": "./src/utils/*.ts"
+      },
+      "default": {
+        "import": "./dist/utils/*.mjs",
+        "require": "./dist/utils/*.js"
+      }
+    },
+    "./controllers/task": {
+      "__vibe-forge__": {
+        "default": "./src/controllers/task/index.ts"
+      },
+      "default": {
+        "import": "./dist/controllers/task/index.mjs",
+        "require": "./dist/controllers/task/index.js"
+      }
+    },
+    "./schema": {
+      "__vibe-forge__": {
+        "default": "./src/schema.ts"
+      },
+      "default": {
+        "import": "./dist/schema.mjs",
+        "require": "./dist/schema.js"
+      }
+    },
+    "./hooks": {
+      "__vibe-forge__": {
+        "default": "./src/hooks/index.ts"
+      },
+      "default": {
+        "import": "./dist/hooks/index.mjs",
+        "require": "./dist/hooks/index.js"
+      }
+    }
+  },
+  "dependencies": {
+    "fast-glob": "^3.3.3",
+    "front-matter": "^4.0.2",
+    "js-yaml": "^4.1.1",
+    "node-notifier": "^10.0.1",
+    "zod": "^3.24.1"
+  },
+  "devDependencies": {
+    "@types/js-yaml": "^4.0.9",
+    "@types/node-notifier": "^8.0.5"
+  }
+}

package/src/adapter/index.ts

@@ -0,0 +1,6 @@
+import type { Adapter } from './type'
+
+export * from './loader'
+export * from './type'
+
+export const defineAdapter = (adapter: Adapter) => adapter

package/src/adapter/loader.ts

@@ -0,0 +1,9 @@
+import type { Adapter } from './type'
+
+export const loadAdapter = async (type: string) =>
+  (
+    // eslint-disable-next-line ts/no-require-imports
+    require(
+      type.startsWith('@') ? type : `@vibe-forge/adapter-${type}`
+    )
+  ).default as Adapter

package/src/adapter/type.ts

@@ -0,0 +1,99 @@
+import type { Cache, Config } from '@vibe-forge/core'
+
+import type { Logger } from '#~/utils/create-logger.js'
+
+import type { ChatMessage, ChatMessageContent } from '../types'
+
+export type AdapterMessageContent = ChatMessageContent
+
+export type AdapterOutputEvent =
+  | { type: 'init'; data: SessionInitInfo }
+  | { type: 'summary'; data: SessionSummaryInfo }
+  | { type: 'message'; data: ChatMessage }
+  | { type: 'exit'; data: { exitCode: number | null; stderr?: string } }
+  | { type: 'stop'; data?: ChatMessage }
+
+export type SessionInfo =
+  | ({ type: 'init' } & SessionInitInfo)
+  | ({ type: 'summary' } & SessionSummaryInfo)
+
+export interface SessionInitInfo {
+  uuid: string
+  model: string
+  version: string
+  tools: string[]
+  slashCommands: string[]
+  cwd: string
+  agents: string[]
+  title?: string
+}
+
+export interface SessionSummaryInfo {
+  summary: string
+  leafUuid: string
+}
+
+export type AdapterEvent =
+  | { type: 'message'; content: AdapterMessageContent[]; parentUuid?: string }
+  | { type: 'interrupt' }
+  | { type: 'stop' }
+
+export interface AdapterCtx {
+  ctxId: string
+
+  cwd: string
+  env: Record<string, string | null | undefined>
+
+  cache: {
+    set: <K extends keyof Cache>(key: K, value: Cache[K]) => Promise<{
+      cachePath: string
+    }>
+    get: <K extends keyof Cache>(key: K) => Promise<Cache[K] | undefined>
+  }
+  logger: Logger
+
+  configs: [Config?, Config?]
+}
+
+export interface AdapterQueryOptions {
+  description?: string
+
+  type: 'create' | 'resume'
+  runtime: 'server' | 'cli' | 'mcp'
+  sessionId: string
+  model?: string
+  mode?: 'stream' | 'direct'
+
+  systemPrompt?: string
+  appendSystemPrompt?: boolean
+
+  mcpServers?: {
+    include?: string[]
+    exclude?: string[]
+  }
+  tools?: {
+    include?: string[]
+    exclude?: string[]
+  }
+  skills?: {
+    include?: string[]
+    exclude?: string[]
+  }
+
+  extraOptions?: string[]
+
+  onEvent: (event: AdapterOutputEvent) => void
+}
+
+export interface AdapterSession {
+  kill: () => void
+  emit: (event: AdapterEvent) => void
+  pid?: number
+}
+
+export interface Adapter {
+  query: (
+    ctx: AdapterCtx,
+    options: AdapterQueryOptions
+  ) => Promise<AdapterSession>
+}

package/src/config.ts

@@ -0,0 +1,375 @@
+import { existsSync } from 'node:fs'
+import { readFile } from 'node:fs/promises'
+import { resolve } from 'node:path'
+import process from 'node:process'
+
+import { load } from 'js-yaml'
+
+import type { PluginConfig } from './hooks'
+
+export interface AdapterMap {}
+
+export interface ModelServiceConfig {
+  /**
+   * 模型服务展示标题
+   */
+  title?: string
+  /**
+   * 模型服务展示描述
+   */
+  description?: string
+  /**
+   * 模型服务 API 基础 URL
+   */
+  apiBaseUrl: string
+  /**
+   * 模型服务 API 密钥
+   */
+  apiKey: string
+  /**
+   * 模型服务支持的模型列表
+   */
+  models?: string[]
+  /**
+   * 模型服务支持的模型别名
+   */
+  modelsAlias?: Record<string, string[]>
+  /**
+   * 拓展配置，由下游自行消费
+   */
+  extra?: Record<string, unknown>
+}
+
+export interface RecommendedModelConfig {
+  service?: string
+  model: string
+  title?: string
+  description?: string
+  placement?: 'modelSelector'
+}
+
+export type LanguageCode = 'zh' | 'en'
+
+export type NotificationTrigger = 'completed' | 'failed' | 'terminated' | 'waiting_input'
+
+export interface NotificationEventConfig {
+  title?: string
+  description?: string
+  disabled?: boolean
+  sound?: string
+}
+
+export interface NotificationConfig {
+  disabled?: boolean
+  volume?: number
+  events?: Partial<Record<NotificationTrigger, NotificationEventConfig>>
+}
+
+export interface Config {
+  /**
+   * 配置目录
+   */
+  baseDir?: string
+  /**
+   * 适配器配置
+   */
+  adapters?: Partial<AdapterMap>
+  /**
+   * 默认适配器名称
+   */
+  defaultAdapter?: keyof AdapterMap
+  /**
+   * 模型服务配置
+   */
+  modelServices?: Record<string, ModelServiceConfig>
+  /**
+   * 默认模型服务名称
+   */
+  defaultModelService?: string
+  /**
+   * 默认模型名称
+   */
+  defaultModel?: string
+  recommendedModels?: RecommendedModelConfig[]
+  interfaceLanguage?: LanguageCode
+  modelLanguage?: LanguageCode
+  /**
+   * MCP 服务器配置
+   */
+  mcpServers?: Record<
+    string,
+    & {
+      /**
+       * 是否启用
+       */
+      enabled?: boolean
+      /**
+       * 环境变量配置
+       */
+      env?: Record<string, string>
+    }
+    & (
+      | {
+        type?: undefined
+        command: string
+        args: string[]
+      }
+      | {
+        type: 'sse'
+        url: string
+        headers: Record<string, string>
+      }
+      | {
+        type: 'http'
+        url: string
+        headers?: Record<string, string>
+      }
+    )
+  >
+  /**
+   * 默认启用的 MCP 服务器列表
+   */
+  defaultIncludeMcpServers?: string[]
+  /**
+   * 默认禁用的 MCP 服务器列表
+   */
+  defaultExcludeMcpServers?: string[]
+  noDefaultVibeForgeMcpServer?: boolean
+  /**
+   * 权限配置
+   */
+  permissions?: {
+    allow?: string[]
+    deny?: string[]
+    ask?: string[]
+  }
+  /**
+   * 环境变量配置
+   */
+  env?: Record<string, string>
+  /**
+   * 公告配置
+   */
+  announcements?: string[]
+  /**
+   * 快捷键配置
+   */
+  shortcuts?: {
+    newSession?: string
+    openConfig?: string
+  }
+  notifications?: NotificationConfig
+  /**
+   * 会话配置
+   */
+  conversation?: {
+    /**
+     * 对话风格
+     * - `friendly`: 友好的对话风格，适合用户与助手交互
+     * - `programmatic`: 程序化的对话风格，适合助手执行任务
+     */
+    style?: 'friendly' | 'programmatic'
+    /**
+     * 自定义对话风格。通过指定提示词约束对话风格。
+     */
+    customInstructions?: string
+  }
+  /**
+   * 插件配置
+   */
+  plugins?: PluginConfig
+  enabledPlugins?: Record<string, boolean>
+  extraKnownMarketplaces?: Record<
+    string,
+    {
+      source:
+        | {
+          source: 'github'
+          repo: string
+        }
+        | {
+          source: 'git'
+          url: string
+        }
+        | {
+          source: 'directory'
+          path: string
+        }
+    }
+  >
+}
+
+export interface AboutInfo {
+  version?: string
+  lastReleaseAt?: string
+  urls?: {
+    repo?: string
+    docs?: string
+    contact?: string
+    issues?: string
+    releases?: string
+  }
+}
+
+export interface ConfigSection {
+  general?: {
+    baseDir?: Config['baseDir']
+    defaultAdapter?: Config['defaultAdapter']
+    defaultModelService?: Config['defaultModelService']
+    defaultModel?: Config['defaultModel']
+    recommendedModels?: Config['recommendedModels']
+    interfaceLanguage?: Config['interfaceLanguage']
+    modelLanguage?: Config['modelLanguage']
+    announcements?: Config['announcements']
+    permissions?: Config['permissions']
+    env?: Config['env']
+    notifications?: Config['notifications']
+  }
+  conversation?: Config['conversation']
+  modelServices?: Config['modelServices']
+  adapters?: Config['adapters']
+  plugins?: {
+    plugins?: Config['plugins']
+    enabledPlugins?: Config['enabledPlugins']
+    extraKnownMarketplaces?: Config['extraKnownMarketplaces']
+  }
+  mcp?: {
+    mcpServers?: Config['mcpServers']
+    defaultIncludeMcpServers?: Config['defaultIncludeMcpServers']
+    defaultExcludeMcpServers?: Config['defaultExcludeMcpServers']
+    noDefaultVibeForgeMcpServer?: Config['noDefaultVibeForgeMcpServer']
+  }
+  shortcuts?: Config['shortcuts']
+}
+
+export interface ConfigResponse {
+  sources?: {
+    project?: ConfigSection
+    user?: ConfigSection
+    merged?: ConfigSection
+  }
+  meta?: {
+    workspaceFolder?: string
+    configPresent?: {
+      project?: boolean
+      user?: boolean
+    }
+    experiments?: Record<string, unknown>
+    about?: AboutInfo
+  }
+}
+
+export const defineConfig = (config: Config) => config
+
+const loadJSConfig = async (paths: string[]) => {
+  for (const path of paths) {
+    try {
+      const configPath = resolve(process.cwd(), path)
+      if (!existsSync(configPath)) {
+        continue
+      }
+      // eslint-disable-next-line ts/no-require-imports
+      return (require(configPath)?.default ?? {}) as Config
+    } catch (e) {
+      console.error(`Failed to load config file ${path}: ${e}`)
+    }
+  }
+}
+
+const loadJSONConfig = async (paths: string[], jsonVariables: Record<string, string | null | undefined>) => {
+  for (const path of paths) {
+    try {
+      const configPath = resolve(process.cwd(), path)
+      if (!existsSync(configPath)) {
+        continue
+      }
+      const configContent = await readFile(configPath, 'utf-8')
+      const configResolvedContent = configContent
+        .replace(/\$\{(\w+)\}/g, (_, key) => jsonVariables[key] ?? `$\{${key}}`)
+      return JSON.parse(configResolvedContent) as Config
+    } catch (e) {
+      console.error(`Failed to load config file ${path}: ${e}`)
+    }
+  }
+}
+
+const loadYAMLConfig = async (paths: string[], jsonVariables: Record<string, string | null | undefined>) => {
+  for (const path of paths) {
+    try {
+      const configPath = resolve(process.cwd(), path)
+      if (!existsSync(configPath)) {
+        continue
+      }
+      const configContent = await readFile(configPath, 'utf-8')
+      const configResolvedContent = configContent
+        .replace(/\$\{(\w+)\}/g, (_, key) => jsonVariables[key] ?? `$\{${key}}`)
+      return load(configResolvedContent) as Config
+    } catch (e) {
+      console.error(`Failed to load config file ${path}: ${e}`)
+    }
+  }
+}
+
+let configCache: Promise<readonly [Config | undefined, Config | undefined]> | null = null
+
+export const resetConfigCache = () => {
+  configCache = null
+}
+
+export const loadConfig = (options: {
+  jsonVariables?: Record<string, string | null | undefined>
+}) => {
+  if (configCache) {
+    return configCache
+  }
+
+  configCache = (async () =>
+    [
+      await loadJSONConfig(
+        [
+          './.ai.config.json',
+          './infra/.ai.config.json'
+        ],
+        options.jsonVariables ?? {}
+      ) ??
+        await loadYAMLConfig(
+          [
+            './.ai.config.yaml',
+            './.ai.config.yml',
+            './infra/.ai.config.yaml',
+            './infra/.ai.config.yml'
+          ],
+          options.jsonVariables ?? {}
+        ),
+      await loadJSONConfig(
+        [
+          './.ai.dev.config.json',
+          './infra/.ai.dev.config.json'
+        ],
+        options.jsonVariables ?? {}
+      ) ??
+        await loadYAMLConfig(
+          [
+            './.ai.dev.config.yaml',
+            './.ai.dev.config.yml',
+            './infra/.ai.dev.config.yaml',
+            './infra/.ai.dev.config.yml'
+          ],
+          options.jsonVariables ?? {}
+        )
+    ] as const)()
+  return configCache
+}
+
+export const loadAdapterConfig = async <
+  K extends keyof AdapterMap,
+>(
+  name: K,
+  options: { jsonVariables?: Record<string, string> }
+) => {
+  const [projectConfig, userConfig] = await loadConfig(options)
+  return {
+    ...(projectConfig?.adapters?.[name] ?? {}),
+    ...(userConfig?.adapters?.[name] ?? {})
+  } as unknown as NonNullable<Config['adapters']>[K]
+}