UNPKG

@builder.io/dev-tools

Version:

Builder.io Visual CMS Devtools

467 lines (466 loc) • 19.8 kB
import type { DevToolsSys } from "../types"; import { type Credentials } from "./credentials"; import type { CodegenFeedback, CodeGenToolMap, CodegenTurn, CustomInstruction, FusionConfig, GenerateCompletionState, GenerateCompletionStep, GenerateCompletionStepGit, GenerateUserMessage, SessionMode, UserContext, UserSource, WorkspaceFolder, LoadWholeSessionOptions, LoadWholeSessionResult, LoadHistoryResult, CodeGenMode, ApplyActionsResult, PrivacyMode, CodeGenPosition, BackupGitRepoResult, SuggestedActionBuildError, PushChangesArgs, CodegenApiResult, CodegenApiTerminal, ConfigureDevOrchestratorOpts, ConfigureDevOrchestratorUpdates, RepoMetrics, FolderWatchEvent, MCPServerConfig, CodegenApiCreateTerminal, SyncChangesFromRemote, SearchFilesOptions, SearchFilesResult, SearchFileTreeOptions, SearchFileTreeResult, ExplorationMetadataToolInput, CodegenAbortOptions, MessageUpdateOptions } from "$/ai-utils"; import { type FusionContext, type ToolResolution } from "./code-tools"; import { type SubAgent } from "./utils/agent-discovery"; import EventEmitter from "node:events"; import { type RunCommandOptions } from "./utils/git"; export interface SyncChangesFromBranches { canPush: boolean; branches: string[]; uncommittedChanges: "stash" | "commit" | "fail"; allowUnrelatedHistory: boolean; fastForward: "never" | "required" | "auto"; requestRefresh?: boolean; } export interface SessionContext { sessionId: string; turns: CodegenTurn[]; customInstructions: CustomInstruction[]; customAgents: SubAgent[]; userContext: UserContext; prettierConfig: Record<string, unknown> | null; state: GenerateCompletionState; title: string | undefined; beforeCommit: string | undefined; createdUnixTime: number; updatedUnixTime: number; canLoadMore: boolean; sessionMode: SessionMode; } export interface CodeGenSessionOptionsBase { sys: DevToolsSys; credentials: Credentials; position: CodeGenPosition; maxTokens?: number; mode: CodeGenMode; privacyMode?: PrivacyMode; builtInCustomInstructions?: CustomInstruction[]; builtInCustomAgents?: SubAgent[]; builtInMCPServerConfig?: MCPServerConfig; autoImportLocalMCPs?: boolean; systemPromptOverride?: string; fusionContext?: FusionContext; fusionConfig?: FusionConfig; workingDirectory?: string; mcpServers?: boolean; enabledTools?: (keyof CodeGenToolMap)[]; modelOverride?: string; } export interface CodeGenSessionOptionsSession extends CodeGenSessionOptionsBase { sessionOrCompletionId?: string; } export interface CodeGenSessionOptionsInitialUrl extends CodeGenSessionOptionsBase { initialUrl: string; } export type CodeGenSessionOptions = CodeGenSessionOptionsSession | CodeGenSessionOptionsInitialUrl; export interface SpawnAgentResult { success: boolean; response: string; metadata: ExplorationMetadataToolInput; lastTurn?: CodegenTurn; } export type CodeGenEventEmitter = EventEmitter<{ step: [GenerateCompletionStep]; idle: []; }>; export declare class CodeGenSession { #private; constructor(options: CodeGenSessionOptions); get fusionConfig(): FusionConfig | undefined; get gitEnabledFolder(): WorkspaceFolder | undefined; get workingDirectory(): string; getSessionMode(): SessionMode; switchSessionMode(newMode: SessionMode): Promise<void>; setPrivacyMode(privacyMode: PrivacyMode | undefined): Promise<void>; initializeSession(opts?: { skipSessionLoading?: boolean; }): Promise<void>; loadHistory(): Promise<LoadHistoryResult>; loadWholeSession(opts?: LoadWholeSessionOptions): Promise<LoadWholeSessionResult>; loadMoreTurns(): Promise<CodegenTurn[]>; setCustomInstructions(instructions: CustomInstruction[]): Promise<void>; setCustomAgents(agents: SubAgent[]): Promise<void>; /** * Spawn a named custom agent by ID or name * @param agentNameOrId - The agent's name or ID * @param options - Additional spawning options * @returns SpawnAgentResult */ spawnNamedAgent(agentName: string | undefined, options: { prompt: string; user: UserSource; onStep?: (step: GenerateCompletionStep, subagent_type: string) => void; signal?: AbortSignal; maxCompletions?: number; sessionId?: string; mcpServers?: boolean; }): Promise<SpawnAgentResult>; pushRepoV2(repoInfo: { repoFullName: string; repoUrl: string; }): Promise<CodegenApiResult>; zipFolder(folderName: string): Promise<string>; archiveProject(): Promise<string>; isIdle(): boolean; needsBackup(): Promise<boolean>; uploadBackup(forcedFullBackup?: boolean): Promise<BackupGitRepoResult | { success: false; error: Error; }>; getCommitMode(): import("$/ai-utils").CommitMode; pushChanges(opts: PushChangesArgs): Promise<CodegenApiResult>; pullLatestFromRemote(opts?: { requestRefresh?: boolean; }): Promise<CodegenApiResult>; abortMerge(emitStatus?: boolean): Promise<CodegenApiResult>; syncChangesFromRemote(opts?: SyncChangesFromRemote): Promise<CodegenApiResult>; /** * Get the current commit hash */ getCurrentCommitHash(branchName?: string): Promise<string | undefined>; getCurrentBranch(): Promise<string>; /** * Get the feature branch name */ getFeatureBranch(): string; /** * Get the AI branch name */ getAiBranch(): string; git(args: string[], opts?: string | RunCommandOptions): Promise<string>; setDebug(debug: boolean): void; createTerminal(options?: CodegenApiCreateTerminal): Promise<CodegenApiTerminal>; emitTerminals(): void; updateTerminal({ terminalId, cols, rows, title, }: { terminalId: string; cols?: number; rows?: number; title?: string; }): boolean; writeTerminal({ terminalId, data }: { terminalId: string; data: string; }): boolean; signalTerminal({ terminalId, signal, }: { terminalId: string; signal: "SIGINT" | "SIGTERM" | "SIGKILL"; }): boolean; disposeTerminal({ terminalId, emitTerminals, }: { terminalId: string; emitTerminals?: boolean; }): boolean; restartTerminal({ terminalId }: { terminalId: string; }): Promise<boolean>; subscribeTerminal({ terminalId, onData, onExit, }: { terminalId: string; onData: (chunk: string) => void; onExit?: (code?: number) => void; }): (() => void) | undefined; getAllFiles(options?: { getDotFiles?: boolean; globbyPattern?: string; includePattern?: string; gitignore?: boolean; deep?: number; truncate?: number; }): Promise<string[]>; searchFiles(options: SearchFilesOptions): Promise<SearchFilesResult>; /** * Search for files by their names/paths (for quick open functionality). * Uses ripgrep's --files flag for speed (like VS Code's Cmd+T quick open). * Always case insensitive. Returns match indices for highlighting in the UI. * Respects access control policies (deny patterns). */ searchFileTree(options: SearchFileTreeOptions): Promise<SearchFileTreeResult>; collectRepoMetrics(opts?: { rootPath?: string; }): Promise<RepoMetrics>; getSessionId(): string; getSpaceId(): string | undefined; revertToCommitHash(commitHash: string): Promise<void>; resetToCommitHash(commitHash: string): Promise<void>; /** * Core function to restore the codebase to a state that matches a predicate. * This is the main function that handles both git-based and file-based restoration. * * @param predicate Function that takes a turn and its index and returns true if we should restore up to that turn * @param dryRun If true, only simulate the restoration without making changes * @returns Array of file paths that were changed */ restore({ location, predicate, revert, dryRun, forceReplay, debug, }: { location: "before" | "after"; predicate: (turn: CodegenTurn | null, index: number) => boolean; dryRun?: boolean; revert?: boolean; forceReplay?: boolean; debug?: string; }): Promise<{ undone: string[] | null; message: string; }>; restoreFromCompletionId({ location, completionId, forceReplay, }: { location: "before" | "after"; completionId: string; forceReplay?: boolean; }): Promise<{ undone: string[] | null; message: string; }>; restoreBeforeCompletionId(completionId: string): Promise<{ undone: string[] | null; message: string; }>; /** * Undo all changes back to the last user message */ undoLastUserMessage(dryRun?: boolean): Promise<{ undone: string[] | null; message: string; }>; getLastUserCompletionId(): string | undefined; getLastCompletionId(): string | undefined; getCurrentState(): GenerateCompletionState; getLastApplyResultsTurn(): CodegenTurn | undefined; getLastTurn(): CodegenTurn | undefined; getNextUrl(): string | undefined; sendFeedback(feedback: Partial<CodegenFeedback>): Promise<void>; lastTurnHasChanges(): Promise<boolean>; waitUntilState(state: GenerateCompletionState, timeout?: number): Promise<void>; clearSession(): Promise<void>; clearMessageQueue(emitUserMessages?: boolean): Promise<void>; updateMessage(opts: MessageUpdateOptions): void; flushMessageQueue(): Promise<void>; sendMessage(message: GenerateUserMessage | GenerateUserMessage[]): Promise<string | null>; spawnAgent(options: { subagent_type: string; prompt: string; user: UserSource; onStep?: (step: GenerateCompletionStep, subagent_type: string) => void; signal?: AbortSignal; maxCompletions?: number; sessionId?: string; mcpServers?: boolean; tools?: string[]; model?: string; mode?: CodeGenMode; includeMemories?: boolean; needDevServer?: boolean; needValidation?: boolean; systemPrompt?: string; }): Promise<SpawnAgentResult>; setProxyOrigin(proxySrc: string | undefined): void; getTurns(): CodegenTurn[]; getSessionContext(): SessionContext; runSetupCommand(): Promise<import("$/ai-utils").SetupCommandResult | undefined>; abortSetupCommand(): Promise<import("$/ai-utils").SetupCommandResult | undefined> | undefined; abortValidateCommand(): Promise<import("$/ai-utils").ValidateCommandResult | undefined> | undefined; toolsRunning(): boolean; abortAllTools(): void; /** * Fulfil a pending tool call (usually AskUser or any passThrough tool) * Exposed via websocket as `toolFullfilment` for the Builder UI to send back * the user's response. */ toolFullfilment(id: string, result: ToolResolution | string): boolean; cancelAllPendingToolFulfilments(): void; fulfillToolCall(id: string, result: ToolResolution): boolean; abortToolCall(id: string): boolean; acceptCode(): Promise<void>; abort(abortOptions?: CodegenAbortOptions | boolean): Promise<boolean>; stopEventLoop(): Promise<void>; requestRefresh(): void; configureDevOrchestrator(opts: ConfigureDevOrchestratorOpts): Promise<ConfigureDevOrchestratorUpdates>; /** * Subscribe to file change events for the entire working directory. * Returns a dispose function to unsubscribe. * The watcher is lazily initialized on first subscription and cleaned up on last unsubscribe. */ subscribeToFileChanges(onEvent: (event: FolderWatchEvent) => void): () => Promise<void>; close(uploadGitBackup?: boolean): Promise<void>; emitGitStatus(): Promise<GenerateCompletionStepGit | null>; manualCommit(options: { add: string; commitMessage: string; }): Promise<boolean>; getLastSuggestedAction(): SuggestedActionBuildError | undefined; connectToEventLoop(shouldReplay: boolean, onStep: (step: GenerateCompletionStep) => void): () => void; waitUntilPendingPromises(): Promise<void>; waitUntilBlockingPendingPromises(): Promise<void>; waitUntilIdle(): Promise<void>; waitForEventLoop(): Promise<void>; commitWorkInProgress(lastTurn: CodegenTurn, changedFiles: string[]): Promise<string | false | undefined>; getChangesReport(): Promise<{ diff: string; files: string[]; } | undefined>; isCleanWorkTree(): Promise<boolean>; /** * Resolves a workspace file path to its actual file system path * @param filePath A file path that may include a workspace prefix (e.g., "workspace1/path/to/file.js") * @param forceWorkspace If true, will try the first workspace as fallback when no workspace folder is found * @returns The actual file system path and the workspace folder it belongs to */ resolveWorkspacePath(filePath: string, forceWorkspace: boolean): { resolvedPath: string; workspaceFolder?: WorkspaceFolder; }; /** * Converts an absolute file system path to a workspace URL * @param absolutePath The absolute file system path to convert * @returns The workspace URL if the path can be converted, undefined otherwise */ absolutePathToWorkspaceUrl(absolutePath: string): string | undefined; /** * Launches the editor for a given file path * @param filePath The file path to launch the editor for * @param line The line number to launch the editor at * @param column The column number to launch the editor at * @returns void */ launchEditor(opts?: { filePath?: string; line?: number; column?: number; }): Promise<{ success: boolean; message?: string; }>; /** * Reads a file from the workspace * @param filePath A file path that may include a workspace prefix * @returns The file content or null if the file doesn't exist */ readFile(filePath: string, skipAclCheck?: boolean): Promise<string | null>; /** * Gets file content at a specific git reference (e.g., origin/main, HEAD~1, commit hash) * @param filePath - The file path relative to the workspace * @param gitRef - The git reference (branch, commit, tag) - defaults to parent branch * @returns The file content at that reference, or null if not found */ getFileAtRef(filePath: string, gitRef?: string): Promise<string | null>; /** * Gets the diff information for a single file including full content * @param filePath - The file path to get diff for * @returns Object with oldContent (from parent branch) and newContent (current) */ getSingleFileDiff(filePath: string): Promise<{ oldContent: string | null; newContent: string | null; action: "create" | "update" | "delete"; }>; /** * Discards changes for a specific file by creating a revert commit * @param filePath - The file path to discard changes for * @returns success status, commit hash for reverting, and optional error message */ discardFileChanges(filePath: string): Promise<{ success: boolean; commitHash?: string; error?: string; }>; /** * Reverts a discard commit using git revert * @param commitHash - The commit hash to revert * @returns success status and optional error message */ revertDiscard(options: { commitHash: string; }): Promise<{ success: boolean; error?: string; }>; /** * Checks if a file exists in the workspace * @param filePath A file path that may include a workspace prefix * @returns True if the file exists, false otherwise */ fileExists(filePath: string): Promise<{ absolutePath: string | undefined; recommendedPath: string | undefined; workspaceFolder: WorkspaceFolder | undefined; virtual: boolean; }>; /** * Reads a file from the workspace synchronously * @param filePath A file path that may include a workspace prefix * @returns The file content or null if the file doesn't exist */ readFileSync(filePath: string, skipAclCheck?: boolean): string | null; /** * Writes content to a file in the workspace * @param filePath A file path that may include a workspace prefix * @param content The content to write * @returns True if the write was successful, false otherwise */ writeFile(filePath: string, content: string | Uint8Array, skipAclCheck?: boolean): Promise<string | null>; /** * Lists files in a directory in the workspace * @param dirPath A directory path that may include a workspace prefix * @returns Array of file names in the directory or empty array if directory doesn't exist */ listDir(dirPath: string): Promise<string[]>; /** * Get stats for a file in the workspace * @param filePath A file path that may include a workspace prefix * @returns The file stats or null if the file doesn't exist */ stat(filePath: string, skipAclCheck?: boolean): Promise<{ isDirectory: () => boolean; isFile: () => boolean; size: number; } | null>; /** * Deletes a file from the workspace * @param filePath A file path that may include a workspace prefix * @returns True if the delete was successful, false otherwise */ deleteFile(filePath: string, skipAclCheck?: boolean): Promise<string | null>; getLinesStats(): { net: number; added: number; removed: number; }; /** * Get git diff between current commit and remote branch * If remote current branch doesn't exist, gets diff between default branch and current branch * @param numberOfContextLines - Optional number of context lines to include in the diff * @param includeFilesOnly - If true, only return filePath and action properties * @param filePaths - Optional array of file paths to limit the diff to specific files */ getDiffFromRemote({ numberOfContextLines, includeFilesOnly, filePaths, }: { numberOfContextLines?: number; includeFilesOnly?: boolean; filePaths?: Array<string>; }): Promise<ApplyActionsResult[]>; /** * Get git diff based on the specified mode * @param mode - The diff mode: 'previous-commit', 'parent-branch', or 'remote' * @param numberOfContextLines - Optional number of context lines to include in the diff (e.g., 999 for -U999) * @param includeFilesOnly - If true, only return filePath and action properties * @param filePaths - Optional array of file paths to limit the diff to specific files */ getDiff({ mode, numberOfContextLines, includeFilesOnly, filePaths, }: { mode: "remote-parent-branch" | "remote-current-branch"; numberOfContextLines?: number; includeFilesOnly?: boolean; filePaths?: Array<string>; }): Promise<{ state: "error" | "success"; diff?: ApplyActionsResult[]; error?: Error; }>; /** * Get the default branch name from remote repository * Falls back to checking common default branch names */ private getDefaultBranch; /** * Get git diff between current branch and its parent branch (main/master) * @param numberOfContextLines - Optional number of context lines to include in the diff * @param includeFilesOnly - If true, only return filePath and action properties * @param filePaths - Optional array of file paths to limit the diff to specific files */ private getDiffFromParentBranch; }