feat(tui): add Claude Code-style --fork flag to duplicate sessions before continuing (resolves #11137) (#11340)

This commit is contained in:
Ariane Emory
2026-02-06 14:15:47 -05:00
committed by GitHub
parent e5b355e458
commit 84c5df19c7
4 changed files with 59 additions and 6 deletions

View File

@@ -236,6 +236,10 @@ export const RunCommand = cmd({
describe: "session id to continue", describe: "session id to continue",
type: "string", type: "string",
}) })
.option("fork", {
describe: "fork the session before continuing (requires --continue or --session)",
type: "boolean",
})
.option("share", { .option("share", {
type: "boolean", type: "boolean",
describe: "share the session", describe: "share the session",
@@ -324,6 +328,11 @@ export const RunCommand = cmd({
process.exit(1) process.exit(1)
} }
if (args.fork && !args.continue && !args.session) {
UI.error("--fork requires --continue or --session")
process.exit(1)
}
const rules: PermissionNext.Ruleset = [ const rules: PermissionNext.Ruleset = [
{ {
permission: "question", permission: "question",
@@ -349,11 +358,17 @@ export const RunCommand = cmd({
} }
async function session(sdk: OpencodeClient) { async function session(sdk: OpencodeClient) {
if (args.continue) { const baseID = args.continue
const result = await sdk.session.list() ? (await sdk.session.list()).data?.find((s) => !s.parentID)?.id
return result.data?.find((s) => !s.parentID)?.id : args.session
if (baseID && args.fork) {
const forked = await sdk.session.fork({ sessionID: baseID })
return forked.data?.id
} }
if (args.session) return args.session
if (baseID) return baseID
const name = title() const name = title()
const result = await sdk.session.create({ title: name, permission: rules }) const result = await sdk.session.create({ title: name, permission: rules })
return result.data?.id return result.data?.id

View File

@@ -250,7 +250,8 @@ function App() {
}) })
local.model.set({ providerID, modelID }, { recent: true }) local.model.set({ providerID, modelID }, { recent: true })
} }
if (args.sessionID) { // Handle --session without --fork immediately (fork is handled in createEffect below)
if (args.sessionID && !args.fork) {
route.navigate({ route.navigate({
type: "session", type: "session",
sessionID: args.sessionID, sessionID: args.sessionID,
@@ -268,10 +269,36 @@ function App() {
.find((x) => x.parentID === undefined)?.id .find((x) => x.parentID === undefined)?.id
if (match) { if (match) {
continued = true continued = true
route.navigate({ type: "session", sessionID: match }) if (args.fork) {
sdk.client.session.fork({ sessionID: match }).then((result) => {
if (result.data?.id) {
route.navigate({ type: "session", sessionID: result.data.id })
} else {
toast.show({ message: "Failed to fork session", variant: "error" })
}
})
} else {
route.navigate({ type: "session", sessionID: match })
}
} }
}) })
// Handle --session with --fork: wait for sync to be fully complete before forking
// (session list loads in non-blocking phase for --session, so we must wait for "complete"
// to avoid a race where reconcile overwrites the newly forked session)
let forked = false
createEffect(() => {
if (forked || sync.status !== "complete" || !args.sessionID || !args.fork) return
forked = true
sdk.client.session.fork({ sessionID: args.sessionID }).then((result) => {
if (result.data?.id) {
route.navigate({ type: "session", sessionID: result.data.id })
} else {
toast.show({ message: "Failed to fork session", variant: "error" })
}
})
})
createEffect( createEffect(
on( on(
() => sync.status === "complete" && sync.data.provider.length === 0, () => sync.status === "complete" && sync.data.provider.length === 0,

View File

@@ -6,6 +6,7 @@ export interface Args {
prompt?: string prompt?: string
continue?: boolean continue?: boolean
sessionID?: string sessionID?: string
fork?: boolean
} }
export const { use: useArgs, provider: ArgsProvider } = createSimpleContext({ export const { use: useArgs, provider: ArgsProvider } = createSimpleContext({

View File

@@ -64,6 +64,10 @@ export const TuiThreadCommand = cmd({
type: "string", type: "string",
describe: "session id to continue", describe: "session id to continue",
}) })
.option("fork", {
type: "boolean",
describe: "fork the session when continuing (use with --continue or --session)",
})
.option("prompt", { .option("prompt", {
type: "string", type: "string",
describe: "prompt to use", describe: "prompt to use",
@@ -73,6 +77,11 @@ export const TuiThreadCommand = cmd({
describe: "agent to use", describe: "agent to use",
}), }),
handler: async (args) => { handler: async (args) => {
if (args.fork && !args.continue && !args.session) {
UI.error("--fork requires --continue or --session")
process.exit(1)
}
// Resolve relative paths against PWD to preserve behavior when using --cwd flag // Resolve relative paths against PWD to preserve behavior when using --cwd flag
const baseCwd = process.env.PWD ?? process.cwd() const baseCwd = process.env.PWD ?? process.cwd()
const cwd = args.project ? path.resolve(baseCwd, args.project) : process.cwd() const cwd = args.project ? path.resolve(baseCwd, args.project) : process.cwd()
@@ -150,6 +159,7 @@ export const TuiThreadCommand = cmd({
agent: args.agent, agent: args.agent,
model: args.model, model: args.model,
prompt, prompt,
fork: args.fork,
}, },
onExit: async () => { onExit: async () => {
await client.call("shutdown", undefined) await client.call("shutdown", undefined)