feat(tui): show current branch in location labels

This commit is contained in:
Aiden Cline 2026-08-03 13:45:33 +00:00
commit 4b58478b6f
16 changed files with 176 additions and 14 deletions

View file

@ -1502,18 +1502,25 @@ export interface ProjectCopyApi<E = never> {
export type Endpoint25_0Input = { export type Endpoint25_0Input = {
readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined
} }
export type Endpoint25_0Output = { readonly location: Location.Info; readonly data: ReadonlyArray<Vcs.FileStatus> } export type Endpoint25_0Output = { readonly location: Location.Info; readonly data: Vcs.Info }
export type VcsStatusOperation<E = never> = (input?: Endpoint25_0Input) => Effect.Effect<Endpoint25_0Output, E> export type VcsGetOperation<E = never> = (input?: Endpoint25_0Input) => Effect.Effect<Endpoint25_0Output, E>
export type Endpoint25_1Input = { export type Endpoint25_1Input = {
readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined
}
export type Endpoint25_1Output = { readonly location: Location.Info; readonly data: ReadonlyArray<Vcs.FileStatus> }
export type VcsStatusOperation<E = never> = (input?: Endpoint25_1Input) => Effect.Effect<Endpoint25_1Output, E>
export type Endpoint25_2Input = {
readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined
readonly mode: Vcs.Mode readonly mode: Vcs.Mode
readonly context?: number | undefined readonly context?: number | undefined
} }
export type Endpoint25_1Output = { readonly location: Location.Info; readonly data: ReadonlyArray<FileDiff.Info> } export type Endpoint25_2Output = { readonly location: Location.Info; readonly data: ReadonlyArray<FileDiff.Info> }
export type VcsDiffOperation<E = never> = (input: Endpoint25_1Input) => Effect.Effect<Endpoint25_1Output, E> export type VcsDiffOperation<E = never> = (input: Endpoint25_2Input) => Effect.Effect<Endpoint25_2Output, E>
export interface VcsApi<E = never> { export interface VcsApi<E = never> {
readonly get: VcsGetOperation<E>
readonly status: VcsStatusOperation<E> readonly status: VcsStatusOperation<E>
readonly diff: VcsDiffOperation<E> readonly diff: VcsDiffOperation<E>
} }

View file

@ -210,6 +210,8 @@ import type {
Endpoint25_0Output, Endpoint25_0Output,
Endpoint25_1Input, Endpoint25_1Input,
Endpoint25_1Output, Endpoint25_1Output,
Endpoint25_2Input,
Endpoint25_2Output,
Endpoint26_0Output, Endpoint26_0Output,
Endpoint26_1Input, Endpoint26_1Input,
Endpoint26_1Output, Endpoint26_1Output,
@ -1182,17 +1184,26 @@ const adaptGroup24 = (raw: RawClient["server.projectCopy"]) => ({
const Endpoint25_0 = (raw: RawClient["server.vcs"]) => (input?: Endpoint25_0Input) => const Endpoint25_0 = (raw: RawClient["server.vcs"]) => (input?: Endpoint25_0Input) =>
preserveEffect<Endpoint25_0Output>()( preserveEffect<Endpoint25_0Output>()(
raw["vcs.get"]({ query: { location: input?.["location"] } }).pipe(Effect.mapError(mapClientError)),
)
const Endpoint25_1 = (raw: RawClient["server.vcs"]) => (input?: Endpoint25_1Input) =>
preserveEffect<Endpoint25_1Output>()(
raw["vcs.status"]({ query: { location: input?.["location"] } }).pipe(Effect.mapError(mapClientError)), raw["vcs.status"]({ query: { location: input?.["location"] } }).pipe(Effect.mapError(mapClientError)),
) )
const Endpoint25_1 = (raw: RawClient["server.vcs"]) => (input: Endpoint25_1Input) => const Endpoint25_2 = (raw: RawClient["server.vcs"]) => (input: Endpoint25_2Input) =>
preserveEffect<Endpoint25_1Output>()( preserveEffect<Endpoint25_2Output>()(
raw["vcs.diff"]({ query: { location: input["location"], mode: input["mode"], context: input["context"] } }).pipe( raw["vcs.diff"]({ query: { location: input["location"], mode: input["mode"], context: input["context"] } }).pipe(
Effect.mapError(mapClientError), Effect.mapError(mapClientError),
), ),
) )
const adaptGroup25 = (raw: RawClient["server.vcs"]) => ({ status: Endpoint25_0(raw), diff: Endpoint25_1(raw) }) const adaptGroup25 = (raw: RawClient["server.vcs"]) => ({
get: Endpoint25_0(raw),
status: Endpoint25_1(raw),
diff: Endpoint25_2(raw),
})
const Endpoint26_0 = (raw: RawClient["server.debug"]) => () => const Endpoint26_0 = (raw: RawClient["server.debug"]) => () =>
preserveEffect<Endpoint26_0Output>()(raw["debug.location"]({}).pipe(Effect.mapError(mapClientError))) preserveEffect<Endpoint26_0Output>()(raw["debug.location"]({}).pipe(Effect.mapError(mapClientError)))

View file

@ -202,6 +202,8 @@ import type {
ProjectCopyRemoveOutput, ProjectCopyRemoveOutput,
ProjectCopyRefreshInput, ProjectCopyRefreshInput,
ProjectCopyRefreshOutput, ProjectCopyRefreshOutput,
VcsGetInput,
VcsGetOutput,
VcsStatusInput, VcsStatusInput,
VcsStatusOutput, VcsStatusOutput,
VcsDiffInput, VcsDiffInput,
@ -1702,6 +1704,18 @@ export function make(options: ClientOptions) {
), ),
}, },
vcs: { vcs: {
get: (input?: VcsGetInput, requestOptions?: RequestOptions) =>
request<VcsGetOutput>(
{
method: "GET",
path: `/api/vcs`,
query: { location: input?.["location"] },
successStatus: 200,
declaredStatuses: [401, 400],
empty: false,
},
requestOptions,
),
status: (input?: VcsStatusInput, requestOptions?: RequestOptions) => status: (input?: VcsStatusInput, requestOptions?: RequestOptions) =>
request<VcsStatusOutput>( request<VcsStatusOutput>(
{ {

View file

@ -530,6 +530,8 @@ export type ReferenceGitSource = {
export type ProjectCopyCopy = { directory: string } export type ProjectCopyCopy = { directory: string }
export type VcsInfo = { branch?: string }
export type VcsFileStatus = { export type VcsFileStatus = {
file: string file: string
additions: number additions: number
@ -4902,6 +4904,17 @@ export type ProjectCopyRefreshInput = {
export type ProjectCopyRefreshOutput = void export type ProjectCopyRefreshOutput = void
export type VcsGetInput = {
readonly location?: {
readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined
}["location"]
}
export type VcsGetOutput = {
location: { directory: string; workspaceID?: string; project: { id: string; directory: string; canonical: string } }
data: VcsInfo
}
export type VcsStatusInput = { export type VcsStatusInput = {
readonly location?: { readonly location?: {
readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined readonly location?: { readonly directory?: string | undefined; readonly workspace?: string | undefined } | undefined

View file

@ -1,14 +1,17 @@
export * as Vcs from "./vcs" export * as Vcs from "./vcs"
import { Context, Effect, Layer } from "effect" import { Context, Effect, Layer, Ref, Stream } from "effect"
import { FileDiff } from "@opencode-ai/schema/file-diff" import { FileDiff } from "@opencode-ai/schema/file-diff"
import { FileSystem } from "@opencode-ai/schema/filesystem"
import { FileStatus, Mode } from "@opencode-ai/schema/vcs" import { FileStatus, Mode } from "@opencode-ai/schema/vcs"
import { VcsEvent } from "@opencode-ai/schema/vcs-event"
import { makeLocationNode } from "@opencode-ai/util/effect/app-node" import { makeLocationNode } from "@opencode-ai/util/effect/app-node"
import { FSUtil } from "@opencode-ai/util/fs-util" import { FSUtil } from "@opencode-ai/util/fs-util"
import { Location } from "./location" import { Location } from "./location"
import { AppProcess } from "@opencode-ai/util/process" import { AppProcess } from "@opencode-ai/util/process"
import { VcsGit } from "./vcs/git" import { VcsGit } from "./vcs/git"
import { VcsHg } from "./vcs/hg" import { VcsHg } from "./vcs/hg"
import { Bus } from "./bus"
export { FileStatus, Mode } export { FileStatus, Mode }
@ -17,6 +20,7 @@ export interface DiffOptions {
} }
export interface Interface { export interface Interface {
readonly branch: () => Effect.Effect<string | undefined>
readonly status: () => Effect.Effect<FileStatus[]> readonly status: () => Effect.Effect<FileStatus[]>
readonly diff: (mode: Mode, options?: DiffOptions) => Effect.Effect<FileDiff.Info[]> readonly diff: (mode: Mode, options?: DiffOptions) => Effect.Effect<FileDiff.Info[]>
} }
@ -38,8 +42,30 @@ const layer = Layer.effect(
const proc = yield* AppProcess.Service const proc = yield* AppProcess.Service
const fs = yield* FSUtil.Service const fs = yield* FSUtil.Service
const location = yield* Location.Service const location = yield* Location.Service
const bus = yield* Bus.Service
const impl = adapter(proc, fs, location) const impl = adapter(proc, fs, location)
const branch = yield* Ref.make(impl ? yield* impl.branch() : undefined)
if (impl && location.vcs?.type === "git") {
yield* bus.subscribe(FileSystem.Event.Changed).pipe(
Stream.filter((event) => event.data.file.endsWith("HEAD")),
Stream.runForEach(() =>
Effect.gen(function* () {
const next = yield* impl.branch()
if (next === (yield* Ref.get(branch))) return
yield* Ref.set(branch, next)
yield* bus.publish(VcsEvent.BranchUpdated, { branch: next }, {
location: { directory: location.directory, workspaceID: location.workspaceID },
})
}),
),
Effect.forkScoped({ startImmediately: true }),
)
}
return Service.of({ return Service.of({
branch: Effect.fn("Vcs.branch")(function* () {
if (!impl) return
return yield* impl.branch()
}),
status: Effect.fn("Vcs.status")(function* () { status: Effect.fn("Vcs.status")(function* () {
if (!impl) return [] if (!impl) return []
return yield* impl.status() return yield* impl.status()
@ -55,5 +81,5 @@ const layer = Layer.effect(
export const node = makeLocationNode({ export const node = makeLocationNode({
service: Service, service: Service,
layer: layer, layer: layer,
deps: [AppProcess.node, FSUtil.node, Location.node], deps: [AppProcess.node, FSUtil.node, Location.node, Bus.node],
}) })

View file

@ -20,6 +20,9 @@ export function make(proc: AppProcess.Interface, input: { directory: string; wor
const ctx: Ctx = { git: makeGit(proc), directory: input.directory, worktree: input.worktree } const ctx: Ctx = { git: makeGit(proc), directory: input.directory, worktree: input.worktree }
return { return {
branch: Effect.fn("VcsGit.branch")(function* () {
return yield* ctx.git.branch(ctx.directory)
}),
status: Effect.fn("VcsGit.status")(function* () { status: Effect.fn("VcsGit.status")(function* () {
const git = ctx.git const git = ctx.git
const ref = (yield* git.hasHead(ctx.directory)) ? "HEAD" : undefined const ref = (yield* git.hasHead(ctx.directory)) ? "HEAD" : undefined

View file

@ -73,6 +73,9 @@ export function make(
}) })
return { return {
branch: Effect.fn("VcsHg.branch")(function* () {
return yield* hg.branch()
}),
status: Effect.fn("VcsHg.status")(function* () { status: Effect.fn("VcsHg.status")(function* () {
const [items, batch] = yield* Effect.all( const [items, batch] = yield* Effect.all(
// Zero-context patches are enough to count changed lines. // Zero-context patches are enough to count changed lines.

View file

@ -53,6 +53,7 @@ describe("Vcs", () => {
withTmp((directory) => withTmp((directory) =>
Effect.gen(function* () { Effect.gen(function* () {
const vcs = yield* Vcs.Service const vcs = yield* Vcs.Service
expect(yield* vcs.branch()).toBeUndefined()
expect(yield* vcs.status()).toEqual([]) expect(yield* vcs.status()).toEqual([])
expect(yield* vcs.diff("working")).toEqual([]) expect(yield* vcs.diff("working")).toEqual([])
expect(yield* vcs.diff("branch")).toEqual([]) expect(yield* vcs.diff("branch")).toEqual([])
@ -155,6 +156,7 @@ describe("Vcs", () => {
await commitAll(directory, "initial") await commitAll(directory, "initial")
}) })
const vcs = yield* Vcs.Service const vcs = yield* Vcs.Service
expect(yield* vcs.branch()).toBe("main")
expect(yield* vcs.diff("branch")).toEqual([]) expect(yield* vcs.diff("branch")).toEqual([])
yield* Effect.promise(async () => { yield* Effect.promise(async () => {
@ -162,6 +164,7 @@ describe("Vcs", () => {
await fs.writeFile(path.join(directory, "file.txt"), "one\ntwo\n") await fs.writeFile(path.join(directory, "file.txt"), "one\ntwo\n")
await commitAll(directory, "feature change") await commitAll(directory, "feature change")
}) })
expect(yield* vcs.branch()).toBe("feature")
const diff = yield* vcs.diff("branch") const diff = yield* vcs.diff("branch")
expect(diff.map((item) => ({ file: item.file, status: item.status }))).toEqual([ expect(diff.map((item) => ({ file: item.file, status: item.status }))).toEqual([
{ file: "file.txt", status: "modified" }, { file: "file.txt", status: "modified" },

View file

@ -19,6 +19,7 @@ import type {
SessionPendingInfo, SessionPendingInfo,
ShellInfo, ShellInfo,
SkillInfo, SkillInfo,
VcsInfo,
} from "@opencode-ai/client" } from "@opencode-ai/client"
import type { ResolvedTheme } from "@opencode-ai/theme/tui" import type { ResolvedTheme } from "@opencode-ai/theme/tui"
import type { CliRenderer, KeyEvent, Renderable } from "@opentui/core" import type { CliRenderer, KeyEvent, Renderable } from "@opentui/core"
@ -113,6 +114,11 @@ export interface Data {
default(): LocationRef default(): LocationRef
sync(location?: LocationRef): Promise<void> sync(location?: LocationRef): Promise<void>
invalidate(location?: LocationRef): void invalidate(location?: LocationRef): void
readonly vcs: {
get(location?: LocationRef): VcsInfo | undefined
sync(location?: LocationRef): Promise<void>
invalidate(location?: LocationRef): void
}
readonly agent: LocationCollection<AgentInfo> readonly agent: LocationCollection<AgentInfo>
readonly command: LocationCollection<CommandInfo> readonly command: LocationCollection<CommandInfo>
readonly integration: LocationCollection<IntegrationInfo> readonly integration: LocationCollection<IntegrationInfo>

View file

@ -13,6 +13,20 @@ const DiffQuery = Schema.Struct({
}) })
export const VcsGroup = HttpApiGroup.make("server.vcs") export const VcsGroup = HttpApiGroup.make("server.vcs")
.add(
HttpApiEndpoint.get("vcs.get", "/api/vcs", {
query: LocationQuery,
success: Location.response(Vcs.Info),
})
.annotateMerge(locationQueryOpenApi)
.annotateMerge(
OpenApi.annotations({
identifier: "v2.vcs.get",
summary: "VCS information",
description: "Get version control information for the requested location.",
}),
),
)
.add( .add(
HttpApiEndpoint.get("vcs.status", "/api/vcs/status", { HttpApiEndpoint.get("vcs.status", "/api/vcs/status", {
query: LocationQuery, query: LocationQuery,

View file

@ -1,7 +1,12 @@
export * as Vcs from "./vcs.js" export * as Vcs from "./vcs.js"
import { Schema } from "effect" import { Schema } from "effect"
import { NonNegativeInt } from "./schema.js" import { NonNegativeInt, optional } from "./schema.js"
export const Info = Schema.Struct({
branch: optional(Schema.String),
}).annotate({ identifier: "Vcs.Info" })
export interface Info extends Schema.Schema.Type<typeof Info> {}
export const Mode = Schema.Literals(["working", "branch"]).annotate({ identifier: "Vcs.Mode" }) export const Mode = Schema.Literals(["working", "branch"]).annotate({ identifier: "Vcs.Mode" })
export type Mode = typeof Mode.Type export type Mode = typeof Mode.Type

View file

@ -7,6 +7,14 @@ import { response } from "../location"
export const VcsHandler = HttpApiBuilder.group(Api, "server.vcs", (handlers) => export const VcsHandler = HttpApiBuilder.group(Api, "server.vcs", (handlers) =>
Effect.gen(function* () { Effect.gen(function* () {
return handlers return handlers
.handle("vcs.get", () =>
response(
Effect.gen(function* () {
const vcs = yield* Vcs.Service
return { branch: yield* vcs.branch() }
}),
),
)
.handle("vcs.status", () => .handle("vcs.status", () =>
response( response(
Effect.gen(function* () { Effect.gen(function* () {

View file

@ -1330,11 +1330,16 @@ export function Prompt(props: PromptProps) {
if (!props.sessionID) { if (!props.sessionID) {
// No session yet: show where the next session will be created. // No session yet: show where the next session will be created.
const directory = currentLocation.ref?.directory ?? data.location.default().directory const directory = currentLocation.ref?.directory ?? data.location.default().directory
return abbreviateHome(directory, paths.home) const branch = data.location.vcs.get(currentLocation.ref)?.branch
const label = abbreviateHome(directory, paths.home)
return branch ? label + ":" + branch : label
} }
if (status() !== "idle") return if (status() !== "idle") return
const directory = data.session.get(props.sessionID)?.location.directory const ref = data.session.get(props.sessionID)?.location
return directory ? abbreviateHome(directory, paths.home) : undefined if (!ref) return
const label = abbreviateHome(ref.directory, paths.home)
const branch = data.location.vcs.get(ref)?.branch
return branch ? label + ":" + branch : label
}) })
const spinnerDef = createMemo(() => { const spinnerDef = createMemo(() => {

View file

@ -28,6 +28,7 @@ import type {
ShellInfo, ShellInfo,
SkillInfo, SkillInfo,
OpenCodeEvent, OpenCodeEvent,
VcsInfo,
WebSearchProvider, WebSearchProvider,
} from "@opencode-ai/client" } from "@opencode-ai/client"
import type { Plugin } from "@opencode-ai/plugin/tui" import type { Plugin } from "@opencode-ai/plugin/tui"
@ -49,6 +50,7 @@ type ShellWithLocation = ShellInfo & { readonly location: LocationRef }
type LocationData = { type LocationData = {
info?: LocationGetOutput info?: LocationGetOutput
vcs?: VcsInfo
agent?: AgentInfo[] agent?: AgentInfo[]
command?: CommandInfo[] command?: CommandInfo[]
integration?: IntegrationInfo[] integration?: IntegrationInfo[]
@ -905,6 +907,15 @@ export const { use: useData, provider: DataProvider } = createSimpleContext({
result.location.mcp.resource.invalidate(event.location) result.location.mcp.resource.invalidate(event.location)
void result.location.mcp.resource.sync(event.location) void result.location.mcp.resource.sync(event.location)
break break
case "vcs.branch.updated": {
const ref = event.location ?? defaultLocation()
const key = locationKey(ref)
setStore("location", key, {
...store.location[key],
vcs: { branch: event.data.branch },
})
break
}
} }
} }
@ -1139,6 +1150,26 @@ export const { use: useData, provider: DataProvider } = createSimpleContext({
default() { default() {
return defaultLocation() return defaultLocation()
}, },
vcs: {
get(ref?: LocationRef) {
return store.location[locationKey(ref ?? defaultLocation())]?.vcs
},
sync(ref?: LocationRef) {
const location = ref ?? defaultLocation()
const id = locationKey(location)
return sync.run(`location.vcs:${id}`, async () => {
const response = await client.api.vcs.get({ location: locationQuery(location) })
const key = locationKey(response.location)
setStore("location", key, {
...store.location[key],
vcs: response.data,
})
})
},
invalidate(ref?: LocationRef) {
sync.invalidate(`location.vcs:${locationKey(ref ?? defaultLocation())}`)
},
},
async sync(ref?: LocationRef) { async sync(ref?: LocationRef) {
const current = ref ?? defaultLocation() const current = ref ?? defaultLocation()
await sync.run(`location:${locationKey(current)}`, async () => { await sync.run(`location:${locationKey(current)}`, async () => {
@ -1161,6 +1192,7 @@ export const { use: useData, provider: DataProvider } = createSimpleContext({
result.location.provider.sync(location), result.location.provider.sync(location),
result.location.reference.sync(location), result.location.reference.sync(location),
result.location.skill.sync(location), result.location.skill.sync(location),
result.location.vcs.sync(location),
result.shell.sync(location), result.shell.sync(location),
result.session.form.sync("global", location), result.session.form.sync("global", location),
]) ])
@ -1177,6 +1209,7 @@ export const { use: useData, provider: DataProvider } = createSimpleContext({
result.location.provider.invalidate(location) result.location.provider.invalidate(location)
result.location.reference.invalidate(location) result.location.reference.invalidate(location)
result.location.skill.invalidate(location) result.location.skill.invalidate(location)
result.location.vcs.invalidate(location)
result.shell.invalidate(location) result.shell.invalidate(location)
result.session.form.invalidate("global", location) result.session.form.invalidate("global", location)
}, },

View file

@ -6,8 +6,14 @@ function View(props: { context: Plugin.Context }) {
const directory = createMemo(() => const directory = createMemo(() =>
props.context.location ? props.context.ui.format.path(props.context.location.directory) : undefined, props.context.location ? props.context.ui.format.path(props.context.location.directory) : undefined,
) )
const value = createMemo(() => {
const path = directory()
if (!path) return
const branch = props.context.data.location.vcs.get(props.context.location)?.branch
return branch ? path + ":" + branch : path
})
return ( return (
<Show when={directory()}> <Show when={value()}>
{(value) => <FilePath value={value()} maxWidth={38} fg={props.context.theme.text.subdued} />} {(value) => <FilePath value={value()} maxWidth={38} fg={props.context.theme.text.subdued} />}
</Show> </Show>
) )

View file

@ -127,6 +127,11 @@ export function createFetch(override?: FetchHandler, events?: ReturnType<typeof
}) })
if (url.pathname === "/api/reference") if (url.pathname === "/api/reference")
return json({ location: { directory, project: { id: "proj_test", directory, canonical: directory } }, data: [] }) return json({ location: { directory, project: { id: "proj_test", directory, canonical: directory } }, data: [] })
if (url.pathname === "/api/vcs")
return json({
location: { directory, project: { id: "proj_test", directory: worktree, canonical: worktree } },
data: { branch: "main" },
})
if (url.pathname === "/api/websearch/provider") { if (url.pathname === "/api/websearch/provider") {
return json({ location: { directory, project: { id: "proj_test", directory, canonical: directory } }, data: [] }) return json({ location: { directory, project: { id: "proj_test", directory, canonical: directory } }, data: [] })
} }